cherry-pick(#16420): feat(driver): add reuse and navigate commands (#…

…16423)
microsoft · Aug 10, 2022 · fdf31cd · fdf31cd
1 parent bbd3fd7
commit fdf31cd
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/packages/playwright-core/src/cli/driver.ts b/packages/playwright-core/src/cli/driver.ts
@@ -92,6 +92,19 @@ class ProtocolHandler {
     this._playwright = playwright;
   }
 
+  async resetForReuse() {
+    const contexts = new Set<BrowserContext>();
+    for (const page of this._playwright.allPages())
+      contexts.add(page.context());
+    for (const context of contexts)
+      await context.resetForReuse(internalMetadata, null);
+  }
+
+  async navigate(params: { url: string }) {
+    for (const p of this._playwright.allPages())
+      await p.mainFrame().goto(internalMetadata, params.url);
+  }
+
   async setMode(params: { mode: Mode, language?: string, file?: string }) {
     await gc(this._playwright);
 

diff --git a/packages/playwright-core/src/server/browserContext.ts b/packages/playwright-core/src/server/browserContext.ts
@@ -157,12 +157,14 @@ export abstract class BrowserContext extends SdkObject {
     return JSON.stringify(paramsCopy);
   }
 
-  async resetForReuse(metadata: CallMetadata, params: channels.BrowserNewContextForReuseParams) {
+  async resetForReuse(metadata: CallMetadata, params: channels.BrowserNewContextForReuseParams | null) {
     this.setDefaultNavigationTimeout(undefined);
     this.setDefaultTimeout(undefined);
 
-    for (const key of paramsThatAllowContextReuse)
-      (this._options as any)[key] = params[key];
+    if (params) {
+      for (const key of paramsThatAllowContextReuse)
+        (this._options as any)[key] = params[key];
+    }
 
     await this._cancelAllRoutesInFlight();