diff --git a/shared/workspaces/test/test_persistence.py b/shared/workspaces/test/test_persistence.py index 122e82230967f13d46a081911a1d0a859b00f3d3..02240d13557a1cd9ee9e2ad55c6ff332ffa0fa7d 100644 --- a/shared/workspaces/test/test_persistence.py +++ b/shared/workspaces/test/test_persistence.py @@ -1,5 +1,6 @@ from schema import create_session from workspaces.capability_interfaces import CapabilityInfoIF +from workspaces.schema import Workflow from workspaces.services import WorkflowInfo, CapabilityInfo from workspaces.workflow_interfaces import WorkflowInfoIF @@ -17,6 +18,14 @@ def get_capability_info() -> CapabilityInfoIF: return CapabilityInfo(SESSION) +def test_get_requests_for_capability(): + """Ensure we can get all requests for a certain capability""" + info = get_capability_info() + info.create_capability("test_get_requests", None, 0) + info.create_capability_request("test_get_requests") + assert len(info.requests_for_capability("test_get_requests")) > 0 + + def test_persisting_workflow_request(): """Ensure we can persist a workflow request""" info = get_workflow_info() @@ -32,6 +41,14 @@ def test_reading_workflows(): assert len(workflows) > 0 +def test_reading_workflow_requests(): + """Ensure we can get workflow requests""" + info = get_workflow_info() + workflow_requests = info.all_workflow_requests() + print(workflow_requests) + assert len(workflow_requests) > 0 + + def test_reading_capability(): """Ensure we can get a capability""" info = get_capability_info()