diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/__init__.py | 0 | ||||
| -rw-r--r-- | tests/cogs/__init__.py | 0 | ||||
| -rw-r--r-- | tests/cogs/sync/__init__.py | 0 | ||||
| -rw-r--r-- | tests/cogs/sync/test_roles.py | 64 | ||||
| -rw-r--r-- | tests/cogs/sync/test_users.py | 61 | 
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)} +    ) | 
