def test_gpg_sign(self):
config_user = MagicMock(return_value="salt")
user_info = MagicMock(return_value={
"name": "salt",
"home": self.gpghome,
"uid": 1000
})
pillar_mock = MagicMock(
return_value={"gpg_passphrase": GPG_TEST_KEY_PASSPHRASE})
with patch.dict(gpg.__salt__, {"config.option": config_user}):
with patch.dict(gpg.__salt__, {"user.info": user_info}):
with patch.dict(gpg.__salt__, {"pillar.get": pillar_mock}):
ret = gpg.import_key(None, self.gpgfile_priv, "salt",
self.gpghome)
self.assertEqual(ret["res"], True)
gpg_text_input = "The quick brown fox jumped over the lazy dog"
gpg_sign_output = gpg.sign(
config_user,
GPG_TEST_KEY_ID,
gpg_text_input,
None,
None,
True,
self.gpghome,
)
self.assertIsNotNone(gpg_sign_output)
def test_gpg_sign(gpghome):
config_user = MagicMock(return_value="salt")
user_info = MagicMock(return_value={
"name": "salt",
"home": str(gpghome.path),
"uid": 1000
})
pillar_mock = MagicMock(
return_value={"gpg_passphrase": GPG_TEST_KEY_PASSPHRASE})
with patch.dict(gpg.__salt__, {"config.option": config_user}):
with patch.dict(gpg.__salt__, {"user.info": user_info}):
with patch.dict(gpg.__salt__, {"pillar.get": pillar_mock}):
ret = gpg.import_key(None, str(gpghome.priv), "salt",
str(gpghome.path))
assert ret["res"] is True
gpg_text_input = "The quick brown fox jumped over the lazy dog"
gpg_sign_output = gpg.sign(
config_user,
GPG_TEST_KEY_ID,
gpg_text_input,
None,
None,
True,
str(gpghome.path),
)
assert gpg_sign_output is not None
def test_gpg_import_priv_key(self):
config_user = MagicMock(return_value='salt')
user_info = MagicMock(return_value={'name': 'salt', 'home': self.gpghome, 'uid': 1000})
with patch.dict(gpg.__salt__, {'config.option': config_user}):
with patch.dict(gpg.__salt__, {'user.info': user_info}):
ret = gpg.import_key(None, self.gpgfile_priv, 'salt', self.gpghome)
self.assertEqual(ret['res'], True)
def test_gpg_import_priv_key(self):
config_user = MagicMock(return_value="salt")
user_info = MagicMock(
return_value={"name": "salt", "home": self.gpghome, "uid": 1000}
)
with patch.dict(gpg.__salt__, {"config.option": config_user}):
with patch.dict(gpg.__salt__, {"user.info": user_info}):
ret = gpg.import_key(None, self.gpgfile_priv, "salt", self.gpghome)
self.assertEqual(ret["res"], True)
def test_gpg_import_priv_key(gpghome):
config_user = MagicMock(return_value="salt")
user_info = MagicMock(
return_value={"name": "salt", "home": str(gpghome.path), "uid": 1000}
)
with patch.dict(gpg.__salt__, {"config.option": config_user}):
with patch.dict(gpg.__salt__, {"user.info": user_info}):
ret = gpg.import_key(None, str(gpghome.priv), "salt", str(gpghome.path))
assert ret["res"] is True
def test_gpg_sign(self):
config_user = MagicMock(return_value='salt')
user_info = MagicMock(return_value={'name': 'salt', 'home': self.gpghome, 'uid': 1000})
pillar_mock = MagicMock(return_value={'gpg_passphrase': GPG_TEST_KEY_PASSPHRASE})
with patch.dict(gpg.__salt__, {'config.option': config_user}):
with patch.dict(gpg.__salt__, {'user.info': user_info}):
with patch.dict(gpg.__salt__, {'pillar.get': pillar_mock}):
ret = gpg.import_key(None, self.gpgfile_priv, 'salt', self.gpghome)
self.assertEqual(ret['res'], True)
gpg_text_input = 'The quick brown fox jumped over the lazy dog'
gpg_sign_output = gpg.sign(config_user, GPG_TEST_KEY_ID, gpg_text_input, None, None, True, self.gpghome)
self.assertIsNotNone(gpg_sign_output)
def test_import_key(self):
'''
Tests if it import a key from text or file.
'''
ret = {'message': 'Unable to import key.', 'res': False}
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.assertRaises(SaltInvocationError, gpg.import_key)
with patch('salt.utils.flopen', mock_open(read_data='')) as fp:
fp.side_effect = IOError()
self.assertRaises(SaltInvocationError, gpg.import_key,
filename='/path/to/public-key-file')
gpg.GPG_1_3_1 = True
self.assertDictEqual(gpg.import_key(text='-BEGIN PGP PUBLIC KEY BLOCK-'), ret)
gpg.GPG_1_3_1 = False
self.assertDictEqual(gpg.import_key(text='-BEGIN PGP PUBLIC KEY BLOCK-'), ret)
def test_import_key(self):
'''
Tests if it import a key from text or file.
'''
ret = {'message': 'Unable to import key.', 'res': False}
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.assertRaises(SaltInvocationError, gpg.import_key)
with patch('salt.utils.flopen', mock_open(read_data='')) as fp:
fp.side_effect = IOError()
self.assertRaises(SaltInvocationError, gpg.import_key,
filename='/path/to/public-key-file')
self.assertDictEqual(gpg.import_key(text='-BEGIN PGP PUBLIC KEY BLOCK-'), ret)
