fix(test): cherry-pick fix for aggregated temporal queries test

stellio-hub · Jan 2, 2024 · fd8f7f1 · fd8f7f1
1 parent f54f1c1
commit fd8f7f1
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/search-service/src/test/kotlin/com/egm/stellio/search/service/AggregatedQueryServiceTests.kt b/search-service/src/test/kotlin/com/egm/stellio/search/service/AggregatedQueryServiceTests.kt
@@ -395,7 +395,13 @@ class AggregatedQueryServiceTests : WithTimescaleContainer, WithKafkaContainer {
         }
 
         val temporalEntitiesQuery = createTemporalEntitiesQuery("avg", aggrPeriodDuration)
-        attributeInstanceService.search(temporalEntitiesQuery, temporalEntityAttribute, startTimestamp)
+        attributeInstanceService.search(
+            temporalEntitiesQuery.copy(
+                temporalQuery = temporalEntitiesQuery.temporalQuery.copy(timeAt = startTimestamp)
+            ),
+            temporalEntityAttribute,
+            startTimestamp
+        )
             .shouldSucceedWith { results ->
                 assertEquals(expectedNumberOfBuckets, results.size)
             }