@@ -1142,7 +1142,10 @@ def test_collection_add(client, cleanup, database):
11421142
11431143
11441144@pytest .mark .parametrize ("database" , [None , FIRESTORE_OTHER_DB ], indirect = True )
1145- def test_list_collections_with_read_time (client , cleanup , database ):
1145+ @pytest .mark .parametrize (
1146+ "use_python_datetime" , [True , False ]
1147+ )
1148+ def test_list_collections_with_read_time (client , cleanup , database , use_python_datetime ):
11461149 # TODO(microgen): list_documents is returning a generator, not a list.
11471150 # Consider if this is desired. Also, Document isn't hashable.
11481151 collection_id = "coll-add" + UNIQUE_RESOURCE_ID
@@ -1152,11 +1155,15 @@ def test_list_collections_with_read_time(client, cleanup, database):
11521155
11531156 data1 = {"foo" : "bar" }
11541157 update_time1 , document_ref1 = collection .add (data1 )
1158+ if use_python_datetime :
1159+ update_time1 = datetime .datetime .now (tz = datetime .timezone .utc )
11551160 cleanup (document_ref1 .delete )
11561161 assert set (collection .list_documents ()) == {document_ref1 }
11571162
11581163 data2 = {"bar" : "baz" }
11591164 update_time2 , document_ref2 = collection .add (data2 )
1165+ if use_python_datetime :
1166+ update_time2 = datetime .datetime .now (tz = datetime .timezone .utc )
11601167 cleanup (document_ref2 .delete )
11611168 assert set (collection .list_documents ()) == {document_ref1 , document_ref2 }
11621169 assert set (collection .list_documents (read_time = update_time1 )) == {document_ref1 }
0 commit comments