docker-autocompose/tests/test_autocompose.py

import unittest
from unittest.mock import patch

from src import autocompose


class FakeContainerSummary:
    def __init__(self, name, short_id):
        self.name = name
        self.short_id = short_id


class FakeContainerWithAttrs:
    def __init__(self, attrs):
        self.attrs = attrs


class FakeContainers:
    def __init__(self, attrs, name="svc", short_id="abc123"):
        self._attrs = attrs
        self._summary = FakeContainerSummary(name=name, short_id=short_id)

    def list(self, all=True):
        return [self._summary]

    def get(self, _cid):
        return FakeContainerWithAttrs(self._attrs)


class FakeNetwork:
    def __init__(self, name, internal=False):
        self.attrs = {"Name": name, "Internal": internal}


class FakeNetworks:
    def __init__(self, networks):
        self._networks = [FakeNetwork(name, internal) for name, internal in networks]

    def list(self):
        return self._networks


class FakeDockerClient:
    def __init__(self, attrs, networks):
        self.containers = FakeContainers(attrs)
        self.networks = FakeNetworks(networks)


def make_attrs(networks, healthcheck=None):
    return {
        "Name": "/svc",
        "HostConfig": {
            "CapDrop": None,
            "CgroupParent": None,
            "Dns": None,
            "DnsSearch": None,
            "ExtraHosts": None,
            "Links": None,
            "LogConfig": {"Type": None, "Config": None},
            "SecurityOpt": None,
            "Ulimits": None,
            "VolumeDriver": None,
            "VolumesFrom": None,
            "IpcMode": None,
            "Privileged": None,
            "RestartPolicy": {"Name": None},
            "ReadonlyRootfs": None,
            "Devices": None,
            "PortBindings": {},
        },
        "Config": {
            "Env": None,
            "Image": "test:latest",
            "Labels": {},
            "Entrypoint": None,
            "User": None,
            "WorkingDir": None,
            "Domainname": None,
            "Hostname": None,
            "OpenStdin": None,
            "Tty": None,
            "Cmd": None,
            "ExposedPorts": {},
            "Healthcheck": healthcheck,
        },
        "NetworkSettings": {"Networks": networks, "MacAddress": None},
        "Mounts": [],
    }


class GenerateTests(unittest.TestCase):
    def test_preserves_network_aliases_and_healthcheck(self):
        attrs = make_attrs(
            {
                "custom_net": {
                    "Aliases": ["svc", "db"],
                }
            },
            healthcheck={
                "Test": ["CMD-SHELL", "echo ok"],
                "Interval": 1000000000,
                "Timeout": 2000000000,
                "Retries": 3,
                "StartPeriod": 3000000000,
            },
        )
        fake_client = FakeDockerClient(attrs, networks=[("custom_net", False)])

        with patch("src.autocompose.docker.from_env", return_value=fake_client):
            cfile, c_networks, _ = autocompose.generate("svc")

        service = cfile["svc"]
        self.assertEqual(service["networks"], {"custom_net": {"aliases": ["svc", "db"]}})
        self.assertEqual(
            service["healthcheck"],
            {
                "test": ["CMD-SHELL", "echo ok"],
                "interval": "1000000000ns",
                "timeout": "2000000000ns",
                "retries": 3,
                "start_period": "3000000000ns",
            },
        )
        self.assertEqual(
            c_networks,
            {"custom_net": {"external": True, "name": "custom_net"}},
        )

    def test_simple_custom_network_uses_list_form(self):
        attrs = make_attrs({"custom_net": {}})
        fake_client = FakeDockerClient(attrs, networks=[("custom_net", True)])

        with patch("src.autocompose.docker.from_env", return_value=fake_client):
            cfile, c_networks, _ = autocompose.generate("svc")

        self.assertEqual(cfile["svc"]["networks"], ["custom_net"])
        self.assertEqual(
            c_networks,
            {"custom_net": {"external": False, "name": "custom_net"}},
        )

    def test_default_network_still_uses_network_mode(self):
        attrs = make_attrs({"bridge": {}})
        fake_client = FakeDockerClient(attrs, networks=[("bridge", False)])

        with patch("src.autocompose.docker.from_env", return_value=fake_client):
            cfile, c_networks, _ = autocompose.generate("svc")

        self.assertEqual(cfile["svc"]["network_mode"], "bridge")
        self.assertNotIn("networks", cfile["svc"])
        self.assertIsNone(c_networks)


if __name__ == "__main__":
    unittest.main()