11import json
22import os
3+ import pytest
34
45from requests import Request
56
@@ -79,8 +80,10 @@ def test_log_request_with_headers(mocked_logger):
7980 "body=%s \n " % request .body )
8081
8182
83+ @pytest .mark .parametrize ('log_level' , ['INFO' , 'CONSOLE' , 'HTML' , 'WARN' , 'ERROR' ])
8284@mock .patch ('RequestsLibrary.log.logger' )
83- def test_log_request_with_headers_auth (mocked_logger ):
85+ def test_log_request_with_headers_auth_with_no_debug_trace_logger (mocked_logger , log_level ):
86+ mocked_logger .LOGLEVEL = log_level
8487 headers = {'User-Agent' : 'python-requests/2.31.0' ,
8588 'Accept-Encoding' : 'gzip, deflate' ,
8689 'Accept' : '*/*' ,
@@ -101,6 +104,28 @@ def test_log_request_with_headers_auth(mocked_logger):
101104 "body=%s \n " % request .body )
102105
103106
107+ @pytest .mark .parametrize ('log_level' , ['DEBUG' , 'TRACE' ])
108+ @mock .patch ('RequestsLibrary.log.logger' )
109+ def test_log_request_with_headers_auth_with_debug_trace_logger (mocked_logger , log_level ):
110+ headers = {'User-Agent' : 'python-requests/2.31.0' ,
111+ 'Accept-Encoding' : 'gzip, deflate' ,
112+ 'Accept' : '*/*' ,
113+ 'Connection' : 'keep-alive' ,
114+ 'Authorization' : 'some_token' }
115+ mocked_logger .LOGLEVEL = log_level
116+ request = Request (method = 'get' , url = 'http://mock.rulezz' , headers = headers )
117+ request = request .prepare ()
118+ response = mock .MagicMock ()
119+ response .history = []
120+ response .request = request
121+ log_request (response )
122+ assert mocked_logger .info .call_args [0 ][0 ] == ("%s Request : " % request .method +
123+ "url=%s \n " % request .url +
124+ "path_url=%s \n " % request .path_url +
125+ "headers=%s \n " % request .headers +
126+ "body=%s \n " % request .body )
127+
128+
104129@mock .patch ('RequestsLibrary.log.logger' )
105130def test_log_request_with_redirect (mocked_logger ):
106131 request = Request (method = 'get' , url = 'http://mock.rulezz/redirected' )
0 commit comments