tests: treat import-form-data responses as multi-input (input_ids)

Author: datalogics-cgreen

tests/live/test_live_import_form_data.py

@@ -65,7 +65,12 @@ def test_live_import_form_data(
     assert response.output_files
     output_file = response.output_file
     assert output_file.type == "application/pdf"
-    assert str(response.input_id) == str(uploaded_pdf_with_forms.id)
+    assert str(uploaded_pdf_with_forms.id) in {
+        str(file_id) for file_id in response.input_ids
+    }
+    assert str(uploaded_form_data_file.id) in {
+        str(file_id) for file_id in response.input_ids
+    }
     if output_name is not None:
         assert output_file.name.startswith(output_name)
     else:
@@ -93,7 +98,12 @@ async def test_live_async_import_form_data_success(
     output_file = response.output_file
     assert output_file.name.startswith("async-imported")
     assert output_file.type == "application/pdf"
-    assert str(response.input_id) == str(uploaded_pdf_with_forms.id)
+    assert str(uploaded_pdf_with_forms.id) in {
+        str(file_id) for file_id in response.input_ids
+    }
+    assert str(uploaded_form_data_file.id) in {
+        str(file_id) for file_id in response.input_ids
+    }
 
 
 def test_live_import_form_data_invalid_data_file_id(

tests/test_import_form_data.py

@@ -49,7 +49,7 @@ def handler(request: httpx.Request) -> httpx.Response:
             return httpx.Response(
                 200,
                 json={
-                    "inputId": [input_file.id],
+                    "inputId": [input_file.id, data_file.id],
                     "outputId": [output_id],
                 },
             )
@@ -79,7 +79,8 @@ def handler(request: httpx.Request) -> httpx.Response:
     assert isinstance(response, PdfRestFileBasedResponse)
     assert response.output_file.name == "filled-form.pdf"
     assert response.output_file.type == "application/pdf"
-    assert str(response.input_id) == str(input_file.id)
+    assert str(input_file.id) in {str(file_id) for file_id in response.input_ids}
+    assert str(data_file.id) in {str(file_id) for file_id in response.input_ids}
     assert response.warning is None
 
 
@@ -108,7 +109,7 @@ def handler(request: httpx.Request) -> httpx.Response:
             return httpx.Response(
                 200,
                 json={
-                    "inputId": [input_file.id],
+                    "inputId": [input_file.id, data_file.id],
                     "outputId": [output_id],
                 },
             )
@@ -177,7 +178,7 @@ def handler(request: httpx.Request) -> httpx.Response:
             return httpx.Response(
                 200,
                 json={
-                    "inputId": [input_file.id],
+                    "inputId": [input_file.id, data_file.id],
                     "outputId": [output_id],
                 },
             )
@@ -203,7 +204,8 @@ def handler(request: httpx.Request) -> httpx.Response:
     assert isinstance(response, PdfRestFileBasedResponse)
     assert response.output_file.name == "async.pdf"
     assert response.output_file.type == "application/pdf"
-    assert str(response.input_id) == str(input_file.id)
+    assert str(input_file.id) in {str(file_id) for file_id in response.input_ids}
+    assert str(data_file.id) in {str(file_id) for file_id in response.input_ids}
 
 
 @pytest.mark.asyncio