Add test for denylist (#1914)

* Add test for denylist

* Mock API in conftest
This commit is contained in:
Joakim Sørensen 2020-08-14 18:09:03 +02:00 committed by GitHub
parent 930eed4500
commit f5f5f829ac
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 29 additions and 3 deletions

View File

@ -4,15 +4,26 @@ from unittest.mock import MagicMock, PropertyMock, patch
import pytest
from supervisor.bootstrap import initialize_coresys
from supervisor.docker import DockerAPI
# pylint: disable=redefined-outer-name, protected-access
@pytest.fixture
def docker():
"""Mock Docker API."""
with patch("supervisor.coresys.DockerAPI") as mock:
yield mock
"""Mock DockerAPI."""
images = [MagicMock(tags=["homeassistant/amd64-hassio-supervisor:latest"])]
with patch("docker.DockerClient", return_value=MagicMock()), patch(
"supervisor.docker.DockerAPI.images", return_value=MagicMock()
), patch("supervisor.docker.DockerAPI.containers", return_value=MagicMock()), patch(
"supervisor.docker.DockerAPI.api", return_value=MagicMock()
), patch(
"supervisor.docker.DockerAPI.images.list", return_value=images
):
docker_obj = DockerAPI()
yield docker_obj
@pytest.fixture

View File

@ -0,0 +1,15 @@
"""Test tags in denylist."""
from unittest.mock import MagicMock, patch
def test_has_images_in_denylist(docker):
"""Test tags in denylist exsist."""
images = [MagicMock(tags=["containrrr/watchtower:latest"])]
with patch("supervisor.docker.DockerAPI.images.list", return_value=images):
assert docker.check_denylist_images()
def test_no_images_in_denylist(docker):
"""Test tags in denylist does not exsist."""
assert not docker.check_denylist_images()