11d0c9a30f
We can rebuild it from the data we already have, but this will allow us to get the function headers in it once the library supports it.
186 lines
5.8 KiB
Python
186 lines
5.8 KiB
Python
# -*- coding: UTF-8 -*-
|
|
#
|
|
# Copyright 2010-2012 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 Diff objects."""
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import unicode_literals
|
|
import unittest
|
|
|
|
from . import utils
|
|
|
|
|
|
COMMIT_SHA1_1 = '5fe808e8953c12735680c257f56600cb0de44b10'
|
|
COMMIT_SHA1_2 = 'c2792cfa289ae6321ecf2cd5806c2194b0fd070c'
|
|
COMMIT_SHA1_3 = '2cdae28389c059815e951d0bb9eed6533f61a46b'
|
|
|
|
PATCH = b"""diff --git a/a b/a
|
|
index 7f129fd..af431f2 100644
|
|
--- a/a
|
|
+++ b/a
|
|
@@ -1 +1 @@
|
|
-a contents 2
|
|
+a contents
|
|
diff --git a/c/d b/c/d
|
|
deleted file mode 100644
|
|
index 297efb8..0000000
|
|
--- a/c/d
|
|
+++ /dev/null
|
|
@@ -1 +0,0 @@
|
|
-c/d contents
|
|
"""
|
|
|
|
DIFF_INDEX_EXPECTED = [
|
|
'staged_changes',
|
|
'staged_changes_file_deleted',
|
|
'staged_changes_file_modified',
|
|
'staged_delete',
|
|
'staged_delete_file_modified',
|
|
'staged_new',
|
|
'staged_new_file_deleted',
|
|
'staged_new_file_modified'
|
|
]
|
|
|
|
DIFF_WORKDIR_EXPECTED = [
|
|
'file_deleted',
|
|
'modified_file',
|
|
'staged_changes',
|
|
'staged_changes_file_deleted',
|
|
'staged_changes_file_modified',
|
|
'staged_delete',
|
|
'staged_delete_file_modified',
|
|
'subdir/deleted_file',
|
|
'subdir/modified_file'
|
|
]
|
|
|
|
class DiffDirtyTest(utils.DirtyRepoTestCase):
|
|
def test_diff_empty_index(self):
|
|
repo = self.repo
|
|
head = repo[repo.lookup_reference('HEAD').resolve().oid]
|
|
diff = head.tree.diff(repo.index)
|
|
|
|
files = [x[1] for x in diff.changes['files']]
|
|
self.assertEqual(DIFF_INDEX_EXPECTED, files)
|
|
|
|
def test_workdir_to_tree(self):
|
|
repo = self.repo
|
|
head = repo[repo.lookup_reference('HEAD').resolve().oid]
|
|
diff = head.tree.diff()
|
|
|
|
files = [x[1] for x in diff.changes['files']]
|
|
self.assertEqual(DIFF_WORKDIR_EXPECTED, files)
|
|
|
|
class DiffTest(utils.BareRepoTestCase):
|
|
|
|
def test_diff_invalid(self):
|
|
commit_a = self.repo[COMMIT_SHA1_1]
|
|
commit_b = self.repo[COMMIT_SHA1_2]
|
|
self.assertRaises(TypeError, commit_a.tree.diff, commit_b)
|
|
|
|
def test_diff_empty_index(self):
|
|
repo = self.repo
|
|
head = repo[repo.lookup_reference('HEAD').resolve().oid]
|
|
diff = head.tree.diff(repo.index)
|
|
|
|
files = [x[0].split('/')[0] for x in diff.changes['files']]
|
|
self.assertEqual([x.name for x in head.tree], files)
|
|
|
|
def test_diff_tree(self):
|
|
commit_a = self.repo[COMMIT_SHA1_1]
|
|
commit_b = self.repo[COMMIT_SHA1_2]
|
|
|
|
diff = commit_a.tree.diff(commit_b.tree)
|
|
|
|
# self.assertIsNotNone is 2.7 only
|
|
self.assertTrue(diff is not None)
|
|
# self.assertIn is 2.7 only
|
|
self.assertTrue(('a','a', 3) in diff.changes['files'])
|
|
self.assertEqual(2, len(diff.changes['hunks']))
|
|
|
|
hunk = diff.changes['hunks'][0]
|
|
self.assertEqual(hunk.old_start, 1)
|
|
self.assertEqual(hunk.old_lines, 0)
|
|
self.assertEqual(hunk.new_start, 1)
|
|
self.assertEqual(hunk.new_lines, 0)
|
|
|
|
self.assertEqual(hunk.old_file, 'a')
|
|
self.assertEqual(hunk.new_file, 'a')
|
|
|
|
#self.assertEqual(hunk.data[0][0], b'a contents 2\n')
|
|
#self.assertEqual(hunk.data[1][0], b'a contents\n')
|
|
|
|
def test_diff_merge(self):
|
|
commit_a = self.repo[COMMIT_SHA1_1]
|
|
commit_b = self.repo[COMMIT_SHA1_2]
|
|
commit_c = self.repo[COMMIT_SHA1_3]
|
|
|
|
diff_b = commit_a.tree.diff(commit_b.tree)
|
|
# self.assertIsNotNone is 2.7 only
|
|
self.assertTrue(diff_b is not None)
|
|
|
|
diff_c = commit_b.tree.diff(commit_c.tree)
|
|
# self.assertIsNotNone is 2.7 only
|
|
self.assertTrue(diff_c is not None)
|
|
|
|
# assertIn / assertNotIn are 2.7 only
|
|
self.assertTrue(('b','b', 3) not in diff_b.changes['files'])
|
|
self.assertTrue(('b','b', 3) in diff_c.changes['files'])
|
|
|
|
diff_b.merge(diff_c)
|
|
|
|
# assertIn is 2.7 only
|
|
self.assertTrue(('b','b', 3) in diff_b.changes['files'])
|
|
|
|
hunk = diff_b.changes['hunks'][1]
|
|
self.assertEqual(hunk.old_start, 1)
|
|
self.assertEqual(hunk.old_lines, 0)
|
|
self.assertEqual(hunk.new_start, 1)
|
|
self.assertEqual(hunk.new_lines, 0)
|
|
|
|
self.assertEqual(hunk.old_file, 'b')
|
|
self.assertEqual(hunk.new_file, 'b')
|
|
|
|
#self.assertEqual(hunk.data[0][0], b'b contents\n')
|
|
#self.assertEqual(hunk.data[1][0], b'b contents 2\n')
|
|
|
|
def test_diff_patch(self):
|
|
commit_a = self.repo[COMMIT_SHA1_1]
|
|
commit_b = self.repo[COMMIT_SHA1_2]
|
|
|
|
diff = commit_a.tree.diff(commit_b.tree)
|
|
self.assertEqual(diff.patch, PATCH)
|
|
|
|
def test_diff_header(self):
|
|
commit_a = self.repo[COMMIT_SHA1_1]
|
|
commit_b = self.repo[COMMIT_SHA1_2]
|
|
diff = commit_a.tree.diff(commit_b.tree)
|
|
|
|
self.assertEqual(diff.changes['hunks'][0].header, "@@ -1 +1 @@\n")
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|