File size: 4,520 Bytes
365de9c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 |
import pytest
from unittest.mock import MagicMock
from app.backend.schemas import SUser
from app.core.utils import construct_collection_name, get_pdf_path, lines_to_markdown, protect_chat, extend_context
from app.backend.models.users import User
# Tests for SUser schema (app/backend/schemas)
# Tests password validation fails when no digit is present.
def test_password_no_digit():
with pytest.raises(ValueError, match="Password must contain at least one number."):
SUser(email="test@example.ru", password="NoDi@gitPassword!")
# Tests password validation fails when password is too short (<8 characters).
def test_password_too_short():
with pytest.raises(ValueError, match="String should have at least 8 characters"):
SUser(email="test@example.ru", password="P1@Sal!")
# Tests password validation fails when password is too long (>32 characters).
def test_password_too_long():
with pytest.raises(ValueError, match="String should have at most 32 characters"):
SUser(
email="test@example.ru",
password="Strong.Password123!Strong.Password123!Strong.Password123!Strong.Password123!Strong.Passwor",
)
# Tests email validation fails when email format is invalid.
def test_email():
with pytest.raises(ValueError, match="value is not a valid email address"):
SUser(email="test.test", password="SoMeGrE@tPa22WoRd!")
# Tests successful creation of SUser with valid email and password.
def test_valid_password():
user = SUser(email="test@example.ru", password="Strong.Password123!")
assert user.email == "test@example.ru"
assert user.password == "Strong.Password123!"
# Tests for app/core/utils
# Tests that construct_collection_name formats user ID and chat ID correctly.
def test_construct_collection_name():
user = User(id=1)
chat_id = 2
result = construct_collection_name(user, chat_id)
expected = "user_id_1_chat_id_2"
assert result == expected
# Tests get_pdf_path extracts correct path segment or returns empty string.
def test_get_pdf_path():
# Case 1: Path contains "chats_storage"
path_with = "/app/chats_storage/user_id=1/chat_id=2/documents/file.pdf"
expected_with = "chats_storage/user_id=1/chat_id=2/documents/file.pdf"
assert get_pdf_path(path_with) == expected_with
# Case 2: Path does not contain "chats_storage"
path_without = "/app/some_other_path/file.pdf"
assert get_pdf_path(path_without) == ""
# Tests lines_to_markdown converts text lines to HTML correctly.
def test_lines_to_markdown():
lines = ["Hello **world**", "Another line"]
result = lines_to_markdown(lines)
assert len(result) == 2
assert "<strong>world</strong>" in result[0] # Check bold formatting
assert "Another line" in result[1] # Check plain text conversion
# Tests protect_chat verifies user ownership of a chat correctly.
def test_protect_chat(monkeypatch):
mock_user = User(id=1)
chat_id = 2
# Case 1: User owns the chat
monkeypatch.setattr("app.core.utils.verify_ownership_rights", lambda user, cid: True)
assert protect_chat(mock_user, chat_id) == 1
# Case 2: User does not own the chat
monkeypatch.setattr("app.core.utils.verify_ownership_rights", lambda user, cid: False)
assert protect_chat(mock_user, chat_id) == 0
# Tests extend_context adds correct navbar, sidebar, and footer data.
def test_extend_context(monkeypatch):
mock_request = MagicMock()
# Case 1: With an authenticated user
mock_user = User(id=1)
monkeypatch.setattr("app.core.utils.get_current_user", lambda req: mock_user)
monkeypatch.setattr("app.core.utils.list_user_chats", lambda uid: [{"id": 1, "name": "Chat1"}])
context_with_user = {"request": mock_request}
result_with_user = extend_context(context_with_user)
assert result_with_user["navbar_context"]["user"]["role"] == "user"
assert result_with_user["navbar_context"]["user"]["instance"] == mock_user
assert result_with_user["sidebar_context"]["chat_groups"] == [{"id": 1, "name": "Chat1"}]
# Case 2: Without a user (guest)
monkeypatch.setattr("app.core.utils.get_current_user", lambda req: None)
context_without_user = {"request": mock_request}
result_without_user = extend_context(context_without_user)
assert result_without_user["navbar_context"]["user"]["role"] == "guest"
assert result_without_user["navbar_context"]["user"]["instance"] is None
assert result_without_user["sidebar_context"]["chat_groups"] == []
|