fix: Print callback syntax in expectation error message

NiGhTTraX · Jul 28, 2022 · 0bf9633 · 0bf9633
1 parent 5e847d9
commit 0bf9633
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/src/errors.spec.ts b/src/errors.spec.ts
@@ -34,7 +34,7 @@ describe('errors', () => {
 
       expectAnsilessEqual(
         pendingExpectation.toJSON(),
-        `when(mock.bar(1, 2, 3))`
+        `when(() => mock.bar(1, 2, 3))`
       );
     });
 
@@ -48,7 +48,7 @@ describe('errors', () => {
       pendingExpectation.args = undefined;
       pendingExpectation.property = 'bar';
 
-      expectAnsilessEqual(pendingExpectation.toJSON(), `when(mock.bar)`);
+      expectAnsilessEqual(pendingExpectation.toJSON(), `when(() => mock.bar)`);
     });
   });
 

diff --git a/src/expectation/strong-expectation.spec.ts b/src/expectation/strong-expectation.spec.ts
@@ -68,7 +68,7 @@ describe('StrongExpectation', () => {
 
     expectAnsilessEqual(
       expectation.toJSON(),
-      `when(mock.baz(4, 5, 6)).thenReturn(42).between(2, 3)`
+      `when(() => mock.baz(4, 5, 6)).thenReturn(42).between(2, 3)`
     );
   });
 

diff --git a/src/print.ts b/src/print.ts
@@ -50,10 +50,10 @@ export const printReturns = (
 
 export const printWhen = (property: Property, args: any[] | undefined) => {
   if (args) {
-    return `when(${EXPECTED_COLOR(`mock${printCall(property, args)}`)})`;
+    return `when(() => ${EXPECTED_COLOR(`mock${printCall(property, args)}`)})`;
   }
 
-  return `when(${EXPECTED_COLOR(`mock${printProperty(property)}`)})`;
+  return `when(() => ${EXPECTED_COLOR(`mock${printProperty(property)}`)})`;
 };
 
 export const printExpectation = (