From e7b5b2048aafe465fd43fa683659c0b60ca3caa5 Mon Sep 17 00:00:00 2001
From: Brian Jacobel <bjacobel@gmail.com>
Date: Wed, 19 Jun 2019 12:54:40 -0400
Subject: [PATCH] Add definitions for jscodeshift/src/testUtils (#36154)

* Add defns for testUtils module

* Import unecessary here, use a reference path instead

* Add testUtils.d.ts to files array instead of reference
---
 types/jscodeshift/index.d.ts                | 1 +
 types/jscodeshift/src/testUtils.d.ts        | 6 ++++++
 types/jscodeshift/test/jscodeshift-tests.ts | 8 ++++++++
 types/jscodeshift/tsconfig.json             | 1 +
 4 files changed, 16 insertions(+)
 create mode 100644 types/jscodeshift/src/testUtils.d.ts

diff --git a/types/jscodeshift/index.d.ts b/types/jscodeshift/index.d.ts
index e19423d928..b0e5b3596a 100644
--- a/types/jscodeshift/index.d.ts
+++ b/types/jscodeshift/index.d.ts
@@ -1,6 +1,7 @@
 // Type definitions for jscodeshift 0.6
 // Project: https://github.com/facebook/jscodeshift#readme
 // Definitions by: Brie Bunge <https://github.com/brieb>
+//                 Brian Jacobel <https://github.com/bjacobel>
 // Definitions: https://github.com/DefinitelyTyped/DefinitelyTyped
 // TypeScript Version: 2.9
 
diff --git a/types/jscodeshift/src/testUtils.d.ts b/types/jscodeshift/src/testUtils.d.ts
new file mode 100644
index 0000000000..4a6a650f8c
--- /dev/null
+++ b/types/jscodeshift/src/testUtils.d.ts
@@ -0,0 +1,6 @@
+export function defineTest(
+  dirName: string,
+  transformName: string,
+  options: any,
+  testFilePrefix?: string
+): () => any;
diff --git a/types/jscodeshift/test/jscodeshift-tests.ts b/types/jscodeshift/test/jscodeshift-tests.ts
index f426eb37b7..d08a640e35 100644
--- a/types/jscodeshift/test/jscodeshift-tests.ts
+++ b/types/jscodeshift/test/jscodeshift-tests.ts
@@ -1,4 +1,5 @@
 import { ASTNode, FileInfo, API, Transform, Parser } from "jscodeshift";
+import * as testUtils from "jscodeshift/src/testUtils";
 
 // Can define transform with `function`.
 function replaceWithFooTransform(fileInfo: FileInfo, api: API) {
@@ -73,3 +74,10 @@ const transformWithRecastParseOptions: Transform = (file, { j }) => {
         }
     }
 }
+
+// Can define a test
+testUtils.defineTest(
+    "directory",
+    "transformName",
+    { opt: true },
+);
diff --git a/types/jscodeshift/tsconfig.json b/types/jscodeshift/tsconfig.json
index 5937bd483a..3887d91c9a 100644
--- a/types/jscodeshift/tsconfig.json
+++ b/types/jscodeshift/tsconfig.json
@@ -24,6 +24,7 @@
         "src/collections/Node.d.ts",
         "src/collections/VariableDeclarator.d.ts",
         "src/template.d.ts",
+        "src/testUtils.d.ts",
         "test/jscodeshift-tests.ts",
         "test/js-transforms/bind-this-to-bind-expression.ts",
         "test/js-transforms/call-expression-bind-this-to-arrow-function-expression.ts",