|
23 | 23 | DEDICATED_SERVER, |
24 | 24 | DEDICATED_RESOURCE, |
25 | 25 | SHARED_RESOURCE, |
| 26 | + SSH_DETAILS, |
26 | 27 | ) |
27 | 28 |
|
28 | 29 |
|
@@ -54,7 +55,7 @@ def test_list_all_servers_no_servers( |
54 | 55 | mocker.patch("faculty_cli.cli._list_user_servers", return_value=[]) |
55 | 56 | result = runner.invoke(cli, ["server", "list"]) |
56 | 57 | assert result.exit_code == 0 |
57 | | - assert result.output == "" |
| 58 | + assert result.output == "No servers.\n" |
58 | 59 |
|
59 | 60 |
|
60 | 61 | def test_list_all_servers( |
@@ -180,6 +181,53 @@ def test_list_servers_verbose( |
180 | 181 | ) |
181 | 182 |
|
182 | 183 |
|
| 184 | +def test_list_all_servers_ssh_config( |
| 185 | + mocker, |
| 186 | + mock_update_check, |
| 187 | + mock_check_credentials, |
| 188 | + mock_profile, |
| 189 | + mock_user_id, |
| 190 | +): |
| 191 | + runner = CliRunner() |
| 192 | + mocker.patch("faculty_cli.cli._list_projects", return_value=[PROJECT]) |
| 193 | + mocker.patch( |
| 194 | + "faculty_cli.cli._list_user_servers", return_value=[DEDICATED_SERVER] |
| 195 | + ) |
| 196 | + mocker.patch("faculty_cli.cli._get_ssh_details", return_value=SSH_DETAILS) |
| 197 | + result = runner.invoke( |
| 198 | + cli, |
| 199 | + ["server", "list", "--format", "ssh-config"], |
| 200 | + ) |
| 201 | + assert result.exit_code == 0 |
| 202 | + assert ( |
| 203 | + result.output |
| 204 | + == f"Host test-server-test-project\n HostName test-server\n User faculty\n Port 1234\n" |
| 205 | + ) |
| 206 | + |
| 207 | + |
| 208 | +def test_list_all_servers_verbose_error( |
| 209 | + mocker, |
| 210 | + mock_update_check, |
| 211 | + mock_check_credentials, |
| 212 | + mock_profile, |
| 213 | + mock_user_id, |
| 214 | +): |
| 215 | + runner = CliRunner() |
| 216 | + result = runner.invoke( |
| 217 | + cli, |
| 218 | + ["server", "list", "--verbose", "--format", "ssh-config"], |
| 219 | + ) |
| 220 | + assert result.exit_code == 2 |
| 221 | + assert ( |
| 222 | + result.output |
| 223 | + == """Usage: cli server list [OPTIONS] PROJECT |
| 224 | +Try 'cli server list --help' for help. |
| 225 | +
|
| 226 | +Error: You can't specify the `--verbose` flag and also pass the `--format` option |
| 227 | +""" |
| 228 | + ) |
| 229 | + |
| 230 | + |
183 | 231 | def test_server_spec_dedicated(): |
184 | 232 | machine_type, cpus, memory_gb = _server_spec(DEDICATED_SERVER) |
185 | 233 | assert machine_type == DEDICATED_RESOURCE.node_type |
|
0 commit comments