summaryrefslogtreecommitdiff
path: root/service/test/unit/adapter/listener_test.py
blob: 284c6b16d80def0c5a346089a29cce6dc95c2d3a (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
#
# 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/>.
import unittest

from mockito import *
import pixelated.adapter.soledad_querier

querier = mock()
when(pixelated.adapter.soledad_querier).get_soledad_querier_instance().thenReturn(querier)

from pixelated.adapter.listener import MailboxListener


class MailboxListenerTest(unittest.TestCase):
    def setUp(self):
        self.account = mock()
        self.account.mailboxes = []

    def test_add_itself_to_mailbox_listeners(self):
        self.account.mailboxes = ['INBOX']
        mailbox = mock()
        when(self.account).getMailbox('INBOX').thenReturn(mailbox)
        mailbox.listeners = set()
        when(mailbox).addListener = lambda x: mailbox.listeners.add(x)

        self.assertNotIn(MailboxListener('INBOX'), mailbox.listeners)

        MailboxListener.listen(self.account, 'INBOX')

        self.assertIn(MailboxListener('INBOX'), mailbox.listeners)

    def test_reindex_missing_idents(self):
        search_engine = mock()
        when(search_engine).search('tag:inbox').thenReturn(['ident1', 'ident2'])

        MailboxListener.SEARCH_ENGINE = search_engine

        listener = MailboxListener('INBOX')
        listener.querier = querier
        when(querier).idents_by_mailbox('INBOX').thenReturn({'ident1', 'ident2', 'missing_ident'})
        querier.used_arguments = []
        querier.mails = lambda x: querier.used_arguments.append(x)
        listener.newMessages(10, 5)

        verify(querier, times=1).idents_by_mailbox('INBOX')
        self.assertIn({'missing_ident'}, querier.used_arguments)