Colin Maudry
553e23dd98
- Import de should_rebuild au niveau module - Suppression d'un setenv DEVELOPMENT inutile (branche db-missing) - Utilisation de os.utime pour un ordre mtime déterministe refs #71 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
68 lines
2.0 KiB
Python
68 lines
2.0 KiB
Python
import os
|
|
import time
|
|
|
|
import pytest
|
|
|
|
from src.db import should_rebuild
|
|
|
|
|
|
@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):
|
|
parquet, db = parquet_and_db
|
|
assert should_rebuild(db, parquet) is True
|
|
|
|
|
|
def test_should_rebuild_prod_when_parquet_newer(parquet_and_db, monkeypatch):
|
|
parquet, db = parquet_and_db
|
|
db.write_bytes(b"x")
|
|
parquet.touch()
|
|
now = time.time()
|
|
os.utime(db, (now, now))
|
|
os.utime(parquet, (now + 10, now + 10))
|
|
monkeypatch.setenv("DEVELOPMENT", "false")
|
|
assert should_rebuild(db, parquet) is True
|
|
|
|
|
|
def test_should_not_rebuild_prod_when_parquet_older(parquet_and_db, monkeypatch):
|
|
parquet, db = parquet_and_db
|
|
parquet.touch()
|
|
db.write_bytes(b"x")
|
|
now = time.time()
|
|
os.utime(parquet, (now, now))
|
|
os.utime(db, (now + 10, now + 10))
|
|
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):
|
|
parquet, db = parquet_and_db
|
|
db.write_bytes(b"x")
|
|
parquet.touch()
|
|
now = time.time()
|
|
os.utime(db, (now, now))
|
|
os.utime(parquet, (now + 10, now + 10))
|
|
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):
|
|
parquet, db = parquet_and_db
|
|
db.write_bytes(b"x")
|
|
parquet.touch()
|
|
now = time.time()
|
|
os.utime(db, (now, now))
|
|
os.utime(parquet, (now + 10, now + 10))
|
|
monkeypatch.setenv("DEVELOPMENT", "true")
|
|
monkeypatch.setenv("REBUILD_DUCKDB", "true")
|
|
assert should_rebuild(db, parquet) is True
|