add tests

tests/test_main.py

@@ -0,0 +1,122 @@
+from tailscalesd.main import group_by_type, matrix_workers_to_sd
+
+
+def test_group_by_type_valid_input():
+    input_list = [
+        {"type": "A", "value": 1},
+        {"type": "B", "value": 2},
+        {"type": "A", "value": 3},
+        {"type": "C", "value": 5},
+    ]
+    expected_output = {
+        "A": [{"type": "A", "value": 1}, {"type": "A", "value": 3}],
+        "B": [{"type": "B", "value": 2}],
+        "C": [{"type": "C", "value": 5}],
+    }
+    assert group_by_type(input_list) == expected_output
+
+
+def test_group_by_type_missing_type_key():
+    input_list = [
+        {"value": 1},
+        {"type": "B", "value": 2},
+        {"value": 3},
+        {"type": "C", "value": 5},
+    ]
+    expected_output = {
+        "B": [{"type": "B", "value": 2}],
+        "C": [{"type": "C", "value": 5}],
+    }
+    assert group_by_type(input_list) == expected_output
+
+
+def test_group_by_type_empty_input():
+    input_list = []
+    expected_output = {}
+    assert group_by_type(input_list) == expected_output
+
+
+def test_matrix_workers_to_sd_valid_input():
+    device = {
+        "addresses": ["10.0.0.1", "192.168.1.1"],
+        "hostname": "test-host",
+        "clientVersion": "1",
+        "tags": ["tag:matrix"],
+        "id": "id",
+        "name": "name",
+        "os": "linux",
+        "authorized": True,
+        "tailnet": "testnet",
+    }
+    workers = {
+        "type1": [
+            {"metrics_port": 1234, "name": "worker1"},
+            {"metrics_port": 1235, "name": "worker2"},
+        ],
+        "type2": [
+            {"metrics_port": 1236, "name": "worker3"},
+        ],
+    }
+    ts_labels = {
+        "__meta_tailscale_device_client_version": "1",
+        "__meta_tailscale_device_hostname": "test-host",
+        "__meta_tailscale_device_authorized": "true",
+        "__meta_tailscale_device_id": "id",
+        "__meta_tailscale_device_name": "name",
+        "__meta_tailscale_device_os": "linux",
+        "__meta_tailscale_tailnet": "testnet",
+    }
+    expected_output = [
+        {
+            "targets": ["10.0.0.1:1234"],
+            "labels": {
+                # Add expected tailscale_labels output here
+                "__meta_matrix_worker_type": "type1",
+                "__meta_matrix_worker_name": "worker1",
+            }
+            | ts_labels,
+        },
+        {
+            "targets": ["10.0.0.1:1235"],
+            "labels": {
+                # Add expected tailscale_labels output here
+                "__meta_matrix_worker_type": "type1",
+                "__meta_matrix_worker_name": "worker2",
+            }
+            | ts_labels,
+        },
+        {
+            "targets": ["10.0.0.1:1236"],
+            "labels": {
+                # Add expected tailscale_labels output here
+                "__meta_matrix_worker_type": "type2",
+                "__meta_matrix_worker_name": "worker3",
+            }
+            | ts_labels,
+        },
+    ]
+    actual_output = matrix_workers_to_sd("testnet", device, workers)
+    assert actual_output == expected_output
+
+
+def test_matrix_workers_to_sd_empty_workers():
+    device = {
+        "addresses": ["10.0.0.1", "192.168.1.1"],
+        "hostname": "test-host",
+    }
+    workers = {}
+    assert matrix_workers_to_sd("testnet", device, workers) == []
+
+
+def test_matrix_workers_to_sd_invalid_port():
+    device = {
+        "addresses": ["10.0.0.1", "192.168.1.1"],
+        "hostname": "test-host",
+    }
+    workers = {
+        "type1": [
+            {"metrics_port": None, "name": "worker1"},
+        ],
+    }
+    # Assuming log.error does not raise any exceptions
+    assert matrix_workers_to_sd("testnet", device, workers) == []