update test

cognitedata · Aug 17, 2023 · 17a067d · 17a067d
1 parent e6425a9
commit 17a067d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/test/scala/cognite/spark/v1/AssetHierarchyBuilderTest.scala b/src/test/scala/cognite/spark/v1/AssetHierarchyBuilderTest.scala
@@ -1110,10 +1110,11 @@ class AssetHierarchyBuilderTest
     ingest(key, grandkids, batchSize = 1000, metricsPrefix = Some(s"ingest.tree.grandkids.$key"))
 
     getNumberOfRowsCreated(s"ingest.tree.grandkids.$key", "assethierarchy") shouldBe 1100
+    // 1 for access token
     // 2 for fetching root parents (for validation)
     // 2 for fetching the roots
     // 2 for creating the roots
-    getNumberOfRequests(s"ingest.tree.grandkids.$key") shouldBe 6
+    getNumberOfRequests(s"ingest.tree.grandkids.$key") shouldBe 7
 
     val grandkidsUpdate = (1 to 1100).map(
       k =>
@@ -1129,10 +1130,11 @@ class AssetHierarchyBuilderTest
     ingest(key, grandkidsUpdate, batchSize = 1000, metricsPrefix = Some(s"ingest.tree.grandkidsU.$key"))
 
     getNumberOfRowsUpdated(s"ingest.tree.grandkidsU.$key", "assethierarchy") shouldBe 1100
+    // 1 for access token
     // 2 for fetching root parents (for validation)
     // 2 for fetching the roots
     // 2 for updating the roots
-    getNumberOfRequests(s"ingest.tree.grandkidsU.$key") shouldBe 6
+    getNumberOfRequests(s"ingest.tree.grandkidsU.$key") shouldBe 7
 
     cleanDB(key)
   }