tests/test-wireproto.py

--- a/tests/test-wireproto.py	Tue Mar 13 22:53:49 2018 -0400
+++ b/tests/test-wireproto.py	Wed Mar 14 15:56:34 2018 -0400
@@ -3,6 +3,7 @@
 from mercurial import (
     error,
     pycompat,
+    ui as uimod,
     util,
     wireproto,
     wireprototypes,
@@ -22,12 +23,13 @@
         pass
 
 class clientpeer(wireproto.wirepeer):
-    def __init__(self, serverrepo):
+    def __init__(self, serverrepo, ui):
         self.serverrepo = serverrepo
+        self._ui = ui
 
     @property
     def ui(self):
-        return self.serverrepo.ui
+        return self._ui
 
     def url(self):
         return b'test'
@@ -84,7 +86,7 @@
 wireproto.commands[b'greet'] = (greet, b'name',)
 
 srv = serverrepo()
-clt = clientpeer(srv)
+clt = clientpeer(srv, uimod.ui())
 
 print(clt.greet(b"Foobar"))
 b = clt.iterbatch()