from behave import step


# Helper
def _visit(context, path, path_args=None):
    if path_args is None:
        path_args = {}
    context.response = context.atlas_um_client.get(path, path_args)


@step("I visit the accounts page")
def step_visit_accounts_page(context):
    _visit(context, "/accounts")


@step("I visit the USM login page")
def step_visit_usm_login_page(context):
    _visit(context, "/usm/login")


@step("I visit the logout page")
def step_visit_logout_page(context):
    _visit(context, "/usm/logout")


@step("I should be redirected and follow")
def step_i_should_be_redirected_and_follow(context):
    assert context.response.status_code == 302, "Not redirected"
    _visit(context, context.response.location)


@step("I visit the account creation page")
def step_visit_te_account_creation_page(context):
    _visit(context, "/accounts/create")


@step('I send "{method}" request to "{path}"')
def step_send_request_to_path(context, method, path):
    if method == "GET":
        context.response = context.atlas_um_client.get(path)
    else:
        raise NotImplementedError("TODO: Add corresponding call")


@step("I visit admin settings page")
def step_visit_admin_settings_page(context):
    _visit(context, "/settings")