示例#1
0
    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
示例#2
0
 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
示例#3
0
    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