Adding invocation log E2E test

Author: fabiocav

test/WebJobs.Script.Tests.E2E/FunctionAppFixture.cs

@@ -36,6 +36,8 @@ public FunctionAppFixture()
 
         public TelemetryClient Telemetry => TelemetryContext.Client;
 
+        public KuduClient KuduClient => _kuduClient;
+
         public string FunctionDefaultKey { get; }
 
         public string FunctionAppMasterKey { get; private set; }

test/WebJobs.Script.Tests.E2E/Functions/wwwroot/GetInvocationId/function.json

@@ -0,0 +1,16 @@
+{
+    "bindings": [
+        {
+            "authLevel": "function",
+            "name": "req",
+            "type": "httpTrigger",
+            "direction": "in"
+        },
+        {
+            "name": "$return",
+            "type": "http",
+            "direction": "out"
+        }
+    ],
+    "disabled": false
+}

test/WebJobs.Script.Tests.E2E/Functions/wwwroot/GetInvocationId/run.csx

@@ -0,0 +1 @@
+public static string Run(HttpRequestMessage req, ExecutionContext context) => context.InvocationId.ToString();

test/WebJobs.Script.Tests.E2E/GeneralEndToEndTests.cs

@@ -3,11 +3,14 @@
 
 using System;
 using System.Collections.Generic;
+using System.Globalization;
 using System.Net.Http;
 using System.Net.Http.Headers;
+using System.Text;
 using System.Threading.Tasks;
 using Microsoft.ApplicationInsights;
 using Microsoft.ApplicationInsights.DataContracts;
+using Newtonsoft.Json.Linq;
 using Xunit;
 
 namespace WebJobs.Script.EndToEndTests
@@ -107,6 +110,85 @@ public async Task ParallelRequests_ReturnExpectedResults()
             }
         }
 
+        [Fact]
+        [TestTrace]
+        public async Task Invocation_Logs_AreReturned()
+        {
+            using (var client = CreateClient())
+            {
+                client.DefaultRequestHeaders.Accept.Add(new MediaTypeWithQualityHeaderValue("text/plain"));
+
+                string invocationId = await client.GetStringAsync($"api/GetInvocationId?code={_fixture.FunctionDefaultKey}");
+
+                string siteName = NormalizeFunctionName(Settings.SiteName);
+                JToken resultToken = null;
+                // We retry for a bit as data may take a while to become available
+                for (int i = 0; i < 20; i++)
+                {
+                    var invocationsRequest = new HttpRequestMessage(HttpMethod.Get, $"azurejobs/api/functions/definitions/{siteName}-GetInvocationId/invocations?limit=10");
+                    var response = await _fixture.KuduClient.SendAsync(invocationsRequest);
+
+                    var results = await response.Content.ReadAsAsync<JObject>();
+
+                    resultToken = results.SelectToken($"$..entries[?(@.id == '{invocationId}')]");
+
+                    if (resultToken != null)
+                    {
+                        break;
+                    }
+                    
+                    await Task.Delay(3000);
+                }
+
+                _fixture.Assert.True(resultToken != null);
+            }
+        }
+
+        // Assumes we have a valid function name.
+        // Function names are case-insensitive, case-preserving. 
+        // Table storage is case-sensitive. So need to normalize case to use as table keys. 
+        // Normalize must be one-to-one to avoid collisions. 
+        // Escape any non-alphanumeric characters so that we 
+        //  a) have a valid rowkey name 
+        //  b) don't have characeters that conflict with separators in the row key (like '-')
+        public static string NormalizeFunctionName(string functionName)
+        {
+            var sb = new StringBuilder();
+            foreach (var ch in functionName)
+            {
+                if (ch >= 'a' && ch <= 'z')
+                {
+                    sb.Append(ch);
+                }
+                else if (ch >= 'A' && ch <= 'Z')
+                {
+                    sb.Append((char)(ch - 'A' + 'a'));
+                }
+                else if (ch >= '0' && ch <= '9')
+                {
+                    sb.Append(ch);
+                }
+                else
+                {
+                    sb.Append(EscapeStorageCharacter(ch));
+                }
+            }
+            return sb.ToString();
+        }
+
+        private static string EscapeStorageCharacter(char character)
+        {
+            var ordinalValue = (ushort)character;
+            if (ordinalValue < 0x100)
+            {
+                return string.Format(CultureInfo.InvariantCulture, ":{0:X2}", ordinalValue);
+            }
+            else
+            {
+                return string.Format(CultureInfo.InvariantCulture, "::{0:X4}", ordinalValue);
+            }
+        }
+
 
         public HttpClient CreateClient()
         {

test/WebJobs.Script.Tests.E2E/KuduClient.cs

@@ -59,6 +59,8 @@ public async Task<string> GetFunctionsMasterKey()
             return result.GetValue("masterKey").ToString();
         }
 
+        public Task<HttpResponseMessage> SendAsync(HttpRequestMessage request) => _client.SendAsync(request);        
+
         protected virtual void Dispose(bool disposing)
         {
             if (!_disposed)

test/WebJobs.Script.Tests.E2E/WebJobs.Script.Tests.E2E.csproj

@@ -100,6 +100,12 @@
   </ItemGroup>
   <ItemGroup>
     <None Include="app.config" />
+    <None Include="Functions\wwwroot\GetInvocationId\function.json">
+      <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+    </None>
+    <None Include="Functions\wwwroot\GetInvocationId\run.csx">
+      <CopyToOutputDirectory>Always</CopyToOutputDirectory>
+    </None>
     <None Include="Functions\wwwroot\host.json">
       <CopyToOutputDirectory>Always</CopyToOutputDirectory>
     </None>