bug fix: allow steps with no parameters

Author: jlwalke2

src/sasctl/_services/microanalytic_score.py

@@ -227,8 +227,8 @@ def define_steps(self, module):
             step = self.get_module_step(module, id)
 
             # Method should have an argument for each parameter of the step
-            arguments = [k['name'] for k in step.inputs]
-            arg_types = [k['type'] for k in step.inputs]
+            arguments = [k['name'] for k in step.get('inputs', [])]
+            arg_types = [k['type'] for k in step.get('inputs', [])]
 
             # Format call to execute_module_step()
             call_params = ['{}={}'.format(i, i) for i in arguments]
@@ -237,28 +237,31 @@ def define_steps(self, module):
             type_string = '    # type: ({})'.format(', '.join(arg_types))
 
             # Method signature
-            signature = 'def _%s_%s(%s, **kwargs):' \
+            input_params = [a for a in arguments] + ['**kwargs']
+            signature = 'def _%s_%s(%s):' \
                         % (module.name,
                            step.id,
-                           ', '.join(a for a in arguments))
+                           ', '.join(input_params))
 
             # MAS always lower-cases variable names
             # Since the original Python variables may have a different case,
             # allow kwargs to be used to input alternative caps
-            arg_checks = ['for k in kwargs.keys():']
-            for arg in arguments:
-                arg_checks.append("    if k.lower() == '%s':" % arg.lower())
-                arg_checks.append("        %s = kwargs[k]" % arg)
-                arg_checks.append("        continue")
+            if len(arguments):
+                arg_checks = ['for k in kwargs.keys():']
+                for arg in arguments:
+                    arg_checks.append("    if k.lower() == '%s':" % arg.lower())
+                    arg_checks.append("        %s = kwargs[k]" % arg)
+                    arg_checks.append("        continue")
+            else:
+                arg_checks = []
 
             # Full method source code
             # Drops 'rc' and 'msg' from return values
             code = (signature,
                     type_string,
                     '    """Execute step %s of module %s."""' % (step, module),
                     '\n'.join(['    %s' % a for a in arg_checks]),
-                    '    r = execute_module_step(module, step, {})'.format(
-                        ', '.join(call_params)),
+                    '    r = execute_module_step(%s)' % ', '.join(['module', 'step'] + call_params),
                     '    r.pop("rc", None)',
                     '    r.pop("msg", None)',
                     '    if len(r) == 1:',

tests/unit/test_microanalytic_score.py

@@ -10,6 +10,7 @@
 from sasctl.services import microanalytic_score as mas
 
 from sasctl import current_session
+from sasctl.core import RestObj
 
 with mock.patch('sasctl.core.requests.Session.request'):
     current_session('example.com', 'username', 'password')
@@ -46,3 +47,126 @@ def test_delete_module(caplog):
         assert mas.delete_module('spam') is None
         assert any("Object 'spam' not found" in r.msg for r in caplog.records)
 
+
+def test_define_steps():
+
+    # Mock module to be returned
+    module = RestObj(name='unittestmodule',
+                     stepIds=['step1', 'step2'])
+
+    # Mock module step with no inputs
+    step1 = RestObj(id='post')
+
+    # Mock module step with multiple inputs
+    step2 = RestObj({
+        "id": "score",
+        "inputs": [
+            {
+                "name": "age",
+                "type": "decimal",
+                "dim": 0,
+                "size": 0
+            },
+            {
+                "name": "b",
+                "type": "decimal",
+                "dim": 0,
+                "size": 0
+            },
+            {
+                "name": "chas",
+                "type": "decimal",
+                "dim": 0,
+                "size": 0
+            },
+            {
+                "name": "crim",
+                "type": "decimal",
+                "dim": 0,
+                "size": 0
+            },
+            {
+                "name": "dis",
+                "type": "decimal",
+                "dim": 0,
+                "size": 0
+            },
+            {
+                "name": "indus",
+                "type": "decimal",
+                "dim": 0,
+                "size": 0
+            },
+            {
+                "name": "lstat",
+                "type": "decimal",
+                "dim": 0,
+                "size": 0
+            },
+            {
+                "name": "nox",
+                "type": "decimal",
+                "dim": 0,
+                "size": 0
+            },
+            {
+                "name": "ptratio",
+                "type": "decimal",
+                "dim": 0,
+                "size": 0
+            },
+            {
+                "name": "rad",
+                "type": "decimal",
+                "dim": 0,
+                "size": 0
+            },
+            {
+                "name": "rm",
+                "type": "decimal",
+                "dim": 0,
+                "size": 0
+            },
+            {
+                "name": "tax",
+                "type": "decimal",
+                "dim": 0,
+                "size": 0
+            },
+            {
+                "name": "zn",
+                "type": "decimal",
+                "dim": 0,
+                "size": 0
+            }
+        ],
+        "outputs": [
+            {
+                "name": "em_prediction",
+                "type": "decimal",
+                "dim": 0,
+                "size": 0
+            },
+            {
+                "name": "p_price",
+                "type": "decimal",
+                "dim": 0,
+                "size": 0
+            },
+            {
+                "name": "_warn_",
+                "type": "string",
+                "dim": 0,
+                "size": 4
+            }
+        ]
+    })
+
+    with mock.patch('sasctl._services.microanalytic_score.MicroAnalyticScore.get_module') as get_module:
+        with mock.patch('sasctl._services.microanalytic_score.MicroAnalyticScore''.get_module_step') as get_step:
+            get_module.return_value = module
+            get_step.side_effect = [step1, step2]
+            result = mas.define_steps(None)
+
+    for step in get_step.side_effect:
+        assert hasattr(result, step.id)