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()