'commit'

DouYin/dy-downloader/tests/test_database.py

@@ -0,0 +1,38 @@
+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),
+    })