Add concurrency test

kevinjqiu · kevinjqiu · commit cd148925a46e · 2017-03-06T17:32:20.000-05:00
diff --git a/couchdb/tests/client.py b/couchdb/tests/client.py
@@ -7,6 +7,7 @@
 # you should have received as part of this distribution.
 
 from datetime import datetime
+import functools
 import multiprocessing
 import os
 import os.path
@@ -847,8 +848,33 @@ def test_startkey(self):
     def test_nullkeys(self):
         self.assertEqual(len(list(self.db.iterview('test/nulls', 10))), self.num_docs)
 
+
+def _get_by_id(db, result, id):
+    result.append(db[id])
+
+
+class TestConcurrent(testutil.TempDatabaseMixin, unittest.TestCase):
+    def test_concurrent_get(self):
+        self.db.save({'_id': 'foo', 'value': 'hello'})
+        self.db.save({'_id': 'bar', 'value': 'world'})
+        processes = []
+        result = multiprocessing.Manager().list()
+        for id in ('foo', 'bar'):
+            process = multiprocessing.Process(target=functools.partial(_get_by_id, self.db, result),
+                                              args=(id,))
+            processes.append(process)
+            process.start()
+
+        for process in processes:
+            process.join()
+
+        self.assertEqual(len(result), 2)
+        self.assertEqual(set(['hello', 'world']), set([r['value'] for r in result]))
+
+
 def suite():
     suite = unittest.TestSuite()
+    suite.addTest(unittest.makeSuite(TestConcurrent, 'test'))
     suite.addTest(unittest.makeSuite(ServerTestCase, 'test'))
     suite.addTest(unittest.makeSuite(DatabaseTestCase, 'test'))
     suite.addTest(unittest.makeSuite(ViewTestCase, 'test'))
