def test_auth_api_server_error_missing_entries(): client = RESTAPIClient(_REMOTE) def callback(request, uri, response_headers): return [200, response_headers, json.dumps({})] httpretty.register_uri( httpretty.HTTPretty.POST, _ENDPOINT + "/create_machine_credentials", body=callback ) with pytest.raises(AuthAPIError) as e: client.create_machine_credentials("AAABBBB", "somepassword") assert "Missing entries" in str(e)
def register_c(username, password, email, remote, accept_tos, skip_inject): """ Register the user on a Splitgraph registry. By default, this registers the user on data.splitgraph.com, obtains a set of machine (API) credentials for the client to communicate with the registry and configures the data.splitgraph.com engine. """ from splitgraph.cloud import RESTAPIClient, DEFAULT_REMOTES from splitgraph.config import CONFIG from splitgraph.config.config import get_all_in_subsection client = RESTAPIClient(remote) tos = client.tos() if tos and not accept_tos: click.echo("%s says: %s" % (client.endpoint, tos)) click.confirm("Do you accept the Terms of Service?", default=False, abort=True) click.echo("Registering the user...") uuid, access, refresh = client.register(username, password, email, accept_tos=True) click.echo("Registration successful. UUID %s" % uuid) key, secret = client.create_machine_credentials(access, password) click.echo("Acquired refresh token and API keys") repo_lookup = _update_repo_lookup(CONFIG, remote) # If remote isn't in the config (is one of the default ones), add the default parameters # to the new config -- otherwise only overwrite the new fields in the existing config. remote_params = (copy(DEFAULT_REMOTES.get(remote, {})) if not get_all_in_subsection(CONFIG, "remotes", remote) else {}) remote_params.update({ "SG_ENGINE_USER": key, "SG_ENGINE_PWD": secret, "SG_NAMESPACE": username, "SG_CLOUD_REFRESH_TOKEN": refresh, "SG_CLOUD_ACCESS_TOKEN": access, "SG_IS_REGISTRY": "true", }) config_patch = { "SG_REPO_LOOKUP": repo_lookup, "remotes": { remote: remote_params }, } config_path = patch_and_save_config(CONFIG, config_patch) if not skip_inject: inject_config_into_engines(CONFIG["SG_ENGINE_PREFIX"], config_path) click.echo("Done.")
def login_c(username, password, remote, overwrite, skip_inject): """Log into a Splitgraph registry with username/password. This will generate a new refresh token (to use the Splitgraph query API) and API keys to let sgr access the registry (if they don't already exist in the configuration file or if the actual username has changed). Note that if you already have generated an API key pair but it's not in the configuration file, this will generate a new pair instead of restoring the existing one, as the API secret is only stored in the configuration file. If you want to log in using an existing API key pair, use `sgr cloud login-api` instead. """ from splitgraph.config import CONFIG from splitgraph.config.config import get_all_in_subsection from splitgraph.cloud import RESTAPIClient, get_token_claim, DEFAULT_REMOTES client = RESTAPIClient(remote) if not password: profile_url = _construct_user_profile_url(client.endpoint) password = click.prompt( text="Password (visit %s if you don't have it)" % profile_url, confirmation_prompt=False, hide_input=True, ) access, refresh = client.get_refresh_token(username, password) # Extract namespace from the access token since we might have logged in with an e-mail. namespace = get_token_claim(access, "username") click.echo("Logged into %s as %s" % (remote, namespace)) config_remote_params = get_all_in_subsection(CONFIG, "remotes", remote) remote_params = copy(DEFAULT_REMOTES.get( remote, {})) if not config_remote_params else {} remote_params.update({ "SG_NAMESPACE": namespace, "SG_CLOUD_REFRESH_TOKEN": refresh, "SG_CLOUD_ACCESS_TOKEN": access, }) config_patch = { "SG_REPO_LOOKUP": _update_repo_lookup(CONFIG, remote), "remotes": { remote: remote_params }, } # Get new tokens in any case if we're logging in under a different username. try: username_changed = namespace != CONFIG["remotes"][remote][ "SG_NAMESPACE"] except KeyError: username_changed = False if ("SG_ENGINE_USER" not in config_remote_params or "SG_ENGINE_PWD" not in config_remote_params or overwrite or username_changed): key, secret = client.create_machine_credentials(access, password) config_patch["remotes"][remote]["SG_ENGINE_USER"] = key config_patch["remotes"][remote]["SG_ENGINE_PWD"] = secret click.echo("Acquired new API keys") config_path = patch_and_save_config(CONFIG, config_patch) if not skip_inject: inject_config_into_engines(CONFIG["SG_ENGINE_PREFIX"], config_path)