import binascii

import unittest

import itertools

import silenttestrunner

from mercurial import manifest as manifestmod

from mercurial import match as matchmod

EMTPY_MANIFEST = ''

HASH_1 = '1' * 40

BIN_HASH_1 = binascii.unhexlify(HASH_1)

HASH_2 = 'f' * 40

BIN_HASH_2 = binascii.unhexlify(HASH_2)

HASH_3 = '1234567890abcdef0987654321deadbeef0fcafe'

BIN_HASH_3 = binascii.unhexlify(HASH_3)

A_SHORT_MANIFEST = (

    'bar/baz/qux.py\0%(hash2)s%(flag2)s\n'

    'foo\0%(hash1)s%(flag1)s\n'

    ) % {'hash1': HASH_1,

         'flag1': '',

         'hash2': HASH_2,

         'flag2': 'l',

         }

A_DEEPER_MANIFEST = (

    'a/b/c/bar.py\0%(hash3)s%(flag1)s\n'

    'a/b/c/bar.txt\0%(hash1)s%(flag1)s\n'

    'a/b/c/foo.py\0%(hash3)s%(flag1)s\n'

    'a/b/c/foo.txt\0%(hash2)s%(flag2)s\n'

    'a/b/d/baz.py\0%(hash3)s%(flag1)s\n'

    'a/b/d/qux.py\0%(hash1)s%(flag2)s\n'

    'a/b/d/ten.txt\0%(hash3)s%(flag2)s\n'

    'a/b/dog.py\0%(hash3)s%(flag1)s\n'

    'a/b/fish.py\0%(hash2)s%(flag1)s\n'

    'a/c/london.py\0%(hash3)s%(flag2)s\n'

    'a/c/paper.txt\0%(hash2)s%(flag2)s\n'

    'a/c/paris.py\0%(hash2)s%(flag1)s\n'

    'a/d/apple.py\0%(hash3)s%(flag1)s\n'

    'a/d/pizza.py\0%(hash3)s%(flag2)s\n'

    'a/green.py\0%(hash1)s%(flag2)s\n'

    'a/purple.py\0%(hash2)s%(flag1)s\n'

    'app.py\0%(hash3)s%(flag1)s\n'

    'readme.txt\0%(hash2)s%(flag1)s\n'

    ) % {'hash1': HASH_1,

         'flag1': '',

         'hash2': HASH_2,

         'flag2': 'l',

         'hash3': HASH_3,

         }

HUGE_MANIFEST_ENTRIES = 200001

A_HUGE_MANIFEST = ''.join(sorted(

    'file%d\0%s%s\n' % (i, h, f) for i, h, f in

    itertools.izip(xrange(200001),

                   itertools.cycle((HASH_1, HASH_2)),

                   itertools.cycle(('', 'x', 'l')))))

def parsemanifest(text):

    return manifestmod.manifestdict(text)

