Add test mode to pydbapi method

SUSE · Mar 23, 2020 · 1b03302 · 1b03302
1 parent 54fd869
commit 1b03302
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/salt/states/hanamod.py b/salt/states/hanamod.py
@@ -790,6 +790,12 @@ def pydbapi_extracted(
             'extraction)'.format(output_dir)
         return ret
 
+    if __opts__['test']:
+        ret['result'] = None
+        ret['comment'] = '{} would be extracted'.format(name)
+        ret['changes']['output_dir'] = output_dir
+        return ret
+
     __salt__['file.mkdir'](output_dir)
 
     try:

diff --git a/tests/unit/states/test_hanamod.py b/tests/unit/states/test_hanamod.py
@@ -1056,6 +1056,16 @@ def test_pydbapi_extracted_already_exists(self):
 
         mock_dir_exists.assert_called_once_with('/tmp/output')
 
+    def test_pydbapi_extracted_test(self):
+        ret = {'name': 'PYDBAPI.tar',
+               'changes': {'output_dir': '/tmp/output'},
+               'result': None,
+               'comment': 'PYDBAPI.tar would be extracted'}
+
+        with patch.dict(hanamod.__opts__, {'test': True}):
+            assert hanamod.pydbapi_extracted(
+                'PYDBAPI.tar', ['1234', '5678'], '/tmp/output', force=True) == ret
+
     def test_pydbapi_extracted_error(self):
         ret = {'name': 'PYDBAPI.tar',
                'changes': {},