mirror of
https://github.com/retspen/webvirtcloud
synced 2026-10-18 09:00:07 +00:00
61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
import json
|
|
from django.contrib.auth import get_user_model
|
|
from django.test import Client, TestCase
|
|
from django.urls import reverse
|
|
|
|
from computes.models import Compute
|
|
from instances.models import Instance
|
|
from logs.models import Logs
|
|
from logs.views import addlogmsg
|
|
|
|
|
|
class LogsTestCase(TestCase):
|
|
def setUp(self):
|
|
User = get_user_model()
|
|
self.superuser = User.objects.create_superuser(
|
|
username="admin_logs", email="admin@example.com", password="adminpassword"
|
|
)
|
|
self.normal_user = User.objects.create_user(
|
|
username="normal_logs", email="user@example.com", password="userpassword"
|
|
)
|
|
self.compute = Compute.objects.create(
|
|
name="local_compute",
|
|
hostname="localhost",
|
|
type=1,
|
|
)
|
|
self.instance = Instance.objects.create(
|
|
compute=self.compute,
|
|
name="test-vm",
|
|
uuid="12345678-1234-1234-1234-123456789012",
|
|
)
|
|
|
|
def test_addlogmsg(self):
|
|
addlogmsg("admin", "localhost", "test-vm", "Started VM")
|
|
log = Logs.objects.filter(instance="test-vm").first()
|
|
self.assertIsNotNone(log)
|
|
self.assertEqual(log.user, "admin")
|
|
self.assertEqual(log.host, "localhost")
|
|
self.assertEqual(log.message, "Started VM")
|
|
|
|
def test_vm_logs_superuser_access(self):
|
|
addlogmsg("admin", "localhost", "test-vm", "Created VM")
|
|
client = Client()
|
|
client.login(username="admin_logs", password="adminpassword")
|
|
|
|
response = client.get(reverse("vm_logs", args=["test-vm"]))
|
|
self.assertEqual(response.status_code, 200)
|
|
|
|
data = json.loads(response.content.decode("utf-8"))
|
|
self.assertIsInstance(data, list)
|
|
self.assertEqual(len(data), 1)
|
|
self.assertEqual(data[0]["user"], "admin")
|
|
self.assertEqual(data[0]["message"], "Created VM")
|
|
self.assertEqual(data[0]["instance"], "test-vm")
|
|
|
|
def test_vm_logs_non_superuser_forbidden(self):
|
|
client = Client()
|
|
client.login(username="normal_logs", password="userpassword")
|
|
|
|
response = client.get(reverse("vm_logs", args=["test-vm"]))
|
|
# superuser_only decorator redirects non-superusers to index or login
|
|
self.assertNotEqual(response.status_code, 200)
|