docker-tidy/dockertidy/test/unit/test_autostop.py

49 lines
1.2 KiB
Python

"""Test Autostop class."""
import docker
import pytest
from dockertidy import autostop
pytest_plugins = [
"dockertidy.test.fixtures.fixtures",
]
@pytest.fixture
def autostop_fixture(mocker):
mocker.patch.object(
autostop.AutoStop,
"_get_docker_client",
return_value=mocker.create_autospec(docker.APIClient)
)
stop = autostop.AutoStop()
return stop
def test_stop_container(autostop_fixture, mocker):
client = mocker.create_autospec(docker.APIClient)
cid = "asdb"
autostop_fixture._stop_container(client, cid)
client.stop.assert_called_once_with(cid)
def test_build_container_matcher(autostop_fixture, mocker):
prefixes = ["one_", "two_"]
matcher = autostop_fixture._build_container_matcher(prefixes)
assert matcher("one_container")
assert matcher("two_container")
assert not matcher("three_container")
assert not matcher("one")
def test_has_been_running_since_true(autostop_fixture, container, later_time):
assert autostop_fixture._has_been_running_since(container, later_time)
def test_has_been_running_since_false(autostop_fixture, container, earlier_time):
assert not autostop_fixture._has_been_running_since(container, earlier_time)