Fix Test (#20791)

Azure · May 3, 2021 · 125a9f8 · 125a9f8
1 parent 2ab5507
commit 125a9f8
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/...nsights/Azure.IoT.TimeSeriesInsights/tests/TimeSeriesInsightsQueryAggregateSeriesTests.cs b/...nsights/Azure.IoT.TimeSeriesInsights/tests/TimeSeriesInsightsQueryAggregateSeriesTests.cs
@@ -239,16 +239,14 @@ await TestRetryHelper.RetryAsync<AsyncPageable<QueryResultPage>>(async () =>
                         TimeSpan.FromSeconds(5),
                         queryAggregateSeriesRequestOptions);
 
-                    var countFound = false;
+                    long? totalCount = 0;
                     await foreach (TimeSeriesPoint point in queryAggregateSeriesPages.GetResultsAsync())
                     {
-                        if ((long?)point.GetValue("Count") == 30)
-                        {
-                            countFound = true;
-                        }
+                        var currentCount = (long?)point.GetValue("Count");
+                        totalCount += currentCount;
                     }
 
-                    countFound.Should().BeTrue();
+                    totalCount.Should().Be(30);
 
                     return null;
                 }, MaxNumberOfRetries, s_retryDelay);