# HG changeset patch
# User Pierre-Yves David <pierre-yves.david@logilab.fr>
# Date 1343232551 -7200
# Node ID 0313cb49fa1a122c25050037cf7ff9f26c557714
# Parent  1d6cc8c22cd9289ef30c547ee4c58189a128b6cc
obsolete: fix visibleheads/branchmap wrapping

diff -r 1d6cc8c22cd9 -r 0313cb49fa1a hgext/obsolete.py
--- a/hgext/obsolete.py	Wed Jul 25 17:53:17 2012 +0200
+++ b/hgext/obsolete.py	Wed Jul 25 18:09:11 2012 +0200
@@ -475,7 +475,7 @@
             def branchmap():
                 newbm = {}
                 oldbm = None
-                if (util.safehasattr(phases, 'visiblebranchmap')
+                if (util.safehasattr(discovery, 'visiblebranchmap')
                     and not util.safehasattr(remote, 'ignorevisiblebranchmap')
                    ):
                     remote.ignorevisiblebranchmap = False
@@ -717,10 +717,10 @@
         extensions.wrapfunction(cmdutil, 'amend', wrapcmdutilamend)
     extensions.wrapfunction(discovery, 'findcommonoutgoing', wrapfindcommonoutgoing)
     extensions.wrapfunction(discovery, 'checkheads', wrapcheckheads)
-    extensions.wrapfunction(phases, 'visibleheads', noextinctsvisibleheads)
+    extensions.wrapfunction(discovery, 'visibleheads', noextinctsvisibleheads)
     extensions.wrapfunction(phases, 'advanceboundary', wrapclearcache)
-    if util.safehasattr(phases, 'visiblebranchmap'):
-        extensions.wrapfunction(phases, 'visiblebranchmap', wrapvisiblebranchmap)
+    if util.safehasattr(discovery, 'visiblebranchmap'):
+        extensions.wrapfunction(discovery, 'visiblebranchmap', wrapvisiblebranchmap)
 
 ### serialisation
 #############################