blob: b045e5246d25864ff91fef949dc8739c5af9bf4b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
from leap.soledad.common.document import SoledadDocument
from leap.soledad.client.shared_db import SoledadSharedDatabase
from test_soledad.util import BaseSoledadTest
class SoledadSharedDBTestCase(BaseSoledadTest):
"""
These tests ensure the functionalities of the shared recovery database.
"""
def setUp(self):
BaseSoledadTest.setUp(self)
self._shared_db = SoledadSharedDatabase(
'https://provider/', document_factory=SoledadDocument,
creds=None)
def tearDown(self):
BaseSoledadTest.tearDown(self)
def test__get_remote_doc(self):
"""
Ensure the shared db is queried with the correct doc_id.
"""
doc_id = self._soledad.secrets.storage._remote_doc_id()
self._soledad.secrets.storage._get_remote_doc()
self._soledad.secrets.storage._shared_db.get_doc.assert_called_with(
doc_id)
def test_save_remote(self):
"""
Ensure recovery document is put into shared recover db.
"""
doc_id = self._soledad.secrets.storage._remote_doc_id()
storage = self._soledad.secrets.storage
storage.save_remote({'content': 'blah'})
storage._shared_db.get_doc.assert_called_with(doc_id)
storage._shared_db.put_doc.assert_called_with(self._doc_put)
self.assertTrue(self._doc_put.doc_id == doc_id)
|
