updates to test cases for onprem

Author: jtsodapop

Makefile

@@ -20,3 +20,9 @@ test-prod: build
 		-e LABELBOX_TEST_ENVIRON="prod" \
 		-e LABELBOX_TEST_API_KEY_PROD=${LABELBOX_TEST_API_KEY_PROD} \
 		local/labelbox-python:test pytest $(PATH_TO_TEST) -svvx
+
+test-onprem: build
+	docker run -it -v ${PWD}:/usr/src -w /usr/src \
+		-e LABELBOX_TEST_ENVIRON="onprem" \
+		-e LABELBOX_TEST_API_KEY_ONPREM=${LABELBOX_TEST_API_KEY_ONPREM} \
+		local/labelbox-python:test pytest $(PATH_TO_TEST) -svvx

tests/integration/annotation_import/test_label_import.py

@@ -28,6 +28,7 @@ def test_create_from_objects(client, project, object_predictions,
                                                    project_id=project.uid,
                                                    name=name,
                                                    labels=object_predictions)
+    label_import.wait_until_done()
     assert label_import.parent_id == project.uid
     annotation_import_test_helpers.check_running_state(label_import, name)
     annotation_import_test_helpers.assert_file_content(

tests/integration/conftest.py

@@ -27,6 +27,7 @@ class Environ(Enum):
     LOCAL = 'local'
     PROD = 'prod'
     STAGING = 'staging'
+    ONPREM = 'onprem'
 
 
 @pytest.fixture(scope="session")
@@ -49,6 +50,9 @@ def graphql_url(environ: str) -> str:
         return 'https://api.labelbox.com/graphql'
     elif environ == Environ.STAGING:
         return 'https://staging-api.labelbox.com/graphql'
+    elif environ == Environ.ONPREM:
+        instance_id = "replicated-1c0e609"
+        return f"https://app.{instance_id}.labelbox.dev/api/_gql"
     return 'http://host.docker.internal:8080/graphql'
 
 
@@ -57,6 +61,8 @@ def testing_api_key(environ: str) -> str:
         return os.environ["LABELBOX_TEST_API_KEY_PROD"]
     elif environ == Environ.STAGING:
         return os.environ["LABELBOX_TEST_API_KEY_STAGING"]
+    elif environ == Environ.ONPREM:
+        return os.environ["LABELBOX_TEST_API_KEY_ONPREM"]
     return os.environ["LABELBOX_TEST_API_KEY_LOCAL"]
 
 

tests/integration/test_label.py

@@ -2,6 +2,7 @@
 
 import pytest
 import requests
+import os
 
 from labelbox import Label
 
@@ -35,6 +36,8 @@ def test_label_export(configured_project_with_label):
     # The new exporter doesn't work with the create_label mutation
 
 
+@pytest.mark.skipif(condition=os.environ['LABELBOX_TEST_ENVIRON'] == "onprem",
+                    reason="does not work for onprem")
 def test_label_update(configured_project_with_label):
     _, _, _, label = configured_project_with_label
     label.update(label="something else")