1
1
import json
2
2
import os
3
+ import pytest
3
4
4
5
from requests import Request
5
6
@@ -79,8 +80,10 @@ def test_log_request_with_headers(mocked_logger):
79
80
"body=%s \n " % request .body )
80
81
81
82
83
+ @pytest .mark .parametrize ('log_level' , ['INFO' , 'CONSOLE' , 'HTML' , 'WARN' , 'ERROR' ])
82
84
@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
84
87
headers = {'User-Agent' : 'python-requests/2.31.0' ,
85
88
'Accept-Encoding' : 'gzip, deflate' ,
86
89
'Accept' : '*/*' ,
@@ -101,6 +104,28 @@ def test_log_request_with_headers_auth(mocked_logger):
101
104
"body=%s \n " % request .body )
102
105
103
106
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
+
104
129
@mock .patch ('RequestsLibrary.log.logger' )
105
130
def test_log_request_with_redirect (mocked_logger ):
106
131
request = Request (method = 'get' , url = 'http://mock.rulezz/redirected' )
0 commit comments