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
