arthexis 0.1.26__py3-none-any.whl → 0.1.28__py3-none-any.whl

nodes/tests.py

@@ -43,9 +43,14 @@ from django.test import (
 from django.urls import reverse
 from django.contrib.auth import get_user_model
 from django.contrib import admin
+from django.contrib import messages
 from django.contrib.admin import helpers
 from django.contrib.auth.models import Permission
 from django_celery_beat.models import IntervalSchedule, PeriodicTask
+from core.celery_utils import (
+    periodic_task_name_variants,
+    slugify_task_name,
+)
 from django.conf import settings
 from django.utils import timezone
 from urllib.parse import urlparse
@@ -54,6 +59,7 @@ from . import dns as dns_utils
 from selenium.common.exceptions import WebDriverException
 from .classifiers import run_default_classifiers
 from .utils import capture_rpi_snapshot, capture_screenshot, save_screenshot
+from .feature_checks import feature_checks
 from django.db.utils import DatabaseError
 
 from .admin import NodeAdmin
@@ -90,7 +96,7 @@ class NodeBadgeColorTests(TestCase):
         node = Node.objects.create(
             hostname="watchtower",
             address="10.1.0.1",
-            port=8000,
+            port=8888,
             mac_address="00:aa:bb:cc:dd:01",
             role=self.watchtower,
         )
@@ -127,6 +133,23 @@ class NodeTests(TestCase):
         NodeRole.objects.get_or_create(name="Terminal")
         NodeRole.objects.get_or_create(name="Interface")
 
+    def test_terminal_role_enables_clipboard_feature_by_default(self):
+        role = NodeRole.objects.get(name="Terminal")
+        feature, _ = NodeFeature.objects.get_or_create(
+            slug="clipboard-poll", defaults={"display": "Clipboard Poll"}
+        )
+        feature.roles.add(role)
+
+        node = Node.objects.create(
+            hostname="terminal-node",
+            address="10.0.0.5",
+            port=8888,
+            mac_address="aa:bb:cc:dd:ee:ff",
+            role=role,
+        )
+
+        self.assertTrue(node.has_feature("clipboard-poll"))
+
 
 class NodeGetLocalDatabaseUnavailableTests(SimpleTestCase):
     def test_get_local_handles_database_errors(self):
@@ -317,7 +340,7 @@ class NodeGetLocalTests(TestCase):
             data={
                 "hostname": "local",
                 "address": "127.0.0.1",
-                "port": 8000,
+                "port": 8888,
                 "mac_address": "00:11:22:33:44:55",
             },
             content_type="application/json",
@@ -437,7 +460,7 @@ class NodeGetLocalTests(TestCase):
             hostname="best-ip",
             address="gateway.local",
             ipv4_address="198.51.100.5",
-            port=8000,
+            port=8888,
             mac_address="00:11:22:33:44:77",
         )
         self.assertEqual(node.get_best_ip(), "198.51.100.5")
@@ -502,7 +525,7 @@ class NodeGetLocalTests(TestCase):
             data={
                 "hostname": "lcd",
                 "address": "127.0.0.1",
-                "port": 8000,
+                "port": 8888,
                 "mac_address": "00:aa:bb:cc:dd:ee",
                 "features": ["clipboard-poll"],
             },
@@ -517,7 +540,7 @@ class NodeGetLocalTests(TestCase):
             data={
                 "hostname": "lcd",
                 "address": "127.0.0.1",
-                "port": 8000,
+                "port": 8888,
                 "mac_address": "00:aa:bb:cc:dd:ee",
                 "features": [],
             },
