From 7bf0153f06bd6dbc070adaa5aa79f59d0ad07a9a Mon Sep 17 00:00:00 2001
From: Jonas <100586436+JonasDov@users.noreply.github.com>
Date: Thu, 13 Feb 2025 17:33:54 +0200
Subject: [PATCH] Adjust e2e tests

---
 .../itwin/tree-widget/src/e2e-tests/CategoriesTree.test.ts   | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/packages/itwin/tree-widget/src/e2e-tests/CategoriesTree.test.ts b/packages/itwin/tree-widget/src/e2e-tests/CategoriesTree.test.ts
index a143c08dd..b2bad0da0 100644
--- a/packages/itwin/tree-widget/src/e2e-tests/CategoriesTree.test.ts
+++ b/packages/itwin/tree-widget/src/e2e-tests/CategoriesTree.test.ts
@@ -107,11 +107,14 @@ test.describe("Categories tree", () => {
       await node.click();
       const treeContainer = page.locator("#tw-tree-renderer-container");
 
+      // ensure checkbox is not disabled
+      const checkbox = node.getByRole("checkbox");
+      await expect(checkbox).toBeEnabled();
+
       // focus on checkbox using keyboard
       await page.keyboard.press("Tab");
 
       // ensure checkbox is focused
-      const checkbox = node.getByRole("checkbox");
       await expect(checkbox).toBeFocused();
 
       await takeScreenshot(page, node, { boundingComponent: treeContainer, expandBy: { top: 10, bottom: 10 } });