Add Mocha.Context to the e2e_non_hosted testing function signature

Bug: none
Change-Id: I973f6d43e19fe3b547be3a22ec73938f44097016
Reviewed-on: https://chromium-review.googlesource.com/c/devtools/devtools-frontend/+/6361534
Commit-Queue: Liviu Rau <[email protected]>
Reviewed-by: Philip Pfaffe <[email protected]>

Author: Liviu Rau

test/e2e_non_hosted/conductor/mocha-interface-helpers.ts

@@ -18,7 +18,7 @@ export interface State {
 }
 
 /* eslint-disable-next-line @typescript-eslint/no-explicit-any */
-export type TestCallbackWithState = (state: State) => PromiseLike<any>;
+export type TestCallbackWithState = (this: Mocha.Context, state: State) => PromiseLike<any>;
 
 async function takeScreenshots(state: State): Promise<{target?: string, frontend?: string}> {
   try {

test/e2e_non_hosted/conductor/mocha-interface.ts

@@ -108,7 +108,7 @@ function iterationSuffix(iteration: number): string {
 function customIt(testImplementation: TestFunctions, suite: Mocha.Suite, file: string) {
   function instrumentWithState(fn: TestCallbackWithState) {
     const fnWithState = async function(this: Mocha.Context) {
-      return await StateProvider.instance.callWithState(suite, fn);
+      return await StateProvider.instance.callWithState(this, suite, fn);
     };
     return makeInstrumentedTestFunction(fnWithState, 'test');
   }

test/e2e_non_hosted/conductor/state-provider.ts

@@ -36,12 +36,12 @@ export class StateProvider {
     this.settingsCallbackMap.set(suite, suiteSettings);
   }
 
-  async callWithState(suite: Mocha.Suite, testFn: TestCallbackWithState) {
+  async callWithState(context: Mocha.Context, suite: Mocha.Suite, testFn: TestCallbackWithState) {
     const {state, browsingContext} = await this.getState(suite);
     try {
       /* eslint-disable-next-line no-debugger */
       debugger;  // If you're paused here while debugging, stepping into the next line will step into your test.
-      return await testFn(state);
+      return await testFn.call(context, state);
     } finally {
       await browsingContext.close();
       dumpCollectedErrors();