Add additional assertions for focus

WordPress · Dec 7, 2023 · cc7f225 · cc7f225
1 parent 220e01f
commit cc7f225
Showing 1 changed file with 22 additions and 2 deletions.
diff --git a/packages/components/src/custom-select-control/test/index.js b/packages/components/src/custom-select-control/test/index.js
@@ -278,6 +278,12 @@ describe.each( [
 			expect( currentSelectedItem ).toHaveFocus();
 
 			await user.keyboard( '{enter}' );
+			expect(
+				screen.getByRole( 'listbox', {
+					name: 'label!',
+				} )
+			).toHaveFocus();
+
 			await user.keyboard( '{arrowdown}' );
 			await user.keyboard( '{enter}' );
 
@@ -295,9 +301,14 @@ describe.each( [
 
 			await user.tab();
 			await user.keyboard( '{enter}' );
+			expect(
+				screen.getByRole( 'listbox', {
+					name: 'label!',
+				} )
+			).toHaveFocus();
+
 			await user.keyboard( '{a}' );
 			await user.keyboard( '{enter}' );
-
 			expect( currentSelectedItem ).toHaveTextContent( 'amber' );
 		} );
 
@@ -311,10 +322,19 @@ describe.each( [
 			} );
 
 			await user.tab();
+			expect( currentSelectedItem ).toHaveFocus();
+
 			await user.keyboard( '{a}' );
 			await user.keyboard( '{q}' );
-			await user.keyboard( '{enter}' );
 
+			expect(
+				screen.queryByRole( 'listbox', {
+					name: 'label!',
+					hidden: true,
+				} )
+			).not.toBeInTheDocument();
+
+			await user.keyboard( '{enter}' );
 			expect( currentSelectedItem ).toHaveTextContent( 'aquamarine' );
 		} );