Fix integration tests

elastic · Jun 13, 2024 · b4e1c8f · b4e1c8f
1 parent f5a61f1
commit b4e1c8f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/handlers/aws/test_integrations.py b/tests/handlers/aws/test_integrations.py
@@ -988,7 +988,7 @@ def test_replay(self) -> None:
 
         self.elasticsearch.put_pipeline(id="test_replay_fail_pipeline", body=processors)
 
-        self.elasticsearch.create_data_stream(index="logs-generic-default")
+        self.elasticsearch.create_data_stream(name="logs-generic-default")
         self.elasticsearch.put_settings(
             index="logs-generic-default", body={"index.default_pipeline": "test_replay_fail_pipeline"}
         )

diff --git a/tests/testcontainers/es.py b/tests/testcontainers/es.py
@@ -211,7 +211,7 @@ def index(self, **kwargs: Any) -> dict[str, Any]:
         return self.es_client.index(**kwargs)
 
     def create_data_stream(self, **kwargs: Any) -> dict[str, Any]:
-        if "index" in kwargs:
-            self._index_indices.add(kwargs["index"])
+        if "name" in kwargs:
+            self._index_indices.add(kwargs["name"])
 
         return self.es_client.indices.create_data_stream(**kwargs)