25
25
class ExpectShelver(shelf_ui.Shelver):
26
26
"""A variant of Shelver that intercepts console activity, for testing."""
28
def __init__(self, work_tree, target_tree, diff_writer=None, path=None,
29
auto=False, auto_apply=False, file_list=None, message=None):
28
def __init__(self, work_tree, target_tree, diff_writer=None,
29
auto=False, auto_apply=False, file_list=None, message=None,
30
31
shelf_ui.Shelver.__init__(self, work_tree, target_tree, diff_writer,
31
auto, auto_apply, file_list, message)
32
auto, auto_apply, file_list, message,
33
35
self.diff_writer = StringIO()