summaryrefslogtreecommitdiff
path: root/service/test/unit/bitmask_libraries/test_session.py
blob: 22c2b3d93ff6305be30ec8f8520b57d06eaef7f7 (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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
#
# 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 mock import patch
from mock import MagicMock

from pixelated.bitmask_libraries.session import LeapSession, SessionCache
from test_abstract_leap import AbstractLeapTest
from leap.common.events.catalog import KEYMANAGER_FINISHED_KEY_GENERATION


class SessionTest(AbstractLeapTest):

    def setUp(self):
        super(SessionTest, self).setUp()
        self.smtp_mock = MagicMock()

    def test_background_jobs_are_started_during_initial_sync(self):
        with patch('pixelated.bitmask_libraries.session.reactor.callFromThread', new=_execute_func) as _:
            with patch('pixelated.bitmask_libraries.session.LeapSession._create_incoming_mail_fetcher') as mail_fetcher_mock:
                session = self._create_session()
                yield session.initial_sync()
                mail_fetcher_mock.startService.assert_called_once()

    def test_that_close_stops_background_jobs(self):
        with patch('pixelated.bitmask_libraries.session.reactor.callFromThread', new=_execute_func) as _:
            with patch('pixelated.bitmask_libraries.session.LeapSession._create_incoming_mail_fetcher') as mail_fetcher_mock:
                session = self._create_session()
                yield session.initial_sync()
                session.close()
                mail_fetcher_mock.stopService.assert_called_once()

    def test_that_sync_deferes_to_soledad(self):
        with patch('pixelated.bitmask_libraries.session.reactor.callFromThread', new=_execute_func) as _:
            with patch('pixelated.bitmask_libraries.session.LeapSession._create_incoming_mail_fetcher') as mail_fetcher_mock:
                session = self._create_session()
                yield session.sync()
                self.soledad_session.sync.assert_called_once()

    def test_session_registers_to_generated_keys(self):
        email = 'someone@somedomain.tld'
        self.provider.address_for.return_value = email
        with patch('pixelated.bitmask_libraries.session.register') as register_mock:
            session = self._create_session()

            register_mock.assert_called_once_with(KEYMANAGER_FINISHED_KEY_GENERATION, session._set_fresh_account, uid=email)

    @patch('pixelated.bitmask_libraries.session.register')
    def test_close_unregisters_from_generate_keys_events(self, _):
        email = 'someone@somedomain.tld'
        self.provider.address_for.return_value = email
        session = self._create_session()

        with patch('pixelated.bitmask_libraries.session.unregister') as unregister_mock:
            session.close()

            unregister_mock.assert_called_once_with(KEYMANAGER_FINISHED_KEY_GENERATION, uid=email)

    @patch('pixelated.bitmask_libraries.session.register')
    def test_close_stops_soledad(self, _):
        email = 'someone@somedomain.tld'
        self.provider.address_for.return_value = email
        session = self._create_session()

        with patch('pixelated.bitmask_libraries.session.unregister') as unregister_mock:
            session.close()

        self.soledad_session.close.assert_called_once_with()

    @patch('pixelated.bitmask_libraries.session.register')
    def test_close_removes_session_from_cache(self, _):
        email = 'someone@somedomain.tld'
        self.provider.address_for.return_value = email
        session = self._create_session()

        key = SessionCache.session_key(self.provider, self.auth.username)
        SessionCache.remember_session(key, session)

        self.assertEqual(session, SessionCache.lookup_session(key))

        with patch('pixelated.bitmask_libraries.session.unregister') as unregister_mock:
            session.close()

        self.assertIsNone(SessionCache.lookup_session(key))

    @patch('pixelated.bitmask_libraries.session.register')
    def test_close_ends_account_session(self, _):
        account_mock = MagicMock()
        email = 'someone@somedomain.tld'
        self.provider.address_for.return_value = email
        session = self._create_session()
        session.account = account_mock

        with patch('pixelated.bitmask_libraries.session.unregister') as unregister_mock:
            session.close()

        account_mock.end_session.assert_called_once_with()

    @patch('pixelated.bitmask_libraries.session.register')
    def test_session_fresh_is_initially_false(self, _):
        session = self._create_session()

        self.assertFalse(session.fresh_account)

    @patch('pixelated.bitmask_libraries.session.register')
    def test_session_sets_status_to_fresh_on_key_generation_event(self, _):
        session = self._create_session()
        self.provider.address_for.return_value = 'someone@somedomain.tld'

        session._set_fresh_account(None, 'someone@somedomain.tld')

        self.assertTrue(session.fresh_account)

    @patch('pixelated.bitmask_libraries.session.register')
    def test_session_does_not_set_status_fresh_for_unkown_emails(self, _):
        session = self._create_session()
        self.provider.address_for.return_value = 'someone@somedomain.tld'

        session._set_fresh_account(None, 'another_email@somedomain.tld')

        self.assertFalse(session.fresh_account)

    def _create_session(self):
        return LeapSession(self.provider, self.auth, self.mail_store, self.soledad_session, self.nicknym, self.smtp_mock)


def _execute_func(func):
    func()