Merge pull request #731 from Labelbox/mkozik/adjust-integration-tests

[X-0] Adjust integration tests to explicitly define queue mode

Author: mkozik-lb

tests/integration/annotation_import/conftest.py

@@ -125,7 +125,8 @@ def ontology():
 
 @pytest.fixture
 def configured_project(client, ontology, rand_gen, image_url):
-    project = client.create_project(name=rand_gen(str))
+    project = client.create_project(name=rand_gen(str),
+                                    queue_mode=QueueMode.Dataset)
     dataset = client.create_dataset(name=rand_gen(str))
     editor = list(
         client.get_labeling_frontends(
@@ -143,7 +144,8 @@ def configured_project(client, ontology, rand_gen, image_url):
 
 @pytest.fixture
 def configured_project_pdf(client, ontology, rand_gen, pdf_url):
-    project = client.create_project(name=rand_gen(str))
+    project = client.create_project(name=rand_gen(str),
+                                    queue_mode=QueueMode.Dataset)
     dataset = client.create_dataset(name=rand_gen(str))
     editor = list(
         client.get_labeling_frontends(

tests/integration/conftest.py

@@ -161,7 +161,8 @@ def pdf_url(client):
 
 @pytest.fixture
 def project(client, rand_gen):
-    project = client.create_project(name=rand_gen(str))
+    project = client.create_project(name=rand_gen(str),
+                                    queue_mode=QueueMode.Dataset)
     yield project
     project.delete()
 
@@ -348,7 +349,8 @@ def create_label():
 
 @pytest.fixture
 def configured_project_with_complex_ontology(client, rand_gen, image_url):
-    project = client.create_project(name=rand_gen(str))
+    project = client.create_project(name=rand_gen(str),
+                                    queue_mode=QueueMode.Dataset)
     dataset = client.create_dataset(name=rand_gen(str), projects=project)
     data_row = dataset.create_data_row(row_data=image_url)
     editor = list(

tests/integration/test_filtering.py

@@ -2,14 +2,15 @@
 
 from labelbox import Project
 from labelbox.exceptions import InvalidQueryError
+from labelbox.schema.queue_mode import QueueMode
 
 
 # Avoid assertions using equality to prevent intermittent failures due to
 # other builds simultaneously adding projects to test org
 def test_where(client):
-    p_a = client.create_project(name="a")
-    p_b = client.create_project(name="b")
-    p_c = client.create_project(name="c")
+    p_a = client.create_project(name="a", queue_mode=QueueMode.Dataset)
+    p_b = client.create_project(name="b", queue_mode=QueueMode.Dataset)
+    p_c = client.create_project(name="c", queue_mode=QueueMode.Dataset)
 
     def _get(f, where=None):
         date_where = Project.created_at >= p_a.created_at

tests/integration/test_relationships.py

@@ -1,10 +1,12 @@
 import pytest
 
 from labelbox.exceptions import InvalidQueryError
+from labelbox.schema.queue_mode import QueueMode
 
 
 def test_project_dataset(client, rand_gen):
-    project = client.create_project(name=rand_gen(str))
+    project = client.create_project(name=rand_gen(str),
+                                    queue_mode=QueueMode.Dataset)
     dataset = client.create_dataset(name=rand_gen(str))
 
     assert len(list(project.datasets())) == 0
@@ -17,7 +19,8 @@ def test_project_dataset(client, rand_gen):
     assert {ds.uid for ds in project.datasets()} == {dataset.uid}
     assert {pr.uid for pr in dataset.projects()} == {project.uid}
 
-    project_2 = client.create_project(name=rand_gen(str))
+    project_2 = client.create_project(name=rand_gen(str),
+                                      queue_mode=QueueMode.Dataset)
 
     # Currently it's not possible to connect a project and dataset
     # by updating dataset.

tests/integration/test_sorting.py

@@ -1,14 +1,21 @@
 import pytest
 
 from labelbox import Project
+from labelbox.schema.queue_mode import QueueMode
 
 
 @pytest.mark.xfail(reason="Relationship sorting not implemented correctly "
                    "on the server-side")
 def test_relationship_sorting(client):
-    a = client.create_project(name="a", description="b")
-    b = client.create_project(name="b", description="c")
-    c = client.create_project(name="c", description="a")
+    a = client.create_project(name="a",
+                              description="b",
+                              queue_mode=QueueMode.Dataset)
+    b = client.create_project(name="b",
+                              description="c",
+                              queue_mode=QueueMode.Dataset)
+    c = client.create_project(name="c",
+                              description="a",
+                              queue_mode=QueueMode.Dataset)
 
     dataset = client.create_dataset(name="Dataset")
     a.datasets.connect(dataset)