From 32db403b5e4fdb96e727e8cda3922d235ad63aa4 Mon Sep 17 00:00:00 2001
From: Jhon Honce <jhonce@redhat.com>
Date: Tue, 24 Apr 2018 18:42:59 -0700
Subject: Refactor unittest for varlink component

- Allow unittest's to run as normal user
- Refactor tests to use unittest features
- Refactor tests to use fixtures to track resources
- Update test runner script to clean up on failure

Signed-off-by: Jhon Honce <jhonce@redhat.com>

Closes: #683
Approved by: rhatdan
---
 test/varlink/podman_testcase.py | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 test/varlink/podman_testcase.py

(limited to 'test/varlink/podman_testcase.py')

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()
-- 
cgit v1.2.3-54-g00ecf