Merge pull request #864 from Labelbox/AL-4817-create-test-case-around-duplicate-global-keys-issue

add tests for recreating the same global key

Author: apollonin

tests/integration/test_data_rows.py

@@ -787,6 +787,42 @@ def test_data_row_bulk_creation_with_same_global_keys(dataset, sample_image):
     assert list(dataset.data_rows())[0].global_key == global_key_1
 
 
+def test_data_row_delete_and_create_with_same_global_key(
+        client, dataset, sample_image):
+    global_key_1 = str(uuid.uuid4())
+    data_row_payload = {
+        DataRow.row_data: sample_image,
+        DataRow.global_key: global_key_1
+    }
+
+    # should successfully insert new datarow
+    task = dataset.create_data_rows([data_row_payload])
+    task.wait_till_done()
+
+    assert task.status == "COMPLETE"
+    assert task.result[0]['global_key'] == global_key_1
+
+    new_data_row_id = task.result[0]['id']
+
+    # same payload should fail due to duplicated global key
+    task = dataset.create_data_rows([data_row_payload])
+    task.wait_till_done()
+
+    assert task.status == "FAILED"
+    assert len(task.failed_data_rows) > 0
+    assert task.errors.startswith("Duplicate global keys found")
+
+    # delete datarow
+    client.get_data_row(new_data_row_id).delete()
+
+    # should successfully insert new datarow now
+    task = dataset.create_data_rows([data_row_payload])
+    task.wait_till_done()
+
+    assert task.status == "COMPLETE"
+    assert task.result[0]['global_key'] == global_key_1
+
+
 def test_data_row_bulk_creation_sync_with_unique_global_keys(
         dataset, sample_image):
     global_key_1 = str(uuid.uuid4())