From: Zack Cerza <zack@redhat.com>
Date: Wed, 20 May 2015 21:11:14 +0000 (-0600)
Subject: Add tests for get_flavor()
X-Git-Tag: 1.1.0~935^2~1
X-Git-Url: http://git-server-git.apps.pok.os.sepia.ceph.com/?a=commitdiff_plain;h=3ed2be0b9a77caedd89612edf71a205fcb3e9db1;p=teuthology.git

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 9233554f5..93b1c1e6e 100644
--- 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'