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 | 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', | ||||
| 'expires_in': 600, | 'expires_in': 600, | ||||
| 'refresh_expires_in': 0, | 'refresh_expires_in': 0, | ||||
| 'refresh_token': 'some-refresh-token', | 'refresh_token': 'some-refresh-token', | ||||
| 'token_type': 'bearer', | 'token_type': 'bearer', | ||||
| 'session_state': 'some-state', | 'session_state': 'some-state', | ||||
| 'scope': 'openid email profile offline_access', | 'scope': 'openid email profile offline_access', | ||||
| } | } | ||||
| def test_auth_login(mocker): | 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 json.loads(result.output) == oidc_profile | ||||
| 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): | def test_auth_refresh(mocker): | ||||
| mock_oidc_session = mocker.patch('swh.web.client.cli.OpenIDConnectSession') | mock_oidc_session = mocker.patch('swh.web.client.cli.OpenIDConnectSession') | ||||
| mock_refresh = mock_oidc_session.return_value.refresh | mock_refresh = mock_oidc_session.return_value.refresh | ||||
| mock_refresh.return_value = _oidc_profile | mock_refresh.return_value = oidc_profile | ||||
| result = runner.invoke(auth, ['refresh', _oidc_profile['refresh_token']]) | result = runner.invoke(auth, ['refresh', oidc_profile['refresh_token']]) | ||||
| assert result.exit_code == 0 | assert result.exit_code == 0 | ||||
| assert json.loads(result.stdout) == _oidc_profile['access_token'] | assert json.loads(result.stdout) == oidc_profile['access_token'] | ||||
| mock_refresh.side_effect = Exception('Auth error') | mock_refresh.side_effect = Exception('Auth error') | ||||
| result = runner.invoke(auth, ['refresh', _oidc_profile['refresh_token']]) | result = runner.invoke(auth, ['refresh', oidc_profile['refresh_token']]) | ||||
| assert result.exit_code == 1 | 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 | ||||