conn: add container/podman/docker connection

next-actions · Jul 11, 2024 · 765432b · 765432b
1 parent 9582fea
commit 765432b
Show file tree

Hide file tree

Showing 2 changed files with 453 additions and 0 deletions.
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}!")