Address flaky TestAppOfflineDebounceTime test (Azure#10128)

* Address flaky test * Fix callback
amih90 · May 30, 2024 · 3d401d3 · 3d401d3
1 parent 80dd177
commit 3d401d3
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/test/WebJobs.Script.Tests/FileMonitoringServiceTests.cs b/test/WebJobs.Script.Tests/FileMonitoringServiceTests.cs
@@ -300,8 +300,15 @@ public static async Task TestAppOfflineDebounceTime(string fileName, int delayIn
                     WatchFiles = { "host.json" }
                 };
                 var loggerFactory = new LoggerFactory();
+
+                TaskCompletionSource stop = new TaskCompletionSource();
                 var mockApplicationLifetime = new Mock<IApplicationLifetime>();
+                mockApplicationLifetime.Setup(m => m.StopApplication()).Callback(() => stop.TrySetResult());
+
+                TaskCompletionSource restart = new TaskCompletionSource();
                 var mockScriptHostManager = new Mock<IScriptHostManager>();
+                mockScriptHostManager.Setup(m => m.RestartHostAsync(default)).Callback(() => restart.TrySetResult());
+
                 var mockEventManager = new ScriptEventManager();
                 var environment = new TestEnvironment();
 
@@ -311,18 +318,19 @@ public static async Task TestAppOfflineDebounceTime(string fileName, int delayIn
                 await fileMonitoringService.StartAsync(new CancellationToken(canceled: false));
 
                 var offlineEventArgs = new FileSystemEventArgs(WatcherChangeTypes.Created, tempDir, fileName);
-                FileEvent offlinefileEvent = new FileEvent("ScriptFiles", offlineEventArgs);
+                FileEvent offlineFileEvent = new FileEvent("ScriptFiles", offlineEventArgs);
 
                 var randomFileEventArgs = new FileSystemEventArgs(WatcherChangeTypes.Created, tempDir, "random.txt");
                 FileEvent randomFileEvent = new FileEvent("ScriptFiles", randomFileEventArgs);
 
-                mockEventManager.Publish(offlinefileEvent);
+                mockEventManager.Publish(offlineFileEvent);
                 await Task.Delay(delayInMs);
                 mockEventManager.Publish(randomFileEvent);
 
                 // Test
                 if (expectShutdown)
                 {
+                    await stop.Task.WaitAsync(TimeSpan.FromSeconds(5));
                     mockApplicationLifetime.Verify(m => m.StopApplication());
                 }
                 else
@@ -332,6 +340,7 @@ public static async Task TestAppOfflineDebounceTime(string fileName, int delayIn
 
                 if (expectRestart)
                 {
+                    await restart.Task.WaitAsync(TimeSpan.FromSeconds(5));
                     mockScriptHostManager.Verify(m => m.RestartHostAsync(default));
                 }
                 else