From ca3602bc7fe972770228be9aa96974e5bffacce9 Mon Sep 17 00:00:00 2001
From: Shrushti Polekar <shrushtipolekar@gmail.com>
Date: Thu, 11 Sep 2025 09:49:21 +0530
Subject: [PATCH] fix test (#22928)

---
 .../Permissions/DomainPermissions.spec.ts        | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/openmetadata-ui/src/main/resources/ui/playwright/e2e/Features/Permissions/DomainPermissions.spec.ts b/openmetadata-ui/src/main/resources/ui/playwright/e2e/Features/Permissions/DomainPermissions.spec.ts
index d54f73795e9..cc23a240e7d 100644
--- a/openmetadata-ui/src/main/resources/ui/playwright/e2e/Features/Permissions/DomainPermissions.spec.ts
+++ b/openmetadata-ui/src/main/resources/ui/playwright/e2e/Features/Permissions/DomainPermissions.spec.ts
@@ -114,7 +114,6 @@ test('Domain allow operations', async ({ testUserPage, browser }) => {
   // Test that domain operation elements are visible
   const directElements = [
     'edit-description',
-    'add-owner',
     'add-tag',
     'edit-icon-right-panel',
     'add-domain',
@@ -139,6 +138,13 @@ test('Domain allow operations', async ({ testUserPage, browser }) => {
     await expect(element).toBeVisible();
   }
 
+  const ownerButton = testUserPage
+    .getByTestId('add-owner')
+    .or(testUserPage.getByTestId('edit-owner'))
+    .first();
+
+  await expect(ownerButton).toBeVisible();
+
   // Click manage button once and test elements inside it
   const manageButton = testUserPage.getByTestId('manage-button');
 
@@ -180,7 +186,7 @@ test('Domain deny operations', async ({ testUserPage, browser }) => {
   // Test that domain operation elements are visible
   const directElements = [
     'edit-description',
-    'add-owner',
+
     'add-tag',
     'edit-icon-right-panel',
     'add-domain',
@@ -203,6 +209,12 @@ test('Domain deny operations', async ({ testUserPage, browser }) => {
 
     await expect(element).not.toBeVisible();
   }
+  const ownerButton = testUserPage
+    .getByTestId('add-owner')
+    .or(testUserPage.getByTestId('edit-owner'))
+    .first();
+
+  await expect(ownerButton).not.toBeVisible();
 
   // Click manage button once and test elements inside it
   const manageButton = testUserPage.getByTestId('manage-button');