|
| 1 | +from itertools import islice |
1 | 2 | from types import SimpleNamespace |
| 3 | +from typing import Generator |
2 | 4 | from unittest.mock import MagicMock, patch |
3 | 5 |
|
4 | 6 | import platform_api_python_client |
@@ -199,6 +201,124 @@ def test_create_hardware_instance_delegates_to_platform_client(): |
199 | 201 | api.create_hardware_instance_hardware_instances_post.assert_called_once_with(request) |
200 | 202 |
|
201 | 203 |
|
| 204 | +def _log_event(event_id, timestamp, message="line"): |
| 205 | + return SimpleNamespace(id=event_id, timestamp=timestamp, message=message) |
| 206 | + |
| 207 | + |
| 208 | +def _log_page(*events): |
| 209 | + return SimpleNamespace(events=list(events)) |
| 210 | + |
| 211 | + |
| 212 | +def test_get_deployment_pods_returns_pod_names(): |
| 213 | + api = MagicMock() |
| 214 | + api.get_deployment_pods_deployments_pods_deployment_id_revision_number_get.return_value = SimpleNamespace( |
| 215 | + pods=["pod-a", "pod-b"] |
| 216 | + ) |
| 217 | + client = CentMLClient(api) |
| 218 | + |
| 219 | + assert client.get_deployment_pods(123, 2) == ["pod-a", "pod-b"] |
| 220 | + |
| 221 | + api.get_deployment_pods_deployments_pods_deployment_id_revision_number_get.assert_called_once_with( |
| 222 | + deployment_id=123, revision_number=2 |
| 223 | + ) |
| 224 | + |
| 225 | + |
| 226 | +def test_get_deployment_logs_paginates_oldest_first_until_no_newer_events(): |
| 227 | + api = MagicMock() |
| 228 | + api.get_deployment_logs_v4_logs_deployment_id_revision_number_get.side_effect = [ |
| 229 | + _log_page(_log_event("1-a", 1000), _log_event("2-b", 2000)), |
| 230 | + _log_page(_log_event("3-c", 3000)), |
| 231 | + _log_page(), |
| 232 | + ] |
| 233 | + client = CentMLClient(api) |
| 234 | + |
| 235 | + events = client.get_deployment_logs(123, 2, pod="pod-a") |
| 236 | + |
| 237 | + assert [e.id for e in events] == ["1-a", "2-b", "3-c"] |
| 238 | + calls = api.get_deployment_logs_v4_logs_deployment_id_revision_number_get.call_args_list |
| 239 | + assert len(calls) == 3 |
| 240 | + # First page scans from the head of the log window; later pages pass the newest |
| 241 | + # seen timestamp verbatim (the boundary is exclusive server-side). |
| 242 | + assert [c.kwargs["timestamp"] for c in calls] == [None, 2000, 3000] |
| 243 | + assert all(c.kwargs["fetch_newer"] is True for c in calls) |
| 244 | + assert all(c.kwargs["deployment_id"] == 123 and c.kwargs["revision_number"] == 2 for c in calls) |
| 245 | + assert all(c.kwargs["pod"] == "pod-a" for c in calls) |
| 246 | + |
| 247 | + |
| 248 | +def test_get_deployment_logs_returns_empty_list_when_pod_has_no_logs(): |
| 249 | + api = MagicMock() |
| 250 | + api.get_deployment_logs_v4_logs_deployment_id_revision_number_get.return_value = _log_page() |
| 251 | + client = CentMLClient(api) |
| 252 | + |
| 253 | + assert client.get_deployment_logs(123, 2, pod="pod-a") == [] |
| 254 | + |
| 255 | + api.get_deployment_logs_v4_logs_deployment_id_revision_number_get.assert_called_once() |
| 256 | + |
| 257 | + |
| 258 | +def test_get_deployment_logs_deduplicates_look_behind_redelivery(): |
| 259 | + api = MagicMock() |
| 260 | + api.get_deployment_logs_v4_logs_deployment_id_revision_number_get.side_effect = [ |
| 261 | + _log_page(_log_event("1-a", 1000), _log_event("2-b", 2000)), |
| 262 | + # Look-behind re-delivers 2000 alongside fresh lines; a late arrival shows up at 1500. |
| 263 | + _log_page(_log_event("2-b", 2000), _log_event("15-l", 1500), _log_event("3-c", 3000)), |
| 264 | + # Re-delivery-only page: no event newer than the boundary means history is done. |
| 265 | + _log_page(_log_event("3-c", 3000)), |
| 266 | + ] |
| 267 | + client = CentMLClient(api) |
| 268 | + |
| 269 | + events = client.get_deployment_logs(123, 2, pod="pod-a") |
| 270 | + |
| 271 | + assert [e.id for e in events] == ["1-a", "2-b", "15-l", "3-c"] |
| 272 | + assert api.get_deployment_logs_v4_logs_deployment_id_revision_number_get.call_count == 3 |
| 273 | + |
| 274 | + |
| 275 | +def test_get_deployment_logs_passes_max_lines_through(): |
| 276 | + api = MagicMock() |
| 277 | + api.get_deployment_logs_v4_logs_deployment_id_revision_number_get.return_value = _log_page() |
| 278 | + client = CentMLClient(api) |
| 279 | + |
| 280 | + client.get_deployment_logs(123, 2, pod="pod-a", max_lines=7) |
| 281 | + |
| 282 | + call = api.get_deployment_logs_v4_logs_deployment_id_revision_number_get.call_args |
| 283 | + assert call.kwargs["max_lines"] == 7 |
| 284 | + |
| 285 | + |
| 286 | +def test_get_deployment_logs_stream_fetches_pages_lazily(): |
| 287 | + api = MagicMock() |
| 288 | + api.get_deployment_logs_v4_logs_deployment_id_revision_number_get.side_effect = [ |
| 289 | + _log_page(_log_event("1-a", 1000), _log_event("2-b", 2000)), |
| 290 | + _log_page(_log_event("3-c", 3000)), |
| 291 | + _log_page(), |
| 292 | + ] |
| 293 | + client = CentMLClient(api) |
| 294 | + |
| 295 | + events = client.get_deployment_logs(123, 2, pod="pod-a", stream=True) |
| 296 | + |
| 297 | + assert isinstance(events, Generator) |
| 298 | + assert [e.id for e in islice(events, 2)] == ["1-a", "2-b"] |
| 299 | + assert api.get_deployment_logs_v4_logs_deployment_id_revision_number_get.call_count == 1 |
| 300 | + assert [e.id for e in events] == ["3-c"] |
| 301 | + assert api.get_deployment_logs_v4_logs_deployment_id_revision_number_get.call_count == 3 |
| 302 | + |
| 303 | + |
| 304 | +def test_get_deployment_logs_follow_polls_for_new_events_and_dedupes(): |
| 305 | + api = MagicMock() |
| 306 | + api.get_deployment_logs_v4_logs_deployment_id_revision_number_get.side_effect = [ |
| 307 | + _log_page(_log_event("1-a", 1000)), |
| 308 | + _log_page(), # nothing new yet -> sleep, poll again with the same boundary |
| 309 | + _log_page(_log_event("1-a", 1000), _log_event("2-b", 2000)), |
| 310 | + ] |
| 311 | + client = CentMLClient(api) |
| 312 | + |
| 313 | + with patch("centml.sdk.api.time.sleep") as mock_sleep: |
| 314 | + events = client.get_deployment_logs(123, 2, pod="pod-a", follow=True, poll_interval=0.5) |
| 315 | + assert [e.id for e in islice(events, 2)] == ["1-a", "2-b"] |
| 316 | + |
| 317 | + calls = api.get_deployment_logs_v4_logs_deployment_id_revision_number_get.call_args_list |
| 318 | + assert [c.kwargs["timestamp"] for c in calls] == [None, 1000, 1000] |
| 319 | + mock_sleep.assert_called_once_with(0.5) |
| 320 | + |
| 321 | + |
202 | 322 | def test_delete_hardware_instance_delegates_to_platform_client(): |
203 | 323 | api = MagicMock() |
204 | 324 | expected_response = MagicMock() |
|
0 commit comments