4b5c23faf6
- Reformatted the error message for database lock detection in main.py for better readability. - Removed unnecessary imports from test_basic.py to streamline the test file.
81 lines
No EOL
2.4 KiB
Python
81 lines
No EOL
2.4 KiB
Python
import os
|
|
import sys
|
|
import pytest
|
|
from unittest.mock import patch, MagicMock
|
|
|
|
# Add the root directory to the path
|
|
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
|
|
|
|
# Mock the telethon client for testing
|
|
@pytest.fixture
|
|
def mock_client():
|
|
with patch('telethon.TelegramClient') as mock:
|
|
# Create a mock instance of TelegramClient
|
|
client_instance = MagicMock()
|
|
mock.return_value = client_instance
|
|
|
|
# Mock basic methods
|
|
client_instance.get_dialogs.return_value = []
|
|
client_instance.get_entity.return_value = MagicMock()
|
|
|
|
yield client_instance
|
|
|
|
# Test functions
|
|
def test_imports():
|
|
"""Test that all necessary imports are available"""
|
|
# This will fail if any import is missing
|
|
from telethon import TelegramClient
|
|
from telethon.sessions import StringSession
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_format_entity():
|
|
"""Test the format_entity function with different entity types"""
|
|
from main import format_entity
|
|
|
|
# Test user entity
|
|
user = MagicMock()
|
|
user.id = 123
|
|
user.first_name = "John"
|
|
user.last_name = "Doe"
|
|
user.username = "johndoe"
|
|
user.phone = "+1234567890"
|
|
|
|
user_result = format_entity(user)
|
|
assert user_result["id"] == 123
|
|
assert user_result["name"] == "John Doe"
|
|
assert user_result["type"] == "user"
|
|
assert user_result["username"] == "johndoe"
|
|
assert user_result["phone"] == "+1234567890"
|
|
|
|
# Test group entity
|
|
from telethon.tl.types import Chat
|
|
group = MagicMock(spec=Chat)
|
|
group.id = 456
|
|
group.title = "Test Group"
|
|
|
|
group_result = format_entity(group)
|
|
assert group_result["id"] == 456
|
|
assert group_result["name"] == "Test Group"
|
|
assert group_result["type"] == "group"
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_format_message():
|
|
"""Test the format_message function"""
|
|
from main import format_message
|
|
from datetime import datetime
|
|
|
|
# Create a mock message
|
|
message = MagicMock()
|
|
message.id = 789
|
|
message.date = datetime(2023, 1, 1, 12, 0, 0)
|
|
message.message = "Hello, world!"
|
|
message.from_id = None
|
|
message.media = None
|
|
|
|
result = format_message(message)
|
|
assert result["id"] == 789
|
|
assert "2023-01-01" in result["date"]
|
|
assert result["text"] == "Hello, world!"
|
|
assert "has_media" not in result
|
|
|
|
# More tests can be added as the project grows |