test(vest): add tests for VestTest.cancel

ealush · Dec 26, 2020 · ede4587 · ede4587
1 parent b8746f9
commit ede4587
Show file tree

Hide file tree

Showing 2 changed files with 57 additions and 0 deletions.
diff --git a/packages/vest/src/core/test/lib/VestTest.js b/packages/vest/src/core/test/lib/VestTest.js
@@ -45,6 +45,10 @@ VestTest.prototype.warn = function () {
   this.isWarning = true;
 };
 
+/**
+ * Marks a test as canceled, removes it from the state.
+ * This function needs to be called within a stateRef context.
+ */
 VestTest.prototype.cancel = function () {
   this.canceled = true;
   removePending(this);

diff --git a/packages/vest/src/core/test/lib/__tests__/VestTest.test.js b/packages/vest/src/core/test/lib/__tests__/VestTest.test.js
@@ -1,8 +1,19 @@
+import runCreateRef from '../../../../../testUtils/runCreateRef';
+
 import VestTest from 'VestTest';
+import addTestToState from 'addTestToState';
+import context from 'ctx';
+import { setPending } from 'pending';
+import usePending from 'usePending';
+import useTestObjects from 'useTestObjects';
 
 const fieldName = 'unicycle';
 const statement = 'I am Root.';
 
+let stateRef;
+
+it.ctx = (str, cb) => it(str, () => context.run({ stateRef }, cb));
+
 describe('VestTest', () => {
   let testObject;
 
@@ -72,4 +83,46 @@ describe('VestTest', () => {
       expect(testObject.valueOf()).toBe(false);
     });
   });
+
+  describe('testObject.cancel', () => {
+    beforeEach(() => {
+      stateRef = runCreateRef();
+
+      context.run({ stateRef }, () => {
+        addTestToState(testObject);
+      });
+    });
+    it.ctx('Should remove a testObject from the state', () => {
+      const [testObjects] = useTestObjects();
+      expect(testObjects).toEqual(expect.arrayContaining([testObject]));
+      testObject.cancel();
+      expect(testObjects).toEqual(expect.not.arrayContaining([testObject]));
+    });
+
+    it.ctx('Should remove a testObject from the pending state', () => {
+      setPending(testObject);
+      {
+        const [{ pending }] = usePending();
+        expect(pending).toEqual(expect.arrayContaining([testObject]));
+      }
+      testObject.cancel();
+      {
+        const [{ pending }] = usePending();
+        expect(pending).toEqual(expect.not.arrayContaining([testObject]));
+      }
+    });
+
+    it.ctx('Should remove a testObject from the lagging state', () => {
+      usePending(state => ({ ...state, lagging: [testObject] }));
+      {
+        const [{ lagging }] = usePending();
+        expect(lagging).toEqual(expect.arrayContaining([testObject]));
+      }
+      testObject.cancel();
+      {
+        const [{ lagging }] = usePending();
+        expect(lagging).toEqual(expect.not.arrayContaining([testObject]));
+      }
+    });
+  });
 });