Differential D2861 Diff 10232 swh/web/client/tests/test_cli.py

Changeset View

Standalone View

swh/web/client/tests/test_cli.py

This file was added.

				# Copyright (C) 2020 The Software Heritage developers
				# See the AUTHORS file at the top-level directory of this distribution
				# License: GNU General Public License version 3, or any later version
				# See top-level LICENSE file for more information

				import json

				from click.testing import CliRunner

				from swh.web.client.cli import auth

				runner = CliRunner()

				_oidc_profile = {
				'access_token': 'some-access-token',
				'expires_in': 600,
				'refresh_expires_in': 0,
				'refresh_token': 'some-refresh-token',
				'token_type': 'bearer',
				'session_state': 'some-state',
				'scope': 'openid email profile offline_access',
				}


				def test_auth_login(mocker):
				mock_getpass = mocker.patch('swh.web.client.cli.getpass')
				mock_getpass.return_value = 'password'
				vlorentzUnsubmitted Not Done Inline Actions You can use the `input=` argument of `runner.invoke` instead vlorentz: You can use the `input=` argument of `runner.invoke` instead
				anlambertAuthorUnsubmitted Done Inline Actions nice, thanks anlambert: nice, thanks
				anlambertAuthorUnsubmitted Done Inline Actions Unfortunately this does not seem to play well with `getpass`, providing `input='password\n'` still makes the prompt appear. Let's keep this mock then. anlambert: Unfortunately this does not seem to play well with `getpass`, providing `input='password\n'`…
				vlorentzUnsubmitted Not Done Inline Actions indeed, getpass, reads from the tty vlorentz: indeed, getpass, reads from the tty
				mock_oidc_session = mocker.patch('swh.web.client.cli.OpenIDConnectSession')
				mock_login = mock_oidc_session.return_value.login
				mock_login.return_value = _oidc_profile

				result = runner.invoke(auth, ['login', 'username'], input='password\n')
				assert result.exit_code == 0
				assert json.loads(result.output) == _oidc_profile

				mock_login.side_effect = Exception('Auth error')

				result = runner.invoke(auth, ['login', 'username'], input='password\n')
				assert result.exit_code == 1

				vlorentzUnsubmitted Not Done Inline Actions can remove this. vlorentz: can remove this.

				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


				vlorentzUnsubmitted Done Inline Actions needs a test of failure for each command vlorentz: needs a test of failure for each command
				anlambertAuthorUnsubmitted Done Inline Actions ack anlambert: ack
				def test_auth_logout(mocker):

				mock_oidc_session = mocker.patch('swh.web.client.cli.OpenIDConnectSession')
				mock_logout = mock_oidc_session.return_value.logout

				result = runner.invoke(auth, ['logout', _oidc_profile['refresh_token']])
				assert result.exit_code == 0

				mock_logout.side_effect = Exception('Auth error')
				result = runner.invoke(auth, ['logout', _oidc_profile['refresh_token']])
				assert result.exit_code == 1