sapling/hggit/hgrepo.py
Augie Fackler c13e8ed89d tests: import test-push-r (with slight changes) from Mercurial
This lets us detect defects in our wrappers which would break
Mercurial when working against normal hg repositories with hg-git
enabled.
2010-08-15 09:45:00 -05:00

71 lines
2.4 KiB
Python

from mercurial.node import bin
from git_handler import GitHandler
from gitrepo import gitrepo
def generate_repo_subclass(baseclass):
class hgrepo(baseclass):
def pull(self, remote, heads=None, force=False):
if isinstance(remote, gitrepo):
git = GitHandler(self, self.ui)
git.fetch(remote.path, heads)
else: #pragma: no cover
return super(hgrepo, self).pull(remote, heads, force)
# TODO figure out something useful to do with the newbranch param
def push(self, remote, force=False, revs=None, newbranch=None):
if isinstance(remote, gitrepo):
git = GitHandler(self, self.ui)
git.push(remote.path, revs, force)
else: #pragma: no cover
# newbranch was added in 1.6
if newbranch is None:
return super(hgrepo, self).push(remote, force, revs)
else:
return super(hgrepo, self).push(remote, force, revs,
newbranch)
def findoutgoing(self, remote, base=None, heads=None, force=False):
if isinstance(remote, gitrepo):
git = GitHandler(self, self.ui)
base, heads = git.get_refs(remote.path)
out, h = super(hgrepo, self).findoutgoing(remote, base, heads, force)
return out
else: #pragma: no cover
return super(hgrepo, self).findoutgoing(remote, base, heads, force)
def _findtags(self):
(tags, tagtypes) = super(hgrepo, self)._findtags()
git = GitHandler(self, self.ui)
for tag, rev in git.tags.iteritems():
if tag in tags:
continue
tags[tag] = bin(rev)
tagtypes[tag] = 'git'
return (tags, tagtypes)
def tags(self):
if not hasattr(self, 'tagscache'):
# mercurial 1.4
return super(hgrepo, self).tags()
if self.tagscache:
return self.tagscache
git = GitHandler(self, self.ui)
tagscache = super(hgrepo, self).tags()
for tag, rev in git.tags.iteritems():
if tag in tagscache:
continue
tagscache[tag] = bin(rev)
self._tagstypecache[tag] = 'git'
return tagscache
return hgrepo