mirror of
https://github.com/simonw/datasette.git
synced 2025-12-10 16:51:24 +01:00
35d6ee2790I've run the black code formatting tool against everything:
black tests datasette setup.py
I also added a new unit test, in tests/test_black.py, which will fail if the code does not
conform to black's exacting standards.
This unit test only runs on Python 3.6 or higher, because black itself doesn't run on 3.5.
76 lines
2.9 KiB
Python
76 lines
2.9 KiB
Python
from click.testing import CliRunner
|
|
from datasette import cli
|
|
from unittest import mock
|
|
|
|
|
|
@mock.patch("shutil.which")
|
|
def test_publish_heroku_requires_heroku(mock_which):
|
|
mock_which.return_value = False
|
|
runner = CliRunner()
|
|
with runner.isolated_filesystem():
|
|
open("test.db", "w").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
|
|
|
|
|
|
@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):
|
|
mock_which.return_value = True
|
|
mock_check_output.side_effect = lambda s: {"['heroku', 'plugins']": b""}[repr(s)]
|
|
runner = CliRunner()
|
|
with runner.isolated_filesystem():
|
|
open("t.db", "w").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
|
|
|
|
|
|
@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):
|
|
mock_which.return_varue = 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()
|
|
with runner.isolated_filesystem():
|
|
open("test.db", "w").write("data")
|
|
result = runner.invoke(cli.cli, ["publish", "heroku", "test.db"])
|
|
assert 0 == result.exit_code, result.output
|
|
mock_call.assert_called_once_with(
|
|
["heroku", "builds:create", "-a", "f", "--include-vcs-ignore"]
|
|
)
|
|
|
|
|
|
@mock.patch("shutil.which")
|
|
@mock.patch("datasette.publish.now.call")
|
|
def test_publish_now_force_token(mock_call, mock_which):
|
|
mock_which.return_value = True
|
|
runner = CliRunner()
|
|
with runner.isolated_filesystem():
|
|
open("test.db", "w").write("data")
|
|
result = runner.invoke(
|
|
cli.cli, ["publish", "now", "test.db", "--force", "--token=X"]
|
|
)
|
|
assert 0 == result.exit_code
|
|
mock_call.assert_called_once_with(["now", "--force", "--token=X"])
|