hide ADAPTERS map from implementation

salesforce · Feb 20, 2018 · 381db1c · 381db1c
1 parent 89973d6
commit 381db1c
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 3 deletions.
diff --git a/packages/lwc-wire-service/src/__tests__/main.test.js b/packages/lwc-wire-service/src/__tests__/main.test.js
@@ -30,5 +30,12 @@ describe("main.js", () => {
             const wireService = registerWireService(register, mockAdapters);
             expect(wireService.unregister).toBeDefined();
         });
+        it("runtime wire adapter registration management", () => {
+            const register = jest.fn();
+            const wireService = registerWireService(register, mockAdapters);
+            const foo = () => {};
+            wireService.register(foo);
+            expect(wireService.unregister(foo.name)).toBe(true);
+        });
     });
 });
diff --git a/packages/lwc-wire-service/src/main.js b/packages/lwc-wire-service/src/main.js
@@ -5,7 +5,12 @@
  * This service is Salesforce-agnostic. The wire adapters must be provided to this module.
  */
 
-import { setWireAdapters, installWiring } from './wire-service';
+import {
+    setWireAdapters,
+    installWiring,
+    registerWireAdapter,
+    unregisterWireAdapter
+} from './wire-service';
 
 /**
  * The wire service.
@@ -46,10 +51,10 @@ export default function registerWireService(register, ...adapterGenerators) {
 
     return {
         register: (adapter) => {
-            ADAPTERS.set(adapter.name, adapter);
+            registerWireAdapter(adapter);
         },
         unregister: (adapterName) => {
-            ADAPTERS.delete(adapterName);
+            return unregisterWireAdapter(adapterName);
         }
     };
 }
diff --git a/packages/lwc-wire-service/src/wire-service.js b/packages/lwc-wire-service/src/wire-service.js
@@ -241,3 +241,19 @@ export function installWiring(cmp, def) {
 export function setWireAdapters(adapterGenerators) {
     ADAPTERS = buildWireAdapterMap(adapterGenerators);
 }
+
+/**
+ * Sets the wire adapter.
+ * @param {Function} adapter Wire adapter function.
+ */
+export function registerWireAdapter(adapter) {
+    ADAPTERS.set(adapter.name, adapter);
+}
+
+/**
+ * Removes the wire adapter.
+ * @param {String} adapterName Wire adapter function name.
+ */
+export function unregisterWireAdapter(adapterName) {
+    return ADAPTERS.delete(adapterName);
+}