1 files changed, 28 insertions, 0 deletions
diff --git a/test/varlink/podman_testcase.py b/test/varlink/podman_testcase.py
new file mode 100644
index 000000000..c8395a5f1
--- /dev/null
+++ b/test/varlink/podman_testcase.py
@@ -0,0 +1,28 @@
+"""Custom TestCase for varlink/podman."""
+import os
+import unittest
+
+import varlink
+
+
+class PodmanTestCase(unittest.TestCase):
+    """Provides varlink setup for podman."""
+
+    def __init__(self, *args, **kwargs):
+        """Initialize class by calling parent."""
+        super(PodmanTestCase, self).__init__(*args, **kwargs)
+        self.address = os.environ.get(
+            'PODMAN_HOST',
+            'unix:/run/podman/io.projectatomic.podman')
+
+    def setUp(self):
+        """Set up the varlink/podman fixture before each test."""
+        super(PodmanTestCase, self).setUp()
+        self.client = varlink.Client(
+            address=self.address)
+        self.podman = self.client.open('io.projectatomic.podman')
+
+    def tearDown(self):
+        """Deconstruct the varlink/podman fixture after each test."""
+        super(PodmanTestCase, self).tearDown()
+        self.podman.close()