def setUp(self): CaptureRequestHandler.__init__ = Mock(return_value=None) self.mock_modifier = Mock() self.mock_storage = Mock() self.mock_send_response = Mock() self.mock_send_header = Mock() self.mock_end_headers = Mock() self.mock_rfile = Mock() self.mock_rfile.read.return_value = b'the body' self.mock_wfile = Mock() self.mock_wfile.write.side_effect = lambda body: setattr(self, 'rbody', body) self.handler = CaptureRequestHandler() self.handler.server = Mock() self.handler.modifier = self.mock_modifier self.handler.storage = self.mock_storage self.handler.options = {} self.handler.headers = {} self.handler.scopes = [] self.handler.command = 'GET' self.handler.send_response = self.mock_send_response self.handler.send_header = self.mock_send_header self.handler.end_headers = self.mock_end_headers self.handler.rfile = self.mock_rfile self.handler.wfile = self.mock_wfile
def setUp(self): CaptureRequestHandler.__init__ = Mock(return_value=None) self.mock_modifier, self.mock_storage = Mock(), Mock() self.handler = CaptureRequestHandler() self.handler.server = Mock() self.handler.id = '12345' self.handler.server.modifier = self.mock_modifier self.handler.server.storage = self.mock_storage self.handler.server.options = {} self.handler.path = 'https://www.google.com/foo/bar?x=y' self.handler.command = 'GET' self.body = None
def setUp(self): CaptureRequestHandler.__init__ = Mock(return_value=None) self.mock_modifier = Mock() self.mock_storage = Mock() self.mock_send_response = Mock() self.mock_send_header = Mock() self.mock_end_headers = Mock() self.mock_rfile = Mock() self.mock_wfile = Mock() self.handler = CaptureRequestHandler() self.handler.server = Mock() self.handler.server.modifier = self.mock_modifier self.handler.server.storage = self.mock_storage self.handler.server.options = {} self.handler.command = 'GET' self.handler.send_response = self.mock_send_response self.handler.send_header = self.mock_send_header self.handler.end_headers = self.mock_end_headers self.handler.rfile = self.mock_rfile self.handler.wfile = self.mock_wfile