def test_handle_user_request(): with patch.multiple('http.handle_requests', process_http_message=DEFAULT, handle_http_request=DEFAULT, HTTPRequest=DEFAULT, ServerRequestLogger=DEFAULT, handle_response=DEFAULT) as values: values['process_http_message'].return_value = request = mock_http_request() values['HTTPRequest'].return_value = request = mock_http_request() values['handle_http_request'].return_value = response = Mock(HTTPResponse) handle_user_request('abc') values['process_http_message'].assert_called_once_with(request, 'abc') values['handle_http_request'].assert_called_once_with(request) values['handle_response'].assert_called_once_with(response) values['ServerRequestLogger'].log.assert_called_once_with(request, response)
def test_handle_user_request_with_exception(): with patch.multiple('http.handle_requests', process_http_message=DEFAULT, handle_http_exception=DEFAULT, ServerRequestLogger=DEFAULT, HTTPRequest=DEFAULT, handle_response=DEFAULT) as values: values['process_http_message'].side_effect = e = BadRequestHttpException() values['handle_http_exception'].return_value = response = Mock(HTTPResponse) values['ServerRequestLogger'].log.return_value = '' values['HTTPRequest'].return_value = request = mock_http_request() handle_user_request('abc') values['ServerRequestLogger'].log.assert_called_once() values['process_http_message'].assert_called_once_with(request, 'abc') values['handle_http_exception'].assert_called_once_with(request, e) values['handle_response'].assert_called_once_with(response)
def test_handle_user_request(): with patch.multiple('http.handle_requests', process_http_message=DEFAULT, handle_http_request=DEFAULT, HTTPRequest=DEFAULT, ServerRequestLogger=DEFAULT, handle_response=DEFAULT) as values: values[ 'process_http_message'].return_value = request = mock_http_request( ) values['HTTPRequest'].return_value = request = mock_http_request() values['handle_http_request'].return_value = response = Mock( HTTPResponse) handle_user_request('abc') values['process_http_message'].assert_called_once_with(request, 'abc') values['handle_http_request'].assert_called_once_with(request) values['handle_response'].assert_called_once_with(response) values['ServerRequestLogger'].log.assert_called_once_with( request, response)
def test_handle_user_request_with_exception(): with patch.multiple('http.handle_requests', process_http_message=DEFAULT, handle_http_exception=DEFAULT, ServerRequestLogger=DEFAULT, HTTPRequest=DEFAULT, handle_response=DEFAULT) as values: values[ 'process_http_message'].side_effect = e = BadRequestHttpException( ) values['handle_http_exception'].return_value = response = Mock( HTTPResponse) values['ServerRequestLogger'].log.return_value = '' values['HTTPRequest'].return_value = request = mock_http_request() handle_user_request('abc') values['ServerRequestLogger'].log.assert_called_once() values['process_http_message'].assert_called_once_with(request, 'abc') values['handle_http_exception'].assert_called_once_with(request, e) values['handle_response'].assert_called_once_with(response)
def client_thread(conn, client_ip, client_port): # Receiving from client, handle request and back the resposne message = "" while True: data = conn.recv(1024) if not data: break message += data if "\r\n\r\n" in message or "\n\n" in message: break reply = handle_user_request(message, client_ip=client_ip, client_port=client_port) if data: conn.sendall(reply) conn.close()
def client_thread(conn, client_ip, client_port): # Receiving from client, handle request and back the resposne message = '' while True: data = conn.recv(1024) if not data: break message += data if '\r\n\r\n' in message or '\n\n' in message: break reply = handle_user_request(message, client_ip=client_ip, client_port=client_port) if data: conn.sendall(reply) conn.close()