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)
