From df7d4038ac717f24eef52b39127b2dd32a7342ce Mon Sep 17 00:00:00 2001
From: Bart Kalisz <bartlomiej.kalisz@gmail.com>
Date: Thu, 3 Oct 2024 16:40:22 +0200
Subject: [PATCH] Add db connection retry

---
 packages/e2e-test-utils-playwright/src/test.ts | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/packages/e2e-test-utils-playwright/src/test.ts b/packages/e2e-test-utils-playwright/src/test.ts
index 6ef614024d7d9..f7a6a13f6f528 100644
--- a/packages/e2e-test-utils-playwright/src/test.ts
+++ b/packages/e2e-test-utils-playwright/src/test.ts
@@ -144,6 +144,24 @@ const test = base.extend<
 	page: async ( { page }, use ) => {
 		page.on( 'console', observeConsoleLogging );
 
+		await expect
+			.poll(
+				async () => {
+					const response = await page.request.get( '/' );
+					const html = await response.text();
+
+					return html.includes(
+						'Error establishing a database connection'
+					);
+				},
+				{
+					message: 'ensure database is connected',
+					timeout: 10000,
+					intervals: [ 2000 ],
+				}
+			)
+			.toBeFalsy();
+
 		await use( page );
 
 		// Clear local storage after each test.