|
|
from __future__ import absolute_import
|
|
|
from __future__ import print_function
|
|
|
|
|
|
import unittest
|
|
|
|
|
|
from mercurial import (
|
|
|
mdiff,
|
|
|
)
|
|
|
from mercurial.context import (
|
|
|
annotateline,
|
|
|
_annotatepair,
|
|
|
)
|
|
|
|
|
|
class AnnotateTests(unittest.TestCase):
|
|
|
"""Unit tests for annotate code."""
|
|
|
|
|
|
def testannotatepair(self):
|
|
|
self.maxDiff = None # camelcase-required
|
|
|
|
|
|
oldfctx = b'old'
|
|
|
p1fctx, p2fctx, childfctx = b'p1', b'p2', b'c'
|
|
|
olddata = b'a\nb\n'
|
|
|
p1data = b'a\nb\nc\n'
|
|
|
p2data = b'a\nc\nd\n'
|
|
|
childdata = b'a\nb2\nc\nc2\nd\n'
|
|
|
diffopts = mdiff.diffopts()
|
|
|
|
|
|
def decorate(text, rev):
|
|
|
return ([annotateline(fctx=rev, lineno=i)
|
|
|
for i in xrange(1, text.count(b'\n') + 1)],
|
|
|
text)
|
|
|
|
|
|
# Basic usage
|
|
|
|
|
|
oldann = decorate(olddata, oldfctx)
|
|
|
p1ann = decorate(p1data, p1fctx)
|
|
|
p1ann = _annotatepair([oldann], p1fctx, p1ann, False, diffopts)
|
|
|
self.assertEqual(p1ann[0], [
|
|
|
annotateline('old', 1),
|
|
|
annotateline('old', 2),
|
|
|
annotateline('p1', 3),
|
|
|
])
|
|
|
|
|
|
p2ann = decorate(p2data, p2fctx)
|
|
|
p2ann = _annotatepair([oldann], p2fctx, p2ann, False, diffopts)
|
|
|
self.assertEqual(p2ann[0], [
|
|
|
annotateline('old', 1),
|
|
|
annotateline('p2', 2),
|
|
|
annotateline('p2', 3),
|
|
|
])
|
|
|
|
|
|
# Test with multiple parents (note the difference caused by ordering)
|
|
|
|
|
|
childann = decorate(childdata, childfctx)
|
|
|
childann = _annotatepair([p1ann, p2ann], childfctx, childann, False,
|
|
|
diffopts)
|
|
|
self.assertEqual(childann[0], [
|
|
|
annotateline('old', 1),
|
|
|
annotateline('c', 2),
|
|
|
annotateline('p2', 2),
|
|
|
annotateline('c', 4),
|
|
|
annotateline('p2', 3),
|
|
|
])
|
|
|
|
|
|
childann = decorate(childdata, childfctx)
|
|
|
childann = _annotatepair([p2ann, p1ann], childfctx, childann, False,
|
|
|
diffopts)
|
|
|
self.assertEqual(childann[0], [
|
|
|
annotateline('old', 1),
|
|
|
annotateline('c', 2),
|
|
|
annotateline('p1', 3),
|
|
|
annotateline('c', 4),
|
|
|
annotateline('p2', 3),
|
|
|
])
|
|
|
|
|
|
# Test with skipchild (note the difference caused by ordering)
|
|
|
|
|
|
childann = decorate(childdata, childfctx)
|
|
|
childann = _annotatepair([p1ann, p2ann], childfctx, childann, True,
|
|
|
diffopts)
|
|
|
self.assertEqual(childann[0], [
|
|
|
annotateline('old', 1),
|
|
|
annotateline('old', 2),
|
|
|
annotateline('p2', 2),
|
|
|
annotateline('p2', 2),
|
|
|
annotateline('p2', 3),
|
|
|
])
|
|
|
|
|
|
childann = decorate(childdata, childfctx)
|
|
|
childann = _annotatepair([p2ann, p1ann], childfctx, childann, True,
|
|
|
diffopts)
|
|
|
self.assertEqual(childann[0], [
|
|
|
annotateline('old', 1),
|
|
|
annotateline('old', 2),
|
|
|
annotateline('p1', 3),
|
|
|
annotateline('p1', 3),
|
|
|
annotateline('p2', 3),
|
|
|
])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
import silenttestrunner
|
|
|
silenttestrunner.main(__name__)
|
|
|
|