"""Test for tortoisehg.util.colormap"""
import sys
from tortoisehg.util import colormap

def setup_module():
    global _origmaxint
    _origmaxint = sys.maxsize
    sys.maxsize = int((1 << 31) - 1)

def teardown_module():
    sys.maxsize = _origmaxint

def fakectx(userhash, date, tz=0):
    # AnnotateColorSaturation uses hash() for mapping user to hue
    class user:
        def __hash__(self):
            return userhash
    class ctx:
        def user(self):
            return user()
        def date(self):
            return float(date), tz
        def __repr__(self):
            return '<fakectx:userhash=%x,date=%d,%d>' % (userhash, date, tz)
    return ctx()

def test_user_hue():
    """Is a user mapped to his own color (hue)?"""
    cm = colormap.AnnotateColorSaturation()

    samples = {0x0: '#ffaaaa', 0x1: '#ffaac9', 0x2: '#ffaae9', 0x3: '#f4aaff',
               0x4: '#d4aaff', 0x5: '#b4aaff', 0x6: '#aabfff', 0x7: '#aadfff',
               0x8: '#aafeff', 0x9: '#aaffdf', 0xa: '#aaffbf', 0xb: '#b4ffaa',
               0xc: '#d4ffaa', 0xd: '#f4ffaa', 0xe: '#ffe9aa', 0xf: '#ffc9aa'}
    for i, c in sorted(samples.items(), key=lambda a: a[0]):
        assert cm.get_color(fakectx(sys.maxsize // 16 * i, 0), 0) == c

def test_user_hue_limit():
    cm = colormap.AnnotateColorSaturation(maxhues=8)

    samples = {0x0: '#ffaaaa', 0x1: '#ffaaaa', 0x2: '#ffaaaa', 0x3: '#ffaae9',
               0x4: '#ffaae9', 0x5: '#d4aaff', 0x6: '#d4aaff', 0x7: '#aabfff',
               0x8: '#aabfff', 0x9: '#aaffff', 0xa: '#aaffff', 0xb: '#aaffbf',
               0xc: '#aaffbf', 0xd: '#d4ffaa', 0xe: '#d4ffaa', 0xf: '#ffe9aa'}
    for i, c in sorted(samples.items(), key=lambda a: a[0]):
        assert cm.get_color(fakectx(sys.maxsize // 16 * i, 0), 0) == c

SECS_PER_DAY = 24 * 60 * 60

def test_age_saturation():
    """Is an age mapped to its saturation?"""
    cm = colormap.AnnotateColorSaturation()

    samples = {0: '#ffaaaa', 50: '#ffd4d4', 100: '#ffe2e2', 150: '#ffe9e9',
               200: '#ffeeee', 250: '#fff0f0', 300: '#fff2f2', 350: '#fff4f4',
               400: '#fff5f5', 450: '#fff6f6', 500: '#fff7f7', 550: '#fff7f7'}
    for i, c in sorted(samples.items(), key=lambda a: a[0]):
        assert cm.get_color(fakectx(0, 0), float(i * SECS_PER_DAY)) == c

def test_age_saturation_limit():
    cm = colormap.AnnotateColorSaturation(maxsaturations=16)

    samples = {0: '#ffaaaa', 3: '#ffaaaa', 4: '#ffafaf', 7: '#ffafaf',
               8: '#ffb4b4', 11: '#ffb4b4', 12: '#ffb9b9', 16: '#ffb9b9',
               17: '#ffbfbf', 22: '#ffbfbf', 23: '#ffc4c4', 29: '#ffc4c4'}
    for i, c in sorted(samples.items(), key=lambda a: a[0]):
        assert cm.get_color(fakectx(0, 0), float(i * SECS_PER_DAY)) == c

def test_age_calc():
    """Color shouldn't depend on the date but the age"""
    cm = colormap.AnnotateColorSaturation()
    age = 50 * SECS_PER_DAY
    for d in (0, SECS_PER_DAY, 365 * SECS_PER_DAY):
        assert cm.get_color(fakectx(0, d), d + age) == '#ffd4d4'

def test_negative_age_calc():
    cm = colormap.AnnotateColorSaturation()
    assert cm.get_color(fakectx(0, 0), 0), '0 days old' == '#ffaaaa'
    assert cm.get_color(fakectx(0, 50 * SECS_PER_DAY), 0) == '#ffaaaa', \
        '-50 days old (should not raise ZeroDivisionError)'
    assert cm.get_color(fakectx(0, 100 * SECS_PER_DAY), 0) == '#ffaaaa', \
        '-100 days old'

def test_makeannotatepalette_latest_wins():
    userstep = sys.maxsize // 16
    filectxs = [fakectx(0 * userstep, 0), fakectx(1 * userstep, 1),
                fakectx(2 * userstep, 2), fakectx(3 * userstep, 3),
                fakectx(4 * userstep, 4)]

    palette = colormap.makeannotatepalette(filectxs, now=4, maxcolors=4)
    palfctxs = set()
    for _color, fctxs in palette.items():
        palfctxs.update(fctxs)
    assert palfctxs == set(filectxs[1:])

def test_makeannotatepalette_fold_same_color():
    userstep = sys.maxsize // 16
    filectxs = [fakectx(0 * userstep, 0), fakectx(1 * userstep, 0),
                fakectx(2 * userstep, 0), fakectx(3 * userstep, 0),
                fakectx(4 * userstep, 0)]
    palette = colormap.makeannotatepalette(filectxs, now=0,
                                           maxcolors=4, maxhues=8)
    assert len(palette) == 2
    assert set(palette['#ffaaaa']) == {filectxs[0], filectxs[1], filectxs[2]}
    assert set(palette['#ffaae9']) == {filectxs[3], filectxs[4]}

def test_makeannotatepalette_mindate_included():
    agestep = 10 * SECS_PER_DAY
    filectxs = [fakectx(0, 0 * agestep), fakectx(0, 1 * agestep),
                fakectx(0, 2 * agestep), fakectx(0, 3 * agestep),
                fakectx(0, 4 * agestep), fakectx(0, 5 * agestep),
                fakectx(0, 6 * agestep), fakectx(0, 7 * agestep)]
    palette = colormap.makeannotatepalette(filectxs, now=7 * agestep,
                                           maxcolors=4, maxhues=4,
                                           maxsaturations=255,
                                           mindate=2 * agestep)
    palfctxs = set()
    for _color, fctxs in palette.items():
        palfctxs.update(fctxs)
    for fctx in filectxs[2:]:
        assert fctx in palfctxs, (fctx, palfctxs)

def test_makeannotatepalette_mindate_earlier_than_rev0():
    agestep = 50 * SECS_PER_DAY
    filectxs = [fakectx(0, 1 * agestep), fakectx(0, 2 * agestep)]
    palette = colormap.makeannotatepalette(filectxs, now=2 * agestep,
                                           maxcolors=1, maxhues=1,
                                           maxsaturations=255, mindate=0)
    palfctxs = set()
    for _color, fctxs in palette.items():
        palfctxs.update(fctxs)
    for fctx in filectxs:
        assert fctx in palfctxs, (fctx, palfctxs)