class testmanifest(unittest.TestCase):

    def assertIn(self, thing, container, msg=None):

        # assertIn new in 2.7, use it if available, otherwise polyfill

        sup = getattr(unittest.TestCase, 'assertIn', False)

        if sup:

            return sup(self, thing, container, msg=msg)

        if not msg:

            msg = 'Expected %r in %r' % (thing, container)

        self.assert_(thing in container, msg)

    def testEmptyManifest(self):

        m = parsemanifest(EMTPY_MANIFEST)

        self.assertEqual(0, len(m))

        self.assertEqual([], list(m))

    def testManifest(self):

        m = parsemanifest(A_SHORT_MANIFEST)

        self.assertEqual(['bar/baz/qux.py', 'foo'], list(m))

        self.assertEqual(BIN_HASH_2, m['bar/baz/qux.py'])

        self.assertEqual('l', m.flags('bar/baz/qux.py'))

        self.assertEqual(BIN_HASH_1, m['foo'])

        self.assertEqual('', m.flags('foo'))

        self.assertRaises(KeyError, lambda : m['wat'])

    def testSetItem(self):

        want = BIN_HASH_1

        m = parsemanifest(EMTPY_MANIFEST)

        m['a'] = want

        self.assertIn('a', m)

        self.assertEqual(want, m['a'])

        self.assertEqual('a\0' + HASH_1 + '\n', m.text())

        m = parsemanifest(A_SHORT_MANIFEST)

        m['a'] = want

        self.assertEqual(want, m['a'])

        self.assertEqual('a\0' + HASH_1 + '\n' + A_SHORT_MANIFEST,

                         m.text())

    def testSetFlag(self):

        want = 'x'

        m = parsemanifest(EMTPY_MANIFEST)

        # first add a file; a file-less flag makes no sense

        m['a'] = BIN_HASH_1

        m.setflag('a', want)

        self.assertEqual(want, m.flags('a'))

        self.assertEqual('a\0' + HASH_1 + want + '\n', m.text())

        m = parsemanifest(A_SHORT_MANIFEST)

        # first add a file; a file-less flag makes no sense

        m['a'] = BIN_HASH_1

        m.setflag('a', want)

        self.assertEqual(want, m.flags('a'))

        self.assertEqual('a\0' + HASH_1 + want + '\n' + A_SHORT_MANIFEST,

                         m.text())

    def testCopy(self):

        m = parsemanifest(A_SHORT_MANIFEST)

        m['a'] =  BIN_HASH_1

        m2 = m.copy()

        del m

        del m2 # make sure we don't double free() anything

    def testCompaction(self):

        unhex = binascii.unhexlify

        h1, h2 = unhex(HASH_1), unhex(HASH_2)

        m = parsemanifest(A_SHORT_MANIFEST)

        m['alpha'] = h1

        m['beta'] = h2

        del m['foo']

        want = 'alpha\0%s\nbar/baz/qux.py\0%sl\nbeta\0%s\n' % (

            HASH_1, HASH_2, HASH_2)

        self.assertEqual(want, m.text())

        self.assertEqual(3, len(m))

        self.assertEqual(['alpha', 'bar/baz/qux.py', 'beta'], list(m))

        self.assertEqual(h1, m['alpha'])

        self.assertEqual(h2, m['bar/baz/qux.py'])

        self.assertEqual(h2, m['beta'])

        self.assertEqual('', m.flags('alpha'))

        self.assertEqual('l', m.flags('bar/baz/qux.py'))

        self.assertEqual('', m.flags('beta'))

        self.assertRaises(KeyError, lambda : m['foo'])

    def testSetGetNodeSuffix(self):

        clean = parsemanifest(A_SHORT_MANIFEST)

        m = parsemanifest(A_SHORT_MANIFEST)

        h = m['foo']

        f = m.flags('foo')

        want = h + 'a'

        # Merge code wants to set 21-byte fake hashes at times

        m['foo'] = want

        self.assertEqual(want, m['foo'])

        self.assertEqual([('bar/baz/qux.py', BIN_HASH_2),

                          ('foo', BIN_HASH_1 + 'a')],

                         list(m.iteritems()))

        # Sometimes it even tries a 22-byte fake hash, but we can

        # return 21 and it'll work out

        m['foo'] = want + '+'

        self.assertEqual(want, m['foo'])

        # make sure the suffix survives a copy

        match = matchmod.match('', '', ['re:foo'])

        m2 = m.matches(match)

        self.assertEqual(want, m2['foo'])

        self.assertEqual(1, len(m2))

        m2 = m.copy()

        self.assertEqual(want, m2['foo'])

        # suffix with iteration

        self.assertEqual([('bar/baz/qux.py', BIN_HASH_2),

                          ('foo', want)],

                         list(m.iteritems()))

        # shows up in diff

        self.assertEqual({'foo': ((want, f), (h, ''))}, m.diff(clean))

        self.assertEqual({'foo': ((h, ''), (want, f))}, clean.diff(m))

    def testMatchException(self):

        m = parsemanifest(A_SHORT_MANIFEST)

        match = matchmod.match('', '', ['re:.*'])

        def filt(path):

            if path == 'foo':

                assert False

            return True

        match.matchfn = filt

        self.assertRaises(AssertionError, m.matches, match)

    def testRemoveItem(self):

        m = parsemanifest(A_SHORT_MANIFEST)

        del m['foo']

        self.assertRaises(KeyError, lambda : m['foo'])

        self.assertEqual(1, len(m))

        self.assertEqual(1, len(list(m)))

        # now restore and make sure everything works right

        m['foo'] = 'a' * 20

        self.assertEqual(2, len(m))

        self.assertEqual(2, len(list(m)))

    def testManifestDiff(self):

        MISSING = (None, '')

        addl = 'z-only-in-left\0' + HASH_1 + '\n'

        addr = 'z-only-in-right\0' + HASH_2 + 'x\n'

        left = parsemanifest(

            A_SHORT_MANIFEST.replace(HASH_1, HASH_3 + 'x') + addl)

        right = parsemanifest(A_SHORT_MANIFEST + addr)

        want = {

            'foo': ((BIN_HASH_3, 'x'),

                    (BIN_HASH_1, '')),

            'z-only-in-left': ((BIN_HASH_1, ''), MISSING),

            'z-only-in-right': (MISSING, (BIN_HASH_2, 'x')),

            }

        self.assertEqual(want, left.diff(right))

        want = {

            'bar/baz/qux.py': (MISSING, (BIN_HASH_2, 'l')),

            'foo': (MISSING, (BIN_HASH_3, 'x')),

            'z-only-in-left': (MISSING, (BIN_HASH_1, '')),

            }

        self.assertEqual(want, parsemanifest(EMTPY_MANIFEST).diff(left))

        want = {

            'bar/baz/qux.py': ((BIN_HASH_2, 'l'), MISSING),

            'foo': ((BIN_HASH_3, 'x'), MISSING),

            'z-only-in-left': ((BIN_HASH_1, ''), MISSING),

            }

        self.assertEqual(want, left.diff(parsemanifest(EMTPY_MANIFEST)))

        copy = right.copy()

        del copy['z-only-in-right']

        del right['foo']

        want = {

            'foo': (MISSING, (BIN_HASH_1, '')),

            'z-only-in-right': ((BIN_HASH_2, 'x'), MISSING),

            }

        self.assertEqual(want, right.diff(copy))

        short = parsemanifest(A_SHORT_MANIFEST)

        pruned = short.copy()

        del pruned['foo']

        want = {

            'foo': ((BIN_HASH_1, ''), MISSING),

            }

        self.assertEqual(want, short.diff(pruned))

        want = {

            'foo': (MISSING, (BIN_HASH_1, '')),

            }

        self.assertEqual(want, pruned.diff(short))

        want = {

            'bar/baz/qux.py': None,

            'foo': (MISSING, (BIN_HASH_1, '')),

            }

        self.assertEqual(want, pruned.diff(short, True))

    def testReversedLines(self):

        backwards = ''.join(

            l + '\n' for l in reversed(A_SHORT_MANIFEST.split('\n')) if l)

        try:

            parsemanifest(backwards)

            self.fail('Should have raised ValueError')

        except ValueError, v:

            self.assertIn('Manifest lines not in sorted order.', str(v))

    def testNoTerminalNewline(self):

        try:

            parsemanifest(A_SHORT_MANIFEST + 'wat')

            self.fail('Should have raised ValueError')

        except ValueError, v:

            self.assertIn('Manifest did not end in a newline.', str(v))

    def testNoNewLineAtAll(self):

        try:

            parsemanifest('wat')

            self.fail('Should have raised ValueError')

        except ValueError, v:

            self.assertIn('Manifest did not end in a newline.', str(v))

    def testHugeManifest(self):

        m = parsemanifest(A_HUGE_MANIFEST)

        self.assertEqual(HUGE_MANIFEST_ENTRIES, len(m))

        self.assertEqual(len(m), len(list(m)))

    def testMatchesMetadata(self):

        '''Tests matches() for a few specific files to make sure that both

        the set of files as well as their flags and nodeids are correct in

        the resulting manifest.'''

        m = parsemanifest(A_HUGE_MANIFEST)

        match = matchmod.match('/', '',

                ['file1', 'file200', 'file300'], exact=True)

        m2 = m.matches(match)

        w = ('file1\0%sx\n'

             'file200\0%sl\n'

             'file300\0%s\n') % (HASH_2, HASH_1, HASH_1)

        self.assertEqual(w, m2.text())

    def testMatchesNonexistentFile(self):

        '''Tests matches() for a small set of specific files, including one

        nonexistent file to make sure in only matches against existing files.

        '''