fix tests

Author: Dylan Huang

eval_protocol/utils/logs_server.py

@@ -95,12 +95,6 @@ def broadcast_row_upserted(self, row: "EvaluationRow"):
             active_connections_count = len(self.active_connections)
         logger.debug(f"[WEBSOCKET_BROADCAST] Active connections: {active_connections_count}")
 
-        if active_connections_count == 0:
-            logger.debug(
-                f"[WEBSOCKET_BROADCAST] No active connections, skipping broadcast for rollout_id: {rollout_id}"
-            )
-            return
-
         try:
             # Serialize pydantic model
             logger.debug(f"[WEBSOCKET_BROADCAST] Serializing row for rollout_id: {rollout_id}")

tests/test_logs_server.py

@@ -373,7 +373,7 @@ def test_serve_logs_port_parameter(self, temp_build_dir):
             serve_logs(port=test_port)
 
             # Verify that LogsServer was created with the correct port
-            mock_logs_server_class.assert_called_once_with(port=test_port, elasticsearch_config=None)
+            mock_logs_server_class.assert_called_once_with(port=test_port, elasticsearch_config=None, debug=False)
             # Verify that the run method was called on the instance
             mock_server_instance.run.assert_called_once()
 
@@ -387,7 +387,7 @@ def test_serve_logs_default_port(self, temp_build_dir):
             serve_logs()
 
             # Verify that LogsServer was created with None port (which will use LogsServer's default of 8000)
-            mock_logs_server_class.assert_called_once_with(port=None, elasticsearch_config=None)
+            mock_logs_server_class.assert_called_once_with(port=None, elasticsearch_config=None, debug=False)
             # Verify that the run method was called on the instance
             mock_server_instance.run.assert_called_once()