import re
from accounts.models import UserAttributes, UserInstance, UserSSHKey
from appsettings.models import AppSettings
from computes.models import Compute
from django.contrib.auth import get_user_model
from django.contrib.auth.models import Permission
from django.http.response import Http404
from django.shortcuts import reverse
from django.test import TestCase
from instances.views import instance
from libvirt import VIR_DOMAIN_UNDEFINE_NVRAM
from vrtManager.create import wvmCreate
from vrtManager.util import randomUUID
from .models import Flavor, Instance
from .utils import refr
class InstancesTestCase(TestCase):
@classmethod
def setUpClass(cls):
super().setUpClass()
# Add users for testing purposes
User = get_user_model()
cls.admin_user = User.objects.get(pk=1)
cls.test_user = User.objects.create(username="test-user")
UserAttributes.objects.create(
user=cls.test_user,
max_instances=1,
max_cpus=1,
max_memory=128,
max_disk_size=1,
)
permission = Permission.objects.get(codename="clone_instances")
cls.test_user.user_permissions.add(permission)
# Add localhost compute
cls.compute = Compute(
name="test-compute",
hostname="localhost",
login="",
password="",
details="local",
type=4,
)
cls.compute.save()
cls.connection = wvmCreate(
cls.compute.hostname,
cls.compute.login,
cls.compute.password,
cls.compute.type,
)
# Add disks for testing
cls.connection.create_volume(
"default",
"test-volume",
1,
"qcow2",
False,
0,
0,
)
# XML for testing vm
with open("conf/test-vm.xml", "r") as f:
cls.xml = f.read()
# Create testing vm from XML
cls.connection._defineXML(cls.xml)
refr(cls.compute)
cls.instance: Instance = Instance.objects.get(pk=1)
@classmethod
def tearDownClass(cls):
# Destroy testing vm
cls.instance.proxy.delete_all_disks()
cls.instance.proxy.delete(VIR_DOMAIN_UNDEFINE_NVRAM)
super().tearDownClass()
def setUp(self):
self.client.login(username="admin", password="admin")
self.rsa_key = "ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAgQC6OOdbfv27QVnSC6sKxGaHb6YFc+3gxCkyVR3cTSXE/n5BEGf8aOgBpepULWa1RZfxYHY14PlKULDygdXSdrrR2kNSwoKz/Oo4d+3EE92L7ocl1+djZbptzgWgtw1OseLwbFik+iKlIdqPsH+IUQvX7yV545ZQtAP8Qj1R+uCqkw== test@test"
def test_index(self):
response = self.client.get(reverse("instances:index"))
self.assertEqual(response.status_code, 200)
self.client.force_login(self.test_user)
response = self.client.get(reverse("instances:index"))
self.assertEqual(response.status_code, 200)
def test_create_select_type(self):
response = self.client.get(
reverse("instances:create_instance_select_type", args=[1])
)
self.assertEqual(response.status_code, 200)
def test_instance_page(self):
response = self.client.get(
reverse("instances:instance", args=[self.instance.id])
)
self.assertEqual(response.status_code, 200)
self.client.force_login(self.test_user)
response = self.client.get(
reverse("instances:instance", args=[self.instance.id])
)
self.assertRaises(Http404)
# def test_create_volume(self):
# response = self.client.post(
# reverse('create_volume', args=[self.compute.id, 'default']),
# {
# 'name': 'test',
# 'format': 'qcow2',
# 'size': '1',
# 'meta_prealloc': False,
# },
# )
# self.assertRedirects(response, reverse('storage', args=[self.compute.id, 'default']))
def test_create_destroy_instance(self):
# Create
response = self.client.get(
reverse(
"instances:create_instance", args=[self.compute.id, "x86_64", "q35"]
)
)
self.assertEqual(response.status_code, 200)
response = self.client.post(
reverse(
"instances:create_instance", args=[self.compute.id, "x86_64", "q35"]
),
{
"name": "test",
"firmware": "BIOS",
"vcpu": 1,
"vcpu_mode": "host-model",
"memory": 128,
"device0": "disk",
"bus0": "virtio",
"images": "test-volume.qcow2",
"storage-control": "default",
"image-control": "test.qcow2",
"networks": "default",
"network-control": "default",
"cache_mode": "directsync",
"nwfilter": "",
"graphics": "spice",
"video": "vga",
"listener_addr": "0.0.0.0",
"console_pass": "",
"qemu_ga": False,
"virtio": True,
"create": True,
},
)
self.assertEqual(response.status_code, 302)
instance_qs: Instance = Instance.objects.filter(name="test")
self.assertEqual(len(instance_qs), 1)
instance = instance_qs[0]
# Destroy
response = self.client.get(reverse("instances:destroy", args=[instance.id]))
self.assertEqual(response.status_code, 200)
response = self.client.post(
reverse("instances:destroy", args=[instance.id]),
{}, # do not destroy disk image
HTTP_REFERER=reverse("index"),
)
self.assertRedirects(response, reverse("instances:index"))
def test_create_from_xml(self):
uuid = randomUUID()
xml = self.xml.replace("test-vm", "test-vm-xml")
xml = re.sub("\s?.*?", f"{uuid}", xml)
response = self.client.post(
reverse("instances:create_instance_select_type", args=[self.compute.id]),
{
"create_xml": True,
"dom_xml": xml,
},
)
self.assertEqual(response.status_code, 302)
xml_instance_qs: Instance = Instance.objects.filter(name="test-vm-xml")
self.assertEqual(len(xml_instance_qs), 1)
xml_instance = xml_instance_qs[0]
# destroy started instance to maximize coverage
xml_instance.proxy.start()
response = self.client.post(
reverse("instances:destroy", args=[xml_instance.id]),
{}, # do not delete disk image
HTTP_REFERER=reverse("index"),
)
self.assertRedirects(response, reverse("instances:index"))
def test_resize_cpu(self):
self.assertEqual(self.instance.vcpu, 1)
self.assertEqual(self.instance.cur_vcpu, 1)
response = self.client.post(
reverse("instances:resizevm_cpu", args=[self.instance.id]),
{
"vcpu": 4,
"cur_vcpu": 2,
},
)
self.assertRedirects(
response, reverse("instances:instance", args=[self.instance.id]) + "#resize"
)
# reset cached properties
del self.instance.vcpu
del self.instance.cur_vcpu
self.assertEqual(self.instance.vcpu, 4)
self.assertEqual(self.instance.cur_vcpu, 2)
def test_resize_cpu_with_quota(self):
# test for non admin user with quotas
vcpu = self.instance.vcpu
cur_vcpu = self.instance.cur_vcpu
UserInstance.objects.create(
user=self.test_user, instance=self.instance, is_change=True
)
self.client.force_login(self.test_user)
response = self.client.post(
reverse("instances:resizevm_cpu", args=[self.instance.id]),
{
"vcpu": 4,
"cur_vcpu": 2,
},
)
self.assertRedirects(
response, reverse("instances:instance", args=[self.instance.id]) + "#resize"
)
del self.instance.vcpu
del self.instance.cur_vcpu
# no changes as user reached quota
self.assertEqual(self.instance.vcpu, vcpu)
self.assertEqual(self.instance.cur_vcpu, cur_vcpu)
def test_resize_memory(self):
self.assertEqual(self.instance.memory, 128)
self.assertEqual(self.instance.cur_memory, 128)
response = self.client.post(
reverse("instances:resize_memory", args=[self.instance.id]),
{"memory": 512, "cur_memory": 256},
)
self.assertRedirects(
response, reverse("instances:instance", args=[self.instance.id]) + "#resize"
)
del self.instance.memory
del self.instance.cur_memory
self.assertEqual(self.instance.memory, 512)
self.assertEqual(self.instance.cur_memory, 256)
response = self.client.post(
reverse("instances:resize_memory", args=[self.instance.id]),
{"memory_custom": 500, "cur_memory_custom": 200},
)
self.assertRedirects(
response, reverse("instances:instance", args=[self.instance.id]) + "#resize"
)
del self.instance.memory
del self.instance.cur_memory
self.assertEqual(self.instance.memory, 500)
self.assertEqual(self.instance.cur_memory, 200)
def test_resize_memory_with_quota(self):
# test for non admin user with quotas
memory = self.instance.memory
cur_memory = self.instance.cur_memory
UserInstance.objects.create(
user=self.test_user, instance=self.instance, is_change=True
)
self.client.force_login(self.test_user)
response = self.client.post(
reverse("instances:resize_memory", args=[self.instance.id]),
{"memory": 512, "cur_memory": 256},
)
self.assertRedirects(
response, reverse("instances:instance", args=[self.instance.id]) + "#resize"
)
del self.instance.memory
del self.instance.cur_memory
# no changes as user reached quota
self.assertEqual(self.instance.memory, memory)
self.assertEqual(self.instance.cur_memory, cur_memory)
def test_resize_disk(self):
self.assertEqual(self.instance.disks[0]["size"], 1024**3)
response = self.client.post(
reverse("instances:resize_disk", args=[self.instance.id]),
{
"disk_size_vda": 2,
},
)
self.assertRedirects(
response, reverse("instances:instance", args=[self.instance.id]) + "#resize"
)
del self.instance.disks
self.assertEqual(self.instance.disks[0]["size"], 2 * 1024**3)
def test_resize_disk_with_quota(self):
# test for non admin user with quotas
disk_size = self.instance.disks[0]["size"]
UserInstance.objects.create(
user=self.test_user, instance=self.instance, is_change=True
)
self.client.force_login(self.test_user)
response = self.client.post(
reverse("instances:resize_disk", args=[self.instance.id]),
{
"disk_size_vda": 3,
},
)
self.assertRedirects(
response, reverse("instances:instance", args=[self.instance.id]) + "#resize"
)
# no changes as user reached quota
del self.instance.disks
self.assertEqual(self.instance.disks[0]["size"], disk_size)
def test_add_delete_new_volume(self):
self.assertEqual(len(self.instance.disks), 1)
response = self.client.post(
reverse("instances:add_new_vol", args=[self.instance.id]),
{
"storage": "default",
"name": "test-volume-2",
"size": 1,
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.disks
self.assertEqual(len(self.instance.disks), 2)
response = self.client.post(
reverse("instances:delete_vol", args=[self.instance.id]),
{
"storage": "default",
"dev": "vdb",
"name": "test-volume-2.qcow2",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.disks
self.assertEqual(len(self.instance.disks), 1)
def test_detach_attach_volume(self):
# detach volume
response = self.client.post(
reverse("instances:detach_vol", args=[self.instance.id]),
{
"dev": "vda",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.disks
self.assertEqual(len(self.instance.disks), 0)
# reattach volume
response = self.client.post(
reverse("instances:add_existing_vol", args=[self.instance.id]),
{
"selected_storage": "default",
"vols": "test-volume.qcow2",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.disks
self.assertEqual(len(self.instance.disks), 1)
def test_edit_volume(self):
response = self.client.post(
reverse("instances:edit_volume", args=[self.instance.id]),
{
"vol_path": "/var/lib/libvirt/images/test-volume.qcow2",
# 'vol_shareable': False,
# 'vol_readonly': False,
"vol_bus": "virtio",
"vol_bus_old": "virtio",
"vol_format": "qcow2",
"dev": "vda",
"edit_volume": True,
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
def test_attach_detach_cdrom(self):
self.assertEqual(len(self.instance.media), 1)
response = self.client.post(
reverse("instances:add_cdrom", args=[self.instance.id]),
{
"bus": "sata",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.media
self.assertEqual(len(self.instance.media), 2)
# create dummy iso
# with tempfile.NamedTemporaryFile() as f:
# f.write(b'\x00' * 1024**2)
# response = self.client.post(
# reverse('storage', args=[instance.compute.id, 'default']),
# {
# 'iso_upload': True,
# 'file': f
# },
# )
# detach CD-ROM drive
response = self.client.post(
reverse("instances:detach_cdrom", args=[self.instance.id, "sda"]),
{},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.media
self.assertEqual(len(self.instance.media), 1)
def test_snapshots(self):
self.assertEqual(len(self.instance.snapshots), 0)
response = self.client.post(
reverse("instances:snapshot", args=[self.instance.id]),
{
"name": "test-snapshot",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.snapshots
self.assertEqual(len(self.instance.snapshots), 1)
response = self.client.post(
reverse("instances:revert_snapshot", args=[self.instance.id]),
{
"name": "test-snapshot",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
response = self.client.post(
reverse("instances:delete_snapshot", args=[self.instance.id]),
{
"name": "test-snapshot",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.snapshots
self.assertEqual(len(self.instance.snapshots), 0)
def test_autostart(self):
self.assertEqual(self.instance.autostart, 0)
response = self.client.post(
reverse("instances:set_autostart", args=[self.instance.id]),
{},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.autostart
self.assertEqual(self.instance.autostart, 1)
response = self.client.post(
reverse("instances:unset_autostart", args=[self.instance.id]),
{},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.autostart
self.assertEqual(self.instance.autostart, 0)
def test_bootmenu(self):
self.assertEqual(self.instance.bootmenu, True)
response = self.client.post(
reverse("instances:unset_bootmenu", args=[self.instance.id]),
{},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.bootmenu
self.assertEqual(self.instance.bootmenu, False)
response = self.client.post(
reverse("instances:set_bootmenu", args=[self.instance.id]),
{},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.bootmenu
self.assertEqual(self.instance.bootmenu, True)
def test_guest_agent(self):
self.assertEqual(self.instance.guest_agent, False)
response = self.client.post(
reverse("instances:set_guest_agent", args=[self.instance.id]),
{"guest_agent": True},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.guest_agent
self.assertEqual(self.instance.guest_agent, True)
response = self.client.post(
reverse("instances:set_guest_agent", args=[self.instance.id]),
{"guest_agent": False},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.guest_agent
self.assertEqual(self.instance.guest_agent, False)
def test_video_model(self):
self.assertEqual(self.instance.video_model, "vga")
response = self.client.post(
reverse("instances:set_video_model", args=[self.instance.id]),
{"video_model": "virtio"},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.video_model
self.assertEqual(self.instance.video_model, "virtio")
def test_owner(self):
self.assertEqual(UserInstance.objects.count(), 0)
response = self.client.post(
reverse("instances:add_owner", args=[self.instance.id]),
{"user_id": self.admin_user.id},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
self.assertEqual(UserInstance.objects.count(), 1)
user_instance: UserInstance = UserInstance.objects.get(id=1)
self.assertEqual(user_instance.instance_id, self.instance.id)
self.assertEqual(user_instance.user_id, self.admin_user.id)
# test when no multiple owners allowed
setting = AppSettings.objects.get(key="ALLOW_INSTANCE_MULTIPLE_OWNER")
setting.value = "False"
setting.save()
response = self.client.post(
reverse("instances:add_owner", args=[self.instance.id]),
{"user_id": self.test_user.id},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
self.assertEqual(UserInstance.objects.count(), 1)
response = self.client.post(
reverse("instances:del_owner", args=[self.instance.id]),
{"userinstance": user_instance.id},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
self.assertEqual(UserInstance.objects.count(), 0)
def test_clone(self):
instance_count = Instance.objects.count()
response = self.client.post(
reverse("instances:clone", args=[self.instance.id]),
{
"name": "test-vm-clone",
"clone-net-mac-0": "de:ad:be:ef:de:ad",
"disk-vda": "test-clone.img",
"clone-title": "",
"clone-description": "",
"clone": "",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
self.assertEqual(Instance.objects.count(), instance_count + 1)
clone_qs = Instance.objects.filter(name="test-vm-clone")
self.assertEqual(len(clone_qs), 1)
clone = clone_qs[0]
self.assertEqual(clone.proxy.get_net_devices()[0]["mac"], "de:ad:be:ef:de:ad")
response = self.client.post(
reverse("instances:snapshot", args=[clone.id]),
{
"name": "test",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
response = self.client.post(
reverse("instances:destroy", args=[clone.id]),
{"delete_disk": True, "delete_nvram": True},
HTTP_REFERER=reverse("index"),
)
self.assertRedirects(response, reverse("instances:index"))
self.assertEqual(Instance.objects.count(), instance_count)
def test_clone_with_quota(self):
# test for non admin user with quotas
instance_count = Instance.objects.count()
UserInstance.objects.create(user=self.test_user, instance=self.instance)
self.client.force_login(self.test_user)
response = self.client.post(
reverse("instances:clone", args=[self.instance.id]),
{
"name": "test-vm-clone",
"clone-net-mac-0": "de:ad:be:ef:de:ad",
"disk-vda": "test-clone.img",
"clone-title": "",
"clone-description": "",
"clone": "",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
# no new instances created as user reached quota
self.assertEqual(Instance.objects.count(), instance_count)
def test_clone_errors(self):
instance_count = Instance.objects.count()
# duplicate name
response = self.client.post(
reverse("instances:clone", args=[self.instance.id]),
{
"name": "test-vm",
"clone-net-mac-0": "de:ad:be:ef:de:ad",
"disk-vda": "test.img",
"clone-title": "",
"clone-description": "",
"clone": "",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
self.assertEqual(Instance.objects.count(), instance_count)
# wrong name
response = self.client.post(
reverse("instances:clone", args=[self.instance.id]),
{
"name": "!@#$",
"clone-net-mac-0": "de:ad:be:ef:de:ad",
"disk-vda": "!@#$.img",
"clone-title": "",
"clone-description": "",
"clone": "",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
self.assertEqual(Instance.objects.count(), instance_count)
# wrong mac
response = self.client.post(
reverse("instances:clone", args=[self.instance.id]),
{
"name": "test-vm-clone",
"clone-net-mac-0": "gh:ad:be:ef:de:ad",
"disk-vda": "test-clone.img",
"clone-title": "",
"clone-description": "",
"clone": "",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
self.assertEqual(Instance.objects.count(), instance_count)
def test_console(self):
response = self.client.post(
reverse("instances:update_console", args=[self.instance.id]),
{"type": "spice", "listen_on": "0.0.0.0", "password": "", "keymap": "auto"},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
def test_status(self):
response = self.client.get(reverse("instances:status", args=[self.instance.id]))
self.assertEqual(response.status_code, 200)
def test_stats(self):
response = self.client.get(reverse("instances:stats", args=[self.instance.id]))
self.assertEqual(response.status_code, 200)
def test_guess_mac_address(self):
response = self.client.get(
reverse("instances:guess_mac_address", args=[self.instance.name])
)
self.assertEqual(response.status_code, 200)
def test_random_mac_address(self):
response = self.client.get(reverse("instances:random_mac_address"))
self.assertEqual(response.status_code, 200)
def test_guess_clone_name(self):
response = self.client.get(reverse("instances:guess_clone_name"))
self.assertEqual(response.status_code, 200)
def test_sshkeys(self):
UserSSHKey.objects.create(
keyname="keyname", keypublic=self.rsa_key, user=self.test_user
)
UserInstance.objects.create(user=self.test_user, instance=self.instance)
response = self.client.get(
reverse("instances:sshkeys", args=[self.instance.id])
)
self.assertEqual(response.status_code, 200)
response = self.client.get(
reverse("instances:sshkeys", args=[self.instance.id]) + "?plain=true"
)
self.assertEqual(response.status_code, 200)
def test_check_instance(self):
response = self.client.get(
reverse("instances:check_instance", args=["test-vm"])
)
self.assertEqual(response.status_code, 200)
self.assertJSONEqual(response.content, {"vname": "test-vm", "exists": True})
def test_start_template(self):
# starting templates must fail
self.assertEqual(self.instance.status, 5)
self.instance.is_template = True
self.instance.save()
response = self.client.get(
reverse("instances:poweron", args=[self.instance.id]),
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.status
self.assertEqual(self.instance.status, 5)
self.instance.is_template = False
self.instance.save()
def test_power(self):
# poweron
self.assertEqual(self.instance.status, 5)
response = self.client.get(
reverse("instances:poweron", args=[self.instance.id]),
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.status
self.assertEqual(self.instance.status, 1)
# suspend
response = self.client.get(
reverse("instances:suspend", args=[self.instance.id]),
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.status
self.assertEqual(self.instance.status, 3)
# resume
response = self.client.get(
reverse("instances:resume", args=[self.instance.id]),
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.status
self.assertEqual(self.instance.status, 1)
# poweroff
response = self.client.get(
reverse("instances:poweroff", args=[self.instance.id]),
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
# as no OS is installed ACPI won't work
del self.instance.status
self.assertEqual(self.instance.status, 1)
# powercycle
response = self.client.get(
reverse("instances:powercycle", args=[self.instance.id]),
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.status
self.assertEqual(self.instance.status, 1)
# force_off
response = self.client.get(
reverse("instances:force_off", args=[self.instance.id]),
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.status
self.assertEqual(self.instance.status, 5)
def test_vv_file(self):
response = self.client.get(
reverse("instances:getvvfile", args=[self.instance.id])
)
self.assertEqual(response.status_code, 200)
def test_vcpu_hotplug(self):
response = self.client.post(
reverse("instances:set_vcpu_hotplug", args=[self.instance.id]),
{"vcpu_hotplug": "True"},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
def test_change_network(self):
self.assertEqual(self.instance.networks[0]["mac"], "52:54:00:a2:3c:e7")
response = self.client.post(
reverse("instances:change_network", args=[self.instance.id]),
{
"net-mac-0": "52:54:00:a2:3c:e8",
"net-source-0": "net:default",
"net-nwfilter-0": "",
"net-model-0": "virtio",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.networks
self.assertEqual(self.instance.networks[0]["mac"], "52:54:00:a2:3c:e8")
def test_add_delete_network(self):
self.assertEqual(len(self.instance.networks), 1)
net_mac = self.instance.networks[0]["mac"]
response = self.client.post(
reverse("instances:add_network", args=[self.instance.id]),
{
"add-net-mac": "52:54:00:a2:3c:e9",
"add-net-network": "net:default",
"add_net-nwfilter": "",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.networks
self.assertEqual(len(self.instance.networks), 2)
self.assertEqual(self.instance.networks[1]["mac"], "52:54:00:a2:3c:e9")
response = self.client.post(
reverse("instances:delete_network", args=[self.instance.id]),
{
"delete_network": "52:54:00:a2:3c:e9",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.networks
self.assertEqual(len(self.instance.networks), 1)
self.assertEqual(self.instance.networks[0]["mac"], net_mac)
def test_set_link_state(self):
self.assertEqual(self.instance.networks[0]["state"], "up")
response = self.client.post(
reverse("instances:set_link_state", args=[self.instance.id]),
{
"mac": self.instance.networks[0]["mac"],
"set_link_state": "up",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.networks
self.assertEqual(self.instance.networks[0]["state"], "down")
def test_set_unset_qos(self):
self.assertEqual(len(self.instance.qos.keys()), 0)
net_mac = self.instance.networks[0]["mac"]
response = self.client.post(
reverse("instances:set_qos", args=[self.instance.id]),
{
"net-mac-0": net_mac,
"qos_direction": "inbound",
"qos_average": 1,
"qos_peak": 1,
"qos_burst": 1,
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.qos
self.assertEqual(len(self.instance.qos.keys()), 1)
response = self.client.post(
reverse("instances:unset_qos", args=[self.instance.id]),
{
"net-mac": net_mac,
"qos_direction": "inbound",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.qos
self.assertEqual(len(self.instance.qos.keys()), 0)
# test on running instance
# self.instance.proxy.start()
# response = self.client.post(
# reverse('instances:set_qos', args=[self.instance.id]),
# {
# 'net-mac-0': net_mac,
# 'qos_direction': 'inbound',
# 'qos_average': 1,
# 'qos_peak': 1,
# 'qos_burst': 1,
# },
# HTTP_REFERER=reverse('index'),
# )
# self.assertEqual(response.status_code, 302)
# self.instance.proxy.force_shutdown()
def test_change_options(self):
self.assertEqual(self.instance.title, "")
self.assertEqual(self.instance.description, "")
response = self.client.post(
reverse("instances:change_options", args=[self.instance.id]),
{
"title": "test-vm-title",
"description": "test-vm description",
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
del self.instance.title
del self.instance.description
self.assertEqual(self.instance.title, "test-vm-title")
self.assertEqual(self.instance.description, "test-vm description")
def test_flavors(self):
response = self.client.get(reverse("instances:flavor_create"))
self.assertEqual(response.status_code, 200)
response = self.client.post(
reverse("instances:flavor_create"),
{
"label": "test_flavor",
"memory": 256,
"vcpu": 1,
"disk": 10,
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
id = Flavor.objects.last().id
response = self.client.get(reverse("instances:flavor_update", args=[id]))
self.assertEqual(response.status_code, 200)
response = self.client.post(
reverse("instances:flavor_update", args=[id]),
{
"label": "test_flavor_",
"memory": 256,
"vcpu": 1,
"disk": 10,
},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
response = self.client.get(reverse("instances:flavor_delete", args=[id]))
self.assertEqual(response.status_code, 200)
response = self.client.post(
reverse("instances:flavor_delete", args=[id]),
{},
HTTP_REFERER=reverse("index"),
)
self.assertEqual(response.status_code, 302)
# def donot_test_instance(self):
# compute: Compute = Compute.objects.get(pk=1)
# user: User = User.objects.get(pk=1)
# # delete started instance with disks
# self.instance.proxy.start()
# del self.instance.status
# self.assertEqual(self.instance.status, 1)
# # create volume
# response = self.client.post(
# reverse('create_volume', args=[compute.id, 'default']),
# {
# 'name': 'test3',
# 'format': 'qcow2',
# 'size': '1',
# 'meta_prealloc': False,
# },
# )
# self.assertRedirects(response, reverse('storage', args=[compute.id, 'default']))
# # delete volume
# response = self.client.post(
# reverse('instances:delete_vol', args=[instance.id]),
# {
# 'storage': 'default',
# 'dev': 'vdb',
# 'name': 'test3.qcow2',
# },
# HTTP_REFERER=reverse('index'),
# )
# self.assertEqual(response.status_code, 302)
# , list(response.context['messages'])[0]