feat: add a setModelStore to force model store

loopingz · Nov 29, 2023 · 5354f46 · 5354f46
1 parent 997e11b
commit 5354f46
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/packages/core/src/core.spec.ts b/packages/core/src/core.spec.ts
@@ -634,6 +634,11 @@ class CoreTest extends WebdaTest {
     assert.throws(() => this.webda.registerOperation("__proto__", undefined), Error);
 
     assert.ok(Core["getSingletonInfo"](this.webda).match(/- file:\/\/.*packages\/core\/src\/core.ts \/ \d+.\d+.\d+/));
+
+    assert.strictEqual(User.store().getName(), "MemoryUsers");
+    const memoryStore = new MemoryStore(this.webda, "test", {});
+    this.webda.setModelStore(User, memoryStore);
+    assert.strictEqual(User.store().getName(), "test");
   }
 
   @test

diff --git a/packages/core/src/core.ts b/packages/core/src/core.ts
@@ -520,6 +520,18 @@ export class Core<E extends CoreEvents = CoreEvents> extends events.EventEmitter
     return singleton;
   }
 
+  /**
+   * Enforce a specific store for a model
+   *
+   * Useful in some specific case when you want to update store dynamically
+   *
+   * @param model
+   * @param store
+   */
+  setModelStore(model: Constructor<CoreModel>, store: Store) {
+    this._modelStoresCache.set(model, store);
+  }
+
   /**
    * Get the store assigned to this model
    * @param model