adding seed unit test case

tests/unit/modules/seed_test.py

@@ -0,0 +1,100 @@
+# -*- coding: utf-8 -*-
+'''
+    :codeauthor: :email:`Jayesh Kariya <jayeshk@saltstack.com>`
+'''
+# Import Python libs
+from __future__ import absolute_import
+
+# Import Salt Testing Libs
+from salttesting import skipIf, TestCase
+from salttesting.mock import (
+    NO_MOCK,
+    NO_MOCK_REASON,
+    MagicMock,
+    mock_open,
+    patch)
+
+from salttesting.helpers import ensure_in_syspath
+
+ensure_in_syspath('../../')
+
+# Import Salt Libs
+from salt.modules import seed
+import os
+import shutil
+
+# Globals
+seed.__salt__ = {}
+seed.__opts__ = {}
+
+
+@skipIf(NO_MOCK, NO_MOCK_REASON)
+class SeedTestCase(TestCase):
+    '''
+    Test cases for salt.modules.seed
+    '''
+    def test_prep_bootstrap(self):
+        '''
+        Test to update and get the random script to a random place
+        '''
+        with patch.dict(seed.__salt__,
+                        {'config.gather_bootstrap_script': MagicMock()}):
+            with patch.object(os.path, 'join', return_value='A'):
+                with patch.object(os.path, 'exists', return_value=True):
+                    with patch.object(os, 'chmod', return_value=None):
+                        with patch.object(shutil, 'copy', return_value=None):
+                            self.assertEqual(seed.prep_bootstrap('mpt'), 'A')
+
+    def test_apply_(self):
+        '''
+        Test to seed a location (disk image, directory, or block device)
+        with the minion config, approve the minion's key, and/or install
+        salt-minion.
+        '''
+        mock = MagicMock(side_effect=[False, {'type': 'type',
+                                              'target': 'target'},
+                                      {'type': 'type', 'target': 'target'},
+                                      {'type': 'type', 'target': 'target'}])
+        with patch.dict(seed.__salt__, {'file.stats': mock}):
+            self.assertEqual(seed.apply_('path'), 'path does not exist')
+
+            with patch.object(seed, '_mount', return_value=False):
+                self.assertEqual(seed.apply_('path'),
+                                 'target could not be mounted')
+
+            with patch.object(seed, '_mount', return_value=True):
+                with patch.object(os.path, 'join', return_value='A'):
+                    with patch.object(os, 'makedirs',
+                                      MagicMock(side_effect=OSError('f'))):
+                        with patch.object(os.path, 'isdir',
+                                          return_value=False):
+                            self.assertRaises(OSError, seed.apply_, 'p')
+
+                    with patch.object(os, 'makedirs', MagicMock()):
+                        with patch.object(seed, 'mkconfig', return_value='A'):
+                            with patch.object(seed, '_check_install',
+                                              return_value=False):
+                                with patch.object(seed, '_umount',
+                                                  return_value=None):
+                                    self.assertFalse(seed.apply_('path',
+                                                                 install=False))
+
+    def test_mkconfig(self):
+        '''
+        Test to generate keys and config and put them in a tmp directory.
+        '''
+        file_data = '\n'.join(['#', 'A B C D,E,F G H'])
+        with patch('salt.utils.fopen', mock_open(read_data=file_data),
+                   create=True) as mfile:
+            mfile.return_value.__iter__.return_value = file_data.splitlines()
+            with patch.dict(seed.__opts__, {'master': 'master'}):
+                with patch.object(os.path, 'join', return_value=False):
+                    with patch.dict(seed.__salt__, {'pillar.ext': MagicMock()}):
+                        self.assertDictEqual(seed.mkconfig(None, 'tmp', 'id'),
+                                             {'pubkey': False, 'config': False,
+                                              'privkey': False})
+
+
+if __name__ == '__main__':
+    from integration import run_tests
+    run_tests(SeedTestCase, needs_daemon=False)