kopia lustrzana https://github.com/simonw/datasette
54 wiersze
1.6 KiB
Python
54 wiersze
1.6 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
|
||
|
|
|
||
|
|
|
||
|
|
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():
|
||
|
|
open("test.db", "w").write("data")
|
||
|
|
result = runner.invoke(cli.cli, ["package", "test.db"])
|
||
|
|
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():
|
||
|
|
open("test.db", "w").write("data")
|
||
|
|
result = runner.invoke(cli.cli, ["package", "test.db", "-p", "8080"])
|
||
|
|
assert 0 == result.exit_code
|
||
|
|
assert EXPECTED_DOCKERFILE.format(port=8080) == capture.captured
|