Add simple encoding test

2 files changed, 37 insertions, 1 deletions

diff --git a/src/leap/soledad/leap.py b/src/leap/soledad/leap.py
index c9243587..41bcf15a 100644
--- a/src/leap/soledad/leap.py
+++ b/src/leap/soledad/leap.py
@@ -18,7 +18,7 @@ class LeapDocument(Document):
 
     def __init__(self, doc_id=None, rev=None, json='{}', has_conflicts=False,
                  encrypted_json=None, default_key=None):
-        super(Document, self).__init__(doc_id, rev, json, has_conflicts)
+        super(LeapDocument, self).__init__(doc_id, rev, json, has_conflicts)
         if encrypted_json:
             self.set_encrypted_json(encrypted_json)
         self._default_key = default_key
diff --git a/src/leap/soledad/tests/__init__.py b/src/leap/soledad/tests/__init__.py
new file mode 100644
index 00000000..88cddef0
--- /dev/null
+++ b/src/leap/soledad/tests/__init__.py
@@ -0,0 +1,36 @@
+try:
+    import simplejson as json
+except ImportError:
+    import json  # noqa
+
+import unittest
+import os
+
+import u1db
+from soledad import leap
+
+class EncryptedSyncTestCase(unittest.TestCase):
+
+    PREFIX = '/var/tmp'
+    db1_path = "%s/db1.u1db" % PREFIX
+    db2_path = "%s/db2.u1db" % PREFIX
+
+    def setUp(self):
+        self.db1 = u1db.open(self.db1_path, create=True,
+                             document_factory=leap.LeapDocument)
+        self.db2 = u1db.open(self.db2_path, create=True,
+                             document_factory=leap.LeapDocument)
+
+    def tearDown(self):
+        os.unlink(self.db1_path)
+        os.unlink(self.db2_path)
+
+    def test_encoding(self):
+        doc1 = self.db1.create_doc({ 'key' : 'val' })
+        enc1 = doc1.get_encrypted_json()
+        doc2 = leap.LeapDocument(doc_id=doc1.doc_id, json=doc1.get_json())
+        enc2 = doc2.get_encrypted_json()
+        self.assertEqual(enc1, enc2, 'incorrect document encoding')
+
+if __name__ == '__main__':
+    unittest.main()