Skip to content

Commit bd22dde

Browse files
committed
fix: update remaining references to privatized domain methods in har_recorder and tests
1 parent abf4557 commit bd22dde

3 files changed

Lines changed: 29 additions & 29 deletions

File tree

pydoll/browser/requests/har_recorder.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -89,8 +89,8 @@ async def start(self) -> None:
8989
Enables network events if not already on, and registers callbacks
9090
for the 7 CDP events needed to build HAR entries.
9191
"""
92-
if not self._tab.network_events_enabled:
93-
await self._tab.enable_network_events()
92+
if not self._tab._network_events_enabled:
93+
await self._tab._enable_network_events()
9494
self._network_was_enabled = True
9595
logger.debug('HAR recorder enabled network events')
9696

@@ -130,7 +130,7 @@ async def stop(self) -> None:
130130
self._flush_pending()
131131

132132
if self._network_was_enabled:
133-
await self._tab.disable_network_events()
133+
await self._tab._disable_network_events()
134134
self._network_was_enabled = False
135135

136136
logger.info('HAR recorder stopped, captured %d entries', len(self._entries))

tests/test_browser/test_har_recorder.py

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,9 @@
1515
async def mock_tab():
1616
"""Create a mock Tab instance for testing."""
1717
tab = Mock()
18-
tab.network_events_enabled = False
19-
tab.enable_network_events = AsyncMock()
20-
tab.disable_network_events = AsyncMock()
18+
tab._network_events_enabled = False
19+
tab._enable_network_events = AsyncMock()
20+
tab._disable_network_events = AsyncMock()
2121
tab.on = AsyncMock(side_effect=lambda *a, **kw: len(tab.on.call_args_list))
2222
tab.remove_callback = AsyncMock()
2323
tab.clear_callbacks = AsyncMock()
@@ -195,16 +195,16 @@ async def test_start_stores_callback_ids(self, recorder, mock_tab):
195195

196196
@pytest.mark.asyncio
197197
async def test_start_enables_network_events_if_not_enabled(self, recorder, mock_tab):
198-
mock_tab.network_events_enabled = False
198+
mock_tab._network_events_enabled = False
199199
await recorder.start()
200-
mock_tab.enable_network_events.assert_called_once()
200+
mock_tab._enable_network_events.assert_called_once()
201201
assert recorder._network_was_enabled is True
202202

203203
@pytest.mark.asyncio
204204
async def test_start_skips_network_enable_if_already_enabled(self, recorder, mock_tab):
205-
mock_tab.network_events_enabled = True
205+
mock_tab._network_events_enabled = True
206206
await recorder.start()
207-
mock_tab.enable_network_events.assert_not_called()
207+
mock_tab._enable_network_events.assert_not_called()
208208
assert recorder._network_was_enabled is False
209209

210210
@pytest.mark.asyncio
@@ -243,17 +243,17 @@ async def test_stop_clears_callback_ids(self, recorder, mock_tab):
243243

244244
@pytest.mark.asyncio
245245
async def test_stop_disables_network_events_if_we_enabled(self, recorder, mock_tab):
246-
mock_tab.network_events_enabled = False
246+
mock_tab._network_events_enabled = False
247247
await recorder.start()
248248
await recorder.stop()
249-
mock_tab.disable_network_events.assert_called_once()
249+
mock_tab._disable_network_events.assert_called_once()
250250

251251
@pytest.mark.asyncio
252252
async def test_stop_does_not_disable_network_events_if_not_ours(self, recorder, mock_tab):
253-
mock_tab.network_events_enabled = True
253+
mock_tab._network_events_enabled = True
254254
await recorder.start()
255255
await recorder.stop()
256-
mock_tab.disable_network_events.assert_not_called()
256+
mock_tab._disable_network_events.assert_not_called()
257257

258258
@pytest.mark.asyncio
259259
async def test_stop_flushes_pending_entries(self, recorder, mock_tab):
@@ -748,10 +748,10 @@ async def test_record_yields_har_recording(self, request_instance):
748748

749749
@pytest.mark.asyncio
750750
async def test_record_enables_network_events(self, request_instance, mock_tab):
751-
mock_tab.network_events_enabled = False
751+
mock_tab._network_events_enabled = False
752752
async with request_instance.record():
753753
pass
754-
mock_tab.enable_network_events.assert_called_once()
754+
mock_tab._enable_network_events.assert_called_once()
755755

756756
@pytest.mark.asyncio
757757
async def test_record_registers_and_removes_callbacks(self, request_instance, mock_tab):
@@ -771,19 +771,19 @@ async def test_record_cleans_up_on_exception(self, request_instance, mock_tab):
771771
async def test_record_disables_network_events_if_enabled_by_recorder(
772772
self, request_instance, mock_tab
773773
):
774-
mock_tab.network_events_enabled = False
774+
mock_tab._network_events_enabled = False
775775
async with request_instance.record():
776776
pass
777-
mock_tab.disable_network_events.assert_called_once()
777+
mock_tab._disable_network_events.assert_called_once()
778778

779779
@pytest.mark.asyncio
780780
async def test_record_does_not_disable_network_events_if_already_enabled(
781781
self, request_instance, mock_tab
782782
):
783-
mock_tab.network_events_enabled = True
783+
mock_tab._network_events_enabled = True
784784
async with request_instance.record():
785785
pass
786-
mock_tab.disable_network_events.assert_not_called()
786+
mock_tab._disable_network_events.assert_not_called()
787787

788788

789789
class TestResourceTypeFiltering:

tests/test_browser/test_requests_request.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,9 @@
2020
async def mock_tab():
2121
"""Create a mock Tab instance for testing."""
2222
tab = Mock()
23-
tab.network_events_enabled = False
24-
tab.enable_network_events = AsyncMock()
25-
tab.disable_network_events = AsyncMock()
23+
tab._network_events_enabled = False
24+
tab._enable_network_events = AsyncMock()
25+
tab._disable_network_events = AsyncMock()
2626
tab.remove_callback = AsyncMock()
2727
tab.on = AsyncMock(side_effect=lambda *a, **kw: len(tab.on.call_args_list))
2828
tab._execute_command = AsyncMock()
@@ -366,21 +366,21 @@ class TestRequestCallbackManagement:
366366
@pytest.mark.asyncio
367367
async def test_register_callbacks_enables_network_events(self, request_instance, mock_tab):
368368
"""Test that registering callbacks enables network events."""
369-
mock_tab.network_events_enabled = False
369+
mock_tab._network_events_enabled = False
370370

371371
await request_instance._register_callbacks()
372372

373-
mock_tab.enable_network_events.assert_called_once()
373+
mock_tab._enable_network_events.assert_called_once()
374374
assert request_instance._network_events_enabled is True
375375

376376
@pytest.mark.asyncio
377377
async def test_register_callbacks_skips_if_already_enabled(self, request_instance, mock_tab):
378378
"""Test that network events are not re-enabled if already active."""
379-
mock_tab.network_events_enabled = True
379+
mock_tab._network_events_enabled = True
380380

381381
await request_instance._register_callbacks()
382382

383-
mock_tab.enable_network_events.assert_not_called()
383+
mock_tab._enable_network_events.assert_not_called()
384384
assert request_instance._network_events_enabled is False
385385

386386
@pytest.mark.asyncio
@@ -409,7 +409,7 @@ async def test_clear_callbacks_disables_network_events(self, request_instance, m
409409

410410
await request_instance._clear_callbacks()
411411

412-
mock_tab.disable_network_events.assert_called_once()
412+
mock_tab._disable_network_events.assert_called_once()
413413
assert mock_tab.remove_callback.call_count == 4
414414
assert request_instance._network_events_enabled is False
415415
assert request_instance._callback_ids == []
@@ -422,7 +422,7 @@ async def test_clear_callbacks_skips_disable_if_not_enabled(self, request_instan
422422

423423
await request_instance._clear_callbacks()
424424

425-
mock_tab.disable_network_events.assert_not_called()
425+
mock_tab._disable_network_events.assert_not_called()
426426
assert mock_tab.remove_callback.call_count == 2
427427
assert request_instance._callback_ids == []
428428

0 commit comments

Comments
 (0)