@@ -684,7 +707,7 @@ class NodeGetLocalTests(TestCase):
         payload = {
             "hostname": "cors",
             "address": "127.0.0.1",
-            "port": 8000,
+            "port": 8888,
             "mac_address": "10:20:30:40:50:60",
         }
         response = self.client.post(
@@ -702,7 +725,7 @@ class NodeGetLocalTests(TestCase):
         payload = {
             "hostname": "visitor",
             "address": "127.0.0.1",
-            "port": 8000,
+            "port": 8888,
             "mac_address": "aa:bb:cc:dd:ee:00",
         }
         response = self.client.post(
@@ -746,7 +769,7 @@ class NodeGetLocalTests(TestCase):
         payload = {
             "hostname": "visitor",
             "address": "127.0.0.1",
-            "port": 8000,
+            "port": 8888,
             "mac_address": "aa:bb:cc:dd:ee:11",
             "public_key": public_bytes,
             "token": token,
@@ -832,7 +855,7 @@ class NodeEnsureKeysTests(TestCase):
         self.node = Node.objects.create(
             hostname="ensure-host",
             address="127.0.0.1",
-            port=8000,
+            port=8888,
             mac_address="00:11:22:33:44:55",
         )
 
@@ -888,7 +911,7 @@ class NodeInfoViewTests(TestCase):
             address="10.0.0.10",
             ipv4_address="10.0.0.10",
             ipv6_address="2001:db8::10",
-            port=8000,
+            port=8888,
             mac_address=self.mac,
             public_endpoint="local",
             current_relation=Node.Relation.SELF,
@@ -952,7 +975,7 @@ class RegisterVisitorNodeMessageTests(TestCase):
         self.visitor = Node.objects.create(
             hostname="visitor-node",
             address="10.0.0.100",
-            port=8000,
+            port=8888,
             mac_address="00:10:20:30:40:50",
             role=self.role,
         )
@@ -1134,7 +1157,7 @@ class NodeRegisterCurrentTests(TestCase):
                     Node.objects.create(
                         hostname="localnode",
                         address="192.168.1.5",
-                        port=8000,
+                        port=8888,
                         mac_address="00:ff:ee:dd:cc:bb",
                         installed_version="1.9.0",
                         installed_revision="oldrev",
@@ -1192,7 +1215,7 @@ class NodeRegisterCurrentTests(TestCase):
                     Node.objects.create(
                         hostname="samever",
                         address="192.168.1.6",
-                        port=8000,
+                        port=8888,
                         mac_address="00:ff:ee:dd:cc:cc",
                         installed_version="1.0.0",
                         installed_revision="rev1",
@@ -1372,7 +1395,7 @@ class NodeRegisterCurrentTests(TestCase):
         sender = Node.objects.create(
             hostname="sender",
             address="10.0.0.1",
-            port=8000,
+            port=8888,
             mac_address="00:11:22:33:44:cc",
             public_key=public_key,
         )
@@ -1614,8 +1637,11 @@ class NodeRegisterCurrentTests(TestCase):
             port=9000,
             mac_address="00:11:22:33:44:99",
         )
-        task_name = f"poll_clipboard_node_{node.pk}"
-        PeriodicTask.objects.filter(name=task_name).delete()
+        raw_name = f"poll_clipboard_node_{node.pk}"
+        task_name = slugify_task_name(raw_name)
+        PeriodicTask.objects.filter(
+            name__in=periodic_task_name_variants(raw_name)
+        ).delete()
         NodeFeatureAssignment.objects.create(node=node, feature=feature)
         self.assertTrue(PeriodicTask.objects.filter(name=task_name).exists())
         NodeFeatureAssignment.objects.filter(node=node, feature=feature).delete()
@@ -1632,8 +1658,11 @@ class NodeRegisterCurrentTests(TestCase):
             port=9100,
             mac_address="00:11:22:33:44:aa",
         )
-        task_name = f"capture_screenshot_node_{node.pk}"
-        PeriodicTask.objects.filter(name=task_name).delete()
+        raw_name = f"capture_screenshot_node_{node.pk}"
+        task_name = slugify_task_name(raw_name)
+        PeriodicTask.objects.filter(
+            name__in=periodic_task_name_variants(raw_name)
+        ).delete()
         NodeFeatureAssignment.objects.create(node=node, feature=feature)
         self.assertTrue(PeriodicTask.objects.filter(name=task_name).exists())
         NodeFeatureAssignment.objects.filter(node=node, feature=feature).delete()
@@ -1652,14 +1681,18 @@ class NodeRegisterCurrentTests(TestCase):
                 "base_path": settings.BASE_DIR,
             },
         )
-        PeriodicTask.objects.filter(name="pages_purge_landing_leads").delete()
+        raw_name = "pages_purge_landing_leads"
+        task_name = slugify_task_name(raw_name)
+        PeriodicTask.objects.filter(
+            name__in=periodic_task_name_variants(raw_name)
+        ).delete()
         NodeFeatureAssignment.objects.get_or_create(node=node, feature=feature)
         self.assertTrue(
-            PeriodicTask.objects.filter(name="pages_purge_landing_leads").exists()
+            PeriodicTask.objects.filter(name=task_name).exists()
         )
         NodeFeatureAssignment.objects.filter(node=node, feature=feature).delete()
         self.assertFalse(
-            PeriodicTask.objects.filter(name="pages_purge_landing_leads").exists()
+            PeriodicTask.objects.filter(name=task_name).exists()
         )
 
     def test_ocpp_session_report_task_syncs_with_feature(self):
@@ -1675,8 +1708,11 @@ class NodeRegisterCurrentTests(TestCase):
                 "base_path": settings.BASE_DIR,
             },
         )
