def patch_now(self): """ Replace distro_tracker.core.utils.now() with a mocked version returning self.current_datetime """ def get_datetime(*args, **kwargs): return self.current_datetime if not hasattr(self, 'current_datetime'): self.current_datetime = datetime.now() patcher = mock.patch('distro_tracker.core.utils.now') now = patcher.start() now.side_effect = get_datetime self.addCleanup(patcher.stop)
def patch_vendor_call(self, return_value=None): patcher = mock.patch('distro_tracker.vendor.call') mocked = patcher.start() mocked.return_value = (return_value, return_value is not None) self.addCleanup(patcher.stop) return mocked
def patch_forward(self): patcher = mock.patch('distro_tracker.mail.dispatch.forward') mocked = patcher.start() self.addCleanup(patcher.stop) return mocked
def patch_mail_processor(self, new=mock.DEFAULT): process_method = 'distro_tracker.mail.processor.MailProcessor.process' patcher = mock.patch(process_method) mock_process = patcher.start() self.addCleanup(patcher.stop) return mock_process