kopia lustrzana https://github.com/simonw/datasette
184 wiersze
5.6 KiB
Python
184 wiersze
5.6 KiB
Python
from click.testing import CliRunner
|
|
from datasette import cli
|
|
from unittest import mock
|
|
import os
|
|
import pathlib
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.serial
|
|
@mock.patch("shutil.which")
|
|
def test_publish_heroku_requires_heroku(mock_which, tmp_path_factory):
|
|
mock_which.return_value = False
|
|
runner = CliRunner()
|
|
os.chdir(tmp_path_factory.mktemp("runner"))
|
|
with open("test.db", "w") as fp:
|
|
fp.write("data")
|
|
result = runner.invoke(cli.cli, ["publish", "heroku", "test.db"])
|
|
assert result.exit_code == 1
|
|
assert "Publishing to Heroku requires heroku" in result.output
|
|
|
|
|
|
@pytest.mark.serial
|
|
@mock.patch("shutil.which")
|
|
@mock.patch("datasette.publish.heroku.check_output")
|
|
@mock.patch("datasette.publish.heroku.call")
|
|
def test_publish_heroku_installs_plugin(
|
|
mock_call, mock_check_output, mock_which, tmp_path_factory
|
|
):
|
|
mock_which.return_value = True
|
|
mock_check_output.side_effect = lambda s: {"['heroku', 'plugins']": b""}[repr(s)]
|
|
runner = CliRunner()
|
|
os.chdir(tmp_path_factory.mktemp("runner"))
|
|
with open("t.db", "w") as fp:
|
|
fp.write("data")
|
|
result = runner.invoke(cli.cli, ["publish", "heroku", "t.db"], input="y\n")
|
|
assert 0 != result.exit_code
|
|
mock_check_output.assert_has_calls(
|
|
[mock.call(["heroku", "plugins"]), mock.call(["heroku", "apps:list", "--json"])]
|
|
)
|
|
mock_call.assert_has_calls(
|
|
[mock.call(["heroku", "plugins:install", "heroku-builds"])]
|
|
)
|
|
|
|
|
|
@mock.patch("shutil.which")
|
|
def test_publish_heroku_invalid_database(mock_which):
|
|
mock_which.return_value = True
|
|
runner = CliRunner()
|
|
result = runner.invoke(cli.cli, ["publish", "heroku", "woop.db"])
|
|
assert result.exit_code == 2
|
|
assert "Path 'woop.db' does not exist" in result.output
|
|
|
|
|
|
@pytest.mark.serial
|
|
@mock.patch("shutil.which")
|
|
@mock.patch("datasette.publish.heroku.check_output")
|
|
@mock.patch("datasette.publish.heroku.call")
|
|
def test_publish_heroku(mock_call, mock_check_output, mock_which, tmp_path_factory):
|
|
mock_which.return_value = True
|
|
mock_check_output.side_effect = lambda s: {
|
|
"['heroku', 'plugins']": b"heroku-builds",
|
|
"['heroku', 'apps:list', '--json']": b"[]",
|
|
"['heroku', 'apps:create', 'datasette', '--json']": b'{"name": "f"}',
|
|
}[repr(s)]
|
|
runner = CliRunner()
|
|
os.chdir(tmp_path_factory.mktemp("runner"))
|
|
with open("test.db", "w") as fp:
|
|
fp.write("data")
|
|
result = runner.invoke(cli.cli, ["publish", "heroku", "test.db", "--tar", "gtar"])
|
|
assert 0 == result.exit_code, result.output
|
|
mock_call.assert_has_calls(
|
|
[
|
|
mock.call(
|
|
[
|
|
"heroku",
|
|
"builds:create",
|
|
"-a",
|
|
"f",
|
|
"--include-vcs-ignore",
|
|
"--tar",
|
|
"gtar",
|
|
]
|
|
),
|
|
]
|
|
)
|
|
|
|
|
|
@pytest.mark.serial
|
|
@mock.patch("shutil.which")
|
|
@mock.patch("datasette.publish.heroku.check_output")
|
|
@mock.patch("datasette.publish.heroku.call")
|
|
def test_publish_heroku_plugin_secrets(
|
|
mock_call, mock_check_output, mock_which, tmp_path_factory
|
|
):
|
|
mock_which.return_value = True
|
|
mock_check_output.side_effect = lambda s: {
|
|
"['heroku', 'plugins']": b"heroku-builds",
|
|
"['heroku', 'apps:list', '--json']": b"[]",
|
|
"['heroku', 'apps:create', 'datasette', '--json']": b'{"name": "f"}',
|
|
}[repr(s)]
|
|
runner = CliRunner()
|
|
os.chdir(tmp_path_factory.mktemp("runner"))
|
|
with open("test.db", "w") as fp:
|
|
fp.write("data")
|
|
result = runner.invoke(
|
|
cli.cli,
|
|
[
|
|
"publish",
|
|
"heroku",
|
|
"test.db",
|
|
"--plugin-secret",
|
|
"datasette-auth-github",
|
|
"client_id",
|
|
"x-client-id",
|
|
],
|
|
)
|
|
assert 0 == result.exit_code, result.output
|
|
mock_call.assert_has_calls(
|
|
[
|
|
mock.call(
|
|
[
|
|
"heroku",
|
|
"config:set",
|
|
"-a",
|
|
"f",
|
|
"DATASETTE_AUTH_GITHUB_CLIENT_ID=x-client-id",
|
|
]
|
|
),
|
|
mock.call(["heroku", "builds:create", "-a", "f", "--include-vcs-ignore"]),
|
|
]
|
|
)
|
|
|
|
|
|
@pytest.mark.serial
|
|
@mock.patch("shutil.which")
|
|
@mock.patch("datasette.publish.heroku.check_output")
|
|
@mock.patch("datasette.publish.heroku.call")
|
|
def test_publish_heroku_generate_dir(
|
|
mock_call, mock_check_output, mock_which, tmp_path_factory
|
|
):
|
|
mock_which.return_value = True
|
|
mock_check_output.side_effect = lambda s: {
|
|
"['heroku', 'plugins']": b"heroku-builds",
|
|
}[repr(s)]
|
|
runner = CliRunner()
|
|
os.chdir(tmp_path_factory.mktemp("runner"))
|
|
with open("test.db", "w") as fp:
|
|
fp.write("data")
|
|
output = str(tmp_path_factory.mktemp("generate_dir") / "output")
|
|
result = runner.invoke(
|
|
cli.cli,
|
|
[
|
|
"publish",
|
|
"heroku",
|
|
"test.db",
|
|
"--generate-dir",
|
|
output,
|
|
],
|
|
)
|
|
assert result.exit_code == 0
|
|
path = pathlib.Path(output)
|
|
assert path.exists()
|
|
file_names = {str(r.relative_to(path)) for r in path.glob("*")}
|
|
assert file_names == {
|
|
"requirements.txt",
|
|
"bin",
|
|
"runtime.txt",
|
|
"Procfile",
|
|
"test.db",
|
|
}
|
|
for name, expected in (
|
|
("requirements.txt", "datasette"),
|
|
("runtime.txt", "python-3.11.0"),
|
|
(
|
|
"Procfile",
|
|
(
|
|
"web: datasette serve --host 0.0.0.0 -i test.db "
|
|
"--cors --port $PORT --inspect-file inspect-data.json"
|
|
),
|
|
),
|
|
):
|
|
with open(path / name) as fp:
|
|
assert fp.read().strip() == expected
|