fix: context binding in addModel (#873)

rematch · Mar 2, 2021 · 7f99a45 · 7f99a45
1 parent 69b44df
commit 7f99a45
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 4 deletions.
diff --git a/packages/core/src/rematchStore.ts b/packages/core/src/rematchStore.ts
@@ -43,16 +43,14 @@ export default function createRematchStore<
 		addModel(model: NamedModel<TModels>) {
 			validateModel(model)
 			createModelReducer(bag, model)
-			prepareModel(this, bag, model)
-			this.replaceReducer(createRootReducer(bag))
+			prepareModel(rematchStore, bag, model)
+			reduxStore.replaceReducer(createRootReducer(bag))
 			reduxStore.dispatch({ type: '@@redux/REPLACE' })
 		},
 	} as RematchStore<TModels, TExtraModels>
 
 	addExposed(rematchStore, config.plugins)
 
-	rematchStore.addModel.bind(rematchStore)
-
 	// generate dispatch[modelName][actionName] for all reducers and effects
 	bag.models.forEach((model) => prepareModel(rematchStore, bag, model))
 

diff --git a/packages/core/test/store.test.ts b/packages/core/test/store.test.ts
@@ -133,4 +133,20 @@ describe('createStore:', () => {
 
 		expect(store.getState()).toEqual({ countA: 0, countB: 0, countC: 0 })
 	})
+
+	test('addModel works when it was used alone', () => {
+		const store = init()
+
+		const { addModel } = store
+
+		addModel({
+			name: 'count',
+			state: 0,
+			reducers: {},
+		})
+
+		expect(store.getState()).toEqual({
+			count: 0,
+		})
+	})
 })