Add tests for get_flavor()

Signed-off-by: Zack Cerza <zack@redhat.com>

diff --git a/teuthology/test/task/test_install.py b/teuthology/test/task/test_install.py
index 9233554f58ea79c6ef0d584e32106156de51e440..93b1c1e6e033688a1d6f808af4be7e464618408f 100644 (file)
--- a/teuthology/test/task/test_install.py
+++ b/teuthology/test/task/test_install.py
@@ -40,3 +40,19 @@ class TestInstall(object):
         config = Mock()
         config.extras = True
         install.verify_package_version(Mock(), config, Mock())
+
+    def test_get_flavor_default(self):
+        config = dict()
+        assert install.get_flavor(config) == 'basic'
+
+    def test_get_flavor_simple(self):
+        config = dict(
+            flavor='notcmalloc'
+        )
+        assert install.get_flavor(config) == 'notcmalloc'
+
+    def test_get_flavor_valgrind(self):
+        config = dict(
+            valgrind=True
+        )
+        assert install.get_flavor(config) == 'notcmalloc'