Joakim Hellsén
be43c468a4
- Updated dependencies and added Ruff configuration for linting in pyproject.toml. - Improved type hinting in DockerComposeManager methods for better clarity. - Created an empty __init__.py file in tests directory.
68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
import pytest
|
|
import yaml
|
|
|
|
from compose import DockerComposeManager
|
|
|
|
if TYPE_CHECKING:
|
|
from pathlib import Path
|
|
|
|
|
|
def test_create_and_save_service(tmp_path: Path) -> None:
|
|
compose_file: Path = tmp_path / "docker-compose.yml"
|
|
manager = DockerComposeManager(str(compose_file))
|
|
manager.create_service(
|
|
name="web",
|
|
image="nginx:latest",
|
|
ports=["80:80"],
|
|
environment={"ENV_VAR": "value"},
|
|
).save()
|
|
# Check file exists and content
|
|
assert compose_file.exists()
|
|
with compose_file.open() as f:
|
|
data = yaml.safe_load(f)
|
|
assert "web" in data["services"]
|
|
assert data["services"]["web"]["image"] == "nginx:latest"
|
|
assert data["services"]["web"]["ports"] == ["80:80"]
|
|
assert data["services"]["web"]["environment"] == {"ENV_VAR": "value"}
|
|
|
|
|
|
def test_modify_service(tmp_path: Path) -> None:
|
|
compose_file: Path = tmp_path / "docker-compose.yml"
|
|
manager = DockerComposeManager(str(compose_file))
|
|
manager.create_service(name="web", image="nginx:latest").save()
|
|
manager.modify_service(name="web", image="nginx:1.19", ports=["8080:80"]).save()
|
|
with compose_file.open() as f:
|
|
data = yaml.safe_load(f)
|
|
assert data["services"]["web"]["image"] == "nginx:1.19"
|
|
assert data["services"]["web"]["ports"] == ["8080:80"]
|
|
|
|
|
|
def test_remove_service(tmp_path: Path) -> None:
|
|
compose_file: Path = tmp_path / "docker-compose.yml"
|
|
manager = DockerComposeManager(str(compose_file))
|
|
manager.create_service(name="web", image="nginx:latest").save()
|
|
manager.remove_service("web").save()
|
|
with compose_file.open() as f:
|
|
data = yaml.safe_load(f)
|
|
assert "web" not in data["services"]
|
|
|
|
|
|
def test_context_manager(tmp_path: Path) -> None:
|
|
compose_file: Path = tmp_path / "docker-compose.yml"
|
|
with DockerComposeManager(str(compose_file)) as manager:
|
|
manager.create_service(name="web", image="nginx:latest")
|
|
with compose_file.open() as f:
|
|
data = yaml.safe_load(f)
|
|
assert "web" in data["services"]
|
|
|
|
|
|
def test_modify_nonexistent_service(tmp_path: Path) -> None:
|
|
compose_file: Path = tmp_path / "docker-compose.yml"
|
|
manager = DockerComposeManager(str(compose_file))
|
|
with pytest.raises(KeyError):
|
|
manager.modify_service("notfound", image="nginx:latest")
|