Add test for parsing outputs

Author: koopmant

app/tests/components_tests/test_tasks.py

@@ -45,6 +45,7 @@
     delete_container_image,
     encode_b64j,
     execute_job,
+    parse_endpoint_invocation_outputs,
     preload_interactive_algorithms,
     remove_container_image_from_registry,
     remove_inactive_container_images,
@@ -1651,3 +1652,32 @@ def test_stop_expired_endpoints(
     assert len(callbacks) == 1
     mock_deprovision.assert_called_once()
     assert endpoint_to_stop.status == EndpointStatusChoices.STOPPED
+
+
+@pytest.mark.django_db
+def test_parse_endpoint_invocation_outputs(settings):
+    socket = ComponentInterfaceFactory(kind=InterfaceKindChoices.STRING)
+    invocation = InvocationFactory(
+        status=InvocationStatusChoices.EXECUTED,
+        algorithm_interface__outputs=[socket],
+    )
+    orchestrator = invocation.orchestrator
+    content = json.dumps("test output content").encode("utf-8")
+    orchestrator._s3_client.upload_fileobj(
+        Fileobj=io.BytesIO(content),
+        Bucket=settings.ALGORITHM_ENDPOINTS_OUTPUT_BUCKET_NAME,
+        Key=f"{orchestrator._io_prefix}/{socket.relative_path}",
+    )
+
+    assert invocation.outputs.count() == 0
+
+    parse_endpoint_invocation_outputs(**invocation.task_kwargs)
+    invocation.refresh_from_db()
+
+    assert invocation.error_message == ""
+    assert invocation.status == InvocationStatusChoices.SUCCESS
+    assert invocation.outputs.count() == 1
+
+    civ = invocation.outputs.first()
+
+    assert civ.value == "test output content"