From dbf139862ff33add2a007704ee3b91c164f9cf56 Mon Sep 17 00:00:00 2001
From: Olga Bulat <obulat@gmail.com>
Date: Tue, 13 Dec 2022 16:57:48 +0300
Subject: [PATCH] Use cookies code from the Playwright PR

---
 test/playwright/utils/navigation.ts              | 16 +++++++++++++++-
 .../pages/pages-with-old-header.spec.ts          | 16 +++++++---------
 2 files changed, 22 insertions(+), 10 deletions(-)

diff --git a/test/playwright/utils/navigation.ts b/test/playwright/utils/navigation.ts
index 16d505eb4e..8889a843d7 100644
--- a/test/playwright/utils/navigation.ts
+++ b/test/playwright/utils/navigation.ts
@@ -1,4 +1,4 @@
-import { expect, Page } from '@playwright/test'
+import { BrowserContext, expect, Page } from '@playwright/test'
 
 import rtlMessages from '~~/test/locales/ar.json'
 
@@ -458,3 +458,17 @@ export const enableNewHeader = async (page: Page) => {
     },
   ])
 }
+
+export const setCookies = async (
+  context: BrowserContext,
+  cookies: Record<string, string | boolean | string[]>
+) => {
+  await context.addCookies(
+    Object.entries(cookies).map(([name, value]) => ({
+      name,
+      value: JSON.stringify(value),
+      domain: 'localhost',
+      path: '/',
+    }))
+  )
+}
diff --git a/test/playwright/visual-regression/pages/pages-with-old-header.spec.ts b/test/playwright/visual-regression/pages/pages-with-old-header.spec.ts
index 1ed0b5ae63..2830ed6005 100644
--- a/test/playwright/visual-regression/pages/pages-with-old-header.spec.ts
+++ b/test/playwright/visual-regression/pages/pages-with-old-header.spec.ts
@@ -6,6 +6,7 @@ import {
   dismissTranslationBanner,
   pathWithDir,
   languageDirections,
+  setCookies,
 } from '~~/test/playwright/utils/navigation'
 
 test.describe.configure({ mode: 'parallel' })
@@ -15,15 +16,12 @@ for (const contentPage of contentPages) {
   for (const dir of languageDirections) {
     test.describe(`${contentPage} ${dir} page snapshots`, () => {
       breakpoints.describeEvery(({ breakpoint, expectSnapshot }) => {
-        test('full page', async ({ page }) => {
-          await page.context().addCookies([
-            {
-              name: 'uiBreakpoint',
-              value: JSON.stringify(breakpoint),
-              domain: 'localhost',
-              path: '/',
-            },
-          ])
+        test('full page', async ({ context, page }) => {
+          await setCookies(context, {
+            uiBreakpoint: breakpoint,
+            uiDismissedBanners: ['translation-ar'],
+          })
+
           await page.goto(pathWithDir(contentPage, dir))
           await dismissTranslationBanner(page)