raw: testing a program

Author: frawa

samples/src/Samples/Program.test.tsx

@@ -0,0 +1,100 @@
+import { mount, ReactWrapper } from 'enzyme';
+import { extendJest, Cmd, Sub, Task, Program, ProgramProps } from "react-tea-cup";
+import React, { ReactNode } from 'react';
+import { Dispatcher } from 'tea-cup-core';
+// import 'jest-enzyme';
+
+extendJest(expect);
+
+const init1: () => [number, Cmd<string>] = () => {
+    return [0, toCmd('go')];
+}
+
+const view1: (dispatch: Dispatcher<string>, model: number) => React.ReactNode = (dispatch: Dispatcher<string>, model: number) => {
+    return (<div className={'count'}>{model}</div>);
+}
+
+const update1: (msg: string, model: number) => [number, Cmd<string>] = (msg: string, model: number) => {
+    return [model + 1, model < 5 ? toCmd('go') : Cmd.none()];
+}
+
+// const toCmd = (msg: string) => Task.perform(Time.in(0), () => msg);
+const toCmd = (msg: string) => Task.perform(Task.succeed(0), () => msg);
+
+describe('Test Program', () => {
+
+    it('first', () => {
+        const props: ProgramProps<number, string> = {
+            init: init1,
+            view: view1,
+            update: update1,
+            subscriptions: () => Sub.none<string>()
+        }
+        return mountWhenIdle(props).then(([model, wrapper]) => {
+            expect(model).toEqual(6)
+            // expect(wrapper).toHaveHTML('')
+            expect(wrapper.find('.count')).toHaveText('6')
+        })
+    })
+})
+
+function mountWhenIdle<Model, Msg>(props: ProgramProps<Model, Msg>) {
+    const testable = new Testable(props);
+    // return Promise.resolve(mount(<Program {...testable.programProps} />));
+    return testable.mountWhenIdle()
+}
+
+class Testable<Model, Msg>{
+    private resolve?: (idle: [Model, ReactWrapper<Program<Model, Msg>, ProgramProps<Model, Msg>, never>]) => void;
+    private cmds: Cmd<Msg>[] = [];
+    private dispatch?: Dispatcher<Msg>;
+    private model?: Model;
+    private viewed?: ReactNode;
+
+    constructor(private readonly props: ProgramProps<Model, Msg>) {
+    }
+
+    get programProps() {
+        const props: ProgramProps<Model, Msg> = {
+            init: this.props.init,
+            view: (dispatch: Dispatcher<Msg>, model: Model) => this.view(dispatch, model),
+            update: (msg: Msg, model: Model) => this.update(msg, model),
+            subscriptions: this.props.subscriptions
+        }
+        return props
+    }
+
+    private view(dispatch: Dispatcher<Msg>, model: Model) {
+        this.dispatch = dispatch;
+        this.viewed = this.props.view(dispatch, model);
+        return this.viewed;
+    }
+
+    private update(msg: Msg, model: Model): [Model, Cmd<Msg>] {
+        const [model1, cmd1] = this.props.update(msg, model);
+        this.model = model1;
+        this.cmds.push(cmd1);
+        setTimeout(() => {
+            this.process();
+        }, 0);
+        return [model1, Cmd.none()];
+    }
+
+    private process() {
+        this.cmds = this.cmds.filter(cmd => cmd.constructor.name !== 'CmdNone')
+        if (this.cmds.length === 0) {
+            this.resolve?.([this.model!, mount(<>{this.viewed}</>)]);
+            return;
+        }
+        const first = this.cmds[0]
+        this.cmds = this.cmds.slice(1)
+        first.execute(this.dispatch!)
+    }
+
+    mountWhenIdle(): Promise<[Model, ReactWrapper<Program<Model, Msg>, ProgramProps<Model, Msg>, never>]> {
+        mount(<Program {...this.programProps} />)
+        return new Promise(resolve => {
+            this.resolve = resolve;
+        });
+    }
+}