conn: add container/podman/docker connection

next-actions · Jul 11, 2024 · 83a907f · 83a907f
1 parent e31f590
commit 83a907f
Show file tree

Hide file tree

Showing 3 changed files with 463 additions and 0 deletions.
diff --git a/docs/api.rst b/docs/api.rst
@@ -10,4 +10,5 @@ API Reference
    pytest_mh.cli
    pytest_mh.conn
    pytest_mh.conn.ssh
+   pytest_mh.conn.container
    pytest_mh.utils
diff --git a/pytest_mh/_private/multihost.py b/pytest_mh/_private/multihost.py
@@ -11,6 +11,7 @@
 
 from ..cli import CLIBuilder
 from ..conn import Bash, Connection, Powershell, Process, ProcessError, ProcessInputBuffer, ProcessResult, Shell
+from ..conn.container import ContainerClient
 from ..conn.ssh import SSHClient
 from .artifacts import (
     MultihostArtifactsCollector,
@@ -619,6 +620,8 @@ def get_connection(self, shell: Shell) -> Connection:
         match conn_type:
             case "ssh":
                 return SSHClient.from_confdict(self, conn_confdict)
+            case "podman" | "docker":
+                return ContainerClient.from_confdict(self, conn_confdict)
             case _:
                 raise ValueError(f"Unknown connection type: {conn_type}!")