summaryrefslogtreecommitdiff
path: root/service/test/support/integration/soledad_test_base.py
blob: 1a2645d1f58dd1b7b7638faea765e17efda674f9 (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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#
# Copyright (c) 2014 ThoughtWorks, Inc.
#
# Pixelated is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Pixelated is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
from uuid import uuid4
from leap.mail.adaptors.soledad import SoledadMailAdaptor
from leap.mail.mail import Message
from twisted.internet import defer
from twisted.trial import unittest
from pixelated.adapter.mailstore import LeapMailStore
from test.support.integration.app_test_client import AppTestClient
from leap.common.events.flags import set_events_enabled


class SoledadTestBase(unittest.TestCase, AppTestClient):
    # these are so long because our CI is so slow at the moment.
    DEFERRED_TIMEOUT = 120
    DEFERRED_TIMEOUT_LONG = 300

    @defer.inlineCallbacks
    def setUp(self):
        set_events_enabled(False)
        super(SoledadTestBase, self).setUp()
        self.adaptor = SoledadMailAdaptor()
        self.mbox_uuid = str(uuid4())
        yield self.start_client()
        self.store = LeapMailStore(self.soledad)

    def tearDown(self):
        set_events_enabled(True)
        self.cleanup()

    @defer.inlineCallbacks
    def _create_mail_in_soledad(self, mail):
        yield self.adaptor.initialize_store(self.soledad)
        mbox = yield self.adaptor.get_or_create_mbox(self.soledad, 'INBOX')
        message = self._convert_mail_to_leap_message(mail, mbox.uuid)
        yield self.adaptor.create_msg(self.soledad, message)

        defer.returnValue(message.get_wrapper().mdoc.doc_id)

    def _convert_mail_to_leap_message(self, mail, mbox_uuid):
        message = self.adaptor.get_msg_from_string(Message, mail.as_string())
        message.get_wrapper().set_mbox_uuid(mbox_uuid)
        return message