[Piglit] [PATCH 02/11] profile_tests.py: Add tests for the TestProfile class

[Dylan Baker]

This adds a good number of tests for various functions of the
TestProfile class.

Signed-off-by: Dylan Baker <baker.dylan.c at gmail.com>
---
 framework/tests/profile_tests.py | 218 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 218 insertions(+)

diff --git a/framework/tests/profile_tests.py b/framework/tests/profile_tests.py
index 7881b24..9119dbd 100644
--- a/framework/tests/profile_tests.py
+++ b/framework/tests/profile_tests.py
@@ -20,7 +20,12 @@
 
 """ Provides test for the framework.profile modules """
 
+import copy
+import platform
 import nose.tools as nt
+from nose.plugins.skip import SkipTest
+import framework.core as core
+import framework.dmesg as dmesg
 import framework.profile as profile
 
 
@@ -45,3 +50,216 @@ def test_load_test_profile_returns():
     """ loadTestProfile returns a TestProfile instance """
     profile_ = profile.loadTestProfile('sanity')
     assert isinstance(profile_, profile.TestProfile)
+
+
+def test_testprofile_default_dmesg():
+    """ Dmesg initializes false """
+    profile_ = profile.TestProfile()
+    assert isinstance(profile_.dmesg, dmesg.DummyDmesg)
+
+
+def test_testprofile_set_dmesg_true():
+    """ Dmesg returns an apropriate dmesg is ste to True """
+    if not platform.platform().startswith('Linux'):
+        raise SkipTest('No dmesg support on this platform')
+    profile_ = profile.TestProfile()
+    profile_.dmesg = True
+    assert isinstance(profile_.dmesg, dmesg.LinuxDmesg)
+
+
+def test_testprofile_set_dmesg_false():
+    """ Dmesg returns a DummyDmesg if set to False """
+    if not platform.platform().startswith('Linux'):
+        raise SkipTest('No dmesg support on this platform')
+    profile_ = profile.TestProfile()
+    profile_.dmesg = True
+    profile_.dmesg = False
+    assert isinstance(profile_.dmesg, dmesg.DummyDmesg)
+
+
+def test_testprofile_flatten():
+    """ TestProfile.flatten_group_hierarchy flattens and empties self.tests """
+    profile_ = profile.TestProfile()
+    profile_.tests['group1'] = {}
+    profile_.tests['group1']['group2'] = {}
+    profile_.tests['group1']['group2']['group3'] = {}
+    profile_.tests['group1']['test1'] = 'thing'
+    profile_.tests['group1']['group2']['test2'] = 'thing'
+    profile_.tests['group1']['group2']['group3']['test3'] = 'thing'
+
+    profile_.flatten_group_hierarchy()
+
+    baseline = {
+        'group1/test1': 'thing',
+        'group1/group2/test2': 'thing',
+        'group1/group2/group3/test3': 'thing'
+    }
+
+    # profile_.tests should have been emptied
+    nt.assert_dict_equal(profile_.tests, {})
+
+    nt.assert_dict_equal(profile_.test_list, baseline)
+
+
+def test_testprofile_update_tests():
+    """ TestProfile.update() updates TestProfile.tests
+
+    TestProfile.tests is deprecated, and this test should be removed eventually
+
+    """
+    profile1 = profile.TestProfile()
+    profile1.tests['group1'] = {}
+    profile1.tests['group1']['test1'] = 'test1'
+
+    profile2 = profile.TestProfile()
+    baseline = {}
+    baseline['group2'] = {}
+    baseline['group2']['test2'] = 'test2'
+    baseline['group1'] = {}
+    baseline['group1']['test1'] = 'test3'
+    profile2.tests = baseline
+
+    profile1.update(profile2)
+
+    nt.assert_dict_equal(profile1.tests, baseline)
+
+
+def test_testprofile_update_test_list():
+    """ TestProfile.update() updates TestProfile.test_list """
+    profile1 = profile.TestProfile()
+    profile1.test_list['group1/test1'] = 'test1'
+
+    baseline = {'group1/test1': 'test3', 'group2/test2': 'test2'}
+
+    profile2 = profile.TestProfile()
+    profile2.test_list = baseline
+
+    profile1.update(profile2)
+
+    nt.assert_dict_equal(profile1.test_list, baseline)
+
+
+def generate_prepare_test_list_flatten():
+    """ Generate tests for TestProfile.prepare_test_list() """
+    tests = {'group1': {'test1': 'thingy', 'group3': {'test2': 'thing'}},
+             'group3': {'test5': 'other'}}
+    test_list = {'group1/test1': 'thingy', 'group1/group3/test2': 'thing',
+                 'group3/test5': 'other'}
+
+    check_flatten.description = \
+        "TestProfile.prepare_test_list flattens TestProfile.tests"
+    yield check_flatten, tests, test_list
+
+    check_mixed_flatten.description = \
+        "TestProfile flattening is correct when tess and test_list are set"
+    yield check_mixed_flatten, tests, test_list
+
+
+ at nt.nottest
+def check_flatten(tests, testlist):
+    """ TestProfile.prepare_test_list flattens TestProfile.tests """
+    profile_ = profile.TestProfile()
+    profile_.tests = tests
+    profile_.flatten_group_hierarchy()
+
+    nt.assert_dict_equal(profile_.test_list, testlist)
+
+
+ at nt.nottest
+def check_mixed_flatten(tests, testlist):
+    """ flattening is correct when tests and test_list are defined """
+    profile_ = profile.TestProfile()
+    profile_.tests = tests
+    profile_.test_list['test8'] = 'other'
+    profile_.flatten_group_hierarchy()
+
+    baseline = {'test8': 'other'}
+    baseline.update(testlist)
+
+    nt.assert_dict_equal(profile_.test_list, baseline)
+
+
+def generate_prepare_test_list_test_test_matches():
+    """ Generate tests for TestProfile.perpare_test_list filtering """
+    data = {'group1/test1': 'thingy', 'group1/group3/test2': 'thing',
+            'group3/test5': 'other'}
+
+    test_matches_filter_mar_1.description = (
+        "TestProfile.prepare_test_list: "
+        "'not env.filter or matches_any_regex() env.filter is False")
+    yield test_matches_filter_mar_1, data
+
+    test_matches_filter_mar_2.description = (
+        "TestProfile.prepare_test_list: "
+        "Tests 'not env.filter or matches_any_regex() mar is False")
+    yield test_matches_filter_mar_2, data
+
+    test_matches_env_exclude.description = (
+        "TestProfile.prepare_test_list: "
+        "Tests 'not path in env.exclude_tests' is True")
+    yield test_matches_env_exclude, data
+
+    test_matches_exclude_mar.description = \
+        "TestProfile.prepare_test_list: Tests 'not matches_any_regex"
+    yield test_matches_exclude_mar, data
+
+
+ at nt.nottest
+def test_matches_filter_mar_1(data):
+    """ Tests 'not env.filter or matches_any_regex() env.filter is False
+
+    Nothing should be filtered.
+
+    """
+    env = core.Environment()
+
+    profile_ = profile.TestProfile()
+    profile_.test_list = data
+    profile_.prepare_test_list(env)
+
+    nt.assert_dict_equal(profile_.test_list, data)
+
+
+ at nt.nottest
+def test_matches_filter_mar_2(data):
+    """ Tests 'not env.filter or matches_any_regex() mar is False"""
+    env = core.Environment(include_filter=['test5'])
+
+    profile_ = profile.TestProfile()
+    profile_.test_list = data
+    profile_.prepare_test_list(env)
+
+    baseline = {'group3/test5': 'other'}
+
+    nt.assert_dict_equal(profile_.test_list, baseline)
+
+
+ at nt.nottest
+def test_matches_env_exclude(data):
+    """ Tests 'not path in env.exclude_tests  """
+    env = core.Environment()
+    env.exclude_tests.add('group3/test5')
+
+    profile_ = profile.TestProfile()
+    profile_.test_list = data
+    profile_.prepare_test_list(env)
+
+    baseline = copy.deepcopy(data)
+    del baseline['group3/test5']
+
+    nt.assert_dict_equal(profile_.test_list, baseline)
+
+
+ at nt.nottest
+def test_matches_exclude_mar(data):
+    """ Tests 'not matches_any_regexp() """
+    env = core.Environment(exclude_filter=['test5'])
+
+    profile_ = profile.TestProfile()
+    profile_.test_list = data
+    profile_.prepare_test_list(env)
+
+    baseline = copy.deepcopy(data)
+    del baseline['group3/test5']
+
+    nt.assert_dict_equal(profile_.test_list, baseline)
-- 
2.0.0.rc0