fix custom image integration test

Author: korgan00

.github/workflows/kubernetes-deploy.yaml

@@ -82,7 +82,6 @@ jobs:
           echo $GATEWAY_HOST
           # basic tests
           cd /home/runner/work/qiskit-serverless/qiskit-serverless/tests/basic
-          rm 06_function.py
           for f in *.py; do echo "TEST: $f" && python "$f"; done
           # experimental tests
           cd /home/runner/work/qiskit-serverless/qiskit-serverless/tests/experimental

tests/basic/06_function.py

@@ -21,19 +21,12 @@
     provider=os.environ.get("PROVIDER_ID", "mockprovider"),
     description=help,
 )
-serverless.upload(function_with_custom_image)
+runnable_function = serverless.upload(function_with_custom_image)
 
-my_functions = serverless.list()
-for function in my_functions:
-    print("Name: " + function.title)
-    print(function.description)
-    print()
-
-my_function = serverless.get("custom-image-function")
-job = my_function.run(message="Argument for the custum function")
+job = runnable_function.run(message="Argument for the custum function")
 
 print(job.result())
 print(job.logs())
 
-jobs = my_function.jobs()
+jobs = runnable_function.jobs()
 print(jobs)

tests/docker/test_docker.py

@@ -141,11 +141,7 @@ def test_multiple_runs(self, base_client: BaseClient):
         assert isinstance(retrieved_job1.logs(), str)
         assert isinstance(retrieved_job2.logs(), str)
 
-    @mark.skip(
-        reason="Images are not working in tests jet and "
-        + "LocalClient does not manage image instead of working_dir+entrypoint"
-    )
-    def test_error(self, base_client: BaseClient):
+    def test_custom_image(self, serverless_client: BaseClient):
         """Integration test to force an error."""
 
         description = """
@@ -164,7 +160,7 @@ def test_error(self, base_client: BaseClient):
             description=description,
         )
 
-        runnable_function = base_client.upload(function_with_custom_image)
+        runnable_function = serverless_client.upload(function_with_custom_image)
 
         job = runnable_function.run(message="Argument for the custum function")