add tests

Author: ThiagoAVicente

tests/conftest.py

@@ -12,6 +12,7 @@ def __init__(self, *topics, hostname=None, port=None):
         self.port = port
         self._closed = False
         self._callbacks = {}
+        self._consumer_task = None  # Added for consumer task tracking
 
     def add_n_topics(self, topics, bind=None):
         """Mock add_n_topics method."""

tests/test_clickhouse_service.py

@@ -39,6 +39,59 @@ def test_connect(self, mock_clickhouse_client):
 
         assert service.client == mock_clickhouse_client
 
+    def test_write_data(self, clickhouse_service, mock_clickhouse_client):
+        """Test writing a single ProcessedLatency record to ClickHouse."""
+        # Prepare test data
+        test_data = {
+            'window_start_time': datetime(2024, 1, 1, 12, 0, 0, tzinfo=timezone.utc),
+            'window_end_time': datetime(2024, 1, 1, 12, 5, 0, tzinfo=timezone.utc),
+            'window_duration_seconds': 300.0,
+            'cell_index': 1,
+            'network': '5G',
+            'rsrp_mean': -80.0,
+            'rsrp_max': -70.0,
+            'rsrp_min': -90.0,
+            'rsrp_std': 5.0,
+            'sinr_mean': 15.0,
+            'sinr_max': 20.0,
+            'sinr_min': 10.0,
+            'sinr_std': 3.0,
+            'rsrq_mean': -10.0,
+            'rsrq_max': -8.0,
+            'rsrq_min': -12.0,
+            'rsrq_std': 1.5,
+            'latency_mean': 20.0,
+            'latency_max': 30.0,
+            'latency_min': 10.0,
+            'latency_std': 5.0,
+            'cqi_mean': 12.0,
+            'cqi_max': 15.0,
+            'cqi_min': 10.0,
+            'cqi_std': 2.0,
+            'primary_bandwidth': 100.0,
+            'ul_bandwidth': 50.0,
+            'sample_count': 100
+        }
+
+        # Execute write
+        clickhouse_service.write_data(test_data)
+
+        # Verify insert was called with correct parameters
+        mock_clickhouse_client.insert.assert_called_once()
+        call_args = mock_clickhouse_client.insert.call_args
+
+        # Check table name
+        assert call_args[0][0] == 'analytics.processed_latency'
+        
+        # Check data format (should be a list with one dict)
+        assert isinstance(call_args[0][1], list)
+        assert len(call_args[0][1]) == 1
+        
+        # Check async insert settings
+        assert call_args[1]['settings']['async_insert'] == 1
+        assert call_args[1]['settings']['wait_for_async_insert'] == 0
+
+
     def test_query_processed_latency_success(self, clickhouse_service, mock_clickhouse_client):
         """Test successful query of processed latency data."""
         # Mock query result