resloved conflict

PalisadoesFoundation · Dec 20, 2024 · 87505d5 · 87505d5
1 parent cbb63c2
commit 87505d5
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/src/components/OrganizationScreen/OrganizationScreen.test.tsx b/src/components/OrganizationScreen/OrganizationScreen.test.tsx
@@ -80,15 +80,13 @@ describe('Testing OrganizationScreen', () => {
     const closeButton = screen.getByTestId('closeMenu');
     fireEvent.click(closeButton);
 
-    // Check for a class indicating the drawer is contracted (use stringContaining in the class)
-    const mainPageElement = screen.getByTestId('mainpageright');
-    expect(mainPageElement.className).toMatch('_expand'); // Simply check if _expand exists
+    expect(screen.getByTestId('mainpageright')).toHaveClass(styles.expand);
 
     const openButton = screen.getByTestId('openMenu');
     fireEvent.click(openButton);
 
-    // Check for a class indicating the drawer is expanded
-    expect(mainPageElement.className).toMatch('_contract');
+    // Check for expand class after opening
+    expect(screen.getByTestId('mainpageright')).toHaveClass(styles.contract);
   });
 
   test('handles window resize', () => {