From 3ed2be0b9a77caedd89612edf71a205fcb3e9db1 Mon Sep 17 00:00:00 2001 From: Zack Cerza Date: Wed, 20 May 2015 15:11:14 -0600 Subject: [PATCH] Add tests for get_flavor() Signed-off-by: Zack Cerza --- teuthology/test/task/test_install.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/teuthology/test/task/test_install.py b/teuthology/test/task/test_install.py index 9233554f58..93b1c1e6e0 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' -- 2.39.5