Fix tests

Author: florijanstamenkovic

tests/integration/test_data_rows.py

@@ -46,10 +46,7 @@ def test_data_row_bulk_creation(client, rand_gen):
     assert res.status_code == 200
     assert res.text == "Test data"
 
-    # Currently can't delete DataRow by setting deleted=true
-    # TODO ensure DataRow can be deleted (server-side) by setting deleted=true
-    with pytest.raises(InvalidQueryError):
-        data_rows[0].delete()
+    data_rows[0].delete()
 
     # Do a longer task and expect it not to be complete immediately
     with NamedTemporaryFile() as fp:

tests/integration/test_dataset.py

@@ -49,12 +49,14 @@ def test_dataset(client, rand_gen):
         dataset = client.get_dataset(dataset.uid)
 
 
-def test_dataset_filtering(client):
-    d1 = client.create_dataset(name="d1")
-    d2 = client.create_dataset(name="d2")
-
-    assert list(client.get_datasets(where=Dataset.name=="d1")) == [d1]
-    assert list(client.get_datasets(where=Dataset.name=="d2")) == [d2]
+def test_dataset_filtering(client, rand_gen):
+    name_1 = rand_gen(str)
+    name_2 = rand_gen(str)
+    d1 = client.create_dataset(name=name_1)
+    d2 = client.create_dataset(name=name_2)
+
+    assert list(client.get_datasets(where=Dataset.name==name_1)) == [d1]
+    assert list(client.get_datasets(where=Dataset.name==name_2)) == [d2]
 
     d1.delete()
     d2.delete()

tests/integration/test_project.py

@@ -39,12 +39,14 @@ def test_project(client, rand_gen):
     project = client.get_project(project.uid)
 
 
-def test_project_filtering(client):
-    p1 = client.create_project(name="p1")
-    p2 = client.create_project(name="p2")
-
-    assert list(client.get_projects(where=Project.name=="p1")) == [p1]
-    assert list(client.get_projects(where=Project.name=="p2")) == [p2]
+def test_project_filtering(client, rand_gen):
+    name_1 = rand_gen(str)
+    name_2 = rand_gen(str)
+    p1 = client.create_project(name=name_1)
+    p2 = client.create_project(name=name_2)
+
+    assert list(client.get_projects(where=Project.name==name_1)) == [p1]
+    assert list(client.get_projects(where=Project.name==name_2)) == [p2]
 
     p1.delete()
     p2.delete()

tests/integration/test_project_setup.py

@@ -22,8 +22,8 @@ def test_project_setup(client, rand_gen):
     project = client.create_project(name=rand_gen(str))
 
     # TODO this can only run against the staging server
-    labeling_frontends = list(client.get_all(
-        LabelingFrontend, where=LabelingFrontend.iframe_url_path ==
+    labeling_frontends = list(client.get_labeling_frontends(
+        where=LabelingFrontend.iframe_url_path ==
         "https://staging-image-segmentation-v4.labelbox.com"))
     assert len(labeling_frontends) == 1
     labeling_frontend = labeling_frontends[0]