def _init_diff(self):
"""Return a minimal diff contains all required samples
header
--- old
+++ new
hunk header
@@ -1,4 +1,4 @@
-hhello
+helloo
+spammm
world
-garb
-Again
+again
"""
hunk = cdiff.Hunk(['hunk header\n'], '@@ -1,4 +1,4 @@\n', (1, 4),
(1, 4))
hunk.append(('-', 'hhello\n'))
hunk.append(('+', 'helloo\n'))
hunk.append(('+', 'spammm\n'))
hunk.append((' ', 'world\n'))
hunk.append(('-', 'garb\n'))
hunk.append(('-', 'Again\n'))
hunk.append(('+', 'again\n'))
diff = cdiff.UnifiedDiff(['header\n'], '--- old\n', '+++ new\n',
[hunk])
return diff
def test_markup_traditional_hunk_header(self):
hunk = cdiff.Hunk(['hunk header\n'], '@@ -0 +0 @@\n', (0, 0), (0, 0))
diff = cdiff.UnifiedDiff([], '--- old\n', '+++ new\n', [hunk])
marker = cdiff.DiffMarker()
out = list(marker._markup_traditional(diff))
self.assertEqual(len(out), 4)
self.assertEqual(out[0], '\x1b[33m--- old\n\x1b[0m')
self.assertEqual(out[1], '\x1b[33m+++ new\n\x1b[0m')
self.assertEqual(out[2], '\x1b[1;36mhunk header\n\x1b[0m')
self.assertEqual(out[3], '\x1b[1;34m@@ -0 +0 @@\n\x1b[0m')
def test_markup_traditional_new_changed(self):
hunk = cdiff.Hunk([], '@@ -0,0 +1 @@\n', (0, 0), (1, 0))
hunk.append(('+', 'spam\n'))
diff = cdiff.UnifiedDiff([], '--- old\n', '+++ new\n', [hunk])
marker = cdiff.DiffMarker()
out = list(marker._markup_traditional(diff))
self.assertEqual(len(out), 4)
self.assertEqual(out[0], '\x1b[33m--- old\n\x1b[0m')
self.assertEqual(out[1], '\x1b[33m+++ new\n\x1b[0m')
self.assertEqual(out[2], '\x1b[1;34m@@ -0,0 +1 @@\n\x1b[0m')
self.assertEqual(out[3], '\x1b[32m+spam\n\x1b[0m')
def test_markup_traditional_both_changed(self):
hunk = cdiff.Hunk([], '@@ -1,2 +1,2 @@\n', (1, 2), (1, 2))
hunk.append(('-', 'hella\n'))
hunk.append(('+', 'hello\n'))
hunk.append((' ', 'common\n'))
diff = cdiff.UnifiedDiff([], '--- old\n', '+++ new\n', [hunk])
marker = cdiff.DiffMarker()
out = list(marker._markup_traditional(diff))
self.assertEqual(len(out), 6)
self.assertEqual(out[0], '\x1b[33m--- old\n\x1b[0m')
self.assertEqual(out[1], '\x1b[33m+++ new\n\x1b[0m')
self.assertEqual(out[2], '\x1b[1;34m@@ -1,2 +1,2 @@\n\x1b[0m')
self.assertEqual(
out[3], '\x1b[1;31m-\x1b[0m\x1b[31mhell'
'\x1b[4m\x1b[31ma\x1b[0m\x1b[31m\n\x1b[0m')
self.assertEqual(
out[4], '\x1b[32m+\x1b[0m\x1b[32mhell'
'\x1b[4m\x1b[32mo\x1b[0m\x1b[32m\n\x1b[0m')
self.assertEqual(out[5], '\x1b[0m common\n\x1b[0m')
def test_get_new_text(self):
hunk = cdiff.Hunk([], '@@ -1,2 +1,2 @@', (1, 2), (1, 2))
hunk.append(('-', 'foo\n'))
hunk.append(('+', 'bar\n'))
hunk.append((' ', 'common\n'))
self.assertEqual(hunk._get_new_text(), ['bar\n', 'common\n'])
