|
6 | 6 |
|
7 | 7 | @pytest.fixture |
8 | 8 | def mock_connection(mocker): |
9 | | - mock_connection = mocker.AsyncMock(asyncssh.SSHClientConnection, name="connection", _username="testuser") |
| 9 | + mock_connection = mocker.AsyncMock(asyncssh.SSHClientConnection, name="connection") |
| 10 | + mock_connection.get_extra_info.return_value = "testuser" |
10 | 11 | mock_connection.run.return_value = mocker.Mock(exit_status=0, stdout="remote output", stderr="") |
11 | 12 | mock_connection.is_closed.return_value = False |
12 | 13 |
|
@@ -55,8 +56,10 @@ async def test_get_connection_different_hosts(mocker, mock_asyncssh_connect): |
55 | 56 | manager = SSHConnectionManager() |
56 | 57 | manager._connections.clear() |
57 | 58 |
|
58 | | - mock_conn1 = mocker.AsyncMock(asyncssh.SSHClientConnection, return_value=False, _username="testuser") |
59 | | - mock_conn2 = mocker.AsyncMock(asyncssh.SSHClientConnection, return_value=False, _username="testuser") |
| 59 | + mock_conn1 = mocker.AsyncMock(asyncssh.SSHClientConnection, return_value=False) |
| 60 | + mock_conn1.get_extra_info.return_value = "testuser" |
| 61 | + mock_conn2 = mocker.AsyncMock(asyncssh.SSHClientConnection, return_value=False) |
| 62 | + mock_conn2.get_extra_info.return_value = "testuser" |
60 | 63 |
|
61 | 64 | async def async_connect(*args, **kwargs): |
62 | 65 | return mock_conn1 if kwargs.get("host") == "host1" else mock_conn2 |
@@ -135,12 +138,10 @@ async def test_close_connections(mocker, mock_asyncssh_connect): |
135 | 138 | manager = SSHConnectionManager() |
136 | 139 | manager._connections.clear() |
137 | 140 |
|
138 | | - mock_conn1 = mocker.AsyncMock( |
139 | | - asyncssh.SSHClientConnection, return_value=False, wait_closed=mocker.AsyncMock(), _username="testuser" |
140 | | - ) |
141 | | - mock_conn2 = mocker.AsyncMock( |
142 | | - asyncssh.SSHClientConnection, return_value=False, wait_closed=mocker.AsyncMock(), _username="testuser" |
143 | | - ) |
| 141 | + mock_conn1 = mocker.AsyncMock(asyncssh.SSHClientConnection, return_value=False, wait_closed=mocker.AsyncMock()) |
| 142 | + mock_conn1.get_extra_info.return_value = "testuser" |
| 143 | + mock_conn2 = mocker.AsyncMock(asyncssh.SSHClientConnection, return_value=False, wait_closed=mocker.AsyncMock()) |
| 144 | + mock_conn2.get_extra_info.return_value = "testuser" |
144 | 145 |
|
145 | 146 | async def async_connect(*args, **kwargs): |
146 | 147 | return mock_conn1 if kwargs.get("host") == "host1" else mock_conn2 |
|
0 commit comments