Add context tests

statelyai · Sep 10, 2021 · 65c7e62 · 65c7e62
1 parent 648435c
commit 65c7e62
Showing 1 changed file with 35 additions and 0 deletions.
diff --git a/packages/core/test/state.test.ts b/packages/core/test/state.test.ts
@@ -615,6 +615,41 @@ describe('State', () => {
       expect(machine.initialState.can({ type: 'NEXT' })).toBe(true);
     });
 
+    it('should return true for an event object that results in a new action', () => {
+      const machine = createMachine({
+        initial: 'a',
+        states: {
+          a: {
+            on: {
+              NEXT: {
+                actions: 'newAction'
+              }
+            }
+          }
+        }
+      });
+
+      expect(machine.initialState.can({ type: 'NEXT' })).toBe(true);
+    });
+
+    it('should return true for an event object that results in a context change', () => {
+      const machine = createMachine({
+        initial: 'a',
+        context: { count: 0 },
+        states: {
+          a: {
+            on: {
+              NEXT: {
+                actions: assign({ count: 1 })
+              }
+            }
+          }
+        }
+      });
+
+      expect(machine.initialState.can({ type: 'NEXT' })).toBe(true);
+    });
+
     it('should return false for an external self-transition without actions', () => {
       const machine = createMachine({
         initial: 'a',