JS: classify x.test.js files with test(...) calls as jest tests

Author: Esben Sparre Andreasen

javascript/ql/src/semmle/javascript/frameworks/Testing.qll

@@ -38,6 +38,34 @@ class BDDTest extends Test, @callexpr {
   }
 }
 
+/**
+ * Gets the test file for `f` with stem extension `stemExt`.
+ * That is, a file named file named `<base>.<stemExt>.<ext>` in the
+ * same directory as `f` which is named `<base>.<ext>`.
+ */
+bindingset[stemExt]
+private File getTestFile(File f, string stemExt) {
+  result = f.getParentContainer().getFile(f.getStem() + "." + stemExt + "." + f.getExtension())
+}
+
+/**
+ * A Jest test, that is, an invocation of a global function named
+ * `test` where the first argument is a string and the second
+ * argument is a function. Additionally, the invocation happens in a file
+ * named `<base>.test.<ext>` in the same directory as a file named
+ * `<base>.<ext>`.
+ */
+class JestTest extends Test, @callexpr {
+  JestTest() {
+    exists(CallExpr call | call = this |
+      call.getCallee().(GlobalVarAccess).getName() = "test" and
+      exists(call.getArgument(0).getStringValue()) and
+      call.getArgument(1).analyze().getAValue() instanceof AbstractFunction
+    ) and
+    getFile() = getTestFile(any(File f), "test")
+  }
+}
+
 /**
  * A xUnit.js fact, that is, a function annotated with an xUnit.js
  * `Fact` annotation.

javascript/ql/test/query-tests/filters/ClassifyFiles/ClassifyFiles.expected

@@ -7,6 +7,7 @@
 | etherpad.html:0:0:0:0 | etherpad.html | generated |
 | exported-data.js:0:0:0:0 | exported-data.js | generated |
 | htmltidy.html:0:0:0:0 | htmltidy.html | generated |
+| implementation.test.js:0:0:0:0 | implementation.test.js | test |
 | jison-lex.js:0:0:0:0 | jison-lex.js | generated |
 | jison.js:0:0:0:0 | jison.js | generated |
 | jquery-datatables.js:0:0:0:0 | jquery-datatables.js | library |

javascript/ql/test/query-tests/filters/ClassifyFiles/implementation.js

@@ -0,0 +1 @@
+var x = 42;

javascript/ql/test/query-tests/filters/ClassifyFiles/implementation.spec.js

@@ -0,0 +1 @@
+test("this should work", function(){});

javascript/ql/test/query-tests/filters/ClassifyFiles/implementation.test.js

@@ -0,0 +1 @@
+test("this should work", function(){});