Fixed test_custom_generate_id_is_called test

django-es · Oct 1, 2021 · 0cc33bb · 0cc33bb
1 parent 79218c8
commit 0cc33bb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_documents.py b/tests/test_documents.py
@@ -485,5 +485,5 @@ def generate_id(cls, article):
 
         # Get the data from the elasticsearch low level API because
         # The generator get executed there.
-        data = json.loads(mock_bulk.call_args[0][0].split("\n")[0])
+        data = json.loads(mock_bulk.call_args[1]['body'].split("\n")[0])
         assert data["index"]["_id"] == article.slug