mirror of
https://github.com/retspen/webvirtcloud
synced 2026-11-08 19:26:48 +00:00
42ab293dc5
- 25 vrtManager test modules configured settings themselves when Django had none (MAC_OUI, and in test_network_ip a fake django module): under manage.py test settings are always configured, so it never ran. - webvirtcloud/tests/test_common_tags.py tested no tag: it fetched an unknown URL, as test_error_pages and test_logging already do. - .gitignore covers the parallel runs' test_db_N.sqlite3 files too.
44 lines
2 KiB
Python
44 lines
2 KiB
Python
"""A running guest's root password and SSH keys through its QEMU guest
|
|
agent: the password as a crypt(3) hash, the key added to those it has. Each
|
|
is one agent command with a timeout: libvirt's own calls for them would wait
|
|
for a hung agent indefinitely, and setting the domain's agent timeout would
|
|
change it for every other agent call too."""
|
|
|
|
import base64
|
|
import json
|
|
import unittest
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from libvirt import libvirtError
|
|
from vrtManager import instance as module
|
|
from vrtManager.instance import wvmInstance
|
|
|
|
|
|
class AgentAccessTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
self.vm = wvmInstance.__new__(wvmInstance) # no connection: only these calls
|
|
self.vm.instance = MagicMock()
|
|
patcher = patch.object(module, "qemuAgentCommand", return_value='{"return":{}}')
|
|
self.agent = patcher.start()
|
|
self.addCleanup(patcher.stop)
|
|
|
|
def sent(self):
|
|
(dom, command, timeout, flags), = [c.args for c in self.agent.call_args_list]
|
|
self.assertIs(dom, self.vm.instance)
|
|
self.assertEqual((timeout, flags), (10, 0))
|
|
return json.loads(command)
|
|
|
|
def test_the_password_goes_as_a_hash(self):
|
|
self.vm.set_user_password("root", "$6$salt$hash")
|
|
self.assertEqual(self.sent(), {"execute": "guest-set-user-password", "arguments": {
|
|
"username": "root", "password": base64.b64encode(b"$6$salt$hash").decode(), "crypted": True}})
|
|
|
|
def test_the_key_is_added_to_the_users_keys(self):
|
|
self.vm.add_authorized_key("root", "ssh-ed25519 AAAA laptop")
|
|
self.assertEqual(self.sent(), {"execute": "guest-ssh-add-authorized-keys", "arguments": {
|
|
"username": "root", "keys": ["ssh-ed25519 AAAA laptop"], "reset": False}})
|
|
|
|
def test_the_agents_error_is_passed_on(self):
|
|
self.agent.side_effect = libvirtError("guest agent didn't respond to command within '10' seconds")
|
|
with self.assertRaisesRegex(libvirtError, "within '10' seconds"):
|
|
self.vm.set_user_password("root", "$6$salt$hash")
|