fixes

Author: rhoban13

core/testcontainers/core/container.py

@@ -74,7 +74,7 @@ def __init__(
         volumes: Optional[list[tuple[str, str, str]]] = None,
         network: Optional[Network] = None,
         network_aliases: Optional[list[str]] = None,
-        transferrables: Optional[list[Transferable]] = None,
+        transferables: Optional[list[Transferable]] = None,
         **kwargs: Any,
     ) -> None:
         self.env = env or {}
@@ -102,7 +102,7 @@ def __init__(
             self.with_network_aliases(*network_aliases)
 
         self._kwargs = kwargs
-        self._transferables: list[Transferable] = transferrables or []
+        self._transferables: list[Transferable] = transferables or []
 
     def with_env(self, key: str, value: str) -> Self:
         self.env[key] = value
@@ -285,15 +285,19 @@ def _configure(self) -> None:
         pass
 
     def with_copy_into_container(
-        self, file_content: bytes | PathLike, destination_in_container: str, mode: int = 0o644
-    ):
+        self, file_content: Union[bytes, PathLike], destination_in_container: str, mode: int = 0o644
+    ) -> Self:
         self._transferables.append(Transferable(file_content, destination_in_container, mode))
         return self
 
-    def copy_into_container(self, file_content: bytes | PathLike, destination_in_container: str, mode: int = 0o644):
+    def copy_into_container(
+        self, file_content: Union[bytes, PathLike], destination_in_container: str, mode: int = 0o644
+    ) -> None:
         return self._transfer_into_container(file_content, destination_in_container, mode)
 
-    def _transfer_into_container(self, source: bytes | PathLike, destination_in_container: str, mode: int):
+    def _transfer_into_container(
+        self, source: Union[bytes, PathLike], destination_in_container: str, mode: int
+    ) -> None:
         if isinstance(source, bytes):
             file_content = source
         elif isinstance(source, PathLike):
@@ -309,10 +313,12 @@ def _transfer_into_container(self, source: bytes | PathLike, destination_in_cont
             tarinfo.mode = mode
             tar.addfile(tarinfo, io.BytesIO(file_content))
         fileobj.seek(0)
+        assert self._container is not None
         rv = self._container.put_archive(path="/", data=fileobj.getvalue())
         assert rv is True
 
-    def copy_from_container(self, source_in_container: str, destination_on_host: PathLike):
+    def copy_from_container(self, source_in_container: str, destination_on_host: PathLike) -> None:
+        assert self._container is not None
         tar_stream, _ = self._container.get_archive(source_in_container)
 
         for chunk in tar_stream:

core/testcontainers/core/transferable.py

@@ -1,10 +1,14 @@
 import dataclasses
-import os
+import pathlib
 from typing import Union
 
 
 @dataclasses.dataclass
 class Transferable:
-    source: Union[bytes, os.PathLike]
+    """
+    Wrapper class enabling copying files into a container
+    """
+
+    source: Union[bytes, pathlib.Path]
     destination_in_container: str
     mode: int = 0o644

core/tests/test_core.py

@@ -88,10 +88,9 @@ def test_copy_file_into_container_via_initializer(tmp_path: Path):
     my_file = tmp_path / "my_file"
     my_file.write_text("hello world")
     destination_in_container = "/tmp/my_file"
+    transferables = [Transferable(my_file, destination_in_container)]
 
-    with DockerContainer(
-        "bash", command="sleep infinity", transferrables=(Transferable(my_file, destination_in_container),)
-    ) as container:
+    with DockerContainer("bash", command="sleep infinity", transferables=transferables) as container:
         # When
         result = container.exec(f"cat {destination_in_container}")
 
@@ -137,10 +136,9 @@ def test_copy_bytes_to_container_via_initializer():
     # Given
     file_content = b"hello world"
     destination_in_container = "/tmp/my_file"
+    transferables = [Transferable(file_content, destination_in_container)]
 
-    with DockerContainer(
-        "bash", command="sleep infinity", transferrables=(Transferable(file_content, destination_in_container),)
-    ) as container:
+    with DockerContainer("bash", command="sleep infinity", transferables=transferables) as container:
         # When
         result = container.exec(f"cat {destination_in_container}")