Added Blame.AddReagions(iterable) method

appengine/findit/util_scripts/git_checkout/test/local_git_parsers_test.py

 # 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.
 
 from datetime import datetime
 from datetime import timedelta
 import os
 import sys
 import textwrap
 import unittest
 
 _SCRIPT_DIR = os.path.join(os.path.dirname(__file__),
                            os.path.pardir, os.path.pardir)
 sys.path.insert(1, _SCRIPT_DIR)
 import script_util
 script_util.SetUpSystemPaths()
 
 from git_checkout import local_git_parsers
-from libs.gitiles import blame
+from libs.gitiles.blame import Blame
+from libs.gitiles.blame import Region
 from libs.gitiles import change_log
 
 
 class LocalGitParsersTest(unittest.TestCase):
 
   def setUp(self):
     super(LocalGitParsersTest, self).setUp()
     self.blame_parser = local_git_parsers.GitBlameParser()
 
   def testGitBlameParser(self):
@@ skipping 15 matching lines @@
         revision_hash 19 19
                      *  blabla line 2
         revision_hash 20 20
                      *  blabla line 3
 
         revision_hash 29 29 2
                      *  blabla line 4
         """
     )
 
-    expected_regions = [blame.Region(18, 3, 'revision_hash', 'test@google.com',
-                                     'test@google.com',
-                                     datetime(2013, 03, 11, 17, 13, 36)),
-                        blame.Region(29, 2, 'revision_hash', 'test@google.com',
-                                     'test@google.com',
-                                     datetime(2013, 03, 11, 17, 13, 36))]
-    expected_blame = blame.Blame('src/core/SkFont.h', 'rev')
-    # TODO(crbug.com/663445): Add methods in blame for this.
-    for expected_region in expected_regions:
-      expected_blame.AddRegion(expected_region)
+    expected_blame = Blame('src/core/SkFont.h', 'rev')
+    mock_email = 'test@google.com'
+    author_time = datetime(2013, 03, 11, 17, 13, 36)
+    expected_blame.AddRegions([
+        Region(18, 3, 'revision_hash', mock_email, mock_email, author_time),
+        Region(29, 2, 'revision_hash', mock_email, mock_email, author_time)])
 
     blame_result = self.blame_parser(output, 'src/core/SkFont.h', 'rev')
     self.assertTrue(blame_result.revision, expected_blame.revision)
     self.assertTrue(blame_result.path, expected_blame.path)
     for region, expected_region in zip(blame_result, expected_blame):
       self.assertTrue(region.ToDict(), expected_region.ToDict())
 
   def testGitBlameParserEmptyOutput(self):
     blame_result = self.blame_parser('', 'src/core/SkFont.h', 'rev')
     self.assertIsNone(blame_result)
@@ skipping 162 matching lines @@
     output = '**Changelog start**\nblablabla'
     self.assertEqual(local_git_parsers.GitChangeLogsParser()(output,
                                                              'http://repo'), [])
 
   def testGitDiffParser(self):
     self.assertEqual('output', local_git_parsers.GitDiffParser()('output'))
 
 
 if __name__ == '__main__':
   unittest.main()