mirror of
https://github.com/retspen/webvirtcloud
synced 2026-11-08 19:26:48 +00:00
0f9bbd3e86
Settings > Console gets a Console type field. Serial removes the VM's graphics device (adding a pty console when there is none); VNC adds the graphics device new VMs get, on localhost, and then applies the form's VNC fields, so an administrator can open it to all interfaces for a TCP compute. Serial is offered only while SERIAL_CONSOLE_ENABLED is on.
42 lines
2.1 KiB
Python
42 lines
2.1 KiB
Python
from unittest.mock import patch
|
|
|
|
from computes.models import Compute
|
|
from django.contrib.auth import get_user_model
|
|
from django.test import TestCase, override_settings
|
|
from django.urls import reverse
|
|
from instances.models import Instance
|
|
|
|
|
|
class UpdateConsoleTestCase(TestCase):
|
|
def setUp(self):
|
|
admin = get_user_model().objects.create_superuser("console_admin", "c@example.com", "pw")
|
|
self.client.force_login(admin)
|
|
compute = Compute.objects.create(name="console-compute", hostname="127.0.0.1", login="", password="", type=4)
|
|
self.instance = Instance.objects.create(
|
|
compute=compute, name="console-vm", uuid="11111111-2222-3333-4444-555555555555"
|
|
)
|
|
|
|
def post(self, data):
|
|
with patch("instances.models.wvmInstance") as wvm:
|
|
self.client.post(reverse("instances:update_console", args=[self.instance.id]), data)
|
|
return wvm.return_value
|
|
|
|
def test_console_settings_are_applied(self):
|
|
proxy = self.post({"listen_on": "127.0.0.1", "password": "", "keymap": "de"})
|
|
proxy.set_console_keymap.assert_called_once_with("de")
|
|
proxy.set_console_listener_addr.assert_called_once_with("127.0.0.1")
|
|
|
|
@override_settings(SERIAL_CONSOLE_ENABLED=True)
|
|
def test_a_new_console_type_and_its_vnc_settings(self):
|
|
data = {"listen_on": "0.0.0.0", "keymap": "de"}
|
|
for old, new, vnc_settings in (("vnc", "pty", False), ("pty", "vnc", True)):
|
|
with self.subTest(new), patch("instances.models.wvmInstance") as wvm:
|
|
wvm.return_value.get_console_type.return_value = old
|
|
self.client.post(reverse("instances:update_console", args=[self.instance.id]), dict(data, console_type=new))
|
|
wvm.return_value.set_console_type.assert_called_once_with(new)
|
|
self.assertEqual(wvm.return_value.set_console_listener_addr.called, vnc_settings)
|
|
|
|
@override_settings(SERIAL_CONSOLE_ENABLED=False)
|
|
def test_serial_is_not_offered_while_the_serial_console_is_off(self):
|
|
proxy = self.post({"console_type": "pty", "listen_on": "127.0.0.1", "keymap": "de"})
|
|
proxy.set_console_type.assert_not_called()
|