205 lines
6.8 KiB
Python
205 lines
6.8 KiB
Python
"""成本分析模块测试 — 标准成本/实际成本CRUD + ABC + 看板"""
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
from sqlalchemy.orm import Session
|
|
from tests.conftest import create_test_user, get_token_for_user, auth_header
|
|
from app.models import StandardCost, ActualCost, AbcActivity
|
|
|
|
|
|
class TestStandardCost:
|
|
"""标准成本卡片CRUD测试"""
|
|
|
|
BASE = "/api/cma/cost"
|
|
|
|
def test_list_standard_costs_empty(self, client: TestClient, db: Session):
|
|
"""空列表"""
|
|
create_test_user(db)
|
|
token = get_token_for_user(client)
|
|
|
|
resp = client.get(f"{self.BASE}/standard-costs", headers=auth_header(token))
|
|
assert resp.status_code == 200
|
|
assert resp.json()["data"] == []
|
|
|
|
def test_create_standard_cost(self, client: TestClient, db: Session):
|
|
"""创建标准成本卡片"""
|
|
create_test_user(db)
|
|
token = get_token_for_user(client)
|
|
|
|
resp = client.post(
|
|
f"{self.BASE}/standard-costs",
|
|
headers=auth_header(token),
|
|
json={
|
|
"product_code": "PROD_A",
|
|
"product_name": "产品A",
|
|
"cost_type": "material",
|
|
"item_name": "原材料X",
|
|
"standard_quantity": 10.0,
|
|
"standard_price": 5.0,
|
|
"unit": "kg",
|
|
},
|
|
)
|
|
assert resp.status_code == 200
|
|
assert resp.json()["message"] == "标准成本已创建"
|
|
|
|
# 验证列表中有数据
|
|
list_resp = client.get(f"{self.BASE}/standard-costs", headers=auth_header(token))
|
|
assert len(list_resp.json()["data"]) == 1
|
|
|
|
def test_update_standard_cost(self, client: TestClient, db: Session):
|
|
"""修改标准成本卡片"""
|
|
create_test_user(db)
|
|
token = get_token_for_user(client)
|
|
|
|
sc = StandardCost(
|
|
product_code="PROD_B", product_name="产品B", cost_type="labor",
|
|
item_name="人工", standard_quantity=100, unit="小时",
|
|
standard_price=50.0, standard_cost=5000.0,
|
|
)
|
|
db.add(sc)
|
|
db.commit()
|
|
|
|
resp = client.put(
|
|
f"{self.BASE}/standard-costs/{sc.id}",
|
|
headers=auth_header(token),
|
|
json={"standard_price": 55.0, "remark": "调薪后更新"},
|
|
)
|
|
assert resp.status_code == 200
|
|
assert resp.json()["message"] == "已更新"
|
|
|
|
def test_delete_standard_cost(self, client: TestClient, db: Session):
|
|
"""删除(归档)标准成本卡片"""
|
|
create_test_user(db)
|
|
token = get_token_for_user(client)
|
|
|
|
sc = StandardCost(
|
|
product_code="PROD_C", product_name="产品C", cost_type="overhead",
|
|
item_name="水电费", standard_quantity=1, unit="项",
|
|
standard_price=1000.0, standard_cost=1000.0,
|
|
)
|
|
db.add(sc)
|
|
db.commit()
|
|
|
|
resp = client.delete(
|
|
f"{self.BASE}/standard-costs/{sc.id}",
|
|
headers=auth_header(token),
|
|
)
|
|
assert resp.status_code == 200
|
|
assert resp.json()["message"] == "已归档"
|
|
|
|
|
|
class TestActualCost:
|
|
"""实际成本CRUD测试"""
|
|
|
|
BASE = "/api/cma/cost"
|
|
|
|
def test_create_actual_cost(self, client: TestClient, db: Session):
|
|
"""录入实际成本"""
|
|
create_test_user(db)
|
|
token = get_token_for_user(client)
|
|
|
|
resp = client.post(
|
|
f"{self.BASE}/actual-costs",
|
|
headers=auth_header(token),
|
|
json={
|
|
"period": "2026-06",
|
|
"product_code": "PROD_A",
|
|
"product_name": "产品A",
|
|
"cost_type": "material",
|
|
"item_name": "原材料X",
|
|
"actual_quantity": 12.0,
|
|
"actual_price": 4.8,
|
|
},
|
|
)
|
|
assert resp.status_code == 200
|
|
assert resp.json()["message"] == "实际成本已录入"
|
|
|
|
def test_list_actual_costs(self, client: TestClient, db: Session):
|
|
"""查询实际成本"""
|
|
create_test_user(db)
|
|
token = get_token_for_user(client)
|
|
|
|
ac = ActualCost(
|
|
period="2026-06", product_code="PROD_A", product_name="产品A",
|
|
cost_type="material", item_name="原材料X",
|
|
actual_quantity=10, actual_price=5.0, actual_cost=50.0,
|
|
)
|
|
db.add(ac)
|
|
db.commit()
|
|
|
|
resp = client.get(f"{self.BASE}/actual-costs?period=2026-06", headers=auth_header(token))
|
|
assert resp.status_code == 200
|
|
assert len(resp.json()["data"]) == 1
|
|
|
|
|
|
class TestAbcCost:
|
|
"""ABC作业成本测试"""
|
|
|
|
BASE = "/api/cma/cost"
|
|
|
|
def test_create_abc_activity(self, client: TestClient, db: Session):
|
|
"""创建ABC作业中心"""
|
|
create_test_user(db)
|
|
token = get_token_for_user(client)
|
|
|
|
resp = client.post(
|
|
f"{self.BASE}/abc/activities",
|
|
headers=auth_header(token),
|
|
json={
|
|
"activity_code": "ACT_001",
|
|
"activity_name": "机器调试",
|
|
"cost_driver": "调试次数",
|
|
"driver_unit": "次",
|
|
"total_cost": 50000.0,
|
|
"driver_volume": 100,
|
|
},
|
|
)
|
|
assert resp.status_code == 200
|
|
assert resp.json()["message"] == "作业中心已创建"
|
|
|
|
def test_list_abc_activities(self, client: TestClient, db: Session):
|
|
"""查询ABC作业中心列表"""
|
|
create_test_user(db)
|
|
token = get_token_for_user(client)
|
|
|
|
act = AbcActivity(
|
|
activity_code="ACT_002", activity_name="品质检验",
|
|
cost_driver="检验批次", driver_unit="批",
|
|
total_cost=30000.0, driver_volume=200, driver_rate=150.0,
|
|
)
|
|
db.add(act)
|
|
db.commit()
|
|
|
|
resp = client.get(f"{self.BASE}/abc/activities", headers=auth_header(token))
|
|
assert resp.status_code == 200
|
|
assert len(resp.json()["data"]) >= 1
|
|
assert resp.json()["data"][0]["activity_code"] == "ACT_002"
|
|
|
|
def test_abc_allocate(self, client: TestClient, db: Session):
|
|
"""执行ABC成本分配"""
|
|
create_test_user(db)
|
|
token = get_token_for_user(client)
|
|
|
|
act = AbcActivity(
|
|
activity_code="ACT_003", activity_name="订单处理",
|
|
cost_driver="订单数", driver_unit="单",
|
|
total_cost=20000.0, driver_volume=400, driver_rate=50.0,
|
|
)
|
|
db.add(act)
|
|
db.commit()
|
|
|
|
resp = client.post(
|
|
f"{self.BASE}/abc/allocate",
|
|
headers=auth_header(token),
|
|
json={
|
|
"activity_id": act.id,
|
|
"period": "2026-06",
|
|
"product_code": "PROD_A",
|
|
"product_name": "产品A",
|
|
"driver_consumed": 30,
|
|
},
|
|
)
|
|
assert resp.status_code == 200
|
|
data = resp.json()
|
|
# 30 * 50 = 1500
|
|
assert data["allocated_cost"] == 1500.0
|