From fd8f7f1e4fe420a398b072b690cbb0b7fbeb5a3f Mon Sep 17 00:00:00 2001
From: Benoit Orihuela <benoit.orihuela@egm.io>
Date: Tue, 2 Jan 2024 11:27:42 +0100
Subject: [PATCH] fix(test): cherry-pick fix for aggregated temporal queries
 test

---
 .../stellio/search/service/AggregatedQueryServiceTests.kt | 8 +++++++-
 1 file changed, 7 insertions(+), 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
index ea80463b1..3510dce8e 100644
--- 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)
             }