1'''Mock classes that imitate idlelib modules or classes. 2 3Attributes and methods will be added as needed for tests. 4''' 5 6from idlelib.idle_test.mock_tk import Text 7 8class Func: 9 '''Record call, capture args, return/raise result set by test. 10 11 When mock function is called, set or use attributes: 12 self.called - increment call number even if no args, kwds passed. 13 self.args - capture positional arguments. 14 self.kwds - capture keyword arguments. 15 self.result - return or raise value set in __init__. 16 self.return_self - return self instead, to mock query class return. 17 18 Most common use will probably be to mock instance methods. 19 Given class instance, can set and delete as instance attribute. 20 Mock_tk.Var and Mbox_func are special variants of this. 21 ''' 22 def __init__(self, result=None, return_self=False): 23 self.called = 0 24 self.result = result 25 self.return_self = return_self 26 self.args = None 27 self.kwds = None 28 def __call__(self, *args, **kwds): 29 self.called += 1 30 self.args = args 31 self.kwds = kwds 32 if isinstance(self.result, BaseException): 33 raise self.result 34 elif self.return_self: 35 return self 36 else: 37 return self.result 38 39 40class Editor: 41 '''Minimally imitate editor.EditorWindow class. 42 ''' 43 def __init__(self, flist=None, filename=None, key=None, root=None, 44 text=None): # Allow real Text with mock Editor. 45 self.text = text or Text() 46 self.undo = UndoDelegator() 47 48 def get_selection_indices(self): 49 first = self.text.index('1.0') 50 last = self.text.index('end') 51 return first, last 52 53 54class UndoDelegator: 55 '''Minimally imitate undo.UndoDelegator class. 56 ''' 57 # A real undo block is only needed for user interaction. 58 def undo_block_start(*args): 59 pass 60 def undo_block_stop(*args): 61 pass 62