def test_create_key(self): ''' Tests if it create a key in the GPG keychain ''' ret = { 'res': True, 'fingerprint': u'F321F', 'message': 'GPG key pair successfully generated.' } ret1 = { 'fingerprint': '', 'res': False, 'message': 'gpg_passphrase not available in pillar.' } mock_conf = MagicMock(return_value='') mock_user = MagicMock(return_value={'home': 'salt'}) mock_item = MagicMock(return_value=False) with patch.dict( gpg.__salt__, { 'config.option': mock_conf, 'user.info': mock_user, 'pillar.item': mock_item }): self.assertDictEqual(gpg.create_key(), ret) self.assertDictEqual(gpg.create_key(use_passphrase=True), ret1)
def test_create_key_with_passphrase_with_gpg_passphrase_in_pillar(self): """ Test gpg.create_key with passphrase and gpg_passphrase pillar """ _user_mock = { "shell": "/bin/bash", "workphone": "", "uid": 0, "passwd": "x", "roomnumber": "", "gid": 0, "groups": ["root"], "home": self.gpghome, "fullname": "root", "homephone": "", "name": "root", } mock_opt = MagicMock(return_value="root") pillar_mock = MagicMock(return_value=GPG_TEST_KEY_PASSPHRASE) with patch.dict( gpg.__salt__, {"user.info": MagicMock(return_value=_user_mock)} ): with patch.dict(gpg.__salt__, {"config.option": mock_opt}), patch.dict( gpg.__salt__, {"pillar.get": pillar_mock} ): ret = gpg.create_key(use_passphrase=True) self.assertTrue(ret["res"]) self.assertTrue(ret["fingerprint"]) self.assertEqual(ret["message"], "GPG key pair successfully generated.")
def test_create_key_without_passphrase(self): """ Test gpg.create_key without a passphrase """ _user_mock = { "shell": "/bin/bash", "workphone": "", "uid": 0, "passwd": "x", "roomnumber": "", "gid": 0, "groups": ["root"], "home": self.gpghome, "fullname": "root", "homephone": "", "name": "root", } mock_opt = MagicMock(return_value="root") with patch.dict( gpg.__salt__, {"user.info": MagicMock(return_value=_user_mock)} ): with patch.dict(gpg.__salt__, {"config.option": mock_opt}): ret = gpg.create_key() self.assertTrue(ret["res"]) self.assertTrue(ret["fingerprint"]) self.assertEqual(ret["message"], "GPG key pair successfully generated.")
def test_create_key_with_passphrase_without_gpg_passphrase_in_pillar(self): """ Test gpg.create_key with passphrase but without gpg_passphrase pillar """ _user_mock = { "shell": "/bin/bash", "workphone": "", "uid": 0, "passwd": "x", "roomnumber": "", "gid": 0, "groups": ["root"], "home": self.gpghome, "fullname": "root", "homephone": "", "name": "root", } mock_opt = MagicMock(return_value="root") pillar_mock = MagicMock(return_value=None) with patch.dict( gpg.__salt__, {"user.info": MagicMock(return_value=_user_mock)} ): with patch.dict(gpg.__salt__, {"config.option": mock_opt}), patch.dict( gpg.__salt__, {"pillar.get": pillar_mock} ): ret = gpg.create_key(use_passphrase=True) self.assertFalse(ret["res"]) self.assertEqual( ret["message"], "gpg_passphrase not available in pillar." )
def test_create_key(self): ''' Tests if it create a key in the GPG keychain ''' ret = {'res': True, 'fingerprint': u'F321F', 'message': 'GPG key pair successfully generated.'} ret1 = {'fingerprint': '', 'res': False, 'message': 'gpg_passphrase not available in pillar.'} mock_conf = MagicMock(return_value='') mock_user = MagicMock(return_value={'home': 'salt'}) mock_item = MagicMock(return_value=False) with patch.dict(gpg.__salt__, {'config.option': mock_conf, 'user.info': mock_user, 'pillar.item': mock_item}): self.assertDictEqual(gpg.create_key(), ret) self.assertDictEqual(gpg.create_key(use_passphrase=True), ret1)