Added exact count for TC mock call

Azure · Jan 10, 2024 · 8890c24 · 8890c24
1 parent e679939
commit 8890c24
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/connector/src/test/scala/com/microsoft/kusto/spark/utils/ContainerProviderTest.scala b/connector/src/test/scala/com/microsoft/kusto/spark/utils/ContainerProviderTest.scala
@@ -37,8 +37,8 @@ class ContainerProviderTest extends AnyFlatSpec with Matchers with MockFactory {
           thenAnswer(_ => List(getMockContainerWithSas(1), getMockContainerWithSas(2)).asJava)
       }
     }
-    // noMoreThanTwice()
-    mockIngestClient.getResourceManager _ expects() atLeastOnce() returning mockIngestionResourceManager
+    // Expecting getResourceManager to be called maxCommandsRetryAttempts i.e. 8 times.
+    mockIngestClient.getResourceManager _ expects() repeated 8 times() returning mockIngestionResourceManager
     // Unfortunately we cannot Mock this class as there is a member variable that is a val and cannot be mocked
     new ExtendedKustoClient(new ConnectionStringBuilder("https://somecluster.eastus.kusto.windows.net/"),
       new ConnectionStringBuilder("https://ingest-somecluster.eastus.kusto.windows.net"), "somecluster") {