update selector and add condition (#56441)

Automattic · Sep 23, 2021 · ba7eb23 · ba7eb23
1 parent f60739b
commit ba7eb23
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/test/e2e/lib/components/site-editor-component.js b/test/e2e/lib/components/site-editor-component.js
@@ -21,10 +21,14 @@ export default class SiteEditorComponent extends AbstractEditorComponent {
 	}
 
 	async waitForTemplatePartsToLoad() {
-		await this.runInCanvas( async () => {
+		return await this.runInCanvas( async () => {
+			await driverHelper.waitUntilElementNotLocated(
+				this.driver,
+				By.css( '.wp-block-template-part > .components-spinner' )
+			);
 			await driverHelper.waitUntilElementNotLocated(
 				this.driver,
-				By.css( '.wp-block-template-part .components-spinner' )
+				By.css( '.wp-block-template-part > .block-editor-warning' )
 			);
 		} );
 	}