mirror of
https://github.com/simonw/dclient.git
synced 2026-07-23 09:24:31 +02:00
120 lines
3.8 KiB
Python
120 lines
3.8 KiB
Python
from click.testing import CliRunner
|
|
from dclient.cli import cli
|
|
import json
|
|
import pathlib
|
|
import pytest
|
|
|
|
|
|
def test_query_error(httpx_mock):
|
|
httpx_mock.add_response(
|
|
json={
|
|
"ok": False,
|
|
"error": "Statement must be a SELECT",
|
|
"status": 400,
|
|
"title": "Invalid SQL",
|
|
},
|
|
status_code=400,
|
|
)
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, ["query", "https://example.com", "hello"])
|
|
assert result.exit_code == 1
|
|
assert (
|
|
result.output
|
|
== "Error: 400 status code. Invalid SQL: Statement must be a SELECT\n"
|
|
)
|
|
|
|
|
|
@pytest.mark.parametrize("with_token", (False, True))
|
|
def test_query(httpx_mock, with_token):
|
|
httpx_mock.add_response(
|
|
json={
|
|
"ok": True,
|
|
"database": "fixtures",
|
|
"query_name": None,
|
|
"rows": [{"5 * 2": 10}],
|
|
"truncated": False,
|
|
"columns": ["5 * 2"],
|
|
"query": {"sql": "select 5 * 2", "params": {}},
|
|
"error": None,
|
|
"private": False,
|
|
"allow_execute_sql": True,
|
|
},
|
|
status_code=200,
|
|
)
|
|
runner = CliRunner()
|
|
args = ["query", "https://example.com", "hello"]
|
|
if with_token:
|
|
args.append("--token")
|
|
args.append("xyz")
|
|
result = runner.invoke(cli, args)
|
|
assert result.exit_code == 0
|
|
assert json.loads(result.output) == [{"5 * 2": 10}]
|
|
|
|
# Check the request
|
|
request = httpx_mock.get_request()
|
|
assert str(request.url) == "https://example.com.json?sql=hello&_shape=objects"
|
|
if with_token:
|
|
assert request.headers["authorization"] == "Bearer xyz"
|
|
else:
|
|
assert "authorization" not in request.headers
|
|
|
|
|
|
def test_aliases(mocker, tmpdir, httpx_mock):
|
|
mocker.patch("dclient.cli.get_config_dir", return_value=pathlib.Path(tmpdir))
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli, ["alias", "list"])
|
|
assert result.exit_code == 0
|
|
assert result.output == ""
|
|
|
|
result = runner.invoke(cli, ["alias", "add", "foo", "https://example.com/foo"])
|
|
assert result.exit_code == 0
|
|
assert result.output == ""
|
|
|
|
result = runner.invoke(cli, ["alias", "list"])
|
|
assert result.exit_code == 0
|
|
assert result.output == "foo = https://example.com/foo\n"
|
|
|
|
# --json mode:
|
|
result = runner.invoke(cli, ["alias", "list", "--json"])
|
|
assert result.exit_code == 0
|
|
assert json.loads(result.output) == {"foo": "https://example.com/foo"}
|
|
|
|
# Check the aliases file
|
|
aliases_file = pathlib.Path(tmpdir) / "aliases.json"
|
|
assert json.loads(aliases_file.read_text()) == {"foo": "https://example.com/foo"}
|
|
|
|
# Try a query against that alias
|
|
httpx_mock.add_response(
|
|
json={
|
|
"ok": True,
|
|
"database": "foo",
|
|
"query_name": None,
|
|
"rows": [{"11 * 3": 33}],
|
|
"truncated": False,
|
|
"columns": ["11 * 3"],
|
|
"query": {"sql": "select 11 * 3", "params": {}},
|
|
"error": None,
|
|
"private": False,
|
|
"allow_execute_sql": True,
|
|
},
|
|
status_code=200,
|
|
)
|
|
result = runner.invoke(cli, ["query", "foo", "select 11 * 3"])
|
|
assert result.exit_code == 0
|
|
assert json.loads(result.output) == [{"11 * 3": 33}]
|
|
|
|
# Should have hit https://example.com/foo.json
|
|
url = httpx_mock.get_request().url
|
|
assert url.host == "example.com"
|
|
assert url.path == "/foo.json"
|
|
assert dict(url.params) == {"sql": "select 11 * 3", "_shape": "objects"}
|
|
|
|
# Remove alias
|
|
result = runner.invoke(cli, ["alias", "remove", "invalid"])
|
|
assert result.exit_code == 1
|
|
assert result.output == "Error: No such alias\n"
|
|
|
|
result = runner.invoke(cli, ["alias", "remove", "foo"])
|
|
assert result.exit_code == 0
|
|
assert result.output == ""
|
|
assert json.loads(aliases_file.read_text()) == {}
|