feat(jasmine): handle spy.throwError (#619)

skovhus · Sep 14, 2024 · 17ca751 · 17ca751
1 parent ba5fbf7
commit 17ca751
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 0 deletions.
diff --git a/src/transformers/jasmine-globals.test.ts b/src/transformers/jasmine-globals.test.ts
@@ -126,6 +126,9 @@ test('jasmine.createSpy', () => {
     jasmine.createSpy().and.resolveTo('lmao');
     jasmine.createSpy().and.rejectWith('oh no');
     const spy3 = jasmine.createSpy().and.resolveTo('lmao');
+    spy.and.throwError('my error message');
+    spy.and.throwError(new Error('awesome error'));
+    spy.and.throwError(42);
     `,
     `
     jest.fn();
@@ -137,6 +140,15 @@ test('jasmine.createSpy', () => {
     jest.fn().mockResolvedValue('lmao');
     jest.fn().mockRejectedValue('oh no');
     const spy3 = jest.fn().mockResolvedValue('lmao');
+    spy.mockImplementation(() => {
+        throw new Error('my error message');
+    });
+    spy.mockImplementation(() => {
+        throw new Error('awesome error');
+    });
+    spy.mockImplementation(() => {
+        throw 42;
+    });
     `
   )
 

diff --git a/src/transformers/jasmine-globals.ts b/src/transformers/jasmine-globals.ts
@@ -341,6 +341,27 @@ export default function jasmineGlobals(fileInfo, api, options) {
           path.node.callee.property.name = 'mockRejectedValue'
           break
         }
+
+        case 'throwError': {
+          const throwArg = path.node.arguments[0]
+          const isStringLiteral =
+            throwArg.type === 'Literal' && typeof throwArg.value === 'string'
+
+          path.node.callee.object = path.node.callee.object.object
+          path.node.callee.property.name = 'mockImplementation'
+
+          path.node.arguments = [
+            j.arrowFunctionExpression(
+              [],
+              isStringLiteral
+                ? j.blockStatement([
+                    j.throwStatement(j.newExpression(j.identifier('Error'), [throwArg])),
+                  ])
+                : j.blockStatement([j.throwStatement(throwArg)])
+            ),
+          ]
+          break
+        }
       }
     })