From 88ab7ed4ea5debd9bae6d82131b1706c99552b63 Mon Sep 17 00:00:00 2001
From: Prashant Mital <5883388+prashantmital@users.noreply.github.com>
Date: Fri, 10 Aug 2018 15:11:34 -0700
Subject: [PATCH] DOC: add manual example for causal consistency (#368)

* DOC: add manual example for causal consistency
---
 test/test_examples.py | 41 ++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 40 insertions(+), 1 deletion(-)

diff --git a/test/test_examples.py b/test/test_examples.py
index e83b68af8..dae1f9d4e 100644
--- a/test/test_examples.py
+++ b/test/test_examples.py
@@ -14,15 +14,18 @@
 
 """MongoDB documentation examples in Python."""
 
-import threading
+import datetime
 import sys
+import threading
 
 sys.path[0:0] = [""]
 
 import pymongo
 from pymongo.errors import ConnectionFailure, OperationFailure
 from pymongo.read_concern import ReadConcern
+from pymongo.read_preferences import ReadPreference
 from pymongo.write_concern import WriteConcern
+
 from test import client_context, unittest, IntegrationTest
 from test.utils import rs_or_single_client
 
@@ -1114,5 +1117,41 @@ class TestTransactionExamples(IntegrationTest):
         # End Beta Transaction Example 3
 
 
+class TestCausalConsistencyExamples(IntegrationTest):
+    @client_context.require_version_min(3, 6, 0)
+    @client_context.require_secondaries_count(1)
+    def test_causal_consistency(self):
+        # Causal consistency examples
+        client = self.client
+        self.addCleanup(client.drop_database, 'test')
+        client.test.drop_collection('items')
+        client.test.items.insert_one({
+            'sku': "111", 'name': 'Peanuts',
+            'start':datetime.datetime.today()})
+
+        # Start Causal Consistency Example 1
+        with client.start_session(causal_consistency=True) as s1:
+            current_date = datetime.datetime.today()
+            items = client.get_database('test').items
+            items.update_one(
+                {'sku': "111", 'end': None},
+                {'$set': {'end': current_date}}, session=s1)
+            items.insert_one(
+                {'sku': "nuts-111", 'name': "Pecans",
+                 'start': current_date}, session=s1)
+        # End Causal Consistency Example 1
+
+        # Start Causal Consistency Example 2
+        with client.start_session(causal_consistency=True) as s2:
+            s2.advance_cluster_time(s1.cluster_time)
+            s2.advance_operation_time(s1.operation_time)
+
+            items = client.get_database(
+                'test', read_preference=ReadPreference.SECONDARY).items
+            for item in items.find({'end': None}, session=s2):
+                print(item)
+        # End Causal Consistency Example 2
+
+
 if __name__ == "__main__":
     unittest.main()