Add models and routes for costs, consumption, equipment, maintenance, and production; implement CRUD operations and unit tests
This commit is contained in:
42
tests/unit/test_consumption.py
Normal file
42
tests/unit/test_consumption.py
Normal file
@@ -0,0 +1,42 @@
|
||||
from fastapi.testclient import TestClient
|
||||
from main import app
|
||||
from config.database import Base, engine
|
||||
|
||||
# Setup and teardown
|
||||
|
||||
|
||||
def setup_module(module):
|
||||
Base.metadata.create_all(bind=engine)
|
||||
|
||||
|
||||
def teardown_module(module):
|
||||
Base.metadata.drop_all(bind=engine)
|
||||
|
||||
|
||||
client = TestClient(app)
|
||||
|
||||
|
||||
def test_create_and_list_consumption():
|
||||
# Create a scenario to attach consumption
|
||||
resp = client.post(
|
||||
"/api/scenarios/", json={"name": "ConsScenario", "description": "consumption scenario"}
|
||||
)
|
||||
assert resp.status_code == 200
|
||||
scenario = resp.json()
|
||||
sid = scenario["id"]
|
||||
|
||||
# Create Consumption item
|
||||
cons_payload = {"scenario_id": sid, "amount": 250.0,
|
||||
"description": "Monthly consumption"}
|
||||
resp2 = client.post("/api/consumption/", json=cons_payload)
|
||||
assert resp2.status_code == 200
|
||||
cons = resp2.json()
|
||||
assert cons["scenario_id"] == sid
|
||||
assert cons["amount"] == 250.0
|
||||
|
||||
# List Consumption items
|
||||
resp3 = client.get("/api/consumption/")
|
||||
assert resp3.status_code == 200
|
||||
data = resp3.json()
|
||||
assert any(item["amount"] == 250.0 and item["scenario_id"]
|
||||
== sid for item in data)
|
||||
Reference in New Issue
Block a user