48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
from sqlalchemy import select
|
|
from sqlalchemy.orm import Session
|
|
|
|
from src.ingest.b4w import upsert_category
|
|
from src.models import Category
|
|
from src.models.b4w import CategoryDb
|
|
|
|
|
|
def test_upsert_category_returns_orm_object(db_session: Session):
|
|
result = upsert_category(
|
|
db_session,
|
|
Category.from_str("dummy_parent_1:dummy_parent_2:dummy"),
|
|
do_commit=False,
|
|
)
|
|
assert result
|
|
assert isinstance(result, CategoryDb)
|
|
assert str(result.name) == "dummy"
|
|
assert str(result.parent.name) == "dummy_parent_2"
|
|
assert str(result.parent.parent.name) == "dummy_parent_1"
|
|
|
|
|
|
def test_upsert_category_writes_new_to_db(db_session: Session):
|
|
upsert_category(
|
|
db_session,
|
|
Category.from_str("grand_father:father:new_category"),
|
|
do_commit=True,
|
|
)
|
|
result = db_session.execute(
|
|
select(CategoryDb).filter_by(name="new_category")
|
|
).scalar_one()
|
|
assert result
|
|
assert str(result.name) == "new_category"
|
|
assert str(result.parent.name) == "father"
|
|
assert str(result.parent.parent.name) == "grand_father"
|
|
|
|
|
|
def test_upsert_category_skips_existing(db_session: Session):
|
|
for _ in range(3):
|
|
upsert_category(
|
|
db_session,
|
|
Category.from_str("grand_father:father:try_again_and_again"),
|
|
do_commit=True,
|
|
)
|
|
result = db_session.execute(
|
|
select(CategoryDb).filter_by(name="try_again_and_again")
|
|
).all()
|
|
assert len(result) == 1
|