def test_writeKnownHostFile(self):
KEY = '|1|abcdefghijklmnopqrstuvwxyz= ecdsa-sha2-nistp256 AAAAABCDEFGHIJKLMNOPQRSTUVWXYZ='
with TemporaryDirectory() as tmp:
knownHosts = os.path.expanduser('~/.ssh/known_hosts')
knownHostsSic = os.path.join(tmp, 'known_hosts')
if os.path.exists(knownHosts):
shutil.copyfile(knownHosts, knownHostsSic)
try:
sshtools.writeKnownHostsFile(KEY)
self.assertTrue(os.path.exists(knownHosts))
with open(knownHosts, 'rt') as f:
self.assertIn(KEY, [x.strip() for x in f.readlines()])
finally:
# restore original known_hosts file
if os.path.exists(knownHostsSic):
shutil.copyfile(knownHostsSic, knownHosts)
def test_writeKnownHostFile(self):
KEY = '|1|abcdefghijklmnopqrstuvwxyz= ecdsa-sha2-nistp256 AAAAABCDEFGHIJKLMNOPQRSTUVWXYZ='
with TemporaryDirectory() as tmp:
knownHosts = os.path.expanduser('~/.ssh/known_hosts')
knownHostsSic = os.path.join(tmp, 'known_hosts')
if os.path.exists(knownHosts):
shutil.copyfile(knownHosts, knownHostsSic)
try:
sshtools.writeKnownHostsFile(KEY)
self.assertExists(knownHosts)
with open(knownHosts, 'rt') as f:
self.assertIn(KEY, [x.strip() for x in f.readlines()])
finally:
# restore original known_hosts file
if os.path.exists(knownHostsSic):
shutil.copyfile(knownHostsSic, knownHosts)