Fix test case for logStreaming

projectkudu · Mar 9, 2016 · 495dc90 · 495dc90
1 parent 8a0983d
commit 495dc90
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/Kudu.FunctionalTests/LogStreamManagerTests.cs b/Kudu.FunctionalTests/LogStreamManagerTests.cs
@@ -120,8 +120,11 @@ public void TestLogStreamNotFound()
             ApplicationManager.Run(appName, appManager =>
             {
                 RemoteLogStreamManager manager = new RemoteLogStreamManager(appManager.ServiceUrl + "/logstream/notfound");
-                var ex = KuduAssert.ThrowsUnwrapped<WebException>(() => manager.GetStream().Wait());
-                Assert.Equal(((HttpWebResponse)ex.Response).StatusCode, HttpStatusCode.NotFound);
+                using (var waitHandle = new LogStreamWaitHandle(manager.GetStream().Result))
+                {
+                    string line = waitHandle.WaitNextLine(10000);
+                    Assert.True(!String.IsNullOrEmpty(line) && line.Contains("Welcome"), "check welcome message: " + line);
+                }
             });
         }
 

diff --git a/Kudu.Services/Diagnostics/LogStreamManager.cs b/Kudu.Services/Diagnostics/LogStreamManager.cs
@@ -276,6 +276,12 @@ private string ParseRequest(HttpContext context)
                 return _logPath;
             }
 
+            var firstPath = routePath.Split(new char[] { '/' }, StringSplitOptions.RemoveEmptyEntries).FirstOrDefault();
+            if (string.Equals(firstPath, "Application", StringComparison.OrdinalIgnoreCase))
+            {
+                _enableTrace = true;
+            }
+
             return FileSystemHelpers.EnsureDirectory(Path.Combine(_logPath, routePath));
         }