10000 issue-1788 fix; added firedbase tests (#11533) · dtest/python-docs-samples@963a3ee · GitHub
[go: up one dir, main page]

Skip to content

Commit 963a3ee

Browse files
authored
issue-1788 fix; added firedbase tests (GoogleCloudPlatform#11533)
1 parent 6fd0dfe commit 963a3ee

File tree

2 files changed

+607
-145
lines changed

2 files changed

+607
-145
lines changed

firestore/cloud-client/snippets.py

Lines changed: 17 additions & 10 deletions
10000
Original file line numberDiff line numberDiff line change
@@ -487,7 +487,7 @@ def compound_query_single_clause():
487487

488488

489489
def compound_query_valid_multi_clause():
490-
db = firestore.Client()
490+
db = firestore.Client(add_unique_string=False) # Flag for testing purposes, needs index to be precreated
491491
# [START firestore_query_filter_compound_multi_eq]
492492
cities_ref = db.collection("cities")
493493

@@ -500,6 +500,7 @@ def compound_query_valid_multi_clause():
500500
# [END firestore_query_filter_compound_multi_eq]
501501
print(denver_query)
502502
print(large_us_cities_query)
503+
return denver_query, large_us_cities_query
503504

504505

505506
def compound_query_valid_single_field():
@@ -525,8 +526,9 @@ def compound_query_invalid_multi_field():
525526
def order_simple_limit():
526527
db = firestore.Client()
527528
# [START firestore_order_simple_limit]
528-
db.collection("cities").order_by("name").limit(3).stream()
529+
query = db.collection("cities").order_by("name").limit(3).stream()
529530
# [END firestore_order_simple_limit]
531+
return query
530532

531533

532534
def order_simple_limit_desc():
@@ -537,16 +539,18 @@ def order_simple_limit_desc():
537539
results = query.stream()
538540
# [END firestore_query_order_desc_limit]
539541
print(results)
542+
return results
540543

541544

542545
def order_multiple():
543-
db = firestore.Client()
546+
db = firestore.Client(add_unique_string=False) # Flag for testing purposes, needs index to be precreated
544547
# [START firestore_query_order_multi]
545548
cities_ref = db.collection("cities")
546-
cities_ref.order_by("state").order_by(
549+
ordered_city_ref = cities_ref.order_by("state").order_by(
547550
"population", direction=firestore.Query.DESCENDING
548551
)
549552
# [END firestore_query_order_multi]
553+
return ordered_city_ref
550554

551555

552556
def order_where_limit():
@@ -561,6 +565,7 @@ def order_where_limit():
561565
results = query.stream()
562566
# [END firestore_query_order_limit_field_valid]
563567
print(results)
568+
return results
564569

565570

566571
def order_limit_to_last():
@@ -571,6 +576,7 @@ def order_limit_to_last():
571576
results = query.get()
572577
# [END firestore_query_order_limit]
573578
print(results)
579+
return results
574580

575581

576582
def order_where_valid():
@@ -583,10 +589,11 @@ def order_where_valid():
583589
results = query.stream()
584590
# [END firestore_query_order_with_filter]
585591
print(results)
592+
return results
586593

587594

588595
def order_where_invalid():
589-
db = firestore.Client()
596+
db = firestore.Client(add_unique_string=False) # Flag for testing purposes, needs index to be precreated
590597
# [START firestore_query_order_field_invalid]
591598
cities_ref = db.collection("cities")
592599
query = cities_ref.where(filter=FieldFilter("population", ">", 2500000)).order_by(
@@ -595,6 +602,7 @@ def order_where_invalid():
595602
results = query.stream()
596603
# [END firestore_query_order_field_invalid]
597604
print(results)
605+
return results
598606

599607

600608
def cursor_simple_start_at():
@@ -628,9 +636,6 @@ def snapshot_cursors():
628636
)
629637
# [END firestore_query_cursor_start_at_document]
630638
results = start_at_snapshot.limit(10).stream()
631-
for doc in results:
632-
print(f"{doc.id}")
633-
634639
return results
635640

636641

@@ -790,7 +795,7 @@ def on_snapshot(col_snapshot, changes, read_time):
790795

791796

792797
def cursor_multiple_conditions():
793-
db = firestore.Client()
798+
db = firestore.Client(add_unique_string=False) # Flag for testing purposes, needs index to be precreated
794799
# [START firestore_query_cursor_start_at_field_value_multi]
795800
start_at_name = (
796801
db.collection("cities").order_by("name").start_at({"name": "Springfield"})
@@ -932,7 +937,8 @@ def update_document_increment(db):
932937
def list_document_subcollections():
933938
db = firestore.Client()
934939
# [START firestore_data_get_sub_collections]
935-
collections = db.collection("cities").document("SF").collections()
940+
city_ref = db.collection("cities").document("SF")
941+
collections = city_ref.collections()
936942
for collection in collections:
937943
for doc in collection.stream():
938944
print(f"{doc.id} => {doc.to_dict()}")
@@ -983,3 +989,4 @@ def regional_endpoint():
983989
for r in cities_query:
984990
print(r)
985991
# [END firestore_regional_endpoint]
992+
return cities_query

0 commit comments

Comments
 (0)
0