Fix tests to adopt new override

microsoft · Dec 4, 2018 · c4e964f · c4e964f
1 parent 8903fd1
commit c4e964f
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/tests/Microsoft.Bot.Builder.ApplicationInsights.Tests/TelemetryWaterfallTests.cs b/tests/Microsoft.Bot.Builder.ApplicationInsights.Tests/TelemetryWaterfallTests.cs
@@ -32,8 +32,10 @@ public async Task Waterfall()
                 async (step, cancellationToken) => { await step.Context.SendActivityAsync("step1"); return Dialog.EndOfTurn; },
                 async (step, cancellationToken) => { await step.Context.SendActivityAsync("step2"); return Dialog.EndOfTurn; },
                 async (step, cancellationToken) => { await step.Context.SendActivityAsync("step3"); return Dialog.EndOfTurn; },
-            })
-            { TelemetryClient = telemetryClient.Object });
+            }));
+
+            dialogs.TelemetryClient = telemetryClient.Object;
+
 
             await new TestFlow(adapter, async (turnContext, cancellationToken) =>
             {
@@ -71,10 +73,11 @@ public async Task WaterfallWithCallback()
                     async (step, cancellationToken) => { await step.Context.SendActivityAsync("step1"); return Dialog.EndOfTurn; },
                     async (step, cancellationToken) => { await step.Context.SendActivityAsync("step2"); return Dialog.EndOfTurn; },
                     async (step, cancellationToken) => { await step.Context.SendActivityAsync("step3"); return Dialog.EndOfTurn; },
-            })
-            { TelemetryClient = telemetryClient.Object };
+            });
+
 
             dialogs.Add(waterfallDialog);
+            dialogs.TelemetryClient = telemetryClient.Object;
 
             await new TestFlow(adapter, async (turnContext, cancellationToken) =>
             {
@@ -127,10 +130,10 @@ public async Task EnsureEndDialogCalled()
                     async (step, cancellationToken) => { await step.Context.SendActivityAsync("step1"); return Dialog.EndOfTurn; },
                     async (step, cancellationToken) => { await step.Context.SendActivityAsync("step2"); return Dialog.EndOfTurn; },
                     async (step, cancellationToken) => { await step.Context.SendActivityAsync("step3"); return Dialog.EndOfTurn; },
-            })
-            { TelemetryClient = telemetryClient.Object };
+            });
 
             dialogs.Add(waterfallDialog);
+            dialogs.TelemetryClient = telemetryClient.Object;
 
             await new TestFlow(adapter, async (turnContext, cancellationToken) =>
             {
@@ -190,10 +193,10 @@ public async Task EnsureCancelDialogCalled()
                     async (step, cancellationToken) => { await step.Context.SendActivityAsync("step1"); return Dialog.EndOfTurn; },
                     async (step, cancellationToken) => { await step.Context.SendActivityAsync("step2"); return Dialog.EndOfTurn; },
                     async (step, cancellationToken) => { await step.CancelAllDialogsAsync(); return Dialog.EndOfTurn; },
-            })
-            { TelemetryClient = telemetryClient.Object };
+            });
 
             dialogs.Add(waterfallDialog);
+            dialogs.TelemetryClient = telemetryClient.Object;
 
             await new TestFlow(adapter, async (turnContext, cancellationToken) =>
             {