From 31d304ddf77b9fdc483a7d923d8f3f490e867261 Mon Sep 17 00:00:00 2001
From: nhertz <nhertz@nrao.edu>
Date: Fri, 16 Oct 2020 16:03:04 -0600
Subject: [PATCH] Added persistence tests
---
shared/workspaces/test/test_persistence.py | 17 +++++++++++++++++
1 file changed, 17 insertions(+)
diff --git a/shared/workspaces/test/test_persistence.py b/shared/workspaces/test/test_persistence.py
index 122e82230..02240d135 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()
--
GitLab