Merge pull request #3 from marklogic/feature/test-get-documents

Added simple tests for getting many documents back at once

Author: rjrudin

test-app/src/main/ml-data/collections.properties

@@ -0,0 +1 @@
+*=test-data

test-app/src/main/ml-data/doc1.json

@@ -0,0 +1,3 @@
+{
+    "hello": "world"
+}

test-app/src/main/ml-data/doc2.xml

@@ -0,0 +1 @@
+<hello>world</hello>

test-app/src/main/ml-data/permissions.properties

@@ -0,0 +1 @@
+*=rest-reader,read,rest-writer,update

tests/test_get_documents.py

@@ -0,0 +1,46 @@
+from requests_toolbelt.multipart.decoder import MultipartDecoder
+
+
+def test_get_docs(test_session):
+    """
+    Possible future client interface:
+    array_of_documents = client.documents.get(uri=[], metadata=True)
+
+    Where each Document in the array would have fields of:
+    uri/content/collections/permissions/quality/properties/metadata_values.
+    """
+    response = test_session.get(
+        "http://localhost:8030/v1/documents",
+        params={
+            "uri": ["/doc1.json", "/doc2.xml"],
+            "category": ["content", "metadata"],
+            "format": "json",  # Applies only to metadata
+        },
+        headers={"Accept": "multipart/mixed"},
+    )
+
+    # Could provide a class for converting a multipart/mixed response into an array
+    # of documents too:
+    # from marklogic import DocumentDecoder
+    # array_of_documents = DocumentDecoder.from_response(response)
+
+    decoder = MultipartDecoder.from_response(response)
+    for part in decoder.parts:
+        print(part.headers)
+        print(part.text)
+
+
+def test_search_docs(test_session):
+    response = test_session.get(
+        "http://localhost:8030/v1/search",
+        params={
+            "collection": "test-data",
+            "category": ["content", "metadata"],
+            "format": "json",  # Applies only to metadata
+        },
+        headers={"Accept": "multipart/mixed"},  # Indicates we want documents back.
+    )
+
+    for part in MultipartDecoder.from_response(response).parts:
+        print(part.headers)
+        print(part.text)