mirror of
https://github.com/retspen/webvirtcloud
synced 2026-11-09 19:56:48 +00:00
a6f9e4b573
The page checked only that the VM's host answers; a VM undefined with virsh then failed at its first read and the page was a server error. It now says the VM is no longer on its host and goes back to the list, whose sync forgets the record (tr and zh too).
54 lines
2.8 KiB
Python
54 lines
2.8 KiB
Python
"""The VM page reads the user's rights from the VM's own record, and a VM
|
|
removed outside the panel sends the user back to the list."""
|
|
|
|
from unittest.mock import PropertyMock, patch
|
|
|
|
from django.contrib.auth import get_user_model
|
|
from django.contrib.messages import get_messages
|
|
from django.test import TestCase
|
|
from django.urls import reverse
|
|
from libvirt import VIR_ERR_NO_DOMAIN, libvirtError
|
|
|
|
from accounts.models import UserInstance
|
|
from computes.models import Compute
|
|
from instances.models import Instance
|
|
|
|
|
|
class VmPageRecordsTestCase(TestCase):
|
|
def setUp(self):
|
|
self.owner = get_user_model().objects.create_user("rows-owner", password="x")
|
|
self.compute = Compute.objects.create(name="rows-c", hostname="192.0.2.70", login="root", password="", type=1)
|
|
self.vm = Instance.objects.create(compute=self.compute, name="web", uuid="aaaaaaaa-0000-0000-0000-000000000071")
|
|
UserInstance.objects.create(instance=self.vm, user=self.owner, is_change=False)
|
|
self.client.force_login(self.owner)
|
|
|
|
def page(self, wvm_error=None):
|
|
with patch("instances.models.wvmInstance", side_effect=wvm_error) as wvm, patch.object(
|
|
Compute, "status", new_callable=PropertyMock, return_value=True
|
|
):
|
|
proxy = wvm.return_value
|
|
proxy.get_memory.return_value = proxy.get_cur_memory.return_value = 1024
|
|
proxy.get_max_memory.return_value = 4096
|
|
proxy.get_hostdevs.return_value = ([], [], [])
|
|
proxy.get_console_type.return_value = "vnc"
|
|
proxy.get_status.return_value = 5
|
|
for getter in ("get_networks", "get_ifaces", "get_storages", "get_disk_devices", "get_media_devices",
|
|
"get_net_devices"):
|
|
getattr(proxy, getter).return_value = []
|
|
return self.client.get(reverse("instances:instance", args=[self.vm.id]))
|
|
|
|
def test_the_rights_come_from_the_vms_own_record(self):
|
|
# another VM of the same name on the compute, owned with more rights:
|
|
# its record was not refreshed after a rename outside the panel
|
|
other = Instance.objects.create(compute=self.compute, name="web", uuid="aaaaaaaa-0000-0000-0000-000000000072")
|
|
UserInstance.objects.create(instance=other, user=self.owner, is_change=True)
|
|
response = self.page()
|
|
self.assertEqual(response.status_code, 200)
|
|
self.assertEqual(response.context["userinstance"].instance, self.vm)
|
|
|
|
def test_a_vm_removed_outside_the_panel_sends_the_user_to_the_list(self):
|
|
error = libvirtError("Domain not found")
|
|
error.get_error_code = lambda: VIR_ERR_NO_DOMAIN
|
|
response = self.page(wvm_error=error)
|
|
self.assertRedirects(response, reverse("instances:index"), fetch_redirect_response=False)
|
|
self.assertEqual([str(m) for m in get_messages(response.wsgi_request)], ["'web' is no longer on its host."])
|