3
3
import pytest
4
4
from google_cloud_logger import GoogleCloudFormatter
5
5
6
+
6
7
# from https://stackoverflow.com/a/19258720
7
8
class FakeCode (object ):
8
9
def __init__ (self , co_filename , co_name ):
@@ -64,7 +65,9 @@ def record_with_extra_attribute(log_record_factory, mocker):
64
65
"lineno" : "88" ,
65
66
"levelname" : "WARNING" ,
66
67
"message" : "farofa" ,
67
- "extra" : {"extra_field" : "extra" },
68
+ "extra" : {
69
+ "extra_field" : "extra"
70
+ },
68
71
}
69
72
record = log_record_factory (** data )
70
73
record .getMessage = mocker .Mock (return_value = data ["message" ])
@@ -95,19 +98,25 @@ def test_add_fields(formatter, record, mocker):
95
98
mocker .patch .object (
96
99
formatter ,
97
100
"make_entry" ,
98
- return_value = OrderedDict (
99
- [
100
- ("timestamp" , "2018-08-30 20:40:57Z" ),
101
- ("severity" , "WARNING" ),
102
- ("message" , "farofa" ),
103
- ("labels" , {"type" : "python-application" }),
104
- ("metadata" , {"userLabels" : {}}),
105
- (
106
- "sourceLocation" ,
107
- {"file" : "_internal.py" , "function" : "_log" , "line" : "88" },
108
- ),
109
- ]
110
- ),
101
+ return_value = OrderedDict ([
102
+ ("timestamp" , "2018-08-30 20:40:57Z" ),
103
+ ("severity" , "WARNING" ),
104
+ ("message" , "farofa" ),
105
+ ("labels" , {
106
+ "type" : "python-application"
107
+ }),
108
+ ("metadata" , {
109
+ "userLabels" : {}
110
+ }),
111
+ (
112
+ "sourceLocation" ,
113
+ {
114
+ "file" : "_internal.py" ,
115
+ "function" : "_log" ,
116
+ "line" : "88"
117
+ },
118
+ ),
119
+ ]),
111
120
)
112
121
formatter .add_fields (log_record , record , {})
113
122
@@ -141,7 +150,8 @@ def test_make_metadata(formatter, record):
141
150
assert metadata ["userLabels" ]["extra_field" ] == "extra"
142
151
143
152
144
- def test_make_metadata_with_extra_attribute (formatter , record_with_extra_attribute ):
153
+ def test_make_metadata_with_extra_attribute (formatter ,
154
+ record_with_extra_attribute ):
145
155
metadata = formatter .make_metadata (record_with_extra_attribute )
146
156
147
157
assert metadata ["userLabels" ]["extra_field" ] == "extra"
@@ -166,10 +176,8 @@ def test_make_source_location(formatter, record):
166
176
167
177
168
178
def test_format_timestamp (formatter ):
169
- assert (
170
- formatter .format_timestamp ("2018-08-30 20:40:57,245" )
171
- == "2018-08-30T20:40:57.245000Z"
172
- )
179
+ assert (formatter .format_timestamp ("2018-08-30 20:40:57,245" ) ==
180
+ "2018-08-30T20:40:57.245000Z" )
173
181
174
182
175
183
@pytest .mark .parametrize (
0 commit comments