test: expand unit test coverage across admin, accounts, vrtManager, logs, appsettings, and datasource

datasource/test_datasource.py

@@ -0,0 +1,57 @@
+import json
+from unittest.mock import MagicMock, patch
+
+from django.test import RequestFactory, TestCase
+from django.urls import reverse
+
+from computes.models import Compute
+from datasource.views import get_client_ip, get_hostname_by_ip, os_index, os_metadata_json, os_userdata
+
+
+class DataSourceTestCase(TestCase):
+    def setUp(self):
+        self.factory = RequestFactory()
+        self.client.login(username="admin", password="admin")
+        self.compute = Compute.objects.create(
+            name="test_compute",
+            hostname="localhost",
+            type=1,
+        )
+
+    def test_os_index(self):
+        response = self.client.get(reverse("ds_openstack_index"))
+        self.assertEqual(response.status_code, 200)
+        self.assertIn("latest", response.content.decode("utf-8"))
+
+    def test_os_metadata_json_latest(self):
+        response = self.client.get(reverse("ds_openstack_metadata", args=["latest"]))
+        self.assertEqual(response.status_code, 200)
+        data = json.loads(response.content.decode("utf-8"))
+        self.assertEqual(data["uuid"], "iid-dswebvirtcloud")
+        self.assertIn("hostname", data)
+
+    def test_os_metadata_json_invalid_version(self):
+        response = self.client.get(reverse("ds_openstack_metadata", args=["v1"]))
+        self.assertEqual(response.status_code, 404)
+
+    def test_os_userdata_invalid_version(self):
+        response = self.client.get(reverse("ds_openstack_userdata", args=["v1"]))
+        self.assertEqual(response.status_code, 404)
+
+    def test_get_client_ip_direct(self):
+        request = self.factory.get("/")
+        request.META["REMOTE_ADDR"] = "192.168.1.10"
+        self.assertEqual(get_client_ip(request), "192.168.1.10")
+
+    def test_get_client_ip_forwarded(self):
+        request = self.factory.get("/")
+        request.META["HTTP_X_FORWARDED_FOR"] = "10.0.0.1, 192.168.1.50"
+        self.assertEqual(get_client_ip(request), "192.168.1.50")
+
+    def test_get_hostname_by_ip_fallback(self):
+        # Invalid IP will cause gethostbyaddr to fail and return the IP string itself
+        self.assertEqual(get_hostname_by_ip("256.256.256.256"), "256.256.256.256")
+
+    def test_vdi_url_not_found_compute(self):
+        response = self.client.get(reverse("vdi_url", args=[9999, "vm1"]))
+        self.assertEqual(response.status_code, 404)