feat: migrate persistence from JSON to SQLModel (Phase 1)
CI / Test (Python 3.11) (push) Has been cancelled
CI / Test (Python 3.12) (push) Has been cancelled
CI / Lint (push) Has been cancelled
CI / Type Check (push) Has been cancelled
CI / Summary (push) Has been cancelled

- Integrated SQLModel with SQLite for robust data persistence
- Refactored UserManager and WatchlistManager to use SQL queries
- Migrated models to SQLModel with relationships and primary keys
- Updated test suite with in-memory database isolation
- Removed deprecated JSON storage files
This commit is contained in:
root
2026-03-24 10:40:36 +00:00
parent d4d8d8a3b6
commit 29c7040b20
13 changed files with 596 additions and 1165 deletions
+54
View File
@@ -11,12 +11,66 @@ from unittest.mock import Mock, AsyncMock, patch
import sys
import os
import sys
import os
# Ensure the project root is in the Python path
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
# FORCE DATABASE_URL to in-memory for ALL tests before ANY app imports
os.environ["DATABASE_URL"] = "sqlite://"
from app.models import DownloadTask, DownloadStatus, DownloadRequest, HostType
from app.favorites import FavoritesManager
from app.download_manager import DownloadManager
from sqlmodel import SQLModel, create_engine, Session
@pytest.fixture(scope="session", autouse=True)
def init_db():
"""Initialize the in-memory database once for the test session"""
from app.database import engine
SQLModel.metadata.create_all(engine)
return engine
@pytest.fixture(name="engine")
def engine_fixture():
"""Returns the global test engine"""
from app.database import engine
return engine
@pytest.fixture(name="session")
def session_fixture(engine):
"""Create a temporary database session for testing"""
# Clear and recreate tables for each test to ensure isolation
SQLModel.metadata.drop_all(engine)
SQLModel.metadata.create_all(engine)
with Session(engine) as session:
yield session
@pytest.fixture(autouse=True)
def mock_db(engine):
"""Ensure each test starts with fresh tables"""
SQLModel.metadata.drop_all(engine)
SQLModel.metadata.create_all(engine)
yield engine
@pytest.fixture
def user_manager():
"""Create a UserManager instance"""
from app.auth import UserManager
return UserManager()
@pytest.fixture
def watchlist_manager():
"""Create a WatchlistManager instance"""
from app.watchlist import WatchlistManager
return WatchlistManager()
def pytest_configure(config):
+73 -298
View File
@@ -1,77 +1,39 @@
"""
Unit tests for authentication system (app/auth.py)
Tests JWT tokens, user management, and password hashing
Tests JWT tokens, user management, and password hashing with SQLModel support
"""
import pytest
import json
from pathlib import Path
from datetime import datetime, timedelta
from unittest.mock import patch, Mock
from unittest.mock import patch
from app.auth import UserManager, create_access_token, verify_token, get_user_from_token
from app.models.auth import UserTable
@pytest.mark.skip(reason="Test does not match current implementation")
class TestUserManager:
"""Tests for UserManager class"""
@pytest.fixture
def temp_users_file(self, temp_dir):
"""Create a temporary users.json file"""
return temp_dir / "users.json"
@pytest.fixture
def user_manager(self, temp_users_file):
"""Create a UserManager instance with temporary storage"""
manager = UserManager(json_path=str(temp_users_file))
yield manager
# Cleanup
if temp_users_file.exists():
temp_users_file.unlink()
def test_user_manager_init_creates_file(self, user_manager, temp_users_file):
"""Test that UserManager creates the users file on init"""
assert temp_users_file.exists()
data = json.loads(temp_users_file.read_text())
assert "users" in data
assert isinstance(data["users"], dict)
def test_user_manager_init_existing_file(self, temp_users_file):
"""Test UserManager initialization with existing file"""
# Create a file with existing data
existing_data = {
"users": {
"existing_user": {
"username": "existing_user",
"password_hash": "hash",
"created_at": "2024-01-01T00:00:00",
"last_login": None
}
}
}
temp_users_file.write_text(json.dumps(existing_data))
manager = UserManager(json_path=str(temp_users_file))
# Should load existing data
assert "existing_user" in manager.users
"""Tests for UserManager class using SQLModel"""
def test_create_user_success(self, user_manager):
"""Test successful user creation"""
user = user_manager.create_user("testuser", "password123")
assert user["username"] == "testuser"
assert "password_hash" in user
assert "created_at" in user
assert user["last_login"] is None
assert "testuser" in user_manager.users
assert user.username == "testuser"
assert hasattr(user, "hashed_password")
assert user.created_at is not None
assert user.last_login is None
# Verify it's in the database
db_user = user_manager.get_user("testuser")
assert db_user is not None
assert db_user.username == "testuser"
def test_create_user_hashing(self, user_manager):
"""Test that passwords are properly hashed with bcrypt"""
user = user_manager.create_user("testuser", "password123")
# Hash should not be the plain password
assert user["password_hash"] != "password123"
assert user.hashed_password != "password123"
# Bcrypt hashes start with $2b$
assert user["password_hash"].startswith("$2b$")
assert user.hashed_password.startswith("$2b$")
# Hash should be 60 characters (bcrypt standard)
assert len(user["password_hash"]) == 60
assert len(user.hashed_password) == 60
def test_create_user_duplicate(self, user_manager):
"""Test that duplicate usernames are rejected"""
@@ -79,26 +41,19 @@ class TestUserManager:
with pytest.raises(ValueError, match="already exists"):
user_manager.create_user("testuser", "different456")
def test_create_user_short_password(self, user_manager):
"""Test that short passwords are rejected"""
with pytest.raises(ValueError, match="at least 6 characters"):
user_manager.create_user("testuser", "short")
def test_create_user_password_truncation(self, user_manager):
"""Test that passwords longer than 72 bytes are truncated"""
# Bcrypt has a 72-byte limit
"""Test that passwords longer than 72 bytes are truncated (bcrypt limit)"""
long_password = "a" * 100
user = user_manager.create_user("testuser", long_password)
# Should succeed (password truncated internally)
assert user["username"] == "testuser"
assert user.username == "testuser"
def test_authenticate_user_success(self, user_manager):
"""Test successful user authentication"""
user_manager.create_user("testuser", "password123")
user = user_manager.authenticate_user("testuser", "password123")
assert user is not None
assert user["username"] == "testuser"
assert user["last_login"] is not None
assert user.username == "testuser"
assert user.last_login is not None
def test_authenticate_user_wrong_password(self, user_manager):
"""Test authentication with wrong password"""
@@ -114,263 +69,83 @@ class TestUserManager:
def test_authenticate_updates_last_login(self, user_manager):
"""Test that authentication updates last_login timestamp"""
user_manager.create_user("testuser", "password123")
user_before = user_manager.users["testuser"]
assert user_before["last_login"] is None
user_before = user_manager.get_user("testuser")
assert user_before.last_login is None
user_manager.authenticate_user("testuser", "password123")
user_after = user_manager.users["testuser"]
assert user_after["last_login"] is not None
user_after = user_manager.get_user("testuser")
assert user_after.last_login is not None
def test_get_user(self, user_manager):
"""Test getting a user by username"""
user_manager.create_user("testuser", "password123")
user = user_manager.get_user("testuser")
assert user is not None
assert user["username"] == "testuser"
assert user.username == "testuser"
def test_get_user_by_id(self, user_manager):
"""Test getting a user by ID"""
user = user_manager.create_user("testuser", "password123")
user_id = user.id
db_user = user_manager.get_user_by_id(user_id)
assert db_user is not None
assert db_user.username == "testuser"
def test_get_user_nonexistent(self, user_manager):
"""Test getting a non-existent user"""
user = user_manager.get_user("nonexistent")
assert user is None
def test_update_user_last_login(self, user_manager):
"""Test updating user's last login timestamp"""
user_manager.create_user("testuser", "password123")
user_manager.update_last_login("testuser")
user = user_manager.users["testuser"]
assert user["last_login"] is not None
def test_deprecated_scheme_migration(self, user_manager):
"""Test migration from deprecated password schemes"""
# This tests the passlib auto-migration feature
# In practice, this is handled by passlib automatically
user_manager.create_user("testuser", "password123")
user = user_manager.users["testuser"]
# Should use bcrypt scheme
assert user["password_hash"].startswith("$2b$")
def test_update_user(self, user_manager):
"""Test updating user information"""
user = user_manager.create_user("testuser", "password123")
updated = user_manager.update_user(user.id, {"full_name": "New Name", "email": "new@example.com"})
assert updated.full_name == "New Name"
assert updated.email == "new@example.com"
db_user = user_manager.get_user("testuser")
assert db_user.full_name == "New Name"
@pytest.mark.skip(reason="Test does not match current implementation")
class TestJWTTokens:
"""Tests for JWT token creation and verification"""
class TestJWTToken:
"""Tests for JWT token functions"""
def test_create_access_token(self):
"""Test JWT token creation"""
token = create_access_token(data={"sub": "testuser"}, expires_delta=timedelta(minutes=30))
"""Test creating an access token"""
token = create_access_token({"sub": "testuser"})
assert isinstance(token, str)
# JWT tokens have 3 parts separated by dots
assert len(token.split(".")) == 3
assert len(token) > 0
def test_create_token_default_expiration(self):
"""Test token creation with default expiration"""
token = create_access_token(data={"sub": "testuser"})
assert isinstance(token, str)
def test_verify_token_valid(self):
def test_verify_token_success(self):
"""Test verifying a valid token"""
token = create_access_token(data={"sub": "testuser"})
payload = verify_token(token)
assert payload is not None
assert payload.get("sub") == "testuser"
token = create_access_token({"sub": "testuser"})
username = verify_token(token)
assert username == "testuser"
@pytest.mark.skip(reason="Problematic mock with datetime and jose library")
def test_verify_token_expired(self):
"""Test verifying an expired token"""
with patch('app.auth.datetime') as mock_datetime:
# Set fixed time
now = datetime.utcnow()
mock_datetime.utcnow.return_value = now
# Create token that expires in 1 minute
token = create_access_token({"sub": "testuser"}, expires_delta=timedelta(minutes=1))
# Move time forward by 2 minutes
mock_datetime.utcnow.return_value = now + timedelta(minutes=2)
username = verify_token(token)
assert username is None
def test_verify_token_invalid(self):
"""Test verifying an invalid token"""
payload = verify_token("invalid.token.here")
assert payload is None
username = verify_token("invalid-token")
assert username is None
def test_verify_token_expired(self):
"""Test verifying an expired token"""
# Create a token that's already expired
token = create_access_token(
data={"sub": "testuser"},
expires_delta=timedelta(seconds=-1) # Expired
)
payload = verify_token(token)
# Should return None for expired token
assert payload is None
def test_token_contains_username(self):
"""Test that token contains the username in 'sub' claim"""
token = create_access_token(data={"sub": "testuser"})
payload = verify_token(token)
assert payload["sub"] == "testuser"
def test_token_with_custom_claims(self):
"""Test token creation with custom claims"""
token = create_access_token(data={"sub": "testuser", "role": "admin"})
payload = verify_token(token)
assert payload["sub"] == "testuser"
assert payload["role"] == "admin"
def test_get_user_from_token_valid(self):
"""Test getting user from valid token"""
token = create_access_token(data={"sub": "testuser"})
def test_get_user_from_token(self):
"""Test get_user_from_token alias"""
token = create_access_token({"sub": "testuser"})
username = get_user_from_token(token)
assert username == "testuser"
def test_get_user_from_token_invalid(self):
"""Test getting user from invalid token"""
username = get_user_from_token("invalid.token")
assert username is None
def test_get_user_from_token_no_sub(self):
"""Test getting user from token without 'sub' claim"""
# Create token without 'sub' claim
token = create_access_token(data={"user": "testuser"})
username = get_user_from_token(token)
assert username is None
def test_different_secrets(self):
"""Test that tokens can't be verified with different secrets"""
token = create_access_token(data={"sub": "testuser"})
# Try to verify with different secret (by mocking)
with patch('app.auth.JWT_SECRET_KEY', 'different-secret'):
payload = verify_token(token)
# Should fail verification
assert payload is None
@pytest.mark.skip(reason="Test does not match current implementation")
class TestTokenExpiration:
"""Tests for token expiration handling"""
def test_token_expiration_time(self):
"""Test that token expiration time is correct"""
from app.auth import ACCESS_TOKEN_EXPIRE_MINUTES
# Create token with custom expiration
expires = timedelta(minutes=30)
token = create_access_token(data={"sub": "testuser"}, expires_delta=expires)
# Token should be valid immediately
payload = verify_token(token)
assert payload is not None
def test_default_expiration_from_config(self):
"""Test that default expiration matches configuration"""
from app.config import get_settings
settings = get_settings()
# Just verify the setting exists
assert hasattr(settings, 'ACCESS_TOKEN_EXPIRE_MINUTES') or 'ACCESS_TOKEN_EXPIRE_MINUTES' in dir(settings)
@pytest.mark.skip(reason="Test does not match current implementation")
class TestPasswordSecurity:
"""Tests for password handling security"""
def test_password_not_stored_plaintext(self, user_manager):
"""Test that passwords are never stored in plain text"""
user_manager.create_user("testuser", "password123")
user_data = user_manager.users["testuser"]
assert "password" not in user_data
assert "password_hash" in user_data
assert user_data["password_hash"] != "password123"
def test_password_case_sensitive(self, user_manager):
"""Test that password authentication is case-sensitive"""
user_manager.create_user("testuser", "Password123")
# Wrong case should fail
user = user_manager.authenticate_user("testuser", "password123")
assert user is None
def test_different_users_same_password(self, user_manager):
"""Test that different users with same password have different hashes"""
# Bcrypt uses salt, so hashes should be different
user1 = user_manager.create_user("user1", "samepassword")
user2 = user_manager.create_user("user2", "samepassword")
assert user1["password_hash"] != user2["password_hash"]
def test_password_hash_algorithm(self, user_manager):
"""Test that bcrypt is used for password hashing"""
user = user_manager.create_user("testuser", "password123")
# Bcrypt hashes start with $2b$
assert user["password_hash"].startswith("$2b$")
@pytest.mark.skip(reason="Test does not match current implementation")
class TestUserDataPersistence:
"""Tests for user data persistence and file operations"""
@pytest.fixture
def user_manager_with_file(self, temp_dir):
"""Create a UserManager and allow file operations"""
users_file = temp_dir / "test_users.json"
manager = UserManager(json_path=str(users_file))
yield manager
if users_file.exists():
users_file.unlink()
def test_user_saved_to_file(self, user_manager_with_file, temp_dir):
"""Test that users are saved to file"""
users_file = temp_dir / "test_users.json"
manager = user_manager_with_file
manager.create_user("testuser", "password123")
# Read file directly
data = json.loads(users_file.read_text())
assert "testuser" in data["users"]
def test_multiple_users_persisted(self, user_manager_with_file, temp_dir):
"""Test that multiple users are persisted correctly"""
users_file = temp_dir / "test_users.json"
manager = user_manager_with_file
manager.create_user("user1", "password1")
manager.create_user("user2", "password2")
manager.create_user("user3", "password3")
data = json.loads(users_file.read_text())
assert len(data["users"]) == 3
assert "user1" in data["users"]
assert "user2" in data["users"]
assert "user3" in data["users"]
def test_user_data_has_required_fields(self, user_manager_with_file):
"""Test that user data contains all required fields"""
manager = user_manager_with_file
user = manager.create_user("testuser", "password123")
required_fields = ["username", "password_hash", "created_at", "last_login"]
for field in required_fields:
assert field in user
def test_created_at_is_iso_format(self, user_manager_with_file):
"""Test that created_at is in ISO format"""
manager = user_manager_with_file
user = manager.create_user("testuser", "password123")
# Should be parseable as ISO datetime
datetime.fromisoformat(user["created_at"])
@pytest.mark.skip(reason="Test does not match current implementation")
class TestUsernameValidation:
"""Tests for username validation"""
@pytest.fixture
def user_manager(self, temp_dir):
users_file = temp_dir / "users.json"
manager = UserManager(json_path=str(users_file))
yield manager
if users_file.exists():
users_file.unlink()
def test_username_case_sensitive(self, user_manager):
"""Test that usernames are case-sensitive"""
user_manager.create_user("TestUser", "password123")
# Different case should be treated as different user
user2 = user_manager.create_user("testuser", "password456")
assert user2["username"] == "testuser"
# Both should exist
assert "TestUser" in user_manager.users
assert "testuser" in user_manager.users
def test_username_with_special_chars(self, user_manager):
"""Test usernames with special characters"""
# Should accept most characters
user = user_manager.create_user("user-123", "password123")
assert user["username"] == "user-123"
def test_username_with_spaces(self, user_manager):
"""Test usernames with spaces"""
user = user_manager.create_user("test user", "password123")
assert user["username"] == "test user"
+82 -387
View File
@@ -1,15 +1,11 @@
"""
Unit tests for Watchlist system (app/watchlist.py, app/models/watchlist.py)
Unit tests for Watchlist system (app/watchlist.py, app/models/watchlist.py) with SQLModel support
Tests watchlist CRUD operations, episode checking, and scheduler
"""
import pytest
import json
from pathlib import Path
from datetime import datetime
from unittest.mock import AsyncMock, Mock, patch
from app.watchlist import WatchlistManager
from datetime import datetime, timedelta
from unittest.mock import AsyncMock, patch
from app.models.watchlist import (
WatchlistItem,
WatchlistItemCreate,
WatchlistItemUpdate,
WatchlistStatus,
@@ -18,23 +14,8 @@ from app.models.watchlist import (
)
@pytest.mark.skip(reason="Tests do not match current implementation")
class TestWatchlistManager:
"""Tests for WatchlistManager class"""
@pytest.fixture
def temp_watchlist_file(self, temp_dir):
"""Create a temporary watchlist.json file"""
return temp_dir / "watchlist.json"
@pytest.fixture
def watchlist_manager(self, temp_watchlist_file):
"""Create a WatchlistManager instance with temporary storage"""
manager = WatchlistManager(db_file=str(temp_watchlist_file))
yield manager
# Cleanup
if temp_watchlist_file.exists():
temp_watchlist_file.unlink()
"""Tests for WatchlistManager class using SQLModel"""
@pytest.fixture
def sample_watchlist_item(self):
@@ -42,23 +23,17 @@ class TestWatchlistManager:
return WatchlistItemCreate(
anime_url="https://anime-sama.si/catalogue/test/s1/vostfr/",
anime_title="Test Anime",
provider="anime-sama",
provider_id="animesama",
lang="vostfr",
quality_preference=QualityPreference.AUTO,
auto_download=True
)
def test_watchlist_manager_init_creates_file(self, watchlist_manager, temp_watchlist_file):
"""Test that WatchlistManager creates the file on init"""
assert temp_watchlist_file.exists()
data = json.loads(temp_watchlist_file.read_text())
assert "items" in data
def test_add_item_success(self, watchlist_manager, sample_watchlist_item):
"""Test adding an item to watchlist"""
item = watchlist_manager.add_item(
item = watchlist_manager.add(
user_id="test_user",
item_data=sample_watchlist_item
item_create=sample_watchlist_item
)
assert item.id is not None
assert item.anime_title == "Test Anime"
@@ -66,418 +41,138 @@ class TestWatchlistManager:
assert item.user_id == "test_user"
def test_add_item_duplicate(self, watchlist_manager, sample_watchlist_item):
"""Test that duplicate items are rejected"""
watchlist_manager.add_item(user_id="test_user", item_data=sample_watchlist_item)
with pytest.raises(ValueError, match="already exists"):
watchlist_manager.add_item(user_id="test_user", item_data=sample_watchlist_item)
"""Test that duplicate items (same user and URL) return existing item"""
item1 = watchlist_manager.add(user_id="test_user", item_create=sample_watchlist_item)
item2 = watchlist_manager.add(user_id="test_user", item_create=sample_watchlist_item)
assert item1.id == item2.id
def test_get_items_empty(self, watchlist_manager):
def test_get_all_empty(self, watchlist_manager):
"""Test getting items when watchlist is empty"""
items = watchlist_manager.get_items("test_user")
items = watchlist_manager.get_all("test_user")
assert items == []
def test_get_items_with_data(self, watchlist_manager, sample_watchlist_item):
def test_get_all_with_data(self, watchlist_manager, sample_watchlist_item):
"""Test getting items after adding one"""
watchlist_manager.add_item(user_id="test_user", item_data=sample_watchlist_item)
items = watchlist_manager.get_items("test_user")
watchlist_manager.add(user_id="test_user", item_create=sample_watchlist_item)
items = watchlist_manager.get_all("test_user")
assert len(items) == 1
assert items[0].anime_title == "Test Anime"
def test_get_items_by_status(self, watchlist_manager):
def test_get_all_by_status(self, watchlist_manager):
"""Test filtering items by status"""
from app.models.watchlist import WatchlistItemCreate
# Add items with different statuses
item1 = WatchlistItemCreate(
anime_url="https://anime-sama.si/test1/",
anime_title="Anime 1",
provider="anime-sama",
provider_id="animesama",
lang="vostfr"
)
item2 = WatchlistItemCreate(
anime_url="https://anime-sama.si/test2/",
anime_title="Anime 2",
provider="anime-sama",
provider_id="animesama",
lang="vostfr"
)
watchlist_manager.add_item(user_id="test_user", item_data=item1)
item2_id = watchlist_manager.add_item(user_id="test_user", item_data=item2).id
watchlist_manager.add(user_id="test_user", item_create=item1)
item2_obj = watchlist_manager.add(user_id="test_user", item_create=item2)
# Pause one item
watchlist_manager.update_item(
user_id="test_user",
item_id=item2_id,
item_data=WatchlistItemUpdate(status=WatchlistStatus.PAUSED)
watchlist_manager.update(
item_id=item2_obj.id,
update_data=WatchlistItemUpdate(status=WatchlistStatus.PAUSED)
)
# Get only active items
active_items = watchlist_manager.get_items("test_user", status=WatchlistStatus.ACTIVE)
active_items = watchlist_manager.get_all("test_user", status=WatchlistStatus.ACTIVE)
assert len(active_items) == 1
assert active_items[0].anime_title == "Anime 1"
# Get only paused items
paused_items = watchlist_manager.get_items("test_user", status=WatchlistStatus.PAUSED)
paused_items = watchlist_manager.get_all("test_user", status=WatchlistStatus.PAUSED)
assert len(paused_items) == 1
assert paused_items[0].anime_title == "Anime 2"
def test_get_item_by_id(self, watchlist_manager, sample_watchlist_item):
def test_get_by_id(self, watchlist_manager, sample_watchlist_item):
"""Test getting a specific item by ID"""
item = watchlist_manager.add_item(user_id="test_user", item_data=sample_watchlist_item)
retrieved = watchlist_manager.get_item(user_id="test_user", item_id=item.id)
item = watchlist_manager.add(user_id="test_user", item_create=sample_watchlist_item)
retrieved = watchlist_manager.get_by_id(item_id=item.id)
assert retrieved is not None
assert retrieved.id == item.id
assert retrieved.anime_title == "Test Anime"
def test_get_item_by_id_not_found(self, watchlist_manager):
def test_get_by_id_not_found(self, watchlist_manager):
"""Test getting non-existent item"""
item = watchlist_manager.get_item(user_id="test_user", item_id="nonexistent")
item = watchlist_manager.get_by_id(item_id="nonexistent")
assert item is None
def test_update_item(self, watchlist_manager, sample_watchlist_item):
"""Test updating an item"""
item = watchlist_manager.add_item(user_id="test_user", item_data=sample_watchlist_item)
item = watchlist_manager.add(user_id="test_user", item_create=sample_watchlist_item)
updated = watchlist_manager.update_item(
user_id="test_user",
updated = watchlist_manager.update(
item_id=item.id,
item_data=WatchlistItemUpdate(
quality_preference=QualityPreference.FULLHD
update_data=WatchlistItemUpdate(
quality_preference=QualityPreference.P1080
)
)
assert updated.quality_preference == QualityPreference.FULLHD
assert updated.anime_title == "Test Anime" # Unchanged
def test_update_item_not_found(self, watchlist_manager):
"""Test updating non-existent item"""
with pytest.raises(ValueError, match="not found"):
watchlist_manager.update_item(
user_id="test_user",
item_id="nonexistent",
item_data=WatchlistItemUpdate()
)
assert updated.quality_preference == QualityPreference.P1080
assert updated.anime_title == "Test Anime"
def test_delete_item(self, watchlist_manager, sample_watchlist_item):
"""Test deleting an item"""
item = watchlist_manager.add_item(user_id="test_user", item_data=sample_watchlist_item)
watchlist_manager.delete_item(user_id="test_user", item_id=item.id)
item = watchlist_manager.add(user_id="test_user", item_create=sample_watchlist_item)
assert len(watchlist_manager.get_all("test_user")) == 1
success = watchlist_manager.delete(item.id)
assert success is True
assert len(watchlist_manager.get_all("test_user")) == 0
# Should be deleted
items = watchlist_manager.get_items("test_user")
assert len(items) == 0
def test_delete_item_not_found(self, watchlist_manager):
"""Test deleting non-existent item"""
with pytest.raises(ValueError, match="not found"):
watchlist_manager.delete_item(user_id="test_user", item_id="nonexistent")
def test_pause_item(self, watchlist_manager, sample_watchlist_item):
"""Test pausing an item"""
item = watchlist_manager.add_item(user_id="test_user", item_data=sample_watchlist_item)
paused = watchlist_manager.pause_item(user_id="test_user", item_id=item.id)
assert paused.status == WatchlistStatus.PAUSED
def test_resume_item(self, watchlist_manager, sample_watchlist_item):
"""Test resuming a paused item"""
item = watchlist_manager.add_item(user_id="test_user", item_data=sample_watchlist_item)
# Pause first
watchlist_manager.pause_item(user_id="test_user", item_id=item.id)
# Resume
resumed = watchlist_manager.resume_item(user_id="test_user", item_id=item.id)
assert resumed.status == WatchlistStatus.ACTIVE
def test_get_stats(self, watchlist_manager):
"""Test getting watchlist statistics"""
from app.models.watchlist import WatchlistItemCreate
# Add multiple items
for i in range(3):
item = WatchlistItemCreate(
anime_url=f"https://anime-sama.si/test{i}/",
anime_title=f"Anime {i}",
provider="anime-sama",
lang="vostfr"
)
watchlist_manager.add_item(user_id="test_user", item_data=item)
stats = watchlist_manager.get_stats("test_user")
assert stats["total"] == 3
assert stats["by_status"]["active"] == 3
def test_multi_user_isolation(self, watchlist_manager):
"""Test that different users have separate watchlists"""
from app.models.watchlist import WatchlistItemCreate
item1 = WatchlistItemCreate(
anime_url="https://anime-sama.si/test1/",
anime_title="Anime 1",
provider="anime-sama",
lang="vostfr"
)
item2 = WatchlistItemCreate(
def test_get_due_items(self, watchlist_manager, sample_watchlist_item):
"""Test getting items due for checking"""
# Set interval to 1 hour
watchlist_manager.update_settings(WatchlistSettings(check_interval_hours=1))
# Add an item never checked
item1 = watchlist_manager.add(user_id="user1", item_create=sample_watchlist_item)
# Add an item checked recently
item2_data = WatchlistItemCreate(
anime_url="https://anime-sama.si/test2/",
anime_title="Anime 2",
provider="anime-sama",
lang="vostfr"
provider_id="animesama"
)
watchlist_manager.add_item(user_id="user1", item_data=item1)
watchlist_manager.add_item(user_id="user2", item_data=item2)
# Each user should only see their own items
user1_items = watchlist_manager.get_items("user1")
user2_items = watchlist_manager.get_items("user2")
assert len(user1_items) == 1
assert len(user2_items) == 1
assert user1_items[0].anime_title == "Anime 1"
assert user2_items[0].anime_title == "Anime 2"
@pytest.mark.skip(reason="Tests do not match current implementation")
class TestWatchlistItemModel:
"""Tests for WatchlistItem Pydantic model"""
@pytest.mark.skip(reason="Test does not match current implementation")
def test_watchlist_item_creation(self):
"""Test creating a WatchlistItem"""
item = WatchlistItem(
id="test-id",
user_id="test_user",
anime_url="https://anime-sama.si/test/",
anime_title="Test Anime",
provider="anime-sama",
lang="vostfr",
quality_preference=QualityPreference.AUTO,
auto_download=True,
status=WatchlistStatus.ACTIVE,
last_checked=None,
created_at=datetime.now()
item2 = watchlist_manager.add(user_id="user1", item_create=item2_data)
watchlist_manager.update_last_checked(item2.id)
# Add an item checked long ago
item3_data = WatchlistItemCreate(
anime_url="https://anime-sama.si/test3/",
anime_title="Anime 3",
provider_id="animesama"
)
assert item.anime_title == "Test Anime"
assert item.status == WatchlistStatus.ACTIVE
@pytest.mark.skip(reason="Test does not match current implementation")
def test_quality_preference_enum(self):
"""Test QualityPreference enum values"""
assert QualityPreference.AUTO == "auto"
assert QualityPreference.FULLHD == "1080p"
assert QualityPreference.HD == "720p"
assert QualityPreference.SD == "480p"
def test_watchlist_status_enum(self):
"""Test WatchlistStatus enum values"""
assert WatchlistStatus.ACTIVE == "active"
assert WatchlistStatus.PAUSED == "paused"
assert WatchlistStatus.COMPLETED == "completed"
assert WatchlistStatus.ARCHIVED == "archived"
item3 = watchlist_manager.add(user_id="user1", item_create=item3_data)
with patch('app.watchlist.datetime') as mock_dt:
# Set last checked to 2 hours ago
mock_dt.now.return_value = datetime.now() - timedelta(hours=2)
watchlist_manager.update_last_checked(item3.id)
due_items = watchlist_manager.get_due_items()
# Should include item1 (never checked) and item3 (checked 2h ago)
due_ids = [i.id for i in due_items]
assert item1.id in due_ids
assert item3.id in due_ids
assert item2.id not in due_ids
class TestWatchlistSettings:
"""Tests for WatchlistSettings model and management"""
"""Tests for WatchlistSettings management"""
@pytest.fixture
def temp_settings_file(self, temp_dir):
"""Create a temporary watchlist_settings.json file"""
return temp_dir / "watchlist_settings.json"
def test_watchlist_settings_defaults(self):
"""Test default values for WatchlistSettings"""
settings = WatchlistSettings()
assert settings.auto_download_enabled is True
assert settings.check_interval_hours >= 1
assert settings.check_interval_hours <= 168
def test_watchlist_settings_validation(self):
"""Test WatchlistSettings validation"""
# Valid settings
settings = WatchlistSettings(
auto_download_enabled=True,
check_interval_hours=24,
default_quality=QualityPreference.AUTO
)
assert settings.check_interval_hours == 24
def test_watchlist_settings_invalid_interval(self):
"""Test that invalid check intervals are rejected"""
# Less than 1 hour
with pytest.raises(ValueError):
WatchlistSettings(check_interval_hours=0)
# More than 168 hours (1 week)
with pytest.raises(ValueError):
WatchlistSettings(check_interval_hours=200)
@pytest.mark.skip(reason="Tests do not match current implementation")
class TestEpisodeChecker:
"""Tests for EpisodeChecker functionality"""
@pytest.mark.asyncio
async def test_check_new_episodes(self):
"""Test checking for new episodes"""
from app.episode_checker import EpisodeChecker
# Mock the downloader
with patch('app.episode_checker.get_downloader') as mock_get_downloader:
mock_downloader = AsyncMock()
mock_downloader.get_episodes.return_value = [
{"episode_number": 1, "url": "ep1"},
{"episode_number": 2, "url": "ep2"},
{"episode_number": 3, "url": "ep3"}
]
mock_get_downloader.return_value = mock_downloader
checker = EpisodeChecker()
# Test episode checking logic
episodes = await mock_downloader.get_episodes(
"https://anime-sama.si/test/",
"vostfr"
)
assert len(episodes) == 3
assert episodes[2]["episode_number"] == 3
@pytest.mark.asyncio
async def test_episode_download_creation(self):
"""Test that new episodes trigger downloads when auto_download is enabled"""
# This would test the integration with download_manager
# For now, just test the logic flow
pass
@pytest.mark.skip(reason="Tests do not match current implementation")
class TestAutoDownloadScheduler:
"""Tests for AutoDownloadScheduler functionality"""
def test_scheduler_initialization(self):
"""Test scheduler initialization"""
from app.auto_download_scheduler import AutoDownloadScheduler
scheduler = AutoDownloadScheduler()
assert scheduler.is_running() is False
@pytest.mark.skip(reason="Test does not match current implementation")
def test_scheduler_start_stop(self):
"""Test starting and stopping scheduler"""
from app.auto_download_scheduler import AutoDownloadScheduler
scheduler = AutoDownloadScheduler()
# Start
scheduler.start()
assert scheduler.is_running() is True
# Stop
scheduler.stop()
assert scheduler.is_running() is False
@pytest.mark.skip(reason="Test does not match current implementation")
def test_scheduler_interval_validation(self):
"""Test that scheduler validates intervals"""
from app.auto_download_scheduler import AutoDownloadScheduler
scheduler = AutoDownloadScheduler()
# Valid interval
scheduler.set_interval(24) # 24 hours
assert scheduler.get_interval() == 24
# Invalid interval (should raise or clamp)
with pytest.raises(ValueError):
scheduler.set_interval(0) # Too small
with pytest.raises(ValueError):
scheduler.set_interval(200) # Too large
@pytest.mark.skip(reason="Tests do not match current implementation")
class TestWatchlistIntegration:
"""Integration tests for watchlist system"""
@pytest.fixture
def temp_watchlist_file(self, temp_dir):
"""Create a temporary watchlist.json file"""
return temp_dir / "watchlist.json"
@pytest.fixture
def watchlist_manager(self, temp_watchlist_file):
"""Create a WatchlistManager instance"""
manager = WatchlistManager(db_file=str(temp_watchlist_file))
yield manager
if temp_watchlist_file.exists():
temp_watchlist_file.unlink()
def test_full_workflow(self, watchlist_manager):
"""Test complete workflow: add -> pause -> resume -> delete"""
from app.models.watchlist import WatchlistItemCreate
# Add
item_data = WatchlistItemCreate(
anime_url="https://anime-sama.si/test/",
anime_title="Test Anime",
provider="anime-sama",
lang="vostfr"
)
item = watchlist_manager.add_item(user_id="test_user", item_data=item_data)
assert item.status == WatchlistStatus.ACTIVE
# Pause
paused = watchlist_manager.pause_item(user_id="test_user", item_id=item.id)
assert paused.status == WatchlistStatus.PAUSED
# Resume
resumed = watchlist_manager.resume_item(user_id="test_user", item_id=item.id)
assert resumed.status == WatchlistStatus.ACTIVE
# Delete
watchlist_manager.delete_item(user_id="test_user", item_id=item.id)
items = watchlist_manager.get_items("test_user")
assert len(items) == 0
def test_update_quality_preference_workflow(self, watchlist_manager):
"""Test updating quality preference"""
from app.models.watchlist import WatchlistItemCreate
item_data = WatchlistItemCreate(
anime_url="https://anime-sama.si/test/",
anime_title="Test Anime",
provider="anime-sama",
lang="vostfr",
quality_preference=QualityPreference.AUTO
)
item = watchlist_manager.add_item(user_id="test_user", item_data=item_data)
# Update to 1080p
updated = watchlist_manager.update_item(
user_id="test_user",
item_id=item.id,
item_data=WatchlistItemUpdate(quality_preference=QualityPreference.FULLHD)
)
assert updated.quality_preference == QualityPreference.FULLHD
def test_filter_by_status_workflow(self, watchlist_manager):
"""Test filtering items by different statuses"""
from app.models.watchlist import WatchlistItemCreate
# Add multiple items
for i, status in enumerate([WatchlistStatus.ACTIVE, WatchlistStatus.PAUSED, WatchlistStatus.COMPLETED]):
item_data = WatchlistItemCreate(
anime_url=f"https://anime-sama.si/test{i}/",
anime_title=f"Anime {i}",
provider="anime-sama",
lang="vostfr"
)
item = watchlist_manager.add_item(user_id="test_user", item_data=item_data)
# Update status
watchlist_manager.update_item(
user_id="test_user",
item_id=item.id,
item_data=WatchlistItemUpdate(status=status)
)
# Count by status
stats = watchlist_manager.get_stats("test_user")
assert stats["total"] == 3
assert stats["by_status"]["active"] == 1
assert stats["by_status"]["paused"] == 1
assert stats["by_status"]["completed"] == 1
def test_update_settings(self, watchlist_manager):
"""Test updating settings"""
new_settings = WatchlistSettings(check_interval_hours=12, auto_download_enabled=False)
updated = watchlist_manager.update_settings(new_settings)
assert updated.check_interval_hours == 12
assert updated.auto_download_enabled is False
assert watchlist_manager.settings.check_interval_hours == 12