Adds focus management test for entity undo (#49236)

* Adds focus management test for entity undo

* removes the extra user story group and adds the test as a case to the existing template part grouping

* undo edit in grouping name

* Adds some comments and a small readability fix

Co-authored-by: Dave Smith <[email protected]>
Co-authored-by: Kai Hao <[email protected]>

---------

Co-authored-by: Dave Smith <[email protected]>
Co-authored-by: Kai Hao <[email protected]>

test/e2e/specs/site-editor/template-part.spec.js

@@ -356,4 +356,52 @@ test.describe( 'Template Part', () => {
 			page.getByRole( 'combobox', { name: 'Import widget area' } )
 		).not.toBeVisible();
 	} );
+
+	test( 'Keeps focus in place on undo in template parts', async ( {
+		admin,
+		editor,
+		page,
+		pageUtils,
+	} ) => {
+		await admin.visitSiteEditor( {
+			postId: 'emptytheme//header',
+			postType: 'wp_template_part',
+		} );
+		await editor.canvas.click( 'body' );
+
+		// Select the site title block.
+		const siteTitle = editor.canvas.getByRole( 'document', {
+			name: 'Site title',
+		} );
+		await editor.selectBlocks( siteTitle );
+
+		// Remove the default site title block.
+		await pageUtils.pressKeys( 'access+z' );
+
+		// Insert a group block with a Site Title block inside.
+		await editor.insertBlock( {
+			name: 'core/group',
+			innerBlocks: [ { name: 'core/site-title' } ],
+		} );
+
+		// Select the Site Title block inside the group.
+		const siteTitleInGroup = editor.canvas.getByRole( 'document', {
+			name: 'Site title',
+		} );
+		await editor.selectBlocks( siteTitleInGroup );
+
+		// Change heading level of the Site Title block.
+		await editor.clickBlockToolbarButton( 'Change heading level' );
+		const Heading3Button = page.getByRole( 'menuitemradio', {
+			name: 'Heading 3',
+		} );
+		await Heading3Button.click();
+
+		// Undo the change.
+		await pageUtils.pressKeys( 'primary+z' );
+
+		await expect(
+			page.locator( 'role=button[name="Change heading level"i]' )
+		).toBeFocused();
+	} );
 } );