Added tests

Author: GlassOfWhiskey

Diff for: tests/test_extensions.py

@@ -14,6 +14,8 @@ def test_cuda_requirement_v1_0() -> None:
     )
     cwl_obj = load_document_by_uri(uri)
     assert isinstance(next(iter(cwl_obj.requirements)), cwl_v1_0.CUDARequirement)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["requirements"][0]["class"] == "cwltool:CUDARequirement"
 
 
 def test_cuda_requirement_v1_1() -> None:
@@ -25,6 +27,8 @@ def test_cuda_requirement_v1_1() -> None:
     )
     cwl_obj = load_document_by_uri(uri)
     assert isinstance(next(iter(cwl_obj.requirements)), cwl_v1_1.CUDARequirement)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["requirements"][0]["class"] == "cwltool:CUDARequirement"
 
 
 def test_cuda_requirement_v1_2() -> None:
@@ -36,6 +40,8 @@ def test_cuda_requirement_v1_2() -> None:
     )
     cwl_obj = load_document_by_uri(uri)
     assert isinstance(next(iter(cwl_obj.requirements)), cwl_v1_2.CUDARequirement)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["requirements"][0]["class"] == "cwltool:CUDARequirement"
 
 
 def test_load_listing_requirement_v1_0() -> None:
@@ -47,6 +53,8 @@ def test_load_listing_requirement_v1_0() -> None:
     )
     cwl_obj = load_document_by_uri(uri)
     assert isinstance(next(iter(cwl_obj.requirements)), cwl_v1_0.LoadListingRequirement)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["requirements"][0]["class"] == "cwltool:LoadListingRequirement"
 
 
 def test_loop_v1_2() -> None:
@@ -61,6 +69,8 @@ def test_loop_v1_2() -> None:
     loop_req = next(iter(cwl_step.requirements))
     assert isinstance(loop_req, cwl_v1_2.Loop)
     assert isinstance(next(iter(loop_req.loop)), cwl_v1_2.LoopInput)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["steps"][0]["requirements"][0]["class"] == "cwltool:Loop"
 
 
 def test_inplace_update_requirement_v1_0() -> None:
@@ -74,6 +84,8 @@ def test_inplace_update_requirement_v1_0() -> None:
     assert isinstance(
         next(iter(cwl_obj.requirements)), cwl_v1_0.InplaceUpdateRequirement
     )
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["requirements"][0]["class"] == "cwltool:InplaceUpdateRequirement"
 
 
 def test_mpi_requirement_v1_0() -> None:
@@ -85,6 +97,8 @@ def test_mpi_requirement_v1_0() -> None:
     )
     cwl_obj = load_document_by_uri(uri)
     assert isinstance(next(iter(cwl_obj.requirements)), cwl_v1_0.MPIRequirement)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["requirements"][0]["class"] == "cwltool:MPIRequirement"
 
 
 def test_mpi_requirement_v1_1() -> None:
@@ -96,6 +110,8 @@ def test_mpi_requirement_v1_1() -> None:
     )
     cwl_obj = load_document_by_uri(uri)
     assert isinstance(next(iter(cwl_obj.requirements)), cwl_v1_1.MPIRequirement)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["requirements"][0]["class"] == "cwltool:MPIRequirement"
 
 
 def test_mpi_requirement_v1_2() -> None:
@@ -107,6 +123,8 @@ def test_mpi_requirement_v1_2() -> None:
     )
     cwl_obj = load_document_by_uri(uri)
     assert isinstance(next(iter(cwl_obj.requirements)), cwl_v1_2.MPIRequirement)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["requirements"][0]["class"] == "cwltool:MPIRequirement"
 
 
 def test_network_access_v1_0() -> None:
@@ -116,6 +134,8 @@ def test_network_access_v1_0() -> None:
     )
     cwl_obj = load_document_by_uri(uri)
     assert isinstance(next(iter(cwl_obj.requirements)), cwl_v1_0.NetworkAccess)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["requirements"][0]["class"] == "cwltool:NetworkAccess"
 
 
 def test_process_generator_v1_0() -> None:
@@ -125,7 +145,10 @@ def test_process_generator_v1_0() -> None:
         .resolve()
         .as_uri()
     )
-    assert isinstance(load_document_by_uri(uri), cwl_v1_0.ProcessGenerator)
+    cwl_obj = load_document_by_uri(uri)
+    assert isinstance(cwl_obj, cwl_v1_0.ProcessGenerator)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["class"] == "cwltool:ProcessGenerator"
 
 
 def test_process_generator_v1_1() -> None:
@@ -135,7 +158,10 @@ def test_process_generator_v1_1() -> None:
         .resolve()
         .as_uri()
     )
