Fixing flaky test

Azure · Apr 13, 2016 · 44b6571 · 44b6571
1 parent 70b86ab
commit 44b6571
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/test/WebJobs.Script.Tests/FileTraceWriterTests.cs b/test/WebJobs.Script.Tests/FileTraceWriterTests.cs
@@ -49,7 +49,7 @@ await Task.WhenAll(
         }
 
         [Fact]
-        public void SetNewLogFile_PurgesOldLogFiles()
+        public async Task SetNewLogFile_PurgesOldLogFiles()
         {
             DirectoryInfo directory = new DirectoryInfo(_logFilePath);
             directory.Create();
@@ -83,9 +83,14 @@ public void SetNewLogFile_PurgesOldLogFiles()
 
             files = directory.GetFiles().OrderByDescending(p => p.LastWriteTime).ToArray();
 
+            await TestHelpers.Await(() =>
+            {
+                files = directory.GetFiles().OrderByDescending(p => p.LastWriteTime).ToArray();
+                return files.Length == 2;
+            }, timeout: 2000);
+
             // verify the correct log files were purged and the 2
             // most recent files were retained
-            Assert.Equal(2, files.Length);
             Assert.True(files[0].Name.StartsWith("4"));
             Assert.True(files[1].Name.StartsWith("3"));
         }