Card #30 - migrates attachments controller to twisted

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)