39 lines
1.0 KiB
Python
39 lines
1.0 KiB
Python
|
import json
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from storage import Database
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.asyncio
|
||
|
|
async def test_database_aweme_lifecycle(tmp_path):
|
||
|
|
db_path = tmp_path / "test.db"
|
||
|
|
database = Database(str(db_path))
|
||
|
|
|
||
|
|
await database.initialize()
|
||
|
|
|
||
|
|
aweme_payload = {
|
||
|
|
'aweme_id': '123',
|
||
|
|
'aweme_type': 'video',
|
||
|
|
'title': 'test',
|
||
|
|
'author_id': 'author',
|
||
|
|
'author_name': 'Author',
|
||
|
|
'create_time': 1700000000,
|
||
|
|
'file_path': '/tmp',
|
||
|
|
'metadata': json.dumps({'a': 1}, ensure_ascii=False),
|
||
|
|
}
|
||
|
|
|
||
|
|
await database.add_aweme(aweme_payload)
|
||
|
|
|
||
|
|
assert await database.is_downloaded('123') is True
|
||
|
|
assert await database.get_aweme_count_by_author('author') == 1
|
||
|
|
assert await database.get_latest_aweme_time('author') == 1700000000
|
||
|
|
|
||
|
|
await database.add_history({
|
||
|
|
'url': 'https://www.douyin.com/video/123',
|
||
|
|
'url_type': 'video',
|
||
|
|
'total_count': 1,
|
||
|
|
'success_count': 1,
|
||
|
|
'config': json.dumps({'path': './Downloaded/'}, ensure_ascii=False),
|
||
|
|
})
|