From 8d250e99e75850769b4e16a700cfa55efcbbcf4f Mon Sep 17 00:00:00 2001
From: Benjamin Pasero <benjamin.pasero@microsoft.com>
Date: Fri, 26 Nov 2021 11:00:11 +0100
Subject: [PATCH] smoke - wait for tab becoming dirty before exit

---
 test/smoke/src/areas/workbench/data-migration.test.ts | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/test/smoke/src/areas/workbench/data-migration.test.ts b/test/smoke/src/areas/workbench/data-migration.test.ts
index a14bb1da5c8..45d85172796 100644
--- a/test/smoke/src/areas/workbench/data-migration.test.ts
+++ b/test/smoke/src/areas/workbench/data-migration.test.ts
@@ -47,11 +47,13 @@ export function setup(opts: ParsedArgs, testDataPath: string) {
 			const untitled = 'Untitled-1';
 			const textToTypeInUntitled = 'Hello from Untitled';
 			await app.workbench.editor.waitForTypeInEditor(untitled, textToTypeInUntitled);
+			await app.workbench.editors.waitForTab(untitled, true);
 
 			const readmeMd = 'readme.md';
 			const textToType = 'Hello, Code';
 			await app.workbench.quickaccess.openFile(readmeMd);
 			await app.workbench.editor.waitForTypeInEditor(readmeMd, textToType);
+			await app.workbench.editors.waitForTab(readmeMd, true);
 
 			await app.restart();
 
@@ -146,11 +148,13 @@ export function setup(opts: ParsedArgs, testDataPath: string) {
 			const untitled = 'Untitled-1';
 			const textToTypeInUntitled = 'Hello from Untitled';
 			await stableApp.workbench.editor.waitForTypeInEditor(untitled, textToTypeInUntitled);
+			await stableApp.workbench.editors.waitForTab(untitled, true);
 
 			const readmeMd = 'readme.md';
 			const textToType = 'Hello, Code';
 			await stableApp.workbench.quickaccess.openFile(readmeMd);
 			await stableApp.workbench.editor.waitForTypeInEditor(readmeMd, textToType);
+			await stableApp.workbench.editors.waitForTab(readmeMd, true);
 
 			await stableApp.stop();
 			stableApp = undefined;