add some tests for dashboard

Author: andrewm4894

dashboard/tests/test_batch_stats.py

@@ -0,0 +1,59 @@
+import pandas as pd
+from datetime import datetime, timedelta, timezone
+import pytest
+from batch_stats import calculate_batch_stats, _format_time_ago
+
+@pytest.fixture
+def sample_df():
+    """Create a sample DataFrame for testing."""
+    now = datetime.now(timezone.utc)
+    return pd.DataFrame({
+        'metric_name': ['test_metric1', 'test_metric1', 'test_metric2'],
+        'metric_timestamp': [
+            now.isoformat(),
+            (now - timedelta(minutes=30)).isoformat(),
+            (now - timedelta(hours=2)).isoformat()
+        ],
+        'metric_value': [1.0, 2.0, 3.0],
+        'metric_score': [0.5, 0.7, 0.3],
+        'metric_alert': [1, 0, 1]
+    })
+
+def test_calculate_batch_stats_empty_df():
+    """Test stats calculation with empty DataFrame."""
+    empty_df = pd.DataFrame()
+    stats = calculate_batch_stats(empty_df, "test_batch")
+    
+    assert stats["unique_metrics"] == 0
+    assert stats["latest_timestamp"] == "No data"
+    assert stats["avg_score"] == 0
+    assert stats["alert_count"] == 0
+
+def test_calculate_batch_stats(sample_df):
+    """Test stats calculation with sample data."""
+    stats = calculate_batch_stats(sample_df, "test_batch")
+    
+    assert stats["unique_metrics"] == 2  # test_metric1 and test_metric2
+    assert "ago" in stats["latest_timestamp"]  # Should contain time ago string
+    assert stats["avg_score"] == 0.5  # Average of [0.5, 0.7, 0.3]
+    assert stats["alert_count"] == 2  # Count of 1s in metric_alert
+
+def test_format_time_ago():
+    """Test time ago formatting."""
+    now = datetime.now(timezone.utc)
+    
+    # Test minutes
+    timestamp = (now - timedelta(minutes=5)).isoformat()
+    assert "minute" in _format_time_ago(timestamp)
+    
+    # Test hours
+    timestamp = (now - timedelta(hours=3)).isoformat()
+    assert "hour" in _format_time_ago(timestamp)
+    
+    # Test days
+    timestamp = (now - timedelta(days=2)).isoformat()
+    assert "day" in _format_time_ago(timestamp)
+
+def test_format_time_ago_no_data():
+    """Test handling of 'No data' timestamp."""
+    assert _format_time_ago("No data") == "No data" 

dashboard/tests/test_components.py

@@ -0,0 +1,31 @@
+import pytest
+from components import create_batch_card, create_header
+
+def test_create_batch_card():
+    """Test batch card creation."""
+    stats = {
+        "unique_metrics": 10,
+        "latest_timestamp": "5 minutes ago",
+        "avg_score": 0.75,
+        "alert_count": 3
+    }
+    
+    card = create_batch_card("test_batch", stats)
+    
+    # Test that the card contains key information
+    card_str = str(card)
+    assert "test_batch" in card_str
+    assert "10 metrics" in card_str
+    assert "5 minutes ago" in card_str
+    assert "75.0%" in card_str  # 0.75 formatted as percentage
+    assert "3 alerts" in card_str
+
+def test_create_header():
+    """Test header creation."""
+    header = create_header()
+    header_str = str(header)
+    
+    # Test that header contains key elements
+    assert "Anomstack" in header_str
+    assert "Painless open source anomaly detection" in header_str
+    assert "github.com/andrewm4894/anomstack" in header_str 

dashboard/tests/test_routes.py

@@ -0,0 +1,50 @@
+import pytest
+from fasthtml.common import Div
+from app import app
+from routes import index, get_batch_view
+
+@pytest.fixture
+def mock_app_state(monkeypatch):
+    """Mock application state for testing."""
+    class MockState:
+        dark_mode = False
+        metric_batches = ["test_batch"]
+        df_cache = {}
+        specs_enabled = {"test_batch": {}}
+        alert_max_n = {"test_batch": 100}
+        stats_cache = {"test_batch": []}
+        
+    monkeypatch.setattr(app, "state", MockState())
+    return app.state
+
+@pytest.fixture
+def mock_request():
+    """Mock HTTP request."""
+    class MockRequest:
+        def __init__(self, htmx=False):
+            self.headers = {"HX-Request": "true"} if htmx else {}
+    return MockRequest
+
+def test_index_route_no_htmx(mock_app_state, mock_request):
+    """Test index route with regular request."""
+    response = index(mock_request())
+    
+    # Should return multiple elements for full page load
+    assert len(response) > 1
+    assert any("Anomstack" in str(elem) for elem in response)
+
+def test_index_route_htmx(mock_app_state, mock_request):
+    """Test index route with HTMX request."""
+    response = index(mock_request(htmx=True))
+    
+    # Should return only main content
+    assert isinstance(response, Div)
+    assert response.attrs.get("id") == "main-content"
+
+def test_batch_view(mock_app_state):
+    """Test batch view."""
+    response = get_batch_view("test_batch", None)
+    
+    # Should return a div containing charts container
+    assert isinstance(response, Div)
+    assert any("charts-container" in str(elem) for elem in response.children)