Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion core/testcontainers/core/network.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,10 +32,13 @@ def connect(self, container_id: str, network_aliases: Optional[list] = None):
def remove(self) -> None:
self._network.remove()

def __enter__(self) -> "Network":
def create(self) -> "Network":
self._network = self._docker.client.networks.create(self.name, **self._docker_network_kw)
self.id = self._network.id
return self

def __enter__(self) -> "Network":
return self.create()

def __exit__(self, exc_type, exc_val, exc_tb) -> None:
self.remove()
29 changes: 29 additions & 0 deletions core/tests/test_network.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,11 @@
from http import HTTPStatus
from testcontainers.core.container import DockerContainer
from testcontainers.core.docker_client import DockerClient
from testcontainers.core.network import Network

import docker.errors
import pytest

NGINX_ALPINE_SLIM_IMAGE = "nginx:1.25.4-alpine-slim"


Expand All @@ -14,6 +18,31 @@ def test_network_gets_created_and_cleaned_up():
assert not docker.client.networks.list(network.name)


def test_network_create_wo_cm():
network = Network()
network.create()
docker = DockerClient()
networks_list = docker.client.networks.list(network.name)
assert networks_list[0].name == network.name
assert networks_list[0].id == network.id

network.remove()
assert not docker.client.networks.list(network.name)


def test_network_create_errors():
network = Network()
network.create()

# calling create the second time should raise an error
with pytest.raises(docker.errors.APIError) as excinfo:
network.create()

assert excinfo.value.response.status_code == HTTPStatus.CONFLICT
excinfo.match(f"network with name {network.name} already exists")
network.remove()


def test_containers_can_communicate_over_network():
with Network() as network:
with (
Expand Down