Ajout de src.db.should_rebuild et ses tests

refs #71
This commit is contained in:
Colin Maudry
2026-04-15 16:07:41 +02:00
parent 0c3b0265b4
commit e83df64261
2 changed files with 95 additions and 0 deletions
+67
View File
@@ -0,0 +1,67 @@
import time
import pytest
@pytest.fixture
def parquet_and_db(tmp_path, monkeypatch):
parquet = tmp_path / "source.parquet"
db = tmp_path / "decp.duckdb"
parquet.write_bytes(b"fake parquet content")
monkeypatch.delenv("REBUILD_DUCKDB", raising=False)
monkeypatch.delenv("DEVELOPMENT", raising=False)
return parquet, db
def test_should_rebuild_when_db_missing(parquet_and_db, monkeypatch):
from src.db import should_rebuild
parquet, db = parquet_and_db
monkeypatch.setenv("DEVELOPMENT", "true")
assert should_rebuild(db, parquet) is True
def test_should_rebuild_prod_when_parquet_newer(parquet_and_db, monkeypatch):
from src.db import should_rebuild
parquet, db = parquet_and_db
db.write_bytes(b"x")
time.sleep(0.01)
parquet.touch()
monkeypatch.setenv("DEVELOPMENT", "false")
assert should_rebuild(db, parquet) is True
def test_should_not_rebuild_prod_when_parquet_older(parquet_and_db, monkeypatch):
from src.db import should_rebuild
parquet, db = parquet_and_db
parquet.touch()
time.sleep(0.01)
db.write_bytes(b"x")
monkeypatch.setenv("DEVELOPMENT", "false")
assert should_rebuild(db, parquet) is False
def test_should_not_rebuild_dev_even_when_parquet_newer(parquet_and_db, monkeypatch):
from src.db import should_rebuild
parquet, db = parquet_and_db
db.write_bytes(b"x")
time.sleep(0.01)
parquet.touch()
monkeypatch.setenv("DEVELOPMENT", "true")
monkeypatch.delenv("REBUILD_DUCKDB", raising=False)
assert should_rebuild(db, parquet) is False
def test_should_rebuild_dev_when_rebuild_forced(parquet_and_db, monkeypatch):
from src.db import should_rebuild
parquet, db = parquet_and_db
db.write_bytes(b"x")
time.sleep(0.01)
parquet.touch()
monkeypatch.setenv("DEVELOPMENT", "true")
monkeypatch.setenv("REBUILD_DUCKDB", "true")
assert should_rebuild(db, parquet) is True