docs: Add collection subcollection examples and tests

Author: HomelessDinosaur

examples/documents/sub_col_query.py

@@ -1,13 +1,13 @@
-# # [START import]
-# from nitric.api import Documents
-# # [END import]
-# async def documents_sub_col_query():
-# # [START snippet]
-#   docs = Documents()
+# [START import]
+from nitric.api import Documents
+# [END import]
+async def documents_sub_col_query():
+# [START snippet]
+  docs = Documents()
 
-#   query = (docs.collection("Customers")
-#           .collection("Orders")
-#           .query())
+  query = (docs.collection("Customers")
+          .collection("Orders")
+          .query())
 
-#   results = await query.fetch()
-# # [END snippet]
+  results = await query.fetch()
+# [END snippet]

tests/examples/test_documents_example.py

@@ -1,3 +1,4 @@
+from examples.documents.sub_col_query import documents_sub_col_query
 from nitric.proto.nitric.document.v1 import Collection, DocumentGetResponse, DocumentQueryStreamResponse, Document, Key
 from examples.documents.set import documents_set
 from examples.documents.get import documents_get
@@ -96,6 +97,14 @@ async def test_sub_doc_query_document(self):
 
         mock_query.assert_called_once()
 
+    async def test_sub_col_query_document(self):
+        mock_query = AsyncMock()
+
+        with patch("nitric.proto.nitric.document.v1.DocumentServiceStub.query", mock_query):
+            await documents_sub_col_query()
+
+        mock_query.assert_called_once()
+
     async def test_streamed_document(self):
         stream_calls = 0
         call_args = {}