maing possible to access target in beforeEach

Farenheith · Dec 14, 2019 · 195a11f · 195a11f
1 parent f80fea0
commit 195a11f
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 8 deletions.
diff --git a/build/src/strict-describe-method.d.ts b/build/src/strict-describe-method.d.ts
@@ -10,11 +10,11 @@ export declare class StaticMethodDescribeHelper<Target> {
 export declare class MethodDescribeHelper<Target> extends StaticMethodDescribeHelper<Target> {
     protected readonly bootstrap: () => Target;
     constructor(bootstrap: () => Target, cls: ClassOf<Target>);
-    createMethodDescribe(suite: (title: string, fn: () => void) => void): (method: keyof Target, fn: (it: MethodTestFunction<Target>) => void) => void;
+    createMethodDescribe(suite: (title: string, fn: () => void) => void): (method: keyof Target, fn: (it: MethodTestFunction<Target>, getTarget: () => Target) => void) => void;
     createDescribe(): MethodSuite<Target>;
 }
 export interface BaseMethodSuite<Target> {
-    (methodName: keyof Target, fn: (it: MethodTestFunction<Target>) => void): any;
+    (methodName: keyof Target, fn: (it: MethodTestFunction<Target>, getTarget: () => Target) => void): any;
 }
 export interface MethodSuite<Target> extends BaseMethodSuite<Target> {
     only: BaseMethodSuite<Target>;

diff --git a/build/src/strict-describe-method.js b/build/src/strict-describe-method.js
diff --git a/build/src/strict-describe-method.js.map b/build/src/strict-describe-method.js.map
diff --git a/src/strict-describe-method.ts b/src/strict-describe-method.ts
@@ -46,7 +46,12 @@ export class MethodDescribeHelper<Target> extends StaticMethodDescribeHelper<Tar
 	}
 
 	createMethodDescribe(suite: (title: string, fn: () => void) => void) {
-		return (method: keyof Target, fn: (it: MethodTestFunction<Target>) => void) => {
+		return (method: keyof Target,
+			fn: (
+					it: MethodTestFunction<Target>,
+					getTarget: () => Target,
+				) => void
+		) => {
 			const wrapper = {} as TestWrapper<Target>;
 			const itHelper = new ItHelper(wrapper);
 			const it = itHelper.createIt();
@@ -66,7 +71,7 @@ export class MethodDescribeHelper<Target> extends StaticMethodDescribeHelper<Tar
 					);
 				});
 
-				fn(it);
+				fn(it, () => wrapper.target);
 
 				afterEach(() => {
 					for (const pair of backup) {
@@ -90,9 +95,8 @@ export class MethodDescribeHelper<Target> extends StaticMethodDescribeHelper<Tar
 	}
 }
 
-
 export interface BaseMethodSuite<Target> {
-	(methodName: keyof Target, fn: (it: MethodTestFunction<Target>) => void);
+	(methodName: keyof Target, fn: (it: MethodTestFunction<Target>, getTarget: () => Target) => void);
 }
 
 export interface MethodSuite<Target> extends BaseMethodSuite<Target> {