def test_set_ssh_key(self):
with self.assertRaises(SystemExit):
ssh_key.set_ssh_key(SMALL_KEY,
authorized_keys = self.authorized_keys)
self.assertEqual(SMALL_KEY, open(self.authorized_keys).read())
self.assertFalse(ssh_key.key_locked(self.authorized_keys))
self.assertSuccess('{"comment": "my@address"}')
def test_key_uploaded_no_login_not_locked(self):
with open(self.authorized_keys, 'w') as f:
self.assertFalse(ssh_key.key_locked(self.authorized_keys))
def test_key_not_uploaded_but_login_success_not_locked(self):
self.assertFalse(ssh_key.key_locked(self.authorized_keys))
def test_key_locked(self):
uci.get = mock.Mock(return_value = 'true')
with open(self.authorized_keys, 'w') as f:
self.assertTrue(ssh_key.key_locked(self.authorized_keys))
