added tests

Author: daniel-sanche

tests/unit/v1/test_async_query.py

@@ -826,3 +826,18 @@ async def test_asynccollectiongroup_get_partitions_w_offset():
     query = _make_async_collection_group(parent).offset(10)
     with pytest.raises(ValueError):
         [i async for i in query.get_partitions(2)]
+
+def test_asyncquery_collection_pipeline_type():
+    from google.cloud.firestore_v1.async_pipeline import AsyncPipeline
+    client = make_async_client()
+    parent = client.collection("test")
+    query = parent._query()
+    ppl = query.pipeline()
+    assert isinstance(ppl, AsyncPipeline)
+
+def test_asyncquery_collectiongroup_pipeline_type():
+    from google.cloud.firestore_v1.async_pipeline import AsyncPipeline
+    client = make_async_client()
+    query = client.collection_group("test")
+    ppl = query.pipeline()
+    assert isinstance(ppl, AsyncPipeline)

tests/unit/v1/test_query.py

@@ -949,3 +949,18 @@ def test_collection_group_get_partitions_w_offset(database):
     query = _make_collection_group(parent).offset(10)
     with pytest.raises(ValueError):
         list(query.get_partitions(2))
+
+def test_asyncquery_collection_pipeline_type():
+    from google.cloud.firestore_v1.pipeline import Pipeline
+    client = make_client()
+    parent = client.collection("test")
+    query = parent._query()
+    ppl = query.pipeline()
+    assert isinstance(ppl, Pipeline)
+
+def test_asyncquery_collectiongroup_pipeline_type():
+    from google.cloud.firestore_v1.pipeline import Pipeline
+    client = make_client()
+    query = client.collection_group("test")
+    ppl = query.pipeline()
+    assert isinstance(ppl, Pipeline)