Fix sqlite test on CentOS

greatcare · Jan 6, 2017 · 192c92f · 192c92f
1 parent 6bdb83e
commit 192c92f
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/tests/test_sqlite.py b/tests/test_sqlite.py
@@ -3,9 +3,17 @@
 testinfra_hosts = testinfra.utils.ansible_runner.AnsibleRunner(
     '.molecule/ansible_inventory').get_hosts('sqlite')
 
+debian_os = ['debian', 'ubuntu']
+rhel_os = ['redhat', 'centos']
+
+
+def test_package(Package, SystemInfo):
+    p = None
+    if SystemInfo.distribution in debian_os:
+        p = Package('pdns-backend-sqlite3')
+    if SystemInfo.distribution in rhel_os:
+        p = Package('pdns-backend-sqlite')
 
-def test_package(Package):
-    p = Package('pdns-backend-sqlite3')
     assert p.is_installed