def test_get_user_error(requests_mock: Mocker, git_repo_manager: GitRepoManagerClient): get_user_mock = requests_mock.get( f'{git_repo_manager.base_url}/api/v1/users/{FAKE_USER}', status_code=500) with pytest.raises(HTTPError): git_repo_manager.get_user(FAKE_USER) assert get_user_mock.call_count == 1
def upgrade(ctx: click.Context): """ Upgrade users after Nauta upgrade. """ with spinner(text=Texts.UPGRADE_IN_PROGRESS): # noinspection PyBroadException try: # noinspection PyTypeChecker users: List[User] = User.list() with K8sProxy(NAUTAAppNames.GIT_REPO_MANAGER, number_of_retries_wait_for_readiness=60) as proxy: grm_client = GitRepoManagerClient(host='127.0.0.1', port=proxy.tunnel_port) for user in users: grm_user = grm_client.get_user(user.name) if not grm_user: grm_client.add_nauta_user(user.name) except Exception: handle_error(logger, Texts.UPGRADE_FAILED, Texts.UPGRADE_FAILED, add_verbosity_msg=ctx.obj.verbosity == 0) sys.exit(1) click.echo(Texts.UPGRADE_SUCCEEDED)
def test_get_user_not_found(requests_mock: Mocker, git_repo_manager: GitRepoManagerClient): get_user_mock = requests_mock.get( f'{git_repo_manager.base_url}/api/v1/users/{FAKE_USER}', status_code=404) response = git_repo_manager.get_user(FAKE_USER) assert not response assert get_user_mock.call_count == 1