def test_add_known_host_existing_valid_key(self):
self.setup_mocks_add_known_host()
self.check_output.return_value = UNIT2_HOST_KEY_1
self.ssh_known_host_key.return_value = UNIT2_HOST_KEY_1
with patch_open() as (mock_open, mock_file):
ssh_migrations.add_known_host('juju-4665be-20180716142533-8',
'nova-compute-lxd')
self.assertFalse(mock_open.called)
self.assertFalse(self.remove_known_host.called)
def test_add_known_host(self):
self.setup_mocks_add_known_host()
self.check_output.return_value = UNIT1_HOST_KEY_1
self.ssh_known_host_key.return_value = ''
with patch_open() as (mock_open, mock_file):
ssh_migrations.add_known_host('juju-4665be-20180716142533-8',
'nova-compute-lxd')
mock_file.write.assert_called_with(UNIT1_HOST_KEY_1 + '\n')
mock_open.assert_called_with('/somedir/known_hosts', 'a')
self.assertFalse(self.remove_known_host.called)