mirror of
https://github.com/valitydev/salt.git
synced 2024-11-08 01:18:58 +00:00
d01b8da9aa
When using `--no-clean` while running tests, some files/dirs still need to be removed, if not, test will fail. This commit tries to addresses those issues.
105 lines
3.9 KiB
Python
105 lines
3.9 KiB
Python
'''
|
|
Test the ssh_known_hosts state
|
|
'''
|
|
import os
|
|
import shutil
|
|
import integration
|
|
|
|
|
|
KNOWN_HOSTS = os.path.join(integration.TMP, 'known_hosts')
|
|
GITHUB_FINGERPRINT = '16:27:ac:a5:76:28:2d:36:63:1b:56:4d:eb:df:a6:48'
|
|
GITHUB_IP = '207.97.227.239'
|
|
|
|
|
|
class SSHKnownHostsStateTest(integration.ModuleCase):
|
|
'''
|
|
Validate the ssh state
|
|
'''
|
|
def tearDown(self):
|
|
if os.path.isfile(KNOWN_HOSTS):
|
|
os.remove(KNOWN_HOSTS)
|
|
super(SSHKnownHostsStateTest, self).tearDown()
|
|
|
|
def test_present(self):
|
|
'''
|
|
ssh_known_hosts.present
|
|
'''
|
|
kwargs = {'name': 'github.com',
|
|
'user': 'root',
|
|
'fingerprint': GITHUB_FINGERPRINT,
|
|
'config': KNOWN_HOSTS
|
|
}
|
|
# test first
|
|
_ret = self.run_state('ssh_known_hosts.present', test=True, **kwargs)
|
|
ret = list(_ret.values())[0]
|
|
self.assertEqual(ret['result'], None, ret)
|
|
# save once, new key appears
|
|
_ret = self.run_state('ssh_known_hosts.present', **kwargs)
|
|
ret = list(_ret.values())[0]
|
|
self.assertEqual(ret['changes']['new']['fingerprint'],
|
|
GITHUB_FINGERPRINT, ret)
|
|
# save twice, no changes
|
|
_ret = self.run_state('ssh_known_hosts.present', **kwargs)
|
|
ret = list(_ret.values())[0]
|
|
self.assertEqual(ret['changes'], {}, ret)
|
|
# test again, nothing is about to be changed
|
|
_ret = self.run_state('ssh_known_hosts.present', test=True, **kwargs)
|
|
ret = list(_ret.values())[0]
|
|
self.assertEqual(ret['result'], None, ret)
|
|
# then add a record for IP address
|
|
_ret = self.run_state('ssh_known_hosts.present',
|
|
**dict(kwargs, name=GITHUB_IP))
|
|
ret = list(_ret.values())[0]
|
|
self.assertEqual(ret['changes']['new']['fingerprint'],
|
|
GITHUB_FINGERPRINT, ret)
|
|
# record for every host must be available
|
|
ret = self.run_function('ssh.get_known_host', ['root', 'github.com'],
|
|
config=KNOWN_HOSTS)
|
|
self.assertFalse(ret is None)
|
|
ret = self.run_function('ssh.get_known_host', ['root', GITHUB_IP],
|
|
config=KNOWN_HOSTS)
|
|
self.assertFalse(ret is None)
|
|
|
|
def test_present_fail(self):
|
|
# save something wrong
|
|
_ret = self.run_state('ssh_known_hosts.present',
|
|
name='github.com',
|
|
user='root',
|
|
fingerprint='aa:bb:cc:dd',
|
|
config=KNOWN_HOSTS)
|
|
ret = list(_ret.values())[0]
|
|
self.assertFalse(ret['result'], ret)
|
|
|
|
def test_absent(self):
|
|
'''
|
|
ssh_known_hosts.absent
|
|
'''
|
|
known_hosts = os.path.join(integration.FILES, 'ssh', 'known_hosts')
|
|
shutil.copyfile(known_hosts, KNOWN_HOSTS)
|
|
if not os.path.isfile(KNOWN_HOSTS):
|
|
self.skipTest("Unable to copy {0} to {1}".format(known_hosts, KNOWN_HOSTS))
|
|
|
|
kwargs = {'name': 'github.com', 'user': 'root', 'config': KNOWN_HOSTS}
|
|
# test first
|
|
_ret = self.run_state('ssh_known_hosts.absent', test=True, **kwargs)
|
|
ret = list(_ret.values())[0]
|
|
self.assertEqual(ret['result'], None, ret)
|
|
# remove once, the key is gone
|
|
_ret = self.run_state('ssh_known_hosts.absent', **kwargs)
|
|
ret = list(_ret.values())[0]
|
|
self.assertEqual(ret['changes']['old']['fingerprint'],
|
|
GITHUB_FINGERPRINT, ret)
|
|
# remove twice, nothing has changed
|
|
_ret = self.run_state('ssh_known_hosts.absent', **kwargs)
|
|
ret = list(_ret.values())[0]
|
|
self.assertEqual(ret['changes'], {}, ret)
|
|
# test again
|
|
_ret = self.run_state('ssh_known_hosts.absent', test=True, **kwargs)
|
|
ret = list(_ret.values())[0]
|
|
self.assertEqual(ret['result'], None, ret)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
from integration import run_tests
|
|
run_tests(SSHKnownHostsStateTest)
|