FazBrowse GitHub Viewer
|
Trending
|
URL:
|
Home
Tools:
[Download Repo ZIP]
[View Raw Code]
[Original HTTPS Page]
openai-python/tests/test_utils/test_logging.py at main · windynet/openai-python · GitHub
windynet
/
openai-python
Public
forked from
openai/openai-python
Notifications
You must be signed in to change notification settings
Fork
0
Star
0
Code
Pull requests
0
Actions
Projects
Security and quality
0
Insights
Additional navigation options
Code
Pull requests
Actions
Projects
Security and quality
Insights
Expand file tree
Breadcrumbs
openai-python
/
tests
/
test_utils
/
test_logging.py
Copy path
More file actions
More file actions
Latest commit
History
History
History
100 lines (82 loc) · 3.72 KB
Breadcrumbs
openai-python
/
tests
/
test_utils
/
test_logging.py
Copy path
File metadata and controls
100 lines (82 loc) · 3.72 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
import
logging
from
typing
import
Any
,
Dict
,
cast
import
pytest
from
openai
.
_utils
import
SensitiveHeadersFilter
@
pytest
.
fixture
def
logger_with_filter
()
->
logging
.
Logger
:
logger
=
logging
.
getLogger
(
"test_logger"
)
logger
.
setLevel
(
logging
.
DEBUG
)
logger
.
addFilter
(
SensitiveHeadersFilter
())
return
logger
def
test_keys_redacted
(
logger_with_filter
:
logging
.
Logger
,
caplog
:
pytest
.
LogCaptureFixture
)
->
None
:
with
caplog
.
at_level
(
logging
.
DEBUG
):
logger_with_filter
.
debug
(
"Request options: %s"
,
{
"method"
:
"post"
,
"url"
:
"chat/completions"
,
"headers"
: {
"api-key"
:
"12345"
,
"Authorization"
:
"Bearer token"
},
},
)
log_record
=
cast
(
Dict
[
str
,
Any
],
caplog
.
records
[
0
].
args
)
assert
log_record
[
"method"
]
==
"post"
assert
log_record
[
"url"
]
==
"chat/completions"
assert
log_record
[
"headers"
][
"api-key"
]
==
"<redacted>"
assert
log_record
[
"headers"
][
"Authorization"
]
==
"<redacted>"
assert
(
caplog
.
messages
[
0
]
==
"Request options: {'method': 'post', 'url': 'chat/completions', 'headers': {'api-key': '<redacted>', 'Authorization': '<redacted>'}}"
)
def
test_keys_redacted_case_insensitive
(
logger_with_filter
:
logging
.
Logger
,
caplog
:
pytest
.
LogCaptureFixture
)
->
None
:
with
caplog
.
at_level
(
logging
.
DEBUG
):
logger_with_filter
.
debug
(
"Request options: %s"
,
{
"method"
:
"post"
,
"url"
:
"chat/completions"
,
"headers"
: {
"Api-key"
:
"12345"
,
"authorization"
:
"Bearer token"
},
},
)
log_record
=
cast
(
Dict
[
str
,
Any
],
caplog
.
records
[
0
].
args
)
assert
log_record
[
"method"
]
==
"post"
assert
log_record
[
"url"
]
==
"chat/completions"
assert
log_record
[
"headers"
][
"Api-key"
]
==
"<redacted>"
assert
log_record
[
"headers"
][
"authorization"
]
==
"<redacted>"
assert
(
caplog
.
messages
[
0
]
==
"Request options: {'method': 'post', 'url': 'chat/completions', 'headers': {'Api-key': '<redacted>', 'authorization': '<redacted>'}}"
)
def
test_no_headers
(
logger_with_filter
:
logging
.
Logger
,
caplog
:
pytest
.
LogCaptureFixture
)
->
None
:
with
caplog
.
at_level
(
logging
.
DEBUG
):
logger_with_filter
.
debug
(
"Request options: %s"
,
{
"method"
:
"post"
,
"url"
:
"chat/completions"
},
)
log_record
=
cast
(
Dict
[
str
,
Any
],
caplog
.
records
[
0
].
args
)
assert
log_record
[
"method"
]
==
"post"
assert
log_record
[
"url"
]
==
"chat/completions"
assert
"api-key"
not
in
log_record
assert
"Authorization"
not
in
log_record
assert
caplog
.
messages
[
0
]
==
"Request options: {'method': 'post', 'url': 'chat/completions'}"
def
test_headers_without_sensitive_info
(
logger_with_filter
:
logging
.
Logger
,
caplog
:
pytest
.
LogCaptureFixture
)
->
None
:
with
caplog
.
at_level
(
logging
.
DEBUG
):
logger_with_filter
.
debug
(
"Request options: %s"
,
{
"method"
:
"post"
,
"url"
:
"chat/completions"
,
"headers"
: {
"custom"
:
"value"
},
},
)
log_record
=
cast
(
Dict
[
str
,
Any
],
caplog
.
records
[
0
].
args
)
assert
log_record
[
"method"
]
==
"post"
assert
log_record
[
"url"
]
==
"chat/completions"
assert
log_record
[
"headers"
]
==
{
"custom"
:
"value"
}
assert
(
caplog
.
messages
[
0
]
==
"Request options: {'method': 'post', 'url': 'chat/completions', 'headers': {'custom': 'value'}}"
)
def
test_standard_debug_msg
(
logger_with_filter
:
logging
.
Logger
,
caplog
:
pytest
.
LogCaptureFixture
)
->
None
:
with
caplog
.
at_level
(
logging
.
DEBUG
):
logger_with_filter
.
debug
(
"Sending HTTP Request: %s %s"
,
"POST"
,
"chat/completions"
)
assert
caplog
.
messages
[
0
]
==
"Sending HTTP Request: POST chat/completions"
Back
|
FazBrowse Home
|
New Git URL