def setUp(self): MockerTestCase.setUp(self) # we do not expect requests, but lets just prevent accidental # requests self.requests = self.mocker.replace('requests') self.expect(self.requests.request(ARGS, KWARGS)).count(0) self.proxy = self.mocker.mock(Proxy, count=False) ProxyMock = self.mocker.mock(count=False) self.expect(ProxyMock(ANY)).result(self.proxy) sm = get_current_registry() sm.registerAdapter(ProxyMock, [IRequest], IProxy, name='') self.stream = StringIO() self.handler = StreamHandler(self.stream) self.handler.setFormatter(Formatter('%(levelname)s %(message)s')) LOG.addHandler(self.handler) self.ori_log_level = LOG.getEffectiveLevel() LOG.setLevel(DEBUG)
def tearDown(self): LOG.removeHandler(self.handler) LOG.setLevel(self.ori_log_level)