mirror of
https://github.com/lrnselfreliance/wrolpi.git
synced 2026-10-26 19:01:39 -04:00
29 lines
862 B
Python
29 lines
862 B
Python
import pytest
|
|
|
|
from modules.inventory import Inventory, Item, DEFAULT_CATEGORIES, DEFAULT_INVENTORIES
|
|
from modules.inventory.common import set_test_inventories_config
|
|
|
|
|
|
@pytest.fixture
|
|
def test_inventory(test_session):
|
|
inventory = Inventory(name='Test Inventory')
|
|
test_session.add(inventory)
|
|
test_session.commit()
|
|
return inventory
|
|
|
|
|
|
@pytest.fixture
|
|
def init_test_inventory(test_session):
|
|
for subcategory, category in DEFAULT_CATEGORIES:
|
|
item = Item(subcategory=subcategory, category=category)
|
|
test_session.add(item)
|
|
|
|
for name in DEFAULT_INVENTORIES:
|
|
inv = Inventory(name=name)
|
|
test_session.add(inv)
|
|
|
|
inventory = test_session.query(Inventory).filter_by(name='Food Storage').one()
|
|
test_session.commit()
|
|
set_test_inventories_config(True)
|
|
yield inventory
|
|
set_test_inventories_config(False)
|