159 lines
5.0 KiB
Python
159 lines
5.0 KiB
Python
from datetime import date
|
|
from io import BytesIO
|
|
from types import SimpleNamespace
|
|
|
|
from openpyxl import load_workbook
|
|
|
|
from app.routers import dashboard
|
|
from app.services.excel_export import export_dashboard_rows
|
|
|
|
|
|
def test_dashboard_rows_from_allocations_groups_by_allocation_date_desc():
|
|
report_date = date(2026, 7, 25)
|
|
report = SimpleNamespace(
|
|
id=1,
|
|
attendance_point_name="总厂",
|
|
report_date=report_date,
|
|
employee_phone="13800000000",
|
|
employee=SimpleNamespace(name="张三", is_temporary=False),
|
|
items=[],
|
|
audit_logs=[],
|
|
session=None,
|
|
is_system_auto_submitted=False,
|
|
is_voided=False,
|
|
is_multi_person_assistant=False,
|
|
review_remark="复核通过",
|
|
)
|
|
item = SimpleNamespace(
|
|
id=10,
|
|
attendance_point_name="总厂",
|
|
device_no="A-01",
|
|
project_no="P1",
|
|
product_name="23#",
|
|
process_name="冲压",
|
|
remark=None,
|
|
raw_material_batch_no="BATCH-1",
|
|
material_code="M-1",
|
|
material_name="铝材",
|
|
profile_no="PF-1",
|
|
stamping_method="单冲",
|
|
operator_count=1,
|
|
process_unit_price_yuan=2,
|
|
standard_beat=1,
|
|
good_qty=650,
|
|
defect_qty=0,
|
|
scrap_qty=0,
|
|
changeover_count=2,
|
|
allocated_minutes=650,
|
|
)
|
|
report.items = [item]
|
|
allocations = [
|
|
SimpleNamespace(
|
|
report=report,
|
|
item=item,
|
|
allocation_date=date(2026, 7, 24),
|
|
night_minutes=350,
|
|
overtime_minutes=0,
|
|
day_minutes=0,
|
|
effective_minutes=350,
|
|
good_qty=350,
|
|
defect_qty=0,
|
|
scrap_qty=0,
|
|
changeover_count=1,
|
|
reference_wage=350,
|
|
),
|
|
SimpleNamespace(
|
|
report=report,
|
|
item=item,
|
|
allocation_date=date(2026, 7, 25),
|
|
night_minutes=0,
|
|
overtime_minutes=60,
|
|
day_minutes=240,
|
|
effective_minutes=300,
|
|
good_qty=300,
|
|
defect_qty=0,
|
|
scrap_qty=0,
|
|
changeover_count=1,
|
|
reference_wage=300,
|
|
),
|
|
]
|
|
|
|
rows = dashboard._dashboard_rows_from_allocations(allocations)
|
|
|
|
assert [row.report_date for row in rows] == [date(2026, 7, 25), date(2026, 7, 24)]
|
|
assert [row.source_report_dates for row in rows] == [[report_date], [report_date]]
|
|
assert [row.source_report_date for row in rows] == [report_date, report_date]
|
|
assert [row.report_count for row in rows] == [1, 1]
|
|
|
|
current, previous = rows
|
|
assert current.effective_minutes == 300
|
|
assert current.shift_day_minutes == 240
|
|
assert current.shift_overtime_minutes == 60
|
|
assert current.shift_night_minutes == 0
|
|
assert current.total_good_qty == 300
|
|
assert current.total_defect_qty == 0
|
|
assert current.total_output_qty == 300
|
|
assert current.changeover_count == 1
|
|
assert current.reference_wage == 300
|
|
|
|
assert previous.effective_minutes == 350
|
|
assert previous.shift_day_minutes == 0
|
|
assert previous.shift_overtime_minutes == 0
|
|
assert previous.shift_night_minutes == 350
|
|
assert previous.total_good_qty == 350
|
|
assert previous.total_defect_qty == 0
|
|
assert previous.total_output_qty == 350
|
|
assert previous.changeover_count == 1
|
|
assert previous.reference_wage == 350
|
|
|
|
|
|
def test_export_dashboard_rows_writes_allocation_and_source_dates_with_warning_columns():
|
|
row = SimpleNamespace(
|
|
attendance_point_name="总厂",
|
|
report_date=date(2026, 7, 24),
|
|
source_report_dates=[date(2026, 7, 25), date(2026, 7, 23)],
|
|
source_report_date=None,
|
|
employee_name="张三",
|
|
employee_phone="13800000000",
|
|
worker_type="正式工",
|
|
project_no="P1",
|
|
product_name="23#",
|
|
process_name="冲压",
|
|
effective_minutes=60,
|
|
shift_distribution_text="白班1小时",
|
|
total_good_qty=50,
|
|
total_defect_qty=2,
|
|
expected_workload=100,
|
|
actual_beat=20,
|
|
standard_beat=10,
|
|
profile_no="PF-1",
|
|
material_code="M-1",
|
|
material_name="铝材",
|
|
raw_material_batch_no="BATCH-1",
|
|
device_no="A-01",
|
|
stamping_method="单冲",
|
|
changeover_count=1,
|
|
operator_count=1,
|
|
process_unit_price_yuan=2,
|
|
reference_wage=100,
|
|
is_voided=False,
|
|
is_multi_person=False,
|
|
is_continuous_die=False,
|
|
correction_pairs=[],
|
|
review_remark="复核通过",
|
|
)
|
|
|
|
workbook = load_workbook(BytesIO(export_dashboard_rows([row])))
|
|
sheet = workbook["报工看板"]
|
|
|
|
assert [sheet.cell(row=1, column=index).value for index in range(1, 5)] == [
|
|
"考勤点",
|
|
"统计归属日期",
|
|
"原始报工日期",
|
|
"员工姓名",
|
|
]
|
|
assert sheet.cell(row=2, column=2).value == "2026-07-24"
|
|
assert sheet.cell(row=2, column=3).value == "2026-07-23、2026-07-25"
|
|
assert sheet.cell(row=2, column=12).fill.fgColor.rgb == "FFFF6666"
|
|
assert sheet.cell(row=2, column=15).fill.fgColor.rgb == "FFFF6666"
|