PB-54921 - Browser extension - toolbar badge counts notes and custom fields suggested for the current tab

Author: ceroazul

src/shared/models/entity/resourceType/resourceTypesCollection.js

@@ -131,6 +131,14 @@ class ResourceTypesCollection extends EntityV2Collection {
     );
   }
 
+  /**
+   * Filter by resource types supported by the quick access.
+   * @return {void} The function alters the collection itself.
+   */
+  filterByQuickAccessResourceTypes() {
+    this.filterByCallback((resourceType) => resourceType.isSupportedByQuickAccess());
+  }
+
   /**
    * Filter by resource type version.
    * @param {string} version the version used to filter the resource types

src/shared/models/entity/resourceType/resourceTypesCollection.test.js

@@ -35,7 +35,9 @@ import {
   resourceTypeV5DefaultDto,
   TEST_RESOURCE_TYPE_V5_STANDALONE_NOTE,
   TEST_RESOURCE_TYPE_V5_STANDALONE_PIN_CODE,
+  TEST_RESOURCE_TYPE_V5_CUSTOM_FIELDS,
 } from "./resourceTypeEntity.test.data";
+import ResourceTypeEntity from "./resourceTypeEntity";
 import CollectionValidationError from "../../entity/abstract/collectionValidationError";
 import { v4 as uuid } from "uuid";
 import {
@@ -183,6 +185,38 @@ describe("ResourceTypesCollection", () => {
     });
   });
 
+  describe("::filterByQuickAccessResourceTypes", () => {
+    it("should keep the password, TOTP, PIN code, note and custom fields resource types.", () => {
+      expect.assertions(12);
+      const resourceTypes = new ResourceTypesCollection(resourceTypesCollectionDto());
+      resourceTypes.filterByQuickAccessResourceTypes();
+      expect(resourceTypes).toHaveLength(11);
+      expect(resourceTypes.getFirstById(TEST_RESOURCE_TYPE_PASSWORD_STRING)).toBeTruthy();
+      expect(resourceTypes.getFirstById(TEST_RESOURCE_TYPE_PASSWORD_AND_DESCRIPTION)).toBeTruthy();
+      expect(resourceTypes.getFirstById(TEST_RESOURCE_TYPE_PASSWORD_DESCRIPTION_TOTP)).toBeTruthy();
+      expect(resourceTypes.getFirstById(TEST_RESOURCE_TYPE_TOTP)).toBeTruthy();
+      expect(resourceTypes.getFirstById(TEST_RESOURCE_TYPE_V5_DEFAULT)).toBeTruthy();
+      expect(resourceTypes.getFirstById(TEST_RESOURCE_TYPE_V5_PASSWORD_STRING)).toBeTruthy();
+      expect(resourceTypes.getFirstById(TEST_RESOURCE_TYPE_V5_DEFAULT_TOTP)).toBeTruthy();
+      expect(resourceTypes.getFirstById(TEST_RESOURCE_TYPE_V5_TOTP)).toBeTruthy();
+      expect(resourceTypes.getFirstById(TEST_RESOURCE_TYPE_V5_STANDALONE_PIN_CODE)).toBeTruthy();
+      expect(resourceTypes.getFirstById(TEST_RESOURCE_TYPE_V5_STANDALONE_NOTE)).toBeTruthy();
+      expect(resourceTypes.getFirstById(TEST_RESOURCE_TYPE_V5_CUSTOM_FIELDS)).toBeTruthy();
+    });
+
+    it("should drop the resource types the quick access does not support.", () => {
+      expect.assertions(2);
+      const spy = jest.spyOn(ResourceTypeEntity.prototype, "isSupportedByQuickAccess").mockImplementation(function () {
+        return !this.isStandaloneNote();
+      });
+      const resourceTypes = new ResourceTypesCollection(resourceTypesCollectionDto());
+      resourceTypes.filterByQuickAccessResourceTypes();
+      expect(resourceTypes).toHaveLength(10);
+      expect(resourceTypes.getFirstById(TEST_RESOURCE_TYPE_V5_STANDALONE_NOTE)).toBeUndefined();
+      spy.mockRestore();
+    });
+  });
+
   describe("::filterByResourceTypeVersion", () => {
     it("should filter the collection by resources types by the version 4.", () => {
       expect.assertions(9);