from __future__ import absolute_import

import test_hgsubversion_util
from edenscm.hgext.hgsubversion import maps, svnmeta


class TestRevMapMigrate(test_hgsubversion_util.TestBase):
    def tearDown(self):
        # revert changes to defaultrevmapclass
        svnmeta.SVNMeta._defaultrevmapclass = maps.RevMap

    def _test_revmap_migrate(self, fromclass, toclass):
        # revmap interfaces to test
        getters = [
            lambda x: x.branchedits("the_branch", 3),
            lambda x: x.branchedits("the_branch", 4),
            lambda x: x.branchedits("the_branch", 5),
            lambda x: x.branchedits("the_branch", 6),
            lambda x: x.branchedits(None, 5),
            lambda x: x.branchedits("non_existed", 10),
            lambda x: x.branchmaxrevnum("the_branch", 3),
            lambda x: x.branchmaxrevnum("the_branch", 4),
            lambda x: x.branchmaxrevnum("the_branch", 5),
            lambda x: x.branchmaxrevnum("the_branch", 6),
            lambda x: x.branchmaxrevnum(None, 5),
            lambda x: x.branchmaxrevnum("non_existed", 10),
            lambda x: list(x.revhashes(3)),
            lambda x: list(x.revhashes(4)),
            lambda x: list(x.revhashes(42)),
            lambda x: list(x.revhashes(105)),
            lambda x: x.firstpulled,
            lambda x: x.lastpulled,
            lambda x: x.lasthash,
        ]

        svnmeta.SVNMeta._defaultrevmapclass = fromclass
        repo = self._load_fixture_and_fetch("two_heads.svndump")
        # set to None to the default revmapclass will be used
        repo.ui.setconfig("hgsubversion", "revmapimpl", None)
        meta = svnmeta.SVNMeta(repo)
        self.assertEqual(meta.revmap.__class__, fromclass)
        origrevmap = meta.revmap

        # insert fake special (duplicated, with '\0') data
        origrevmap[103, None] = b"\0" * 20
        origrevmap[104, None] = b"\0" * 18 + b"cd"
        origrevmap[105, None] = b"ab\0cdefghijklmnopqrs"
        origrevmap[104, None] = b"\0" * 18 + b"\xff\0"
        origrevmap[105, "ab"] = origrevmap[105, None]

        origvalues = [f(meta.revmap) for f in getters]

        # migrate to another format (transparently)
        svnmeta.SVNMeta._defaultrevmapclass = toclass
        meta = svnmeta.SVNMeta(repo)
        self.assertEqual(meta.revmap.__class__, toclass)

        # enable iteration otherwise we cannot use iteritems
        origrevmap._allowiter = True
        for k, v in origrevmap.iteritems():
            newv = meta.revmap[k]
            self.assertEqual(newv, v)
            self.assertEqual(len(newv), 20)
            self.assertEqual(meta.revmap[meta.revmap.hashes()[v]], v)

        newvalues = [f(meta.revmap) for f in getters]
        self.assertEqual(origvalues, newvalues)

    def test_revmap_migrate_up(self):
        self._test_revmap_migrate(maps.RevMap, maps.SqliteRevMap)

    def test_revmap_migrate_down(self):
        self._test_revmap_migrate(maps.SqliteRevMap, maps.RevMap)


if __name__ == "__main__":
    import silenttestrunner

    silenttestrunner.main(__name__)