Improve script to find user actions.

tools/metrics/actions/extract_actions_test.py

 #!/usr/bin/env python
 # Copyright 2014 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.
 
 import unittest
 
 import extract_actions
 
 # Empty value to be inserted to |ACTIONS_MOCK|.
@@ skipping 176 matching lines @@
   def testAddNewActions(self):
     xml_result = self._GetProcessedAction(TWO_OWNERS, DESCRIPTION, NO_VALUE,
                                           new_actions=['action2'])
     self.assertEqual(ADD_ACTION_EXPECTED_XML, xml_result)
 
   def testComment(self):
     xml_result = self._GetProcessedAction(TWO_OWNERS, DESCRIPTION, NO_VALUE,
                                           comment=COMMENT)
     self.assertEqual(COMMENT_EXPECTED_XML, xml_result)
 
+  def testUserMetricsActionSpanningTwoLines(self):
+    code = 'base::UserMetricsAction(\n"Foo.Bar"));'
+    finder = extract_actions.ActionNameFinder('dummy', code)
+    self.assertEqual('Foo.Bar', finder.FindNextAction())
+    self.assertFalse(finder.FindNextAction())
+
+  def testUserMetricsActionAsAParam(self):
+    code = 'base::UserMetricsAction("Test.Foo"), "Test.Bar");'
+    finder = extract_actions.ActionNameFinder('dummy', code)
+    self.assertEqual('Test.Foo', finder.FindNextAction())
+    self.assertFalse(finder.FindNextAction())
+
+  def testNonLiteralUserMetricsAction(self):
+    code = 'base::UserMetricsAction(FOO)'
+    finder = extract_actions.ActionNameFinder('dummy', code)
+    with self.assertRaises(Exception):
+      finder.FindNextAction()
+
+  def testTernaryUserMetricsAction(self):
+    code = 'base::UserMetricsAction(foo ? "Foo.Bar" : "Bar.Foo"));'
+    finder = extract_actions.ActionNameFinder('dummy', code)
+    with self.assertRaises(Exception):
+      finder.FindNextAction()
+
+  def testTernaryUserMetricsActionWithNewLines(self):
+    code = """base::UserMetricsAction(
+      foo_bar ? "Bar.Foo" :
+      "Foo.Car")"""
+    finder = extract_actions.ActionNameFinder('dummy', code)
+    with self.assertRaises(extract_actions.InvalidStatementException):
+      finder.FindNextAction()
+
 
 if __name__ == '__main__':
   unittest.main()