diff options
| author | 2020-01-02 19:31:08 -0800 | |
|---|---|---|
| committer | 2020-02-12 10:07:48 -0800 | |
| commit | 7a8c71b7cd5b446188b053aef139255af7bf0154 (patch) | |
| tree | a31dd85e87bd9010c28a119f7288ff30e678ec5d | |
| parent | Sync tests: rename user sync test case (diff) | |
Sync tests: add fixture to get a guild with members
| -rw-r--r-- | tests/bot/cogs/sync/test_users.py | 20 |
1 files changed, 19 insertions, 1 deletions
diff --git a/tests/bot/cogs/sync/test_users.py b/tests/bot/cogs/sync/test_users.py index 509b703ae..83a9cdaf0 100644 --- a/tests/bot/cogs/sync/test_users.py +++ b/tests/bot/cogs/sync/test_users.py @@ -1,6 +1,7 @@ import unittest -from bot.cogs.sync.syncers import User, get_users_for_sync +from bot.cogs.sync.syncers import UserSyncer +from tests import helpers def fake_user(**kwargs): @@ -16,6 +17,23 @@ def fake_user(**kwargs): class UserSyncerDiffTests(unittest.TestCase): """Tests for determining differences between users in the DB and users in the Guild cache.""" + def setUp(self): + self.bot = helpers.MockBot() + self.syncer = UserSyncer(self.bot) + + @staticmethod + def get_guild(*members): + """Fixture to return a guild object with the given members.""" + guild = helpers.MockGuild() + guild.members = [] + + for member in members: + roles = (helpers.MockRole(id=role_id) for role_id in member.pop("roles")) + mock_member = helpers.MockMember(roles, **member) + guild.members.append(mock_member) + + return guild + def test_get_users_for_sync_returns_nothing_for_empty_params(self): """When no users are given, none are returned.""" self.assertEqual( |