Dome
b2541c4054
feat: add configurable matrix permalink base, unify user placeholders, and refactor notification rendering
224 lines
8.5 KiB
Python
Executable File
224 lines
8.5 KiB
Python
Executable File
"""Tests for user utility functions."""
|
|
|
|
import pytest
|
|
from unittest.mock import Mock, AsyncMock, patch
|
|
from mautrix.types import EventType, UserID
|
|
from mautrix.errors import MNotFound
|
|
|
|
from community.helpers.user_utils import (
|
|
check_if_banned, get_banlist_roomids, ban_user_from_rooms, user_permitted
|
|
)
|
|
|
|
|
|
class TestUserUtils:
|
|
"""Test cases for user utility functions."""
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_check_if_banned_success(self):
|
|
"""Test successful ban check."""
|
|
client = Mock()
|
|
client.get_joined_rooms = AsyncMock(return_value=["!room1:example.com", "!room2:example.com"])
|
|
|
|
# Mock state events
|
|
ban_rule = Mock()
|
|
ban_rule.type.t = "m.policy.rule.user"
|
|
ban_rule.content = {
|
|
"entity": "@banned:example.com",
|
|
"recommendation": "ban"
|
|
}
|
|
|
|
client.get_state = AsyncMock(return_value=[ban_rule])
|
|
|
|
# Mock get_banlist_roomids to return the room ID directly
|
|
with patch('community.helpers.user_utils.get_banlist_roomids') as mock_get_banlists:
|
|
mock_get_banlists.return_value = ["!room1:example.com"]
|
|
|
|
logger = Mock()
|
|
result = await check_if_banned(client, "@banned:example.com", ["!room1:example.com"], logger)
|
|
|
|
assert result == True
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_check_if_banned_not_banned(self):
|
|
"""Test ban check when user is not banned."""
|
|
client = Mock()
|
|
client.get_joined_rooms = AsyncMock(return_value=["!room1:example.com"])
|
|
|
|
with patch('community.helpers.user_utils.get_banlist_roomids') as mock_get_banlists:
|
|
mock_get_banlists.return_value = ["!room1:example.com"]
|
|
|
|
# Mock state events with no ban rules
|
|
client.get_state = AsyncMock(return_value=[])
|
|
|
|
logger = Mock()
|
|
result = await check_if_banned(client, "@user:example.com", ["!room1:example.com"], logger)
|
|
|
|
assert result == False
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_check_if_banned_room_not_joined(self):
|
|
"""Test ban check when bot is not in banlist room."""
|
|
client = Mock()
|
|
client.get_joined_rooms = AsyncMock(return_value=["!room2:example.com"])
|
|
|
|
with patch('community.helpers.user_utils.get_banlist_roomids') as mock_get_banlists:
|
|
mock_get_banlists.return_value = ["!room1:example.com"]
|
|
|
|
logger = Mock()
|
|
result = await check_if_banned(client, "@user:example.com", ["!room1:example.com"], logger)
|
|
|
|
assert result == False
|
|
logger.error.assert_called()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_banlist_roomids_aliases(self):
|
|
"""Test getting banlist room IDs with aliases."""
|
|
client = Mock()
|
|
client.resolve_room_alias = AsyncMock(return_value={"room_id": "!room1:example.com"})
|
|
|
|
banlists = ["#banlist1:example.com", "!room2:example.com"]
|
|
logger = Mock()
|
|
|
|
result = await get_banlist_roomids(client, banlists, logger)
|
|
|
|
assert "!room1:example.com" in result
|
|
assert "!room2:example.com" in result
|
|
client.resolve_room_alias.assert_called_once_with("#banlist1:example.com")
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_get_banlist_roomids_alias_error(self):
|
|
"""Test getting banlist room IDs with alias resolution error."""
|
|
client = Mock()
|
|
client.resolve_room_alias = AsyncMock(side_effect=Exception("Network error"))
|
|
|
|
banlists = ["#banlist1:example.com", "!room2:example.com"]
|
|
logger = Mock()
|
|
|
|
result = await get_banlist_roomids(client, banlists, logger)
|
|
|
|
assert "!room2:example.com" in result
|
|
assert "!room1:example.com" not in result
|
|
logger.error.assert_called()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_ban_user_from_rooms_success(self):
|
|
"""Test successful user banning from rooms."""
|
|
client = Mock()
|
|
client.ban_user = AsyncMock()
|
|
client.get_state_event = AsyncMock(return_value={"name": "Test Room"})
|
|
|
|
roomlist = ["!room1:example.com", "!room2:example.com"]
|
|
logger = Mock()
|
|
|
|
result = await ban_user_from_rooms(
|
|
client, "@user:example.com", roomlist, "banned", False, False, None, None, 0.1, logger
|
|
)
|
|
|
|
assert "ban_list" in result
|
|
assert "error_list" in result
|
|
assert "@user:example.com" in result["ban_list"]
|
|
assert len(result["ban_list"]["@user:example.com"]) == 2
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_ban_user_from_rooms_with_redaction(self):
|
|
"""Test user banning with message redaction."""
|
|
client = Mock()
|
|
client.ban_user = AsyncMock()
|
|
client.get_state_event = AsyncMock(return_value={"name": "Test Room"})
|
|
|
|
# Mock message redaction
|
|
mock_msg = Mock()
|
|
mock_msg.event_id = "!msg123:example.com"
|
|
get_messages_func = AsyncMock(return_value=[mock_msg])
|
|
|
|
database = Mock()
|
|
database.execute = AsyncMock()
|
|
|
|
roomlist = ["!room1:example.com"]
|
|
logger = Mock()
|
|
|
|
result = await ban_user_from_rooms(
|
|
client, "@user:example.com", roomlist, "banned", False, True,
|
|
get_messages_func, database, 0.1, logger
|
|
)
|
|
|
|
assert "ban_list" in result
|
|
database.execute.assert_called()
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_ban_user_from_rooms_error(self):
|
|
"""Test user banning with errors."""
|
|
client = Mock()
|
|
client.ban_user = AsyncMock(side_effect=Exception("Ban failed"))
|
|
client.get_state_event = AsyncMock(return_value={"name": "Test Room"})
|
|
|
|
roomlist = ["!room1:example.com"]
|
|
logger = Mock()
|
|
|
|
result = await ban_user_from_rooms(
|
|
client, "@user:example.com", roomlist, "banned", False, False, None, None, 0.1, logger
|
|
)
|
|
|
|
assert "error_list" in result
|
|
assert "@user:example.com" in result["error_list"]
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_user_permitted_unlimited_power(self):
|
|
"""Test user permission check with unlimited power."""
|
|
client = Mock()
|
|
|
|
with patch('community.helpers.room_utils.user_has_unlimited_power') as mock_unlimited:
|
|
mock_unlimited.return_value = True
|
|
|
|
result = await user_permitted(client, "@user:example.com", "!parent:example.com", 50, None, None)
|
|
|
|
assert result == True
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_user_permitted_sufficient_level(self):
|
|
"""Test user permission check with sufficient power level."""
|
|
client = Mock()
|
|
|
|
with patch('community.helpers.room_utils.user_has_unlimited_power') as mock_unlimited:
|
|
mock_unlimited.return_value = False
|
|
|
|
power_levels = Mock()
|
|
power_levels.get_user_level.return_value = 75
|
|
|
|
client.get_state_event = AsyncMock(return_value=power_levels)
|
|
|
|
result = await user_permitted(client, "@user:example.com", "!parent:example.com", 50, None, None)
|
|
|
|
assert result == True
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_user_permitted_insufficient_level(self):
|
|
"""Test user permission check with insufficient power level."""
|
|
client = Mock()
|
|
|
|
with patch('community.helpers.room_utils.user_has_unlimited_power') as mock_unlimited:
|
|
mock_unlimited.return_value = False
|
|
|
|
power_levels = Mock()
|
|
power_levels.get_user_level.return_value = 25
|
|
|
|
client.get_state_event = AsyncMock(return_value=power_levels)
|
|
|
|
result = await user_permitted(client, "@user:example.com", "!parent:example.com", 50, None, None)
|
|
|
|
assert result == False
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_user_permitted_error(self):
|
|
"""Test user permission check with error."""
|
|
client = Mock()
|
|
|
|
with patch('community.helpers.room_utils.user_has_unlimited_power') as mock_unlimited:
|
|
mock_unlimited.side_effect = Exception("Network error")
|
|
|
|
logger = Mock()
|
|
result = await user_permitted(client, "@user:example.com", "!parent:example.com", 50, None, logger)
|
|
|
|
assert result == False
|
|
logger.error.assert_called()
|