test get_or_create_code

tests/conftest.py

@@ -30,7 +30,9 @@ def fixture_localhost(aiida_localhost):
 def add_code(fixture_localhost):
     from aiida.orm import InstalledCode
 
-    code = InstalledCode(computer=fixture_localhost, filepath_executable="/bin/bash")
+    code = InstalledCode(
+        label="add", computer=fixture_localhost, filepath_executable="/bin/bash"
+    )
     code.store()
     return code
 

tests/test_utils.py

@@ -73,3 +73,25 @@ def test_validate_task_inout_dict_with_extra_keys():
     ]
     result = validate_task_inout(input_list, "input")
     assert result == input_list
+
+
+def test_get_or_create_code(fixture_localhost, add_code):
+    from aiida_workgraph.utils import get_or_create_code
+    from aiida.orm import Code
+
+    # use already created code
+    code = get_or_create_code(
+        computer="localhost",
+        code_label="add",
+        code_path="/bin/bash",
+        prepend_text='echo "Hello, World!"',
+    )
+    assert code.uuid == add_code.uuid
+    # create a new code
+    code = get_or_create_code(
+        computer="localhost",
+        code_label="test_code",
+        code_path="/bin/bash",
+        prepend_text='echo "Hello, World!"',
+    )
+    assert isinstance(code, Code)