IBX-6827: Allow to add label aggregation ranges (unit tests)

ibexa · Dec 8, 2023 · a23236b · a23236b
1 parent 008045c
commit a23236b
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/...Search/ResultExtractor/AggregationResultExtractor/RangeAggregationResultExtractorTest.php b/...Search/ResultExtractor/AggregationResultExtractor/RangeAggregationResultExtractorTest.php
@@ -63,6 +63,11 @@ public function dataProviderForTestExtract(): iterable
     {
         $aggregation = $this->createMock(AbstractRangeAggregation::class);
         $aggregation->method('getName')->willReturn(self::EXAMPLE_AGGREGATION_NAME);
+        $aggregation->method('getRanges')->willReturn([
+            new Range(Range::INF, 10, 'a'),
+            new Range(10, 100, 'b'),
+            new Range(100, Range::INF, 'c'),
+        ]);
 
         yield 'default' => [
             $aggregation,
@@ -78,9 +83,9 @@ public function dataProviderForTestExtract(): iterable
             new RangeAggregationResult(
                 self::EXAMPLE_AGGREGATION_NAME,
                 [
-                    new RangeAggregationResultEntry(new Range(null, '10'), 10),
-                    new RangeAggregationResultEntry(new Range('10', '100'), 100),
-                    new RangeAggregationResultEntry(new Range('100', null), 1000),
+                    new RangeAggregationResultEntry(new Range(Range::INF, '10', 'a'), 10),
+                    new RangeAggregationResultEntry(new Range('10', '100', 'b'), 100),
+                    new RangeAggregationResultEntry(new Range('100', Range::INF, 'c'), 1000),
                 ]
             ),
         ];