From 9290076b26a9c0d68c631dd0a23d5b0aabd212f0 Mon Sep 17 00:00:00 2001
From: "github-actions[bot]" <github-actions[bot]@users.noreply.github.com>
Date: Wed, 25 Sep 2024 23:31:21 +0000
Subject: [PATCH] Fix segment count for optimized agg rest test (#16067)

Signed-off-by: Finn Carroll <carrofin@amazon.com>
(cherry picked from commit a42e51ddb93d50a9bb0455bb6bda9fe461fe8f5b)
Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
---
 .../search.aggregation/360_date_histogram.yml  | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/rest-api-spec/src/main/resources/rest-api-spec/test/search.aggregation/360_date_histogram.yml b/rest-api-spec/src/main/resources/rest-api-spec/test/search.aggregation/360_date_histogram.yml
index 52d68f4e32812..4d39ad7039a0f 100644
--- a/rest-api-spec/src/main/resources/rest-api-spec/test/search.aggregation/360_date_histogram.yml
+++ b/rest-api-spec/src/main/resources/rest-api-spec/test/search.aggregation/360_date_histogram.yml
@@ -109,6 +109,19 @@ setup:
       version: " - 2.16.99"
       reason: "Bugfix not backported to 2.16"
 
+  - do:
+      indices.create:
+        index: dhisto-agg-w-query
+        body:
+          settings:
+            number_of_shards: 1
+            number_of_replicas: 0
+            refresh_interval: -1
+          mappings:
+            properties:
+              date:
+                type: date
+
   - do:
       bulk:
         refresh: true
@@ -127,6 +140,11 @@ setup:
           - '{"index": {}}'
           - '{"date": "2025-02-14"}'
 
+  - do:
+      indices.forcemerge:
+        index: dhisto-agg-w-query
+        max_num_segments: 1
+
   - do:
       search:
         index: dhisto-agg-w-query