Created workflow unit test and added test to model_management for workflow error flow

Author: Cloud User

.coverage.jakochsasctl.fsl.sashq-r.openstack.sas.com.31477.236928

@@ -0,0 +1 @@
+!coverage.py: This is a private format, don't read it directly!{"lines":{"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/__init__.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/__main__.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/core.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/exceptions.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/services.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/tasks.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/_services/__init__.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/_services/cas_management.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/_services/concepts.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/_services/data_sources.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/_services/files.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/_services/folders.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/_services/microanalytic_score.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/_services/model_management.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/_services/model_publish.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/_services/model_repository.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/_services/projects.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/_services/sentiment_analysis.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/_services/service.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/_services/text_categorization.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/_services/text_parsing.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/_services/workflow.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/utils/__init__.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/utils/astore.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/utils/cli.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/utils/pymas/__init__.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/utils/pymas/core.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/utils/pymas/ds2.py":[],"/home/centos/jamesfork/python-sasctl/.tox/py37-tests/lib/python3.7/site-packages/sasctl/utils/pymas/python.py":[]}}

tests/unit/test_model_management.py

@@ -82,4 +82,20 @@ def test_table_prefix_format():
             # Underscores should not be allowed
             _ = mm.create_performance_definition('model',
                                                  'TestLibrary',
-                                                 'invalid_name')
+                                                 'invalid_name')
+
+def test_execute_model_workflow_definition_invalidworkflow():
+
+    PROJECT = RestObj({'name': 'Test Project', 'id': '98765'})
+    WORKFLOWS = [{'name': 'Test W', 'id': '12345'},{'name': 'TestW2', 'id': '98765', 'prompts': [{'id': '98765', 'variableName': 'projectId', 'variableType': 'string'}]}]
+
+    with mock.patch('sasctl._services.workflow.Workflow'
+                    '.list_workflow_enableddefinitions') as list_workflow_enableddefinitions:
+        with mock.patch('sasctl._services.model_repository.ModelRepository'
+                        '.get_project') as get_project:
+            list_workflow_enableddefinitions.return_value = WORKFLOWS
+            get_project.return_value = PROJECT
+            with pytest.raises(ValueError):
+                # Project missing 
+                _ = mm.execute_model_workflow_definition('TestLibrary','badworkflow')
+

tests/unit/test_workflow.py

@@ -0,0 +1,66 @@
+#!/usr/bin/env python
+# encoding: utf-8
+#
+# Copyright © 2019, SAS Institute Inc., Cary, NC, USA.  All Rights Reserved.
+# SPDX-License-Identifier: Apache-2.0
+
+import pytest
+from six.moves import mock
+
+from sasctl import RestObj
+from sasctl._services import workflow 
+
+
+def test_list_workflow_prompt_invalidworkflow():
+
+    WORKFLOWS = [{'name': 'Test W', 'id': '12345'},{'name': 'TestW2', 'id': '98765', 'prompts': [{'id': '98765', 'variableName': 'projectId', 'variableType': 'string'}]}]
+    wf = workflow.Workflow()
+
+    with mock.patch('sasctl._services.workflow.Workflow'
+                    '.list_workflow_enableddefinitions') as list_workflow_enableddefinitions:
+        list_workflow_enableddefinitions.return_value = WORKFLOWS
+        with pytest.raises(ValueError):
+            # Project missing 
+            _ = wf.list_workflow_prompt('bad')
+
+def test_list_workflow_prompt_workflownoprompt():
+
+    WORKFLOWS = [{'name': 'Test W', 'id': '12345'},{'name': 'TestW2', 'id': '98765', 'prompts': [{'id': '98765', 'variableName': 'projectId', 'variableType': 'string'}]}]
+    wf = workflow.Workflow()
+
+    with mock.patch('sasctl._services.workflow.Workflow'
+                    '.list_workflow_enableddefinitions') as list_workflow_enableddefinitions:
+        list_workflow_enableddefinitions.return_value = WORKFLOWS
+
+        #Testing no prompts on workflow with name
+        testresult = wf.list_workflow_prompt('Test W')
+        print(testresult)
+        assert testresult is None
+
+        #Testing no prompts on workflow with id
+        testresult = wf.list_workflow_prompt('12345')
+        print(testresult)
+        assert testresult is None
+
+
+
+def test_list_workflow_prompt_workflowprompt():
+
+    WORKFLOWS = [{'name': 'Test W', 'id': '12345'},{'name': 'TestW2', 'id': '98765', 'prompts': [{'id': '98765', 'variableName': 'projectId', 'variableType': 'string'}]}]
+    wf = workflow.Workflow()
+
+    with mock.patch('sasctl._services.workflow.Workflow'
+                    '.list_workflow_enableddefinitions') as list_workflow_enableddefinitions:
+        list_workflow_enableddefinitions.return_value = WORKFLOWS
+
+        #Testing workflow with id and prompts
+        testresult = wf.list_workflow_prompt('TestW2')
+        print(testresult)
+        assert testresult is not None
+
+
+        #Testing workflow with id and prompts
+        testresult = wf.list_workflow_prompt('98765')
+        print(testresult)
+        assert testresult is not None
+