From 245179103e4dced4b5a28821642433b9d1dc6da4 Mon Sep 17 00:00:00 2001
From: Max Schmitt <max@schmitt.mx>
Date: Wed, 19 Jun 2024 09:26:26 +0200
Subject: [PATCH] cherry-pick(#31357): fix(clock): under reused context

We uninstall all the setInitScript but forgot to mark `installed` as
`false`.

Fixes https://github.com/microsoft/playwright/issues/31353
---
 .../playwright-core/src/server/browserContext.ts    |  1 +
 packages/playwright-core/src/server/clock.ts        |  4 ++++
 tests/library/browsercontext-reuse.spec.ts          | 13 +++++++++++++
 3 files changed, 18 insertions(+)

diff --git a/packages/playwright-core/src/server/browserContext.ts b/packages/playwright-core/src/server/browserContext.ts
index c1d70372a2f42..404f94fe4fac2 100644
--- a/packages/playwright-core/src/server/browserContext.ts
+++ b/packages/playwright-core/src/server/browserContext.ts
@@ -216,6 +216,7 @@ export abstract class BrowserContext extends SdkObject {
     await this._resetStorage();
     await this._removeExposedBindings();
     await this._removeInitScripts();
+    this.clock.markAsUninstalled();
     // TODO: following can be optimized to not perform noops.
     if (this._options.permissions)
       await this.grantPermissions(this._options.permissions);
diff --git a/packages/playwright-core/src/server/clock.ts b/packages/playwright-core/src/server/clock.ts
index 58a1f7272038e..1e0822ff04b2d 100644
--- a/packages/playwright-core/src/server/clock.ts
+++ b/packages/playwright-core/src/server/clock.ts
@@ -26,6 +26,10 @@ export class Clock {
     this._browserContext = browserContext;
   }
 
+  markAsUninstalled() {
+    this._scriptInstalled = false;
+  }
+
   async fastForward(ticks: number | string) {
     await this._installIfNeeded();
     const ticksMillis = parseTicks(ticks);
diff --git a/tests/library/browsercontext-reuse.spec.ts b/tests/library/browsercontext-reuse.spec.ts
index c25d66d5885db..14590a3ae33cb 100644
--- a/tests/library/browsercontext-reuse.spec.ts
+++ b/tests/library/browsercontext-reuse.spec.ts
@@ -252,6 +252,19 @@ test('should reset tracing', async ({ reusedContext, trace }, testInfo) => {
   expect(error.message).toContain('Must start tracing before stopping');
 });
 
+test('should work with clock emulation', async ({ reusedContext, trace }, testInfo) => {
+  let context = await reusedContext();
+
+  let page = await context.newPage();
+  await page.clock.setFixedTime(new Date('2020-01-01T00:00:00.000Z'));
+  expect(await page.evaluate('new Date().toISOString()')).toBe('2020-01-01T00:00:00.000Z');
+
+  context = await reusedContext();
+  page = context.pages()[0];
+  await page.clock.setFixedTime(new Date('2020-01-01T00:00:00Z'));
+  expect(await page.evaluate('new Date().toISOString()')).toBe('2020-01-01T00:00:00.000Z');
+});
+
 test('should continue issuing events after closing the reused page', async ({ reusedContext, server }) => {
   test.info().annotations.push({ type: 'issue', description: 'https://github.com/microsoft/playwright/issues/24574' });