feat(firestore): add snipped for IN & ARRAY_CONTAINS_ANY (GoogleCloudPlatform#2526)

bradmiro · BenWhitehead · commit 7b13271a9547 · 2019-11-14T15:30:14.000-05:00
diff --git a/firestore/cloud-client/requirements.txt b/firestore/cloud-client/requirements.txt
@@ -1 +1 @@
-google-cloud-firestore==1.4.0
+google-cloud-firestore==1.6.0
diff --git a/firestore/cloud-client/snippets.py b/firestore/cloud-client/snippets.py
@@ -877,6 +877,37 @@ def collection_group_query(db):
     return docs
 
 
+def array_contains_any_queries(db):
+    # [START fs_query_filter_array_contains_any]
+    cities_ref = db.collection(u'cities')
+
+    query = cities_ref.where(
+        u'regions', u'array_contains_any', [u'west_coast', u'east_coast']
+    )
+    return query
+    # [END fs_query_filter_array_contains_any]
+
+
+def in_query_without_array(db):
+    # [START fs_query_filter_in]
+    cities_ref = db.collection(u'cities')
+
+    query = cities_ref.where(u'country', u'in', [u'USA', u'Japan'])
+    return query
+    # [END fs_query_filter_in]
+
+
+def in_query_with_array(db):
+    # [START fs_query_filter_in_with_array]
+    cities_ref = db.collection(u'cities')
+
+    query = cities_ref.where(
+        u'regions', u'in', [[u'west_coast'], [u'east_coast']]
+    )
+    return query
+    # [END fs_query_filter_in_with_array]
+
+
 def update_document_increment(db):
     # [START fs_update_document_increment]
     washington_ref = db.collection(u'cities').document(u'DC')
diff --git a/firestore/cloud-client/snippets_test.py b/firestore/cloud-client/snippets_test.py
@@ -54,6 +54,33 @@ def test_add_example_data():
     snippets.add_example_data()
 
 
+def test_array_contains_any(db):
+    query = snippets.array_contains_any_queries(db)
+
+    expected = {'SF', 'LA', 'DC'}
+    actual = {document.id for document in query.stream()}
+
+    assert expected == actual
+
+
+def test_query_filter_in_query_without_array(db):
+    query = snippets.in_query_without_array(db)
+
+    expected = {'SF', 'LA', 'DC', 'TOK'}
+    actual = {document.id for document in query.stream()}
+
+    assert expected == actual
+
+
+def test_query_filter_in_query_with_array(db):
+    query = snippets.in_query_with_array(db)
+
+    expected = {'DC'}
+    actual = {document.id for document in query.stream()}
+
+    assert expected == actual
+
+
 def test_add_custom_class_with_id():
     snippets.add_custom_class_with_id()
 

Original file line number	Diff line number	Diff line change
`@@ -1 +1 @@`
`1`		`-google-cloud-firestore==1.4.0`
	`1`	`+google-cloud-firestore==1.6.0`