pub/aerich
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve type hints for tests/

Browse Source
This commit is contained in:
Waket Zheng
2024-06-01 20:58:25 +08:00
parent 51117867a6
commit 1addda8178
5 changed files with 39 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/test_migrate.py
View File

@@ -1,4 +1,3 @@
import tempfile
from pathlib import Path
import pytest
@@ -971,14 +970,13 @@ def test_sort_all_version_files(mocker):
]
async def test_empty_migration(mocker) -> None:
async def test_empty_migration(mocker, tmp_path: Path) -> None:
mocker.patch("os.listdir", return_value=[])
Migrate.app = "foo"
expected_content = MIGRATE_TEMPLATE.format(upgrade_sql="", downgrade_sql="")
with tempfile.TemporaryDirectory() as temp_dir:
Migrate.migrate_location = temp_dir
Migrate.migrate_location = tmp_path
migration_file = await Migrate.migrate("update", True)
migration_file = await Migrate.migrate("update", True)
with open(Path(temp_dir, migration_file), "r") as f:
assert f.read() == expected_content
f = tmp_path / migration_file
assert f.read_text() == expected_content