mirror of
https://github.com/retspen/webvirtcloud
synced 2026-11-08 19:26:48 +00:00
c55f52f025
131 test modules sat next to the code they test; instances alone had 51 of them beside views.py and utils.py. They move into a tests/ package per app, and the JavaScript checks into webvirtcloud/tests/js. Each non-empty tests.py becomes tests/test_views.py; the empty generated stubs are removed. Imports of the app's own modules become absolute, paths built from __file__ go one level up, the live helper is instances.tests.livetest, and coverage omits */tests/*. The suite discovers the same 960 tests as before.
57 lines
2.4 KiB
Python
57 lines
2.4 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
|
|
from django.urls import reverse
|
|
from instances.models import Instance
|
|
|
|
|
|
class ChangeNetworkViewTestCase(TestCase):
|
|
def setUp(self):
|
|
admin = get_user_model().objects.create_superuser("net_admin", "n@example.com", "pw")
|
|
self.client.force_login(admin)
|
|
compute = Compute.objects.create(name="net-compute", hostname="127.0.0.1", login="", password="", type=4)
|
|
self.instance = Instance.objects.create(
|
|
compute=compute, name="net-vm", uuid="11111111-2222-3333-4444-555555555555"
|
|
)
|
|
self.url = reverse("instances:change_network", args=[self.instance.id])
|
|
|
|
def post(self, data):
|
|
with patch("instances.models.wvmInstance") as wvm:
|
|
wvm.return_value.get_status.return_value = 5
|
|
self.client.post(self.url, data, HTTP_REFERER="/instances/%d/" % self.instance.id)
|
|
return wvm.return_value.change_network
|
|
|
|
def test_changes_the_nic_of_the_submitted_form(self):
|
|
change = self.post({
|
|
"net-old-mac-1": "52:54:00:00:00:02",
|
|
"net-mac-1": "52:54:00:00:00:03",
|
|
"net-source-1": "net:default",
|
|
"net-nwfilter-1": "",
|
|
"net-model-1": "e1000",
|
|
})
|
|
change.assert_called_once_with("52:54:00:00:00:02", "52:54:00:00:00:03", "default", "net", "e1000", "")
|
|
|
|
def test_invalid_mac_is_rejected(self):
|
|
change = self.post({
|
|
"net-old-mac-0": "52:54:00:00:00:01",
|
|
"net-mac-0": "52:54:00:00:00:01'/><evil/>",
|
|
"net-source-0": "net:default",
|
|
})
|
|
change.assert_not_called()
|
|
|
|
def test_missing_old_mac_is_rejected(self):
|
|
change = self.post({"net-mac-0": "52:54:00:00:00:03", "net-source-0": "net:default"})
|
|
change.assert_not_called()
|
|
|
|
def test_missing_referer_redirects_to_the_instance(self):
|
|
with patch("instances.models.wvmInstance") as wvm:
|
|
wvm.return_value.get_status.return_value = 5
|
|
response = self.client.post(self.url, {
|
|
"net-old-mac-0": "52:54:00:00:00:01",
|
|
"net-source-0": "net:default",
|
|
"net-model-0": "virtio",
|
|
})
|
|
self.assertEqual(response.status_code, 302)
|
|
self.assertEqual(response.url, f"/instances/{self.instance.id}/#network")
|