Add very basic replication support to client.Server.

djc · djc · commit 55efa5ca878e · 2009-12-08T17:09:49.000+01:00
diff --git a/couchdb/client.py b/couchdb/client.py
@@ -209,6 +209,16 @@ def delete(self, name):
         """
         del self[name]
 
+    def replicate(self, source, target):
+        """Replicate changes from the source database to the target database.
+
+        :param source: URL of the source database
+        :param target: URL of the target database
+        """
+        data = {'source': source, 'target': target}
+        resp, data = self.resource.post('/_replicate', data)
+        return data
+
 
 class Database(object):
     """Representation of a database on a CouchDB server.
diff --git a/couchdb/tests/client.py b/couchdb/tests/client.py
@@ -25,6 +25,10 @@ def tearDown(self):
             self.server.delete('python-tests')
         except client.ResourceNotFound:
             pass
+        try:
+            self.server.delete('python-tests-a')
+        except client.ResourceNotFound:
+            pass
 
     def test_server_vars(self):
         version = self.server.version
@@ -49,6 +53,20 @@ def test_delete_db_missing(self):
         self.assertRaises(client.ResourceNotFound, self.server.delete,
                           'python-tests')
 
+    def test_replicate(self):
+        a = self.server.create('python-tests')
+        id = a.create({'test': 'a'})
+        b = self.server.create('python-tests-a')
+        result = self.server.replicate('python-tests', 'python-tests-a')
+        self.assertEquals(result['ok'], True)
+        self.assertEquals(b[id]['test'], 'a')
+
+        doc = b[id]
+        doc['test'] = 'b'
+        b.update([doc])
+        self.server.replicate(client.DEFAULT_BASE_URI + 'python-tests-a',
+                              'python-tests')
+        self.assertEquals(b[id]['test'], 'b')
 
 class DatabaseTestCase(unittest.TestCase):
 
