Changeset View
Changeset View
Standalone View
Standalone View
swh/web/client/tests/test_cli.py
| # Copyright (C) 2020 The Software Heritage developers | # Copyright (C) 2020 The Software Heritage developers | ||||
| # See the AUTHORS file at the top-level directory of this distribution | # See the AUTHORS file at the top-level directory of this distribution | ||||
| # License: GNU General Public License version 3, or any later version | # License: GNU General Public License version 3, or any later version | ||||
| # See top-level LICENSE file for more information | # See top-level LICENSE file for more information | ||||
| import json | |||||
| from click.testing import CliRunner | from click.testing import CliRunner | ||||
| from swh.web.client.cli import auth | from swh.web.client.cli import auth | ||||
| runner = CliRunner() | runner = CliRunner() | ||||
| oidc_profile = { | oidc_profile = { | ||||
| "access_token": "some-access-token", | "access_token": "some-access-token", | ||||
| Show All 10 Lines | def test_auth_login(mocker): | ||||
| mock_getpass = mocker.patch("swh.web.client.cli.getpass") | mock_getpass = mocker.patch("swh.web.client.cli.getpass") | ||||
| mock_getpass.return_value = "password" | mock_getpass.return_value = "password" | ||||
| mock_oidc_session = mocker.patch("swh.web.client.cli.OpenIDConnectSession") | mock_oidc_session = mocker.patch("swh.web.client.cli.OpenIDConnectSession") | ||||
| mock_login = mock_oidc_session.return_value.login | mock_login = mock_oidc_session.return_value.login | ||||
| mock_login.return_value = oidc_profile | mock_login.return_value = oidc_profile | ||||
| result = runner.invoke(auth, ["login", "username"], input="password\n") | result = runner.invoke(auth, ["login", "username"], input="password\n") | ||||
| assert result.exit_code == 0 | assert result.exit_code == 0 | ||||
| assert json.loads(result.output) == oidc_profile | assert result.output[:-1] == oidc_profile["refresh_token"] | ||||
| mock_login.side_effect = Exception("Auth error") | mock_login.side_effect = Exception("Auth error") | ||||
| result = runner.invoke(auth, ["login", "username"], input="password\n") | result = runner.invoke(auth, ["login", "username"], input="password\n") | ||||
| assert result.exit_code == 1 | assert result.exit_code == 1 | ||||
| def test_auth_refresh(mocker): | |||||
| mock_oidc_session = mocker.patch("swh.web.client.cli.OpenIDConnectSession") | |||||
| mock_refresh = mock_oidc_session.return_value.refresh | |||||
| mock_refresh.return_value = oidc_profile | |||||
| result = runner.invoke(auth, ["refresh", oidc_profile["refresh_token"]]) | |||||
| assert result.exit_code == 0 | |||||
| assert json.loads(result.stdout) == oidc_profile["access_token"] | |||||
| mock_refresh.side_effect = Exception("Auth error") | |||||
| result = runner.invoke(auth, ["refresh", oidc_profile["refresh_token"]]) | |||||
| assert result.exit_code == 1 | |||||
| def test_auth_logout(mocker): | def test_auth_logout(mocker): | ||||
| mock_oidc_session = mocker.patch("swh.web.client.cli.OpenIDConnectSession") | mock_oidc_session = mocker.patch("swh.web.client.cli.OpenIDConnectSession") | ||||
| mock_logout = mock_oidc_session.return_value.logout | mock_logout = mock_oidc_session.return_value.logout | ||||
| result = runner.invoke(auth, ["logout", oidc_profile["refresh_token"]]) | result = runner.invoke(auth, ["logout", oidc_profile["refresh_token"]]) | ||||
| assert result.exit_code == 0 | assert result.exit_code == 0 | ||||
| mock_logout.side_effect = Exception("Auth error") | mock_logout.side_effect = Exception("Auth error") | ||||
| result = runner.invoke(auth, ["logout", oidc_profile["refresh_token"]]) | result = runner.invoke(auth, ["logout", oidc_profile["refresh_token"]]) | ||||
| assert result.exit_code == 1 | assert result.exit_code == 1 | ||||