diff options
Diffstat (limited to 'tests/notcapable')
| -rw-r--r-- | tests/notcapable | 11 |
1 files changed, 3 insertions, 8 deletions
diff --git a/tests/notcapable b/tests/notcapable index ef8dc22..31c1858 100644 --- a/tests/notcapable +++ b/tests/notcapable @@ -6,18 +6,13 @@ then fi cat > notcapable-$CAP.py << EOF -from mercurial import extensions, peer, localrepo +from mercurial import extensions, repo def extsetup(): - extensions.wrapfunction(peer.peerrepository, 'capable', wrapcapable) - extensions.wrapfunction(localrepo.localrepository, 'peer', wrappeer) -def wrapcapable(orig, self, name, *args, **kwargs): + extensions.wrapfunction(repo.repository, 'capable', wrapper) +def wrapper(orig, self, name, *args, **kwargs): if name in '$CAP'.split(' '): return False return orig(self, name, *args, **kwargs) -def wrappeer(orig, self): - # Since we're disabling some newer features, we need to make sure local - # repos add in the legacy features again. - return localrepo.locallegacypeer(self) EOF echo '[extensions]' >> $HGRCPATH |