Add tests for the mailmap APIs

Author: mystor

test/test_mailmap.py

@@ -0,0 +1,103 @@
+# -*- coding: UTF-8 -*-
+#
+# Copyright 2010-2017 The pygit2 contributors
+#
+# This file is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License, version 2,
+# as published by the Free Software Foundation.
+#
+# In addition to the permissions in the GNU General Public License,
+# the authors give you unlimited permission to link the compiled
+# version of this file into combinations with other programs,
+# and to distribute those combinations without any restriction
+# coming from the use of this file.  (The General Public License
+# restrictions do apply in other respects; for example, they cover
+# modification of the file, and distribution when not linked into
+# a combined executable.)
+#
+# This file is distributed in the hope that it will be useful, but
+# WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+# General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; see the file COPYING.  If not, write to
+# the Free Software Foundation, 51 Franklin Street, Fifth Floor,
+# Boston, MA 02110-1301, USA.
+
+"""Tests for Mailmap."""
+
+from __future__ import absolute_import
+from __future__ import unicode_literals
+import os
+import unittest
+import tempfile
+
+import pygit2
+from pygit2 import Mailmap
+from . import utils
+
+
+TEST_MAILMAP = """\
+# Simple Comment line
+<[email protected]>                       <[email protected]>
+Some Dude <[email protected]>         nick1 <[email protected]>
+Other Author <[email protected]>   nick2 <[email protected]>
+Other Author <[email protected]>         <[email protected]>
+Phil Hill <[email protected]>  # Comment at end of line
+<[email protected]>             Joseph <[email protected]>
+Santa Claus <[email protected]> <[email protected]>
+"""
+
+TEST_ENTRIES = [
+    (None, "[email protected]", None, "[email protected]"),
+    ("Some Dude", "[email protected]", "nick1", "[email protected]"),
+    ("Other Author", "[email protected]", "nick2", "[email protected]"),
+    ("Other Author", "[email protected]", None, "[email protected]"),
+    ("Phil Hill", None, None, "[email protected]"),
+    (None, "[email protected]", "Joseph", "[email protected]"),
+    ("Santa Claus", "[email protected]", None, "[email protected]")
+]
+
+TEST_RESOLVE = [
+    ("Brad", "[email protected]", "Brad", "[email protected]"),
+    ("Brad L", "[email protected]", "Brad L", "[email protected]"),
+    ("Some Dude", "[email protected]", "nick1", "[email protected]"),
+    ("Other Author", "[email protected]", "nick2", "[email protected]"),
+    ("nick3", "[email protected]", "nick3", "[email protected]"),
+    ("Other Author", "[email protected]", "Some Garbage", "[email protected]"),
+    ("Phil Hill", "[email protected]", "unknown", "[email protected]"),
+    ("Joseph", "[email protected]", "Joseph", "[email protected]"),
+    ("Santa Claus", "[email protected]", "Clause", "[email protected]"),
+    ("Charles", "[email protected]", "Charles", "[email protected]")
+]
+
+class ParsingTest(utils.NoRepoTestCase):
+
+    def test_empty(self):
+        mailmap = Mailmap()
+
+        for (_, _, name, email) in TEST_RESOLVE:
+            self.assertEqual(mailmap.resolve(name, email), (name, email))
+
+    def test_new(self):
+        mailmap = Mailmap()
+
+        # Add entries to the mailmap
+        for entry in TEST_ENTRIES:
+            mailmap.add_entry(*entry)
+
+        for (real_name, real_email, name, email) in TEST_RESOLVE:
+            self.assertEqual(mailmap.resolve(name, email), (real_name, real_email))
+
+    def test_parsed(self):
+        mailmap = Mailmap.from_buffer(TEST_MAILMAP)
+
+        for (real_name, real_email, name, email) in TEST_RESOLVE:
+            self.assertEqual(mailmap.resolve(name, email), (real_name, real_email))
+
+
+# TODO: Add a testcase which uses .mailmap in a repo
+
+if __name__ == '__main__':
+    unittest.main()