blob: ec66c795d5a7475b6ba174016718c50b39688821 (
plain) (
blame)
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
|
import unittest
from unittest import mock
from bot.cogs import sync
from bot.cogs.sync.syncers import Syncer
from tests import helpers
class MockSyncer(helpers.CustomMockMixin, mock.MagicMock):
"""
A MagicMock subclass to mock Syncer objects.
Instances of this class will follow the specifications of `bot.cogs.sync.syncers.Syncer`
instances. For more information, see the `MockGuild` docstring.
"""
def __init__(self, **kwargs) -> None:
super().__init__(spec_set=Syncer, **kwargs)
class SyncExtensionTests(unittest.TestCase):
"""Tests for the sync extension."""
@staticmethod
def test_extension_setup():
"""The Sync cog should be added."""
bot = helpers.MockBot()
sync.setup(bot)
bot.add_cog.assert_called_once()
class SyncCogTests(unittest.TestCase):
"""Tests for the Sync cog."""
def setUp(self):
self.bot = helpers.MockBot()
# These patch the type. When the type is called, a MockSyncer instanced is returned.
# MockSyncer is needed so that our custom AsyncMock is used.
# TODO: Use autospec instead in 3.8, which will automatically use AsyncMock when needed.
self.role_syncer_patcher = mock.patch(
"bot.cogs.sync.syncers.RoleSyncer",
new=mock.MagicMock(return_value=MockSyncer())
)
self.user_syncer_patcher = mock.patch(
"bot.cogs.sync.syncers.UserSyncer",
new=mock.MagicMock(return_value=MockSyncer())
)
self.RoleSyncer = self.role_syncer_patcher.start()
self.UserSyncer = self.user_syncer_patcher.start()
self.cog = sync.Sync(self.bot)
def tearDown(self):
self.role_syncer_patcher.stop()
self.user_syncer_patcher.stop()
@mock.patch.object(sync.Sync, "sync_guild")
def test_sync_cog_init(self, sync_guild):
"""Should instantiate syncers and run a sync for the guild."""
# Reset because a Sync cog was already instantiated in setUp.
self.RoleSyncer.reset_mock()
self.UserSyncer.reset_mock()
self.bot.loop.create_task.reset_mock()
mock_sync_guild_coro = mock.MagicMock()
sync_guild.return_value = mock_sync_guild_coro
sync.Sync(self.bot)
self.RoleSyncer.assert_called_once_with(self.bot)
self.UserSyncer.assert_called_once_with(self.bot)
sync_guild.assert_called_once_with()
self.bot.loop.create_task.assert_called_once_with(mock_sync_guild_coro)
|
