support test macros in typescript definitions (#975)

avajs · Aug 11, 2016 · 8816faf · 8816faf
1 parent ea25b20
commit 8816faf
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/types/base.d.ts b/types/base.d.ts
@@ -120,5 +120,13 @@ export interface ContextualCallbackTestContext extends CallbackTestContext {
 	context: any;
 }
 
+export interface Macro<T> {
+	(t: T, ...args: any[]): void;
+	title? (providedTitle: string, ...args: any[]): string;
+}
+export type Macros<T> = Macro<T> | Macro<T>[];
+
 export function test(name: string, run: ContextualTest): void;
 export function test(run: ContextualTest): void;
+export function test(name: string, run: Macros<ContextualTestContext>, ...args: any[]): void;
+export function test(run: Macros<ContextualTestContext>, ...args: any[]): void;
diff --git a/types/make.js b/types/make.js
@@ -53,6 +53,8 @@ function generatePrefixed(prefix) {
 				const type = testType(parts);
 				output += '\t' + writeFunction(part, 'name: string, implementation: ' + type);
 				output += '\t' + writeFunction(part, 'implementation: ' + type);
+				output += '\t' + writeFunction(part, 'name: string, implementation: Macros<' + type + 'Context>, ...args: any[]');
+				output += '\t' + writeFunction(part, 'implementation: Macros<' + type + 'Context>, ...args: any[]');
 			}
 		}