unit tests for install function

Signed-off-by: Vasu Kulkarni <vasu@redhat.com>

diff --git a/teuthology/test/task/test_install.py b/teuthology/test/task/test_install.py
index 34345eaf8be0a533111c15b414b8a074ef14d20d..a772b5f58202b138de96eccb8eacc6c2dc396be0 100644 (file)
--- a/teuthology/test/task/test_install.py
+++ b/teuthology/test/task/test_install.py
@@ -264,3 +264,36 @@ class TestInstall(object):
             },
         }
         assert install.upgrade_remote_to_config(ctx, config) == expected_config
+
+
+    @patch("teuthology.task.install.packaging.get_package_version")
+    @patch("teuthology.task.install.redhat.set_deb_repo")
+    def test_rh_install_deb_pkgs(self, m_set_rh_deb_repo, m_get_pkg_version):
+        ctx = Mock()
+        remote = Mock()
+        version = '1.3.2'
+        rh_ds_yaml = dict()
+        rh_ds_yaml = {'versions': {'deb': {'mapped': {'1.3.2': '0.94.5'}}},
+                      'pkgs': { 'deb': ['pkg1', 'pkg2'] }}
+        m_get_pkg_version.return_value = "0.94.5"
+        deb_repo=Mock()
+        deb_gpg_key=Mock()
+        install.redhat.install_deb_pkgs(ctx, remote, version, rh_ds_yaml,
+                                    deb_repo, deb_gpg_key)
+
+
+    @patch("teuthology.task.install.packaging.get_package_version")
+    def test_rh_install_pkgs(self, m_get_pkg_version):
+        ctx = Mock()
+        remote = Mock()
+        version = '1.3.2'
+        rh_ds_yaml = dict()
+        rh_ds_yaml = {'versions': {'rpm': {'mapped': {'1.3.2': '0.94.5',
+                                                      '1.3.1': '0.94.3'}}},
+                      'pkgs': { 'rpm': ['pkg1', 'pkg2'] }}
+        m_get_pkg_version.return_value = "0.94.5"
+        install.redhat.install_pkgs(ctx, remote, version, rh_ds_yaml)
+        version = '1.3.1'
+        with pytest.raises(RuntimeError) as e:
+            install.redhat.install_pkgs(ctx, remote, version, rh_ds_yaml)
+        assert "Version check failed" in str(e)