-        task_name = "ocpp_send_daily_session_report"
-        PeriodicTask.objects.filter(name=task_name).delete()
+        raw_name = "ocpp_send_daily_session_report"
+        task_name = slugify_task_name(raw_name)
+        PeriodicTask.objects.filter(
+            name__in=periodic_task_name_variants(raw_name)
+        ).delete()
 
         with patch("nodes.models.mailer.can_send_email", return_value=True):
             NodeFeatureAssignment.objects.get_or_create(node=node, feature=feature)
@@ -1699,8 +1735,11 @@ class NodeRegisterCurrentTests(TestCase):
                 "base_path": settings.BASE_DIR,
             },
         )
-        task_name = "ocpp_send_daily_session_report"
-        PeriodicTask.objects.filter(name=task_name).delete()
+        raw_name = "ocpp_send_daily_session_report"
+        task_name = slugify_task_name(raw_name)
+        PeriodicTask.objects.filter(
+            name__in=periodic_task_name_variants(raw_name)
+        ).delete()
 
         with patch("nodes.models.mailer.can_send_email", return_value=False):
             NodeFeatureAssignment.objects.get_or_create(node=node, feature=feature)
@@ -1736,7 +1775,7 @@ class NodeAdminTests(TestCase):
         return Node.objects.create(
             hostname="localnode",
             address="127.0.0.1",
-            port=8000,
+            port=8888,
             mac_address=Node.get_current_mac(),
         )
 
@@ -1811,7 +1850,7 @@ class NodeAdminTests(TestCase):
         Node.objects.create(
             hostname=hostname,
             address="127.0.0.1",
-            port=8000,
+            port=8888,
             mac_address=None,
         )
 
@@ -1942,7 +1981,7 @@ class NodeAdminTests(TestCase):
         local_node = Node.objects.create(
             hostname="local",
             address="127.0.0.1",
-            port=8000,
+            port=8888,
             mac_address="00:11:22:33:44:aa",
             public_key="LOCAL-PUB",
         )
@@ -2189,6 +2228,34 @@ class NodeAdminTests(TestCase):
             response, "Completed 0 of 1 feature check(s) successfully.", html=False
         )
 
+    @pytest.mark.feature("audio-capture")
+    @patch("nodes.models.Node._has_audio_capture_device", return_value=False)
+    def test_check_features_for_eligibility_audio_capture_requires_device(
+        self, mock_device
+    ):
+        self._create_local_node()
+        feature, _ = NodeFeature.objects.get_or_create(
+            slug="audio-capture", defaults={"display": "Audio Capture"}
+        )
+        changelist_url = reverse("admin:nodes_nodefeature_changelist")
+        response = self.client.post(
+            changelist_url,
+            {
+                "action": "check_features_for_eligibility",
+                "_selected_action": [str(feature.pk)],
+            },
+            follow=True,
+        )
+        self.assertEqual(response.status_code, 200)
+        self.assertContains(
+            response,
+            "No audio recording device detected on localnode for Audio Capture. This feature can be enabled manually.",
+            html=False,
+        )
+        self.assertContains(
+            response, "Completed 0 of 1 feature check(s) successfully.", html=False
+        )
+
     @pytest.mark.feature("screenshot-poll")
     def test_enable_selected_features_enables_manual_feature(self):
         node = self._create_local_node()
@@ -2756,7 +2823,7 @@ class NodeProxyGatewayTests(TestCase):
         self.node = Node.objects.create(
             hostname="requester",
             address="127.0.0.1",
-            port=8000,
+            port=8888,
             mac_address="aa:bb:cc:dd:ee:aa",
             public_key=public_key,
         )
@@ -2981,7 +3048,7 @@ class RFIDExportViewTests(TestCase):
         self.local_node = Node.objects.create(
             hostname="local",
             address="127.0.0.1",
-            port=8000,
+            port=8888,
             mac_address=Node.get_current_mac(),
         )
         self.remote_key = rsa.generate_private_key(public_exponent=65537, key_size=2048)
