mirror of
https://github.com/simonw/datasette.git
synced 2025-12-10 16:51:24 +01:00
8e18c79431Context manager with open closes the files after usage. When the object is already a pathlib.Path i used read_text write_text functions In some cases pathlib.Path.open were used in context manager, it is basically the same as builtin open. Thanks, Konstantin Baikov!
57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
from click.testing import CliRunner
|
|
from datasette import cli
|
|
from unittest import mock
|
|
import pathlib
|
|
import json
|
|
|
|
|
|
class CaptureDockerfile:
|
|
def __call__(self, _):
|
|
self.captured = (pathlib.Path() / "Dockerfile").read_text()
|
|
|
|
|
|
EXPECTED_DOCKERFILE = """
|
|
FROM python:3.8
|
|
COPY . /app
|
|
WORKDIR /app
|
|
|
|
ENV DATASETTE_SECRET 'sekrit'
|
|
RUN pip install -U datasette
|
|
RUN datasette inspect test.db --inspect-file inspect-data.json
|
|
ENV PORT {port}
|
|
EXPOSE {port}
|
|
CMD datasette serve --host 0.0.0.0 -i test.db --cors --inspect-file inspect-data.json --port $PORT
|
|
""".strip()
|
|
|
|
|
|
@mock.patch("shutil.which")
|
|
@mock.patch("datasette.cli.call")
|
|
def test_package(mock_call, mock_which):
|
|
mock_which.return_value = True
|
|
runner = CliRunner()
|
|
capture = CaptureDockerfile()
|
|
mock_call.side_effect = capture
|
|
with runner.isolated_filesystem():
|
|
with open("test.db", "w") as fp:
|
|
fp.write("data")
|
|
result = runner.invoke(cli.cli, ["package", "test.db", "--secret", "sekrit"])
|
|
assert 0 == result.exit_code
|
|
mock_call.assert_has_calls([mock.call(["docker", "build", "."])])
|
|
assert EXPECTED_DOCKERFILE.format(port=8001) == capture.captured
|
|
|
|
|
|
@mock.patch("shutil.which")
|
|
@mock.patch("datasette.cli.call")
|
|
def test_package_with_port(mock_call, mock_which):
|
|
mock_which.return_value = True
|
|
capture = CaptureDockerfile()
|
|
mock_call.side_effect = capture
|
|
runner = CliRunner()
|
|
with runner.isolated_filesystem():
|
|
with open("test.db", "w") as fp:
|
|
fp.write("data")
|
|
result = runner.invoke(
|
|
cli.cli, ["package", "test.db", "-p", "8080", "--secret", "sekrit"]
|
|
)
|
|
assert 0 == result.exit_code
|
|
assert EXPECTED_DOCKERFILE.format(port=8080) == capture.captured
|