aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorGravatar MarkKoz <[email protected]>2020-01-18 09:33:40 -0800
committerGravatar MarkKoz <[email protected]>2020-02-12 10:07:54 -0800
commit8fab4db24939d6d7dd9256c0faf13395e7caddb7 (patch)
tree53981502495bacd043f3a693fb1394dfa006fae3
parentSync tests: adjust sync test to account for _get_confirmation_result (diff)
Sync tests: test diff size calculation
-rw-r--r--tests/bot/cogs/sync/test_base.py22
1 files changed, 22 insertions, 0 deletions
diff --git a/tests/bot/cogs/sync/test_base.py b/tests/bot/cogs/sync/test_base.py
index f82984157..6d784d0de 100644
--- a/tests/bot/cogs/sync/test_base.py
+++ b/tests/bot/cogs/sync/test_base.py
@@ -281,3 +281,25 @@ class SyncerSyncTests(unittest.TestCase):
self.syncer._sync.assert_called_once_with(diff)
else:
self.syncer._sync.assert_not_called()
+
+ def test_sync_diff_size(self):
+ """The diff size should be correctly calculated."""
+ subtests = (
+ (6, _Diff({1, 2}, {3, 4}, {5, 6})),
+ (5, _Diff({1, 2, 3}, None, {4, 5})),
+ (0, _Diff(None, None, None)),
+ (0, _Diff(set(), set(), set())),
+ )
+
+ for size, diff in subtests:
+ with self.subTest(size=size, diff=diff):
+ self.syncer._get_diff.reset_mock()
+ self.syncer._get_diff.return_value = diff
+ self.syncer._get_confirmation_result = helpers.AsyncMock(return_value=(False, None))
+
+ guild = helpers.MockGuild()
+ asyncio.run(self.syncer.sync(guild))
+
+ self.syncer._get_diff.assert_called_once_with(guild)
+ self.syncer._get_confirmation_result.assert_called_once()
+ self.assertEqual(self.syncer._get_confirmation_result.call_args[0][0], size)