Add post_processing code to update copyright year automatically.

tests/mangle_test.py

+#!/usr/bin/env python
+# Copyright (c) 2011 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.
+
+"""Unit tests for post_process/mangle.py."""
+
+import os
+import sys
+import unittest
+
+ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
+PROJECT_DIR = os.path.join(ROOT_DIR, '..')
+sys.path.insert(0, PROJECT_DIR)
+
+import find_depot_tools  # pylint: disable=W0611
+from tests import trial_dir
+import projects
+
+
+class MangleTest(trial_dir.TestCase):
+  def setUp(self):
+    super(MangleTest, self).setUp()
+    class FakeCheckout(object):
+      project_path = self.root_dir
+    self.checkout = FakeCheckout()
+    self.filename = 'foo'
+    self.filepath = os.path.join(self.root_dir, self.filename)
+
+    class FakePatches(object):
+      class FakePatch(object):
+        filename = self.filename
+        is_delete = False
+        is_binary = False
+      patches = [FakePatch()]
+    self.patches = FakePatches()
+
+  def full_check(self, content, expected):
+    """End-to-end test. That's all that matters."""
+    open(self.filepath, 'w').write(content)
+    projects.COPYRIGHT_MANGLER.process(self.checkout, self.patches)
+    self.assertEquals(expected, open(self.filepath).read())
+
+  def test_2_times(self):
+    content = (
+        'Copyright (c) 2010 The Chromium Authors. All rights reserved.\n'
+        'Copyright (c) 2010 The Chromium Authors. All rights reserved.\n')
+    expected = (
+        'Copyright (c) 2011 The Chromium Authors. All rights reserved.\n'
+        'Copyright (c) 2010 The Chromium Authors. All rights reserved.\n')
+    self.full_check(content, expected)
+
+  def test_5_lines(self):
+    content = (
+        '0\n'
+        '1\n'
+        '2\n'
+        '3\n'
+        'Copyright (c) 2010 The Chromium Authors. All rights reserved.\n')
+    expected = (
+        '0\n'
+        '1\n'
+        '2\n'
+        '3\n'
+        'Copyright (c) 2011 The Chromium Authors. All rights reserved.\n')
+    self.full_check(content, expected)
+
+  def test_6_lines(self):
+    content = (
+        '0\n'
+        '1\n'
+        '2\n'
+        '3\n'
+        '4\n'
+        'Copyright (c) 2010 The Chromium Authors. All rights reserved.\n')
+    expected = content
+    self.full_check(content, expected)
+
+  def test_default_lines(self):
+    self.assertEquals(5, projects.COPYRIGHT_MANGLER.max_lines_search)
+
+  def test_re(self):
+    # It's easy to get the regexp wrong so leave this there to confirm it does
+    # the right thing.
+    def check(line, expected):
+      self.assertEquals(expected, projects.COPYRIGHT_MANGLER.process_line(line))
+
+    check(
+        'Copyright (c) 2010 The Chromium Authors. All rights reserved.',
+        'Copyright (c) 2011 The Chromium Authors. All rights reserved.')
+    check(
+        'a Copyright (c) 2010 The Chromium Authors. All rights reserved.',
+        'a Copyright (c) 2011 The Chromium Authors. All rights reserved.')
+    check(
+        '// Copyright (c) 2010 The Chromium Authors. All rights reserved.',
+        '// Copyright (c) 2011 The Chromium Authors. All rights reserved.')
+    check(
+        '// Copyright (c) 2010 The Chromium Authors. All rights reserved.\n',
+        '// Copyright (c) 2011 The Chromium Authors. All rights reserved.\n')
+    check(
+        'Copyright (c) 2010 The Chromium Authors. All rights reserved.\n',
+        'Copyright (c) 2011 The Chromium Authors. All rights reserved.\n')
+    ## \r are not supported.
+    #check(
+    #    '// Copyright (c) 2010 The Chromium Authors. All rights reserved.\r\n',
+    #    '// Copyright (c) 2011 The Chromium Authors. All rights reserved.\r\n')
+    check(
+        'Copyright 2010 The Chromium Authors. All rights reserved.',
+        'Copyright 2010 The Chromium Authors. All rights reserved.')
+
+
+if __name__ == '__main__':
+  unittest.main()