mirror of
https://github.com/facebook/sapling.git
synced 2024-10-06 06:47:41 +03:00
a50d0da7fe
Summary: Blame can use a templater which doesn't support bytes. Let's just force all blame output to be unicode, since it doesn't make a ton of sense to blame binary files anyway. Also fix test-annotate.py Reviewed By: quark-zju Differential Revision: D19907530 fbshipit-source-id: a7a47246368ed50f65486e824f93552872adc09a
119 lines
3.5 KiB
Python
119 lines
3.5 KiB
Python
from __future__ import absolute_import, print_function
|
|
|
|
import unittest
|
|
|
|
from edenscm.mercurial import mdiff
|
|
from edenscm.mercurial.context import _annotatepair, annotateline
|
|
from hghave import require
|
|
|
|
|
|
try:
|
|
xrange(0)
|
|
except NameError:
|
|
xrange = range
|
|
|
|
|
|
class AnnotateTests(unittest.TestCase):
|
|
"""Unit tests for annotate code."""
|
|
|
|
def testannotatepair(self):
|
|
self.maxDiff = None # camelcase-required
|
|
|
|
oldfctx = "old"
|
|
p1fctx, p2fctx, childfctx = "p1", "p2", "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, True),
|
|
# note that this line was carried over from earlier so it is *not*
|
|
# marked skipped
|
|
annotateline("p2", 2),
|
|
annotateline("p2", 2, True),
|
|
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, True),
|
|
annotateline("p1", 3),
|
|
annotateline("p1", 3, True),
|
|
annotateline("p2", 3),
|
|
],
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import silenttestrunner
|
|
|
|
silenttestrunner.main(__name__)
|