[Predator] Add flag manager and flag manager test.

appengine/findit/crash/test/flag_manager_test.py

+# Copyright 2016 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+from crash.flag_manager import Flag
+from crash.flag_manager import FlagManager
+from crash.test.stacktrace_test_suite import StacktraceTestSuite
+
+
+class FlagTest(StacktraceTestSuite):
+
+  def setUp(self):
+    super(FlagTest, self).setUp()
+    self.flag = Flag('test_flag', lambda line: 'turn on flag' in line,
+                     val=False)
+
+  def testFlagNameProperty(self):
+    self.assertEqual(self.flag.name, 'test_flag')
+
+  def testFlagValProperty(self):
+    self.assertFalse(self.flag.val)
+    self.flag.val = True
+    self.assertTrue(self.flag.val)
+    self.assertTrue(bool(self.flag))
+
+  def testConditionallyTurOn(self):
+    self.assertFalse(bool(self.flag))
+    self.flag.ConditionallyTurnOn('line: turn on flag')
+    self.assertTrue(bool(self.flag))
+
+
+class FlagManagerTest(StacktraceTestSuite):
+
+  def setUp(self):
+    super(FlagManagerTest, self).setUp()
+    self.flag_manager = FlagManager()
+
+  def testClearFlags(self):
+    self.flag_manager.Register('test', Flag('test_flag', val=True))
+    self.assertEqual(len(self.flag_manager.flags), 1)
+    self.flag_manager.ClearFlags()
+    self.assertEqual(len(self.flag_manager.flags), 0)
+
+  def testRegister(self):
+    self.flag_manager.Register('dummy', Flag('dummy_flag', val=True))
+    self.assertEqual(len(self.flag_manager.flags), 1)
+
+  def testGetFlagGroup(self):
+    flag = Flag('group_flag', val=True)
+    self.flag_manager.Register('group1', flag)
+    self.flag_manager.Register('group2', flag)
+    self.assertListEqual(self.flag_manager.GetFlagGroup('group1'), [flag])
+    self.assertListEqual(self.flag_manager.GetFlagGroup(), [flag, flag])
+
+  def testReset(self):
+    self.flag_manager.Register('group', Flag('dummy_flag1', val=True))
+    self.flag_manager.Register('group', Flag('dummy_flag2', val=True))
+    self.flag_manager.Reset()
+    for flag in self.flag_manager.GetFlagGroup():
+      self.assertFalse(flag.val)
+
+  def testConditionallyTurOn(self):
+    self.flag_manager.Register(
+        'group',
+        Flag('flag1', condition=lambda line: 'flag1 marker' in line, val=False))
+    self.flag_manager.Register(
+        'group',
+        Flag('flag2', condition=lambda line: 'flag2 marker' in line, val=False))
+    line = 'line: flag1 marker flag2 marker'
+    self.flag_manager.ConditionallyTurnOn(line)
+    self.assertTrue(bool(self.flag_manager.Get('flag1')))
+    self.assertTrue(bool(self.flag_manager.Get('flag2')))
+
+  def testConditionallyTurOnWhenConditionsNotMet(self):
+    self.flag_manager.Register(
+        'group',
+        Flag('flag', condition=lambda line: 'flag marker' in line, val=False))
+    line = 'dummy line'
+    self.flag_manager.ConditionallyTurnOn(line)
+    self.assertFalse(bool(self.flag_manager.Get('flag')))
+
+  def testSetFlagVal(self):
+    self.flag_manager.Register('group', Flag('flag', val=False))
+    self.flag_manager.Set('flag', True)
+    self.flag_manager.Set('non_flag', True)
+    self.assertTrue(self.flag_manager.Get('flag').val)