Spaces:
Sleeping
Sleeping
File size: 2,906 Bytes
fdcd9e5 | 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 | """
Tests for app.core.database — SQLite wrapper.
"""
import sqlite3
from datetime import date, datetime
from unittest.mock import patch
import pytest
from app.core import database
from app.core.models import Digest, Paper
class TestInitialize:
def test_creates_tables(self, tmp_db):
conn = database.get_connection(tmp_db)
tables = conn.execute(
"SELECT name FROM sqlite_master WHERE type='table'"
).fetchall()
names = {r['name'] for r in tables}
assert 'papers' in names
assert 'digests' in names
assert 'digest_papers' in names
assert 'meta' in names
conn.close()
def test_sets_db_version(self, tmp_db):
conn = database.get_connection(tmp_db)
row = conn.execute(
"SELECT value FROM meta WHERE key = 'db_version'"
).fetchone()
assert row is not None
assert int(row['value']) == 1
conn.close()
class TestSaveAndLoadDigest:
def test_round_trip(self, tmp_db, sample_digest):
database.save_digest(tmp_db, sample_digest)
loaded = database.get_latest_digest(tmp_db)
assert loaded is not None
assert loaded.digest_id == sample_digest.digest_id
assert loaded.week_start == sample_digest.week_start
assert loaded.total_fetched == 1
assert 'ml' in loaded.papers
assert len(loaded.papers['ml']) == 1
assert loaded.papers['ml'][0].title == 'Attention Is All You Need (Again)'
def test_load_empty_db(self, tmp_db):
result = database.get_latest_digest(tmp_db)
assert result is None
class TestBookmark:
def test_toggle_bookmark(self, tmp_db, sample_digest):
database.save_digest(tmp_db, sample_digest)
paper_id = 'arxiv:2401.12345'
# Initially False
state = database.toggle_bookmark(tmp_db, paper_id)
assert state is True
# Toggle back
state = database.toggle_bookmark(tmp_db, paper_id)
assert state is False
class TestMarkRead:
def test_mark_read(self, tmp_db, sample_digest):
database.save_digest(tmp_db, sample_digest)
database.mark_read(tmp_db, 'arxiv:2401.12345')
papers = database.get_papers(tmp_db, 'ml', limit=10)
assert len(papers) == 1
assert papers[0].is_read is True
class TestGetPapers:
def test_get_by_category(self, tmp_db, sample_digest):
database.save_digest(tmp_db, sample_digest)
papers = database.get_papers(tmp_db, 'ml')
assert len(papers) == 1
assert papers[0].app_category == 'ml'
def test_get_nonexistent_category(self, tmp_db, sample_digest):
database.save_digest(tmp_db, sample_digest)
papers = database.get_papers(tmp_db, 'nonexistent')
assert papers == []
|