From 77e26a9bc79257be3866145de21bc6ef6c7f337a Mon Sep 17 00:00:00 2001
From: Gil Raphaelli <gil@elastic.co>
Date: Mon, 23 Aug 2021 11:52:18 -0400
Subject: [PATCH] add --apm-server-enable-data-streams option (#1229)

---
 scripts/modules/elastic_stack.py | 8 ++++++++
 scripts/tests/service_tests.py   | 4 ++++
 2 files changed, 12 insertions(+)

diff --git a/scripts/modules/elastic_stack.py b/scripts/modules/elastic_stack.py
index 2b6a26e48..ab0dbbc9b 100644
--- a/scripts/modules/elastic_stack.py
+++ b/scripts/modules/elastic_stack.py
@@ -247,6 +247,9 @@ def add_es_config(args, prefix="output", tls=self.es_tls):
                     ("output.file.path", self.options.get("apm_server_output_file", os.devnull)),
                 ])
 
+        if self.options.get("apm_server_enable_data_streams"):
+            self.apm_server_command_args.append(("apm-server.data_streams.enabled", "true"))
+
         for opt in options.get("apm_server_opt", []):
             self.apm_server_command_args.append(opt.split("=", 1))
 
@@ -387,6 +390,11 @@ def add_arguments(cls, parser):
             action="store_false",
             help="make apm-server act as a Jaeger collector (HTTP and gRPC).",
         )
+        parser.add_argument(
+            "--apm-server-enable-data-streams",
+            action="store_true",
+            help='enable writing to data streams'
+        )
         parser.add_argument(
             '--apm-server-enable-tls',
             action="store_true",
diff --git a/scripts/tests/service_tests.py b/scripts/tests/service_tests.py
index 405a71c04..804ce6954 100644
--- a/scripts/tests/service_tests.py
+++ b/scripts/tests/service_tests.py
@@ -917,6 +917,10 @@ def test_apm_server_custom_pipeline(self):
         self.assertIn("foo:/usr/share/apm-server/ingest/pipeline/definition.json", apm_server["volumes"])
         self.assertIn("apm-server.register.ingest.pipeline.overwrite=true", apm_server["command"])
 
+    def test_data_streams(self):
+        apm_server = ApmServer(version="7.16.0", apm_server_enable_data_streams=True).render()["apm-server"]
+        self.assertIn("apm-server.data_streams.enabled=true", apm_server["command"])
+
     def test_debug(self):
         apm_server = ApmServer(version="6.8.0", apm_server_enable_debug=True).render()["apm-server"]
         self.assertTrue("-d" in apm_server["command"])