1
0
Fork 0
mirror of https://github.com/retspen/webvirtcloud synced 2026-11-08 11:16:47 +00:00
webvirtcloud/instances/tests/test_resize_input.py
catborise 42a1783178 fix(instances): the Resize tab changes only what was asked
Disk sizes were pre-filled rounded to whole GiB, so a 3.5 GiB disk showed
4 and was grown, for good, whenever another disk was resized; an empty
field or a disk outside a pool was a 500. A disk's field is now empty with
its size as the placeholder: an empty field leaves the disk alone, a
larger whole number grows it, anything else is refused with a message.

vCPUs change only on a shut-off VM, also on a direct request: a running
VM got them at its next start, past the quota counted against its live
count. Current vCPUs or memory above the maximum are refused instead of
failing (and dropping vCPU hot-plug) or being capped by libvirt. A running
VM's maximum memory select is disabled, since its maximum is kept, and a
custom memory value is sent only while its box is ticked.
2026-10-26 13:49:37 +03:00

78 lines
4.1 KiB
Python

"""The Resize tab changes only what was asked, and refuses what libvirt would
not apply as asked."""
from unittest.mock import patch
from accounts.models import UserInstance
from appsettings.settings import get_settings
from computes.models import Compute
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 instances.models import Instance
from libvirt import VIR_DOMAIN_RUNNING
GIB = 1 << 30
class ResizeInputTests(TestCase):
def setUp(self):
get_settings()
self.user = get_user_model().objects.create_user("resize-user", password="pw")
self.client.force_login(self.user)
compute = Compute.objects.create(name="rs", hostname="192.0.2.1", login="u", password="p", type=1)
self.vm = Instance.objects.create(compute=compute, name="rs-vm", uuid="13131313-0000-0000-0000-000000000000")
UserInstance.objects.create(user=self.user, instance=self.vm, is_change=True)
def post(self, view, data, status=5, disks=()):
with patch("instances.views.utils.check_user_quota", return_value=""), patch(
"instances.models.wvmInstance"
) as wvm:
proxy = wvm.return_value
proxy.get_vcpu.return_value = 2
proxy.get_memory.return_value = 2048
proxy.get_cur_memory.return_value = 2048
proxy.get_status.return_value = status
proxy.instance.isActive.return_value = status != 5
proxy.get_disk_devices.return_value = [dict(disk) for disk in disks]
response = self.client.post(reverse(view, args=[self.vm.id]), data)
self.assertEqual(response.status_code, 302)
return [str(m) for m in get_messages(response.wsgi_request)], proxy
def test_only_a_disk_given_a_larger_size_grows(self):
# a 3.5 GiB disk left as it was is not grown; a disk outside a pool has no size
disks = [{"dev": "vda", "size": 3 * GIB + GIB // 2}, {"dev": "vdb", "size": 20 * GIB}, {"dev": "vdc", "size": None}]
messages, proxy = self.post("instances:resize_disk", {"disk_size_vda": "", "disk_size_vdb": "30"}, disks=disks)
proxy.resize_disk.assert_called_once()
self.assertEqual([(d["dev"], d["size_new"]) for d in proxy.resize_disk.call_args.args[0]], [("vdb", 30 * GIB)])
self.assertEqual(messages, ["Disk is resized: vdb"])
def test_a_disk_size_that_is_not_a_whole_number_is_refused(self):
messages, proxy = self.post("instances:resize_disk", {"disk_size_vda": "4.5"}, disks=[{"dev": "vda", "size": GIB}])
proxy.resize_disk.assert_not_called()
self.assertEqual(messages, ["Enter disk sizes as whole GiB"])
def test_vcpus_change_only_on_a_shut_off_vm(self):
messages, proxy = self.post("instances:resizevm_cpu", {"vcpu": "4", "cur_vcpu": "4"}, status=VIR_DOMAIN_RUNNING)
proxy.resize_cpu.assert_not_called()
self.assertEqual(messages, ["Shut off the VM to change its vCPUs"])
def test_current_vcpus_above_the_maximum_are_refused(self):
messages, proxy = self.post("instances:resizevm_cpu", {"vcpu": "2", "cur_vcpu": "4"})
proxy.resize_cpu.assert_not_called()
self.assertEqual(messages, ["The current vCPUs cannot be more than the maximum"])
def test_current_memory_above_the_maximum_is_refused(self):
messages, proxy = self.post("instances:resize_memory", {"memory": "4096", "cur_memory": "8192"})
proxy.resize_mem.assert_not_called()
self.assertEqual(messages, ["The current memory cannot be more than the maximum"])
def test_a_running_vm_offers_no_maximum_memory(self):
with patch("instances.models.wvmInstance"):
self.vm.__dict__["status"] = VIR_DOMAIN_RUNNING
from django.template.loader import render_to_string
html = render_to_string("instances/resize_tab.html", {"instance": self.vm, "can_change": True,
"memory_range": [1024, 2048]})
self.assertRegex(html, r'<select name="memory" class="form-select js-custom__toggle" disabled>')
self.assertNotIn('name="memory_custom"', html)