diff options
Diffstat (limited to 'crosperf/flag_test_unittest.py')
-rwxr-xr-x | crosperf/flag_test_unittest.py | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/crosperf/flag_test_unittest.py b/crosperf/flag_test_unittest.py new file mode 100755 index 00000000..9f2a7136 --- /dev/null +++ b/crosperf/flag_test_unittest.py @@ -0,0 +1,41 @@ +#!/usr/bin/env python2 +# +# Copyright 2014 Google Inc. All Rights Reserved. + +"""The unittest of flags.""" + +from __future__ import print_function +import test_flag + +import unittest + + +class FlagTestCase(unittest.TestCase): + """The unittest class.""" + def test_test_flag(self): + # Verify that test_flag.is_test exists, that it is a list, + # and that it contains 1 element. + self.assertTrue(type(test_flag.is_test) is list) + self.assertEqual(len(test_flag.is_test), 1) + + # Verify that the getting the flag works and that the flag + # contains False, its starting value. + save_flag = test_flag.GetTestMode() + self.assertFalse(save_flag) + + # Verify that setting the flat to True, then getting it, works. + test_flag.SetTestMode(True) + self.assertTrue(test_flag.GetTestMode()) + + # Verify that setting the flag to False, then getting it, works. + test_flag.SetTestMode(save_flag) + self.assertFalse(test_flag.GetTestMode()) + + # Verify that test_flag.is_test still exists, that it still is a + # list, and that it still contains 1 element. + self.assertTrue(type(test_flag.is_test) is list) + self.assertEqual(len(test_flag.is_test), 1) + + +if __name__ == '__main__': + unittest.main() |