From 44c22d115baa4ab9f00ecbd22d3fe1575287c636 Mon Sep 17 00:00:00 2001
From: loks0n <22452787+loks0n@users.noreply.github.com>
Date: Mon, 2 Jun 2025 21:11:07 +0100
Subject: [PATCH] tests: fix more flaky

---
 tests/e2e/Services/Account/AccountCustomClientTest.php | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/e2e/Services/Account/AccountCustomClientTest.php b/tests/e2e/Services/Account/AccountCustomClientTest.php
index 0cc2eb893a..cee23d5188 100644
--- a/tests/e2e/Services/Account/AccountCustomClientTest.php
+++ b/tests/e2e/Services/Account/AccountCustomClientTest.php
@@ -1864,6 +1864,9 @@ class AccountCustomClientTest extends Scope
             'failure' => 'http://localhost/v1/mock/tests/general/oauth2/failure',
         ]);
 
+        $this->assertArrayHasKey('cookies', $response, 'Failed to assert response has cookies, response: ', json_encode($response, JSON_PRETTY_PRINT));
+        $this->assertArrayHasKey('a_session_' . $this->getProject()['$id'], $response['cookies'], 'Failed to assert response had session cookie: ' . json_encode($response, JSON_PRETTY_PRINT));
+
         $session = $response['cookies']['a_session_' . $this->getProject()['$id']];
 
         $this->assertEquals(200, $response['headers']['status-code']);