Introduce delay in event route lifecycle test. (#22752)

Azure · Jul 20, 2021 · 132760c · 132760c
1 parent 5762543
commit 132760c
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/sdk/digitaltwins/Azure.DigitalTwins.Core/tests/E2eTestBase.cs b/sdk/digitaltwins/Azure.DigitalTwins.Core/tests/E2eTestBase.cs
@@ -107,5 +107,17 @@ protected static async Task CreateAndListModelsAsync(DigitalTwinsClient client,
                 Console.WriteLine($"{model.Id}");
             }
         }
+
+        /// <summary>
+        /// Injects delay in the process when the test mode is live.
+        /// </summary>
+        /// <param name="delayDuration">Delay duration.</param>
+        protected async Task WaitIfLiveAsync(TimeSpan delayDuration)
+        {
+            if (TestEnvironment.Mode == RecordedTestMode.Live)
+            {
+                await Task.Delay(delayDuration);
+            }
+        }
     }
 }
diff --git a/sdk/digitaltwins/Azure.DigitalTwins.Core/tests/EventRouteTests.cs b/sdk/digitaltwins/Azure.DigitalTwins.Core/tests/EventRouteTests.cs
@@ -21,6 +21,7 @@ public EventRouteTests(bool isAsync)
 
         // Infrastructure setup script uses this hard-coded value when linking the test eventhub to the test digital twins instance
         private const string EndpointName = "someEventHubEndpoint";
+        private readonly TimeSpan _creationDelay = TimeSpan.FromSeconds(10);
 
         [Test]
         public async Task EventRoutes_Lifecycle()
@@ -38,6 +39,10 @@ public async Task EventRoutes_Lifecycle()
             Response createEventRouteResponse = await client.CreateOrReplaceEventRouteAsync(eventRouteId, eventRoute).ConfigureAwait(false);
             createEventRouteResponse.Status.Should().Be((int)HttpStatusCode.NoContent);
 
+            // Wait certain amount of time to test if the issue is with propagation.
+            // TODO: azabbasi: remove this logic once experiment is over.
+            await WaitIfLiveAsync(_creationDelay);
+
             // Test GetEventRoute
             DigitalTwinsEventRoute getEventRouteResult = await client.GetEventRouteAsync(eventRouteId);
             eventRoute.EndpointName.Should().Be(getEventRouteResult.EndpointName);