PR refactor test

Author: Val Brodsky

tests/integration/annotation_import/conftest.py

@@ -177,24 +177,6 @@ def dataset_pdf_entity(client, rand_gen, pdf_entity_row_data):
     dataset.delete()
 
 
-@pytest.fixture
-def configured_project_pdf_entity(client, ontology, rand_gen,
-                                  dataset_pdf_entity):
-    project = client.create_project(name=rand_gen(str),
-                                    queue_mode=QueueMode.Dataset)
-
-    editor = list(
-        client.get_labeling_frontends(
-            where=LabelingFrontend.name == "editor"))[0]
-    project.setup(editor, ontology)
-
-    dataset, data_row_ids = dataset_pdf_entity
-    project.datasets.connect(dataset)
-    project.data_row_ids = data_row_ids
-    yield project
-    project.delete()
-
-
 @pytest.fixture
 def configured_project_without_data_rows(client, configured_project, rand_gen):
     project = client.create_project(name=rand_gen(str),

tests/integration/annotation_import/test_bulk_import_request.py

@@ -293,8 +293,8 @@ def test_pdf_mal_bbox(client, configured_project_pdf):
     assert import_annotations.errors == []
 
 
-def test_pdf_document_entity(client, configured_project_pdf_entity,
-                             dataset_pdf_entity):
+def test_pdf_document_entity(client, configured_project_without_data_rows,
+                             dataset_pdf_entity, rand_gen):
     # for content "Metal-insulator (MI) transitions have been one of the" in OCR JSON extract tests/assets/arxiv-pdf_data_99-word-token-pdfs_0801.3483-lb-textlayer.json
     document_text_selection = DocumentTextSelection(
         groupId="2f4336f4-a07e-4e0a-a9e1-5629b03b719b",
@@ -317,6 +317,12 @@ def test_pdf_document_entity(client, configured_project_pdf_entity,
 
     labels = []
     _, data_row_uids = dataset_pdf_entity
+    configured_project_without_data_rows.create_batch(
+        rand_gen(str),
+        data_row_uids,  # sample of data row objects
+        5  # priority between 1(Highest) - 5(lowest)
+    )
+
     for data_row_uid in data_row_uids:
         labels.append(
             Label(data=TextData(uid=data_row_uid),
@@ -326,7 +332,7 @@ def test_pdf_document_entity(client, configured_project_pdf_entity,
 
     import_annotations = MALPredictionImport.create_from_objects(
         client=client,
-        project_id=configured_project_pdf_entity.uid,
+        project_id=configured_project_without_data_rows.uid,
         name=f"import {str(uuid.uuid4())}",
         predictions=labels)
     import_annotations.wait_until_done()

tests/integration/conftest.py

@@ -173,7 +173,8 @@ def image_url(client):
 
 @pytest.fixture(scope="session")
 def pdf_url(client):
-    return client.upload_file('tests/assets/loremipsum.pdf')
+    pdf_url = client.upload_file('tests/assets/loremipsum.pdf')
+    return {"row_data": {"pdf_url": pdf_url,}, "global_key": str(uuid.uuid4())}
 
 
 @pytest.fixture(scope="session")