@@ -3107,7 +3174,7 @@ class NetMessageReachTests(TestCase):
             self.nodes[name] = Node.objects.create(
                 hostname=name.lower(),
                 address=f"10.0.0.{idx}",
-                port=8000 + idx,
+                port=8888 + idx,
                 mac_address=f"00:11:22:33:44:{idx:02x}",
                 role=self.roles[name],
             )
@@ -3318,7 +3385,7 @@ class NetMessagePropagationTests(TestCase):
                 Node.objects.create(
                     hostname=f"n{idx}",
                     address=f"10.0.0.{idx}",
-                    port=8000 + idx,
+                    port=8888 + idx,
                     mac_address=f"00:11:22:33:44:{idx:02x}",
                     role=self.role,
                     public_endpoint=f"n{idx}",
@@ -3365,7 +3432,7 @@ class NetMessagePropagationTests(TestCase):
                 Node.objects.create(
                     hostname=f"n{idx}",
                     address=f"10.0.0.{idx}",
-                    port=8000 + idx,
+                    port=8888 + idx,
                     mac_address=f"00:11:22:33:44:{idx:02x}",
                     role=self.role,
                     public_endpoint=f"n{idx}",
@@ -3454,7 +3521,7 @@ class NetMessageQueueTests(TestCase):
         local = Node.objects.create(
             hostname="local",
             address="10.0.0.1",
-            port=8000,
+            port=8888,
             mac_address="00:11:22:33:44:10",
             role=self.role,
             public_endpoint="local",
@@ -3753,7 +3820,7 @@ class NetworkChargerActionSecurityTests(TestCase):
         self.local_node = Node.objects.create(
             hostname="local-node",
             address="127.0.0.1",
-            port=8000,
+            port=8888,
             mac_address="00:aa:bb:cc:dd:10",
             public_endpoint="local-endpoint",
         )
@@ -4026,7 +4093,7 @@ class ContentSampleAdminTests(TestCase):
         Node.objects.create(
             hostname="host",
             address="127.0.0.1",
-            port=8000,
+            port=8888,
             mac_address=Node.get_current_mac(),
         )
         url = reverse("admin:nodes_contentsample_from_clipboard")
@@ -4052,7 +4119,7 @@ class EmailOutboxTests(TestCase):
         node = Node.objects.create(
             hostname="outboxhost",
             address="127.0.0.1",
-            port=8000,
+            port=8888,
             mac_address="00:11:22:33:aa:bb",
         )
         outbox = EmailOutbox.objects.create(
@@ -4068,7 +4135,7 @@ class EmailOutboxTests(TestCase):
         node = Node.objects.create(
             hostname="host",
             address="127.0.0.1",
-            port=8000,
+            port=8888,
             mac_address="00:11:22:33:cc:dd",
         )
         node.send_mail("sub", "msg", ["to@example.com"])
@@ -4181,7 +4248,7 @@ class ClipboardTaskTests(TestCase):
         Node.objects.create(
             hostname="host",
             address="127.0.0.1",
-            port=8000,
+            port=8888,
             mac_address=Node.get_current_mac(),
         )
         sample_clipboard()
@@ -4205,7 +4272,7 @@ class ClipboardTaskTests(TestCase):
         node = Node.objects.create(
             hostname="host",
             address="127.0.0.1",
-            port=8000,
+            port=8888,
             mac_address=Node.get_current_mac(),
         )
         screenshot_dir = settings.LOG_DIR / "screenshots"
@@ -4228,7 +4295,7 @@ class ClipboardTaskTests(TestCase):
         Node.objects.create(
             hostname="host",
             address="127.0.0.1",
-            port=8000,
+            port=8888,
             mac_address=Node.get_current_mac(),
         )
         mock_capture.side_effect = RuntimeError("boom")
@@ -4290,14 +4357,14 @@ class NodeRoleAdminTests(TestCase):
         node1 = Node.objects.create(
             hostname="n1",
             address="127.0.0.1",
-            port=8000,
+            port=8888,
             mac_address="00:11:22:33:44:55",
             role=role,
         )
         node2 = Node.objects.create(
             hostname="n2",
             address="127.0.0.2",
-            port=8000,
+            port=8888,
             mac_address="00:11:22:33:44:66",
         )
         url = reverse("admin:nodes_noderole_change", args=[role.pk])
@@ -4316,7 +4383,7 @@ class NodeRoleAdminTests(TestCase):
         Node.objects.create(
             hostname="n1",
             address="127.0.0.1",
-            port=8000,
+            port=8888,
             mac_address="00:11:22:33:44:77",
             role=role,
         )
@@ -4379,7 +4446,7 @@ class NodeFeatureTests(TestCase):
             self.node = Node.objects.create(
                 hostname="local",
                 address="127.0.0.1",
-                port=8000,
+                port=8888,
                 mac_address="00:11:22:33:44:55",
                 role=self.role,
             )
@@ -4552,7 +4619,7 @@ class NodeFeatureTests(TestCase):
             node = Node.objects.create(
                 hostname="control",
                 address="127.0.0.1",
-                port=8000,
+                port=8888,
                 mac_address=mac,
                 role=control_role,
             )
@@ -4576,7 +4643,7 @@ class NodeFeatureTests(TestCase):
                 node = Node.objects.create(
                     hostname="control",
                     address="127.0.0.1",
-                    port=8000,
+                    port=8888,
                     mac_address=mac,
                     role=control_role,
                     base_path=str(Path(tmp)),
@@ -4597,7 +4664,7 @@ class NodeFeatureTests(TestCase):
             node = Node.objects.create(
                 hostname="control",
                 address="127.0.0.1",
-                port=8000,
+                port=8888,
                 mac_address=mac,
                 role=control_role,
             )
@@ -4612,6 +4679,72 @@ class NodeFeatureTests(TestCase):
         )
 
 
+class AudioCaptureDetectionTests(TestCase):
+    def test_has_audio_capture_device_true(self):
+        with TemporaryDirectory() as tmp:
+            pcm_path = Path(tmp) / "pcm"
+            pcm_path.write_text(
+                "00-00: USB Audio : USB Audio : playback 1 : capture 1\n",
+                encoding="utf-8",
+            )
+            with patch.object(Node, "AUDIO_CAPTURE_PCM_PATH", pcm_path):
+                self.assertTrue(Node._has_audio_capture_device())
+
+    def test_has_audio_capture_device_false_without_capture(self):
+        with TemporaryDirectory() as tmp:
+            pcm_path = Path(tmp) / "pcm"
+            pcm_path.write_text(
+                "00-00: USB Audio : USB Audio : playback 1\n",
+                encoding="utf-8",
+            )
+            with patch.object(Node, "AUDIO_CAPTURE_PCM_PATH", pcm_path):
+                self.assertFalse(Node._has_audio_capture_device())
+
+    def test_has_audio_capture_device_false_when_file_missing(self):
+        with TemporaryDirectory() as tmp:
+            pcm_path = Path(tmp) / "pcm"
+        with patch.object(Node, "AUDIO_CAPTURE_PCM_PATH", pcm_path):
+            self.assertFalse(Node._has_audio_capture_device())
+
+
+class AudioCaptureFeatureCheckTests(TestCase):
+    def setUp(self):
+        self.node = Node.objects.create(
+            hostname="localnode",
+            address="127.0.0.1",
+            port=8888,
+            mac_address=Node.get_current_mac(),
+        )
+        self.feature, _ = NodeFeature.objects.get_or_create(
+            slug="audio-capture", defaults={"display": "Audio Capture"}
+        )
+
+    @pytest.mark.feature("audio-capture")
+    @patch("nodes.models.Node._has_audio_capture_device", return_value=False)
+    def test_feature_check_warns_without_device(self, mock_device):
+        result = feature_checks.run(self.feature, node=self.node)
+        self.assertFalse(result.success)
+        self.assertIn("No audio recording device detected", result.message)
+        self.assertEqual(result.level, messages.WARNING)
+
+    @pytest.mark.feature("audio-capture")
+    @patch("nodes.models.Node._has_audio_capture_device", return_value=True)
+    def test_feature_check_warns_when_feature_disabled(self, mock_device):
+        result = feature_checks.run(self.feature, node=self.node)
+        self.assertFalse(result.success)
+        self.assertIn("is not enabled", result.message)
+        self.assertEqual(result.level, messages.WARNING)
+
+    @pytest.mark.feature("audio-capture")
+    @patch("nodes.models.Node._has_audio_capture_device", return_value=True)
+    def test_feature_check_passes_when_enabled(self, mock_device):
+        NodeFeatureAssignment.objects.get_or_create(node=self.node, feature=self.feature)
+        result = feature_checks.run(self.feature, node=self.node)
+        self.assertTrue(result.success)
+        self.assertIn("recording device is available", result.message)
+        self.assertEqual(result.level, messages.SUCCESS)
+
+
 class CeleryReportAdminViewTests(TestCase):
     def setUp(self):
         User = get_user_model()