adding more tests and fixing an error with tar archives

3 files changed, 20 insertions, 1 deletions

diff --git a/tests/archive_tests.py b/tests/archive_tests.py
index 01c46395..01c46395 100755..100644
--- a/tests/archive_tests.py
+++ b/tests/archive_tests.py
diff --git a/tests/importer_tests.py b/tests/importer_tests.py
new file mode 100644
index 00000000..91f82cc0
--- /dev/null
+++ b/tests/importer_tests.py
@@ -0,0 +1,13 @@
+import unittest
+
+from sos.utilities import ImporterHelper
+
+class ImporterHelperTests(unittest.TestCase):
+
+    def test_runs(self):
+        h = ImporterHelper(unittest)
+        modules = h.get_modules()
+        self.assertTrue('main' in modules)
+
+if __name__ == "__main__":
+    unittest.main()
diff --git a/tests/policy_tests.py b/tests/policy_tests.py
index 1c192bdf..2654c82c 100644
--- a/tests/policy_tests.py
+++ b/tests/policy_tests.py
@@ -1,6 +1,6 @@
 import unittest
 
-from sos.policies import Policy
+from sos.policies import Policy, import_policy
 from sos.plugins import Plugin, IndependentPlugin, RedHatPlugin, DebianPlugin
 
 class FauxPolicy(Policy):
@@ -41,5 +41,11 @@ class PolicyTests(unittest.TestCase):
 
         self.assertFalse(p.validatePlugin(FauxDebianPlugin))
 
+    def test_can_import(self):
+        self.assertTrue(import_policy('redhat') is not None)
+
+    def test_cant_import(self):
+        self.assertTrue(import_policy('notreal') is None)
+
 if __name__ == "__main__":
     unittest.main()