Add role and user differ unit tests.

5 files changed, 125 insertions, 0 deletions

diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 000000000..e69de29bb
--- /dev/null
+++ b/tests/__init__.py
diff --git a/tests/cogs/__init__.py b/tests/cogs/__init__.py
new file mode 100644
index 000000000..e69de29bb
--- /dev/null
+++ b/tests/cogs/__init__.py
diff --git a/tests/cogs/sync/__init__.py b/tests/cogs/sync/__init__.py
new file mode 100644
index 000000000..e69de29bb
--- /dev/null
+++ b/tests/cogs/sync/__init__.py
diff --git a/tests/cogs/sync/test_roles.py b/tests/cogs/sync/test_roles.py
new file mode 100644
index 000000000..7def815cc
--- /dev/null
+++ b/tests/cogs/sync/test_roles.py
@@ -0,0 +1,64 @@
+from bot.cogs.sync.syncers import get_roles_for_sync, Role
+
+
+def test_get_roles_for_sync_empty_return_for_equal_roles():
+    api_roles = {Role(id=41, name='name', colour=33, permissions=0x8)}
+    guild_roles = {Role(id=41, name='name', colour=33, permissions=0x8)}
+
+    assert get_roles_for_sync(guild_roles, api_roles) == (set(), set())
+
+
+def test_get_roles_for_sync_returns_roles_to_update_with_non_id_diff():
+    api_roles = {Role(id=41, name='old name', colour=35, permissions=0x8)}
+    guild_roles = {Role(id=41, name='new name', colour=33, permissions=0x8)}
+
+    assert get_roles_for_sync(guild_roles, api_roles) == (
+        set(),
+        guild_roles
+    )
+
+
+def test_get_roles_only_returns_roles_that_require_update():
+    api_roles = {
+        Role(id=41, name='old name', colour=33, permissions=0x8),
+        Role(id=53, name='other role', colour=55, permissions=0)
+    }
+    guild_roles = {
+        Role(id=41, name='new name', colour=35, permissions=0x8),
+        Role(id=53, name='other role', colour=55, permissions=0)
+    }
+
+    assert get_roles_for_sync(guild_roles, api_roles) == (
+        set(),
+        {Role(id=41, name='new name', colour=35, permissions=0x8)},
+    )
+
+
+def test_get_roles_returns_new_roles_in_first_tuple_element():
+    api_roles = {
+        Role(id=41, name='name', colour=35, permissions=0x8),
+    }
+    guild_roles = {
+        Role(id=41, name='name', colour=35, permissions=0x8),
+        Role(id=53, name='other role', colour=55, permissions=0)
+    }
+
+    assert get_roles_for_sync(guild_roles, api_roles) == (
+        {Role(id=53, name='other role', colour=55, permissions=0)},
+        set()
+    )
+
+
+def test_get_roles_returns_roles_to_update_and_new_roles():
+    api_roles = {
+        Role(id=41, name='old name', colour=35, permissions=0x8),
+    }
+    guild_roles = {
+        Role(id=41, name='new name', colour=40, permissions=0x16),
+        Role(id=53, name='other role', colour=55, permissions=0)
+    }
+
+    assert get_roles_for_sync(guild_roles, api_roles) == (
+        {Role(id=53, name='other role', colour=55, permissions=0)},
+        {Role(id=41, name='new name', colour=40, permissions=0x16)}
+    )
diff --git a/tests/cogs/sync/test_users.py b/tests/cogs/sync/test_users.py
new file mode 100644
index 000000000..6869f89c8
--- /dev/null
+++ b/tests/cogs/sync/test_users.py
@@ -0,0 +1,61 @@
+from bot.cogs.sync.syncers import User, get_users_for_sync
+
+def fake_user(**kwargs):
+    kwargs.setdefault('id', 43)
+    kwargs.setdefault('name', 'bob the test man')
+    kwargs.setdefault('discriminator', 1337)
+    kwargs.setdefault('avatar_hash', None)
+    kwargs.setdefault('roles', (666,))
+    kwargs.setdefault('in_guild', True)
+    return User(**kwargs)
+
+
+def test_get_users_for_sync_returns_nothing_for_empty_params():
+    assert get_users_for_sync({}, {}) == (set(), set())
+
+
+def test_get_users_for_sync_returns_nothing_for_equal_users():
+    api_users = {43: fake_user()}
+    guild_users = {43: fake_user()}
+
+    assert get_users_for_sync(guild_users, api_users) == (set(), set())
+
+
+def test_get_users_for_sync_returns_users_to_update_on_non_id_field_diff():
+    api_users = {43: fake_user()}
+    guild_users = {43: fake_user(name='new fancy name')}
+
+    assert get_users_for_sync(guild_users, api_users) == (
+        set(),
+        {fake_user(name='new fancy name')}
+    )
+
+
+def test_get_users_for_sync_returns_users_to_create_with_new_ids_on_guild():
+    api_users = {43: fake_user()}
+    guild_users = {43: fake_user(), 63: fake_user(id=63)}
+
+    assert get_users_for_sync(guild_users, api_users) == (
+        {fake_user(id=63)},
+        set()
+    )
+
+
+def test_get_users_for_sync_updates_in_guild_field_on_user_leave():
+    api_users = {43: fake_user(), 63: fake_user(id=63)}
+    guild_users = {43: fake_user()}
+
+    assert get_users_for_sync(guild_users, api_users) == (
+        set(),
+        {fake_user(id=63, in_guild=False)}
+    )
+
+
+def test_get_users_for_sync_updates_and_creates_users_as_needed():
+    api_users = {43: fake_user()}
+    guild_users = {63: fake_user(id=63)}
+
+    assert get_users_for_sync(guild_users, api_users) == (
+        {fake_user(id=63)},
+        {fake_user(in_guild=False)}
+    )