def test_export_key_without_passphrase(self):
"""
Test gpg.export_key without 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}):
with patch(
"salt.modules.gpg.gnupg.GPG.export_keys",
MagicMock(return_value=GPG_TEST_PUB_KEY),
) as gnupg_export_keys:
ret = gpg.export_key("xxxxxxxxxxxxxxxx")
self.assertEqual(ret, GPG_TEST_PUB_KEY)
gnupg_export_keys.assert_called_with(
["xxxxxxxxxxxxxxxx"], False, expect_passphrase=False,
)
def test_export_key_with_passphrase_without_gpg_passphrase_in_pillar(self):
"""
Test gpg.export_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}
):
with patch(
"salt.modules.gpg.gnupg.GPG.export_keys",
MagicMock(return_value=GPG_TEST_PUB_KEY),
) as gnupg_export_keys:
with self.assertRaises(SaltInvocationError):
assert gpg.export_key("xxxxxxxxxxxxxxxx", use_passphrase=True)
gnupg_export_keys.assert_not_called()
def test_export_key(self):
'''
Tests if it export a key from the GPG keychain
'''
mock_conf = MagicMock(return_value='')
mock_user = MagicMock(return_value={'home': 'salt'})
with patch.dict(gpg.__salt__, {'config.option': mock_conf,
'user.info': mock_user}):
self.assertTrue(gpg.export_key(keyids='3F0C8E90D459D89A'))
def test_export_key(self):
'''
Tests if it export a key from the GPG keychain
'''
mock_conf = MagicMock(return_value='')
mock_user = MagicMock(return_value={'home': 'salt'})
with patch.dict(gpg.__salt__, {'config.option': mock_conf,
'user.info': mock_user}):
self.assertTrue(gpg.export_key(keyids='3F0C8E90D459D89A'))
def test_export_key_with_passphrase_with_gpg_passphrase_in_pillar(self):
"""
Test gpg.export_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}
):
with patch(
"salt.modules.gpg.gnupg.GPG.export_keys",
MagicMock(return_value=GPG_TEST_PUB_KEY),
) as gnupg_export_keys:
ret = gpg.export_key("xxxxxxxxxxxxxxxx", use_passphrase=True)
self.assertEqual(ret, GPG_TEST_PUB_KEY)
gnupg_export_keys.assert_called_with(
["xxxxxxxxxxxxxxxx"], False, passphrase=GPG_TEST_KEY_PASSPHRASE,
)
