test: fix test for callBase

Author: atheck

src/Types.ts

@@ -37,20 +37,20 @@ type InitFunction<TProps, TModel, TMessage> = (props: TProps) => InitResult<TMod
 type UpdateReturnType<TModel, TMessage> = [Partial<TModel>, ...(Cmd<TMessage> | undefined)[]];
 
 type DeferFunction<TModel, TMessage> = (model: Partial<TModel>, ...commands: (Cmd<TMessage> | undefined)[]) => void;
-type UpdateMapFunction<TProps, TModel, TMessage extends Message> = (
+type UpdateMapFunction<TProps, TModel, TMessage> = (
 	msg: TMessage,
 	model: TModel,
 	props: TProps,
 	options: UpdateFunctionOptions<TProps, TModel, TMessage>,
 ) => UpdateReturnType<TModel, TMessage>;
 
-interface UpdateFunctionOptions<TProps, TModel, TMessage extends Message> {
+interface UpdateFunctionOptions<TProps, TModel, TMessage> {
 	defer: DeferFunction<TModel, TMessage>;
 	// biome-ignore lint/suspicious/noExplicitAny: Didn't find a better solution for now.
-	callBase: (fn: UpdateMapFunction<TProps, TModel, any>) => UpdateReturnType<TModel, TMessage>;
+	callBase: (fn: UpdateMapFunction<TProps, TModel, any>) => UpdateReturnType<TModel, any>;
 }
 
-type UpdateFunction<TProps, TModel, TMessage extends Message> = (
+type UpdateFunction<TProps, TModel, TMessage> = (
 	model: TModel,
 	msg: TMessage,
 	props: TProps,

src/createCallBase.spec.ts

@@ -1,28 +1,48 @@
+import { cmd } from "./cmd";
 import type { UpdateMap } from "./Types";
 
 type Message1 = { name: "test"; param: string } | { name: "second" };
 type Message2 = { name: "first" } | Message1;
 
-const updateMap1: UpdateMap<{}, {}, Message1> = {
+interface Model {}
+
+interface Props {}
+
+const updateMap1: UpdateMap<Props, Model, Message1> = {
 	test() {
-		return [{}];
+		return [{}, cmd.ofMsg({ name: "second" })];
 	},
 
 	second() {
 		return [{}];
 	},
 };
 
-const updateMap2: UpdateMap<{}, {}, Message2> = {
+const updateMap2: UpdateMap<Props, Model, Message2> = {
 	first() {
 		return [{}];
 	},
 
 	...updateMap1,
 
-	test(_msg, _model, _props, { callBase }) {
-		callBase(updateMap1.test);
+	test(_msg, _model, _props, { defer, callBase }) {
+		defer(...callBase(updateMap1.test));
 
 		return [{}];
 	},
 };
+
+describe("createCallBase", () => {
+	it("works", () => {
+		// arrange
+		const mockDefer = jest.fn();
+		const mockCallBase = jest.fn().mockReturnValue([{ foo: "bar" }, { name: "second" }]);
+
+		// act
+		updateMap2.test({ name: "test", param: "test" }, {}, {}, { defer: mockDefer, callBase: mockCallBase });
+
+		// assert
+		expect(mockCallBase).toHaveBeenCalledWith(updateMap1.test);
+		expect(mockDefer).toHaveBeenCalledWith({ foo: "bar" }, { name: "second" });
+	});
+});