# HG changeset patch
# User Martijn Pieters <mj@zopatista.com>
# Date 1240864788 -7200
# Node ID dc2bb6faf904912e9a7362a7cf973f9bff6d47e1
# Parent  552deb1351ce22e4741eb441b2c846c532762cd4
Implement the copy method on the mock ui to maintain it across tests.

See mercurial revision d2899a856f9f.

diff --git a/tests/test_util.py b/tests/test_util.py
--- a/tests/test_util.py
+++ b/tests/test_util.py
@@ -86,6 +86,9 @@ class MockUI(object):
     def write(self, *args):
         self.stream.write(*args)
 
+    def copy(self):
+        return self.__class__(self.inner_ui)
+
     def __getattr__(self, attr):
         return getattr(self.inner_ui, attr)