Fixes a number of tests given new design, plus adds new one

Author: WesCossick

test/puppeteer/demo.test.ts

@@ -28,18 +28,27 @@ it('focuses on the first menu item when the enter key is pressed', async () => {
 	expect(await currentFocusID()).toBe('menu-item-1');
 });
 
-it('focuses on the menu button after pressing escape', async () => {
+it('leaves focus on the button if the button is clicked', async () => {
 	await page.click('#menu-button');
 	await menuOpen();
 
+	expect(await currentFocusID()).toBe('menu-button');
+});
+
+it('focuses on the menu button after pressing escape', async () => {
+	await page.focus('#menu-button');
+	await keyboard.down('Enter');
+	await menuOpen();
+
 	await keyboard.down('Escape');
 	await menuClosed();
 
 	expect(await currentFocusID()).toBe('menu-button');
 });
 
 it('focuses on the next item in the tab order after pressing tab', async () => {
-	await page.click('#menu-button');
+	await page.focus('#menu-button');
+	await keyboard.down('Enter');
 	await menuOpen();
 
 	await keyboard.down('Tab');
@@ -49,7 +58,8 @@ it('focuses on the next item in the tab order after pressing tab', async () => {
 });
 
 it('focuses on the previous item in the tab order after pressing shift-tab', async () => {
-	await page.click('#menu-button');
+	await page.focus('#menu-button');
+	await keyboard.down('Enter');
 	await menuOpen();
 
 	await keyboard.down('Shift');
@@ -60,7 +70,8 @@ it('focuses on the previous item in the tab order after pressing shift-tab', asy
 });
 
 it('closes the menu if you click outside of it', async () => {
-	await page.click('#menu-button');
+	await page.focus('#menu-button');
+	await keyboard.down('Enter');
 	await menuOpen();
 
 	await page.click('body');
@@ -70,7 +81,8 @@ it('closes the menu if you click outside of it', async () => {
 });
 
 it('leaves the menu open if you click inside of it', async () => {
-	await page.click('#menu-button');
+	await page.focus('#menu-button');
+	await keyboard.down('Enter');
 	await menuOpen();
 
 	await page.click('#menu-item-1');
@@ -87,7 +99,8 @@ it('leaves the menu open if you click inside of it', async () => {
 it('reroutes enter presses on menu items as clicks', async () => {
 	let alertAppeared = false;
 
-	await page.click('#menu-button');
+	await page.focus('#menu-button');
+	await keyboard.down('Enter');
 	await menuOpen();
 
 	// eslint-disable-next-line @typescript-eslint/no-misused-promises