Updated test with new create_workflow() parameter

ICRAR · Sep 24, 2020 · a508db0 · a508db0
1 parent 6ed1086
commit a508db0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/daliuge-translator/test/dropmake/test_pg_gen.py b/daliuge-translator/test/dropmake/test_pg_gen.py
@@ -168,6 +168,7 @@ def test_pg_eagle(self):
 
     def test_cwl_translate(self):
         import git
+        import io
         import os
         import shutil
         import uuid
@@ -205,11 +206,12 @@ def test_cwl_translate(self):
                 cwl_out_zip = cwl_output_dir + '/workflow.zip'
                 output_list.append((cwl_out, cwl_out_zip))
 
-                cwl_contents = create_workflow(pgt, 'workflow.cwl')
+                buffer = io.BytesIO()
+                create_workflow(pgt, 'workflow.cwl', buffer)
 
                 # write output
                 with open(cwl_out_zip, 'wb') as f:
-                    f.write(cwl_contents)
+                    f.write(buffer.getvalue())
 
         for out, zip in output_list:
             zip_ref = zipfile.ZipFile(zip)