def setUp(self): # unregister the original plugin self.stackoverflow_regex, self.stackoverflow_function = registry.plugins['stackoverflow'] registry.unregister('stackoverflow') # register a mock, just to verify if the regex is working self.called = False self.tag = None self.limit = 50 def stackoverflow_mock(limit=None, tag=None): self.called = True self.tag = tag self.limit = int(limit) if limit else 50 registry.register('stackoverflow', self.stackoverflow_regex, stackoverflow_mock) self.protocol = EizzekProtocol()
def tearDown(self): # undo de mock registry.unregister('stackoverflow') registry.register('stackoverflow', self.stackoverflow_regex, self.stackoverflow_function)
def __call__(self, func): name = self.name or func.__name__ registry.register(name, self.regex, func) return func