diff options
Diffstat (limited to 'tests/server/test_incoming_server.py')
-rw-r--r-- | tests/server/test_incoming_server.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/server/test_incoming_server.py b/tests/server/test_incoming_server.py index 23c0aa90..16d5d5e6 100644 --- a/tests/server/test_incoming_server.py +++ b/tests/server/test_incoming_server.py @@ -18,13 +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 +from twisted.web.test.requesthelper import DummyRequest from leap.soledad.server._incoming import IncomingResource from leap.soledad.server._blobs import BlobsServerState @@ -83,10 +83,10 @@ class IncomingOnCouchServerTestCase(CouchDBTestCase): yield treq.put(incoming_endpoint, BytesIO(content), persistent=False) db = self.state.open_database(user_id) - consumer = mock.Mock() + consumer = DummyRequest(['']) yield db.read_blob(user_id, doc_id, consumer, namespace='MX') flags = yield db.get_flags(user_id, doc_id, namespace='MX') - data = consumer.write.call_args[0][0] + data = consumer.written.pop() expected_preamble = formatter.preamble(content, doc_id) expected_preamble = decode_preamble(expected_preamble, True) written_preamble, written_content = data.split() |