@@ -169,13 +169,18 @@ def _firestore_endpoint_handler(
169
169
firestore_event_data .old_value .create_time ,
170
170
firestore_event_data .old_value .update_time ,
171
171
)
172
- if event_type == _event_type_deleted :
172
+
173
+ if event_type in (_event_type_deleted ,
174
+ _event_type_deleted_with_auth_context ):
173
175
firestore_event_data = _typing .cast (_firestore .DocumentEventData ,
174
176
old_value_snapshot )
175
- if event_type == _event_type_created :
177
+ if event_type in (_event_type_created ,
178
+ _event_type_created_with_auth_context ):
176
179
firestore_event_data = _typing .cast (_firestore .DocumentEventData ,
177
180
value_snapshot )
178
- if event_type in (_event_type_written , _event_type_updated ):
181
+ if event_type in (_event_type_written , _event_type_updated ,
182
+ _event_type_written_with_auth_context ,
183
+ _event_type_updated_with_auth_context ):
179
184
firestore_event_data = _typing .cast (
180
185
_firestore .DocumentEventData ,
181
186
Change (
@@ -306,7 +311,7 @@ def on_document_written_with_auth_context_wrapped(raw: _ce.CloudEvent):
306
311
_util .set_func_endpoint_attr (
307
312
on_document_written_with_auth_context_wrapped ,
308
313
options ._endpoint (
309
- event_type = _event_type_written ,
314
+ event_type = _event_type_written_with_auth_context ,
310
315
func_name = func .__name__ ,
311
316
document_pattern = document_pattern ,
312
317
),
0 commit comments