diff options
Diffstat (limited to 'service/test/support/integration_helper.py')
-rw-r--r-- | service/test/support/integration_helper.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/service/test/support/integration_helper.py b/service/test/support/integration_helper.py index 6ab96e52..36d297d6 100644 --- a/service/test/support/integration_helper.py +++ b/service/test/support/integration_helper.py @@ -183,6 +183,14 @@ class SoledadTestBase: response = json.loads(request.getWrittenData()) return ResponseMail(response) + def get_attachment(self, ident, encoding): + request = requestMock(path='/attachment/' + ident) + request.args = { + 'encoding': encoding + } + _render(self.resource, request) + return request.getWrittenData() + def put_mail(self, data): request = requestMock('/mails', method="PUT", body=data, headers={'Content-Type': ['application/json']}) _render(self.resource, request) @@ -226,6 +234,9 @@ class SoledadTestBase: _render(self.resource, request) return request + def add_document_to_soledad(self, _dict): + self.soledad_querier.soledad.create_doc(_dict) + def add_mail_to_inbox(self, input_mail): mail = self.mailboxes.inbox().add(input_mail) mail.update_tags(input_mail.tags) |