test: More tests

Author: unlight

src/example.ts

@@ -6,6 +6,10 @@ it('hello world test', () => {
     expect(hello()).toBe('hello world');
 });
 
+jest.mock('foo');
+
 describe('hello world test', () => {
-    jest.mock('fs');
+    beforeAll(() => {
+        jest.mock('foo');
+    });
 });

src/index.spec.ts

@@ -21,10 +21,29 @@ function transform(sourceText: string) {
 
 it('smoke test', () => {
     expect(lib).toBeTruthy();
-    expect(transform(';')).toEqual(';');
+    expect(transform(';')).toBe(';');
 });
 
 it('should be removed it call expression', () => {
-    debugger;
-    expect(transform('it()')).toEqual('');
+    expect(transform('it()')).toBe('');
+    expect(transform('xit()')).toBe('');
+    expect(transform('fit()')).toBe('');
+});
+
+it('should be removed describe call expression', () => {
+    expect(transform('describe()')).toBe('');
+    expect(transform('xdescribe()')).toBe('');
+    expect(transform('fdescribe()')).toBe('');
+});
+
+it('functions with modifiers', () => {
+    expect(transform('describe.only()')).toBe('');
+    expect(transform('describe.skip()')).toBe('');
+    expect(transform('describe.any()')).toBe('');
+    expect(transform('test.only()')).toBe('');
+    expect(transform('test.skip()')).toBe('');
+    expect(transform('test.any()')).toBe('');
+    expect(transform('it.only()')).toBe('');
+    expect(transform('it.skip()')).toBe('');
+    expect(transform('it.any()')).toBe('');
 });

src/index.ts

@@ -15,10 +15,47 @@ function visitor(node: ts.Node): ts.Node | undefined {
     return node;
 }
 
+// todo: delete only global
+const testMethodList = [
+    'jasmine',
+    'jest',
+    'before',
+    'after',
+    'beforeAll',
+    'beforeEach',
+    'afterAll',
+    'afterEach',
+    'describe',
+    'fdescribe',
+    'xdescribe',
+    'it',
+    'fit',
+    'xit',
+    'test',
+    'xtest',
+    'spyOn',
+];
+
+function isTestExpressionText(text: string | ts.__String) {
+    return testMethodList.includes(String(text));
+}
+
 function isSpecExpressionStatement(node: ts.Node) {
-    return (ts.isExpressionStatement(node) && ts.isCallExpression(node.expression)
-        && ts.isIdentifier(node.expression.expression)
-        && node.expression.expression.escapedText === 'it');
+    if (ts.isExpressionStatement(node) && ts.isCallExpression(node.expression)) {
+        const callExpr = node.expression.expression;
+        if (ts.isIdentifier(callExpr) && isTestExpressionText(callExpr.escapedText)) {
+            console.log("callExpr", callExpr.flags);
+            return true;
+        }
+        if (ts.isPropertyAccessExpression(callExpr)) {
+            const propertyAccessExpr = callExpr.expression;
+            if (ts.isIdentifier(propertyAccessExpr) && isTestExpressionText(propertyAccessExpr.escapedText)) {
+                console.log("propertyAccessExpr", propertyAccessExpr.flags);
+                return true;
+            }
+        }
+        return false;
+    }
 }
 
 module.exports = typescriptTransformUnspec;

src/test.ts

@@ -47,6 +47,7 @@
     "no-reserved-keywords": false,
     "no-increment-decrement": false,
     "no-unnecessary-bind": false,
+    "newline-per-chained-call": false,
     ".": false
   }
 }