fix: cleanup tests

Author: tech4242

codecov.yml

@@ -3,5 +3,5 @@ coverage:
   status:
     project:
       default:
-        target: 78  # Automatically set coverage target
+        target: 85  # Automatically set coverage target
         threshold: 5% # Allow 5% drop in coverage

pyproject.toml

@@ -82,4 +82,7 @@ dummy-variable-rgx = "^(_+|(_+[a-zA-Z0-9_]*[a-zA-Z0-9]+?))$"
 "__init__.py" = ["F401", "E402"]
 # Ignore assert statements in tests
 "test_*.py" = ["S101", "E712"]
-"tests/*.py" = ["S101", "E712"]
+"tests/*.py" = ["S101", "E712"]
+
+[tool.coverage.run]
+omit = ["mcphawk/models.py"]

tests/test_broadcast_ui.py

@@ -0,0 +1,75 @@
+import asyncio
+from unittest.mock import AsyncMock, MagicMock
+
+import pytest
+
+from mcphawk.web.broadcaster import active_clients, broadcast_new_log
+
+
+@pytest.mark.asyncio
+async def test_broadcast_new_log_with_clients():
+    # Create mock websocket clients
+    mock_client1 = AsyncMock()
+    mock_client2 = AsyncMock()
+    
+    # Add to active clients
+    active_clients.append(mock_client1)
+    active_clients.append(mock_client2)
+    
+    try:
+        # Broadcast a message
+        test_log = {
+            "timestamp": "2024-01-01T12:00:00Z",
+            "src_ip": "127.0.0.1",
+            "message": "test"
+        }
+        
+        await broadcast_new_log(test_log)
+        
+        # Verify both clients received the message
+        mock_client1.send_json.assert_called_once_with(test_log)
+        mock_client2.send_json.assert_called_once_with(test_log)
+        
+    finally:
+        # Clean up
+        active_clients.clear()
+
+
+@pytest.mark.asyncio
+async def test_broadcast_new_log_with_disconnected_client():
+    # Create mock websocket clients
+    mock_client1 = AsyncMock()
+    mock_client2 = AsyncMock()
+    
+    # Make client1 throw an exception (disconnected)
+    mock_client1.send_json.side_effect = Exception("Disconnected")
+    
+    # Add to active clients
+    active_clients.append(mock_client1)
+    active_clients.append(mock_client2)
+    
+    try:
+        # Broadcast a message
+        test_log = {"test": "data"}
+        
+        await broadcast_new_log(test_log)
+        
+        # Client1 should be removed from active_clients
+        assert mock_client1 not in active_clients
+        assert mock_client2 in active_clients
+        
+        # Client2 should still receive the message
+        mock_client2.send_json.assert_called_once_with(test_log)
+        
+    finally:
+        # Clean up
+        active_clients.clear()
+
+
+@pytest.mark.asyncio 
+async def test_broadcast_new_log_no_clients():
+    # Ensure list is empty
+    active_clients.clear()
+    
+    # Should not raise any errors
+    await broadcast_new_log({"test": "data"})

tests/test_broadcaster.py

@@ -86,10 +86,6 @@ def test_packet_callback(dummy_server):
     assert "Berlin" in logged_msg
 
 
-def test_import():
-    import mcphawk
-    assert hasattr(mcphawk, "__version__")
-
 
 class TestAutoDetect:
     """Test auto-detect mode functionality."""