-    assert isinstance(load_document_by_uri(uri), cwl_v1_1.ProcessGenerator)
+    cwl_obj = load_document_by_uri(uri)
+    assert isinstance(cwl_obj, cwl_v1_1.ProcessGenerator)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["class"] == "cwltool:ProcessGenerator"
 
 
 def test_process_generator_v1_2() -> None:
@@ -145,60 +171,79 @@ def test_process_generator_v1_2() -> None:
         .resolve()
         .as_uri()
     )
-    assert isinstance(load_document_by_uri(uri), cwl_v1_2.ProcessGenerator)
+    cwl_obj = load_document_by_uri(uri)
+    assert isinstance(cwl_obj, cwl_v1_2.ProcessGenerator)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["class"] == "cwltool:ProcessGenerator"
 
 
 def test_secrets_v1_0() -> None:
     """Test that Secrets objects are correctly loaded for CWL v1.0."""
     uri = Path(get_data("testdata/extensions/secrets_v1_0.cwl")).resolve().as_uri()
     cwl_obj = load_document_by_uri(uri)
     assert isinstance(next(iter(cwl_obj.requirements)), cwl_v1_0.Secrets)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["requirements"][0]["class"] == "cwltool:Secrets"
 
 
 def test_secrets_v1_1() -> None:
     """Test that Secrets objects are correctly loaded for CWL v1.1."""
     uri = Path(get_data("testdata/extensions/secrets_v1_1.cwl")).resolve().as_uri()
     cwl_obj = load_document_by_uri(uri)
     assert isinstance(next(iter(cwl_obj.requirements)), cwl_v1_1.Secrets)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["requirements"][0]["class"] == "cwltool:Secrets"
 
 
 def test_secrets_v1_2() -> None:
     """Test that Secrets objects are correctly loaded for CWL v1.2."""
     uri = Path(get_data("testdata/extensions/secrets_v1_2.cwl")).resolve().as_uri()
     cwl_obj = load_document_by_uri(uri)
     assert isinstance(next(iter(cwl_obj.requirements)), cwl_v1_2.Secrets)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["requirements"][0]["class"] == "cwltool:Secrets"
 
 
 def test_shm_size_v1_0() -> None:
     """Test that ShmSize objects are correctly loaded for CWL v1.0."""
     uri = Path(get_data("testdata/extensions/shm-size_v1_0.cwl")).resolve().as_uri()
     cwl_obj = load_document_by_uri(uri)
     assert isinstance(next(iter(cwl_obj.requirements)), cwl_v1_0.ShmSize)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["requirements"][0]["class"] == "cwltool:ShmSize"
 
 
 def test_shm_size_v1_1() -> None:
     """Test that ShmSize objects are correctly loaded for CWL v1.1."""
     uri = Path(get_data("testdata/extensions/shm-size_v1_1.cwl")).resolve().as_uri()
     cwl_obj = load_document_by_uri(uri)
     assert isinstance(next(iter(cwl_obj.requirements)), cwl_v1_1.ShmSize)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["requirements"][0]["class"] == "cwltool:ShmSize"
 
 
 def test_shm_size_v1_2() -> None:
     """Test that ShmSize objects are correctly loaded for CWL v1.2."""
     uri = Path(get_data("testdata/extensions/shm-size_v1_2.cwl")).resolve().as_uri()
     cwl_obj = load_document_by_uri(uri)
     assert isinstance(next(iter(cwl_obj.requirements)), cwl_v1_2.ShmSize)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["requirements"][0]["class"] == "cwltool:ShmSize"
 
 
 def test_time_limit_v1_0() -> None:
     """Test that TimeLimit objects are correctly loaded for CWL v1.0."""
     uri = Path(get_data("testdata/extensions/time-limit_v1_0.cwl")).resolve().as_uri()
     cwl_obj = load_document_by_uri(uri)
     assert isinstance(next(iter(cwl_obj.requirements)), cwl_v1_0.TimeLimit)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["requirements"][0]["class"] == "cwltool:TimeLimit"
 
 
 def test_work_reuse_v1_0() -> None:
     """Test that WorkReuse objects are correctly loaded for CWL v1.0."""
     uri = Path(get_data("testdata/extensions/work-reuse_v1_0.cwl")).resolve().as_uri()
     cwl_obj = load_document_by_uri(uri)
     assert isinstance(next(iter(cwl_obj.requirements)), cwl_v1_0.WorkReuse)
+    cwl_dict = cwl_obj.save(top=True)
+    assert cwl_dict["requirements"][0]["class"] == "cwltool:WorkReuse"