diff options
| -rw-r--r-- | common/src/leap/soledad/common/tests/test_encdecpool.py | 44 |
1 files changed, 40 insertions, 4 deletions
diff --git a/common/src/leap/soledad/common/tests/test_encdecpool.py b/common/src/leap/soledad/common/tests/test_encdecpool.py index 0f12a526..7dfa2bbe 100644 --- a/common/src/leap/soledad/common/tests/test_encdecpool.py +++ b/common/src/leap/soledad/common/tests/test_encdecpool.py @@ -20,6 +20,7 @@ Tests for encryption and decryption pool. import json from twisted.internet.defer import inlineCallbacks +from twisted.trial.unittest import TestCase from leap.soledad.client.encdecpool import SyncEncrypterPool from leap.soledad.client.encdecpool import SyncDecrypterPool @@ -32,7 +33,7 @@ DOC_ID = "mydoc" DOC_REV = "rev" DOC_CONTENT = {'simple': 'document'} -class TestSyncEncrypterPool(BaseSoledadTest): +class TestSyncEncrypterPool(TestCase, BaseSoledadTest): def setUp(self): BaseSoledadTest.setUp(self) @@ -66,7 +67,7 @@ class TestSyncEncrypterPool(BaseSoledadTest): self.assertIsNotNone(encrypted) self.assertTrue(attempts < 10) -class TestSyncDecrypterPool(BaseSoledadTest): +class TestSyncDecrypterPool(TestCase, BaseSoledadTest): def _insert_doc_cb(self, doc, gen, trans_id): self._inserted_docs.append((doc, gen, trans_id)) @@ -79,15 +80,16 @@ class TestSyncDecrypterPool(BaseSoledadTest): self._soledad._sync_db, source_replica_uid=self._soledad._dbpool.replica_uid, insert_doc_cb=self._insert_doc_cb) - self._pool.start(1) # reset the inserted docs mock self._inserted_docs = [] def tearDown(self): - self._pool.stop() + if self._pool.running: + self._pool.stop() BaseSoledadTest.tearDown(self) def test_insert_received_doc(self): + self._pool.start(1) self._pool.insert_received_doc( DOC_ID, DOC_REV, "{}", 1, "trans_id", 1) @@ -98,3 +100,37 @@ class TestSyncDecrypterPool(BaseSoledadTest): self._pool.deferred.addCallback(_assert_doc_was_inserted) return self._pool.deferred + + def test_insert_received_doc_many(self): + many = 100 + self._pool.start(many) + + for i in xrange(many): + gen = idx = i + 1 + doc_id = "doc_id: %d" % idx + rev = "rev: %d" % idx + content = {'idx': idx} + trans_id = "trans_id: %d" % idx + self._pool.insert_received_doc( + doc_id, rev, content, gen, trans_id, idx) + + def _assert_doc_was_inserted(_): + self.assertEqual(many, len(self._inserted_docs)) + idx = 1 + for doc, gen, trans_id in self._inserted_docs: + expected_gen = idx + expected_doc_id = "doc_id: %d" % idx + expected_rev = "rev: %d" % idx + expected_content = json.dumps({'idx': idx}) + expected_trans_id = "trans_id: %d" % idx + + self.assertEqual(expected_doc_id, doc.doc_id) + self.assertEqual(expected_rev, doc.rev) + self.assertEqual(expected_content, json.dumps(doc.content)) + self.assertEqual(expected_gen, gen) + self.assertEqual(expected_trans_id, trans_id) + + idx += 1 + + self._pool.deferred.addCallback(_assert_doc_was_inserted) + return self._pool.deferred |