test with waitUntil

Author: osulzhenko

src/test/kotlin/org/prebid/cache/functional/SecondaryCacheSpec.kt

@@ -34,8 +34,6 @@ class SecondaryCacheSpec : ShouldSpec({
         specPrebidCacheConfig = BaseSpec.prebidCacheConfig.getBaseAerospikeConfig(true) +
                 BaseSpec.prebidCacheConfig.getSecondaryCacheConfig(webCacheContainerUri)
         prebidCacheApi = BaseSpec.getPrebidCacheApi(specPrebidCacheConfig)
-        val requestObject = RequestObject.getDefaultJsonRequestObject().apply { puts[0].key = getRandomUuid() }
-        prebidCacheApi.postCache(requestObject, "no")
     }
 
     afterSpec {
@@ -59,7 +57,7 @@ class SecondaryCacheSpec : ShouldSpec({
 
         // and: Request to secondary cache was sent
         val secondaryCacheRecordedRequests =
-            webCacheContainerClient.getSecondaryCacheRecordedRequests(requestObject.puts[0].key!!)
+            webCacheContainerClient.checkRecordedRequests(requestObject.puts[0].key!!)
         secondaryCacheRecordedRequests?.size shouldBe 1
 
         // and: Request contained secondaryCache=yes query parameter
@@ -89,7 +87,7 @@ class SecondaryCacheSpec : ShouldSpec({
 
         // and: Request to secondary cache was sent
         val secondaryCacheRecordedRequests =
-            webCacheContainerClient.getSecondaryCacheRecordedRequests(requestObject.puts[0].key!!)
+            webCacheContainerClient.checkRecordedRequests(requestObject.puts[0].key!!)
         secondaryCacheRecordedRequests?.size shouldBe 1
 
         // and: Secondary cache request 'expiry' parameter matches to the PBC request 'ttlseconds' parameter
@@ -116,7 +114,7 @@ class SecondaryCacheSpec : ShouldSpec({
 
         // and: Request to secondary cache was sent
         val secondaryCacheRecordedRequests =
-            webCacheContainerClient.getSecondaryCacheRecordedRequests(requestObject.puts[0].key!!)
+            webCacheContainerClient.checkRecordedRequests(requestObject.puts[0].key!!)
         secondaryCacheRecordedRequests?.size shouldBe 1
 
         // and: Secondary cache request 'expiry' parameter matches to the Prebid Cache 'cache.expiry.sec' config property
@@ -146,7 +144,7 @@ class SecondaryCacheSpec : ShouldSpec({
 
         // and: Request to secondary cache was sent
         val secondaryCacheRecordedRequests =
-            webCacheContainerClient.getSecondaryCacheRecordedRequests(requestObject.puts[0].key!!)
+            webCacheContainerClient.checkRecordedRequests(requestObject.puts[0].key!!)
         secondaryCacheRecordedRequests?.size shouldBe 1
 
         // and: Secondary cache request 'expiry' parameter matches to the Prebid Cache 'cache.max.expiry' config property
@@ -176,7 +174,7 @@ class SecondaryCacheSpec : ShouldSpec({
 
         // and: Request to secondary cache was sent
         val secondaryCacheRecordedRequests =
-            webCacheContainerClient.getSecondaryCacheRecordedRequests(requestObject.puts[0].key!!)
+            webCacheContainerClient.checkRecordedRequests(requestObject.puts[0].key!!)
         secondaryCacheRecordedRequests?.size shouldBe 1
 
         // and: Secondary cache request 'expiry' parameter matches to the Prebid Cache 'cache.min.expiry' config property

src/test/kotlin/org/prebid/cache/functional/testcontainers/client/WebCacheContainerClient.kt

@@ -59,4 +59,29 @@ class WebCacheContainerClient(mockServerHost: String, mockServerPort: Int) {
         request().withMethod(POST.name())
             .withPath("/$WEB_CACHE_PATH")
             .withBody(jsonPath("\$.puts[?(@.key == '$uuidKey')]"))
+
+    private fun waitUntil(closure: () -> Boolean, timeoutMs: Long = 5000, pollInterval: Long = 100) {
+        var isConditionFulfilled = false
+        val waiterStartTime = System.currentTimeMillis()
+        var waiterElapsedTime = 0L
+
+        while (waiterElapsedTime <= timeoutMs) {
+            if (closure()) {
+                isConditionFulfilled = true
+                break
+            } else {
+                waiterElapsedTime = System.currentTimeMillis() - waiterStartTime
+                Thread.sleep(pollInterval)
+            }
+        }
+
+        if (!isConditionFulfilled) {
+            throw IllegalStateException("Condition was not fulfilled within $timeoutMs ms.")
+        }
+    }
+
+    fun checkRecordedRequests(uuidKey: String): Array<out HttpRequest>? {
+        waitUntil({ getSecondaryCacheRecordedRequests(uuidKey)?.isNotEmpty() == true }, timeoutMs = 5000)
+        return getSecondaryCacheRecordedRequests(uuidKey)
+    }
 }