diff options
| author | drebs <drebs@leap.se> | 2017-12-12 20:11:13 -0200 |
|---|---|---|
| committer | drebs <drebs@leap.se> | 2017-12-13 13:52:30 -0200 |
| commit | 383a19aaf87bd8e2665112c1350627140958eedf (patch) | |
| tree | 402ba69a04158e6293ca3533d82b5515bba3c433 /tests/server | |
| parent | cc480a2e6c11856759a5297e94a0ff128d0a1593 (diff) | |
[refactor] use producer/consumer on write/read_blob respectivelly
Diffstat (limited to 'tests/server')
| -rw-r--r-- | tests/server/test_incoming_server.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/tests/server/test_incoming_server.py b/tests/server/test_incoming_server.py index f278be74..23c0aa90 100644 --- a/tests/server/test_incoming_server.py +++ b/tests/server/test_incoming_server.py @@ -18,12 +18,13 @@ Integration tests for incoming API """ import pytest +import mock +import treq from io import BytesIO from uuid import uuid4 from twisted.web.server import Site from twisted.internet import reactor from twisted.internet import defer -import treq from leap.soledad.server._incoming import IncomingResource from leap.soledad.server._blobs import BlobsServerState @@ -82,9 +83,10 @@ class IncomingOnCouchServerTestCase(CouchDBTestCase): yield treq.put(incoming_endpoint, BytesIO(content), persistent=False) db = self.state.open_database(user_id) - fd = yield db.read_blob(user_id, doc_id, namespace='MX') + consumer = mock.Mock() + yield db.read_blob(user_id, doc_id, consumer, namespace='MX') flags = yield db.get_flags(user_id, doc_id, namespace='MX') - data = fd.read() + data = consumer.write.call_args[0][0] expected_preamble = formatter.preamble(content, doc_id) expected_preamble = decode_preamble(expected_preamble, True) written_preamble, written_content = data.split() |