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
|
########################################################################
# File name: test_utils.py
# This file is part of: aioxmpp
#
# LICENSE
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program 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
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this program. If not, see
# <http://www.gnu.org/licenses/>.
#
########################################################################
import unittest
import unittest.mock
from aioxmpp.chatstates import (ChatState, ChatStateManager, # NOQA
DoNotEmit, AlwaysEmit, DiscoverSupport)
class TestStrategies(unittest.TestCase):
def test_DoNotEmit(self):
strategy = DoNotEmit()
self.assertFalse(strategy.sending)
strategy.reset()
self.assertFalse(strategy.sending)
strategy.no_reply()
self.assertFalse(strategy.sending)
def test_DiscoverSupport(self):
strategy = DiscoverSupport()
self.assertTrue(strategy.sending)
strategy.no_reply()
self.assertFalse(strategy.sending)
strategy.reset()
self.assertTrue(strategy.sending)
def test_AlwaysEmit(self):
strategy = AlwaysEmit()
self.assertTrue(strategy.sending)
strategy.reset()
self.assertTrue(strategy.sending)
strategy.no_reply()
self.assertTrue(strategy.sending)
class TestChatStateManager(unittest.TestCase):
def test_no_reply(self):
manager = ChatStateManager(unittest.mock.Mock())
manager.no_reply()
manager._strategy.no_reply.assert_called_once_with()
def test_reset(self):
manager = ChatStateManager(unittest.mock.Mock())
manager.reset()
manager._strategy.reset.assert_called_once_with()
def test_handle(self):
manager = ChatStateManager(unittest.mock.sentinel)
self.assertIs(manager.handle(ChatState.ACTIVE), False)
self.assertIs(manager.handle(ChatState.INACTIVE),
unittest.mock.sentinel.sending)
self.assertIs(manager.handle(ChatState.INACTIVE), False)
self.assertIs(manager.handle(ChatState.ACTIVE),
unittest.mock.sentinel.sending)
self.assertIs(manager.handle(ChatState.ACTIVE, message=True),
unittest.mock.sentinel.sending)
def test_handle_message_other_than_active_raises(self):
manager = ChatStateManager(unittest.mock.sentinel)
for state in ChatState:
if state != ChatState.ACTIVE:
with self.assertRaises(ValueError):
manager.handle(state, message=True)
|
