|
8 | 8 | ########################################################################### |
9 | 9 | """Tests for remote utilities.""" |
10 | 10 |
|
11 | | -from unittest.mock import Mock |
| 11 | +from unittest.mock import patch |
12 | 12 |
|
13 | 13 | import pytest |
14 | 14 |
|
@@ -39,23 +39,24 @@ def test_clean_mapping_remote_paths_skips_unconfigured_computer(tmp_path, monkey |
39 | 39 | ).store() |
40 | 40 | configured.configure(user=user) |
41 | 41 |
|
42 | | - folder_unconfigured = Mock(spec=RemoteData) |
43 | | - folder_configured = Mock(spec=RemoteData) |
| 42 | + folder_unconfigured = RemoteData(remote_path=str(tmp_path / 'unconfigured-folder'), computer=unconfigured) |
| 43 | + folder_configured = RemoteData(remote_path=str(tmp_path / 'configured-folder'), computer=configured) |
44 | 44 |
|
45 | 45 | warnings: list[str] = [] |
46 | 46 | monkeypatch.setattr(remote.echo, 'echo_warning', lambda message, **kwargs: warnings.append(message)) |
47 | 47 | monkeypatch.setattr(remote.echo, 'echo_success', lambda *args, **kwargs: None) |
48 | 48 |
|
49 | | - remote.clean_mapping_remote_paths( |
50 | | - { |
51 | | - unconfigured.uuid: [folder_unconfigured], |
52 | | - configured.uuid: [folder_configured], |
53 | | - }, |
54 | | - silent=True, |
55 | | - ) |
56 | | - |
57 | | - folder_unconfigured._clean.assert_not_called() |
58 | | - folder_configured._clean.assert_called_once() |
| 49 | + with patch.object(RemoteData, '_clean', autospec=True) as mock_clean: |
| 50 | + remote.clean_mapping_remote_paths( |
| 51 | + { |
| 52 | + unconfigured.uuid: [folder_unconfigured], |
| 53 | + configured.uuid: [folder_configured], |
| 54 | + }, |
| 55 | + silent=True, |
| 56 | + ) |
| 57 | + |
| 58 | + mock_clean.assert_called_once() |
| 59 | + assert mock_clean.call_args.args[0] is folder_configured |
59 | 60 |
|
60 | 61 | assert warnings == [ |
61 | 62 | f'Skipping 1 remote folders on `{unconfigured.label}`: ' f'computer is not configured for user `{user.email}`' |
|
0 commit comments