功能：新增托盘业务接口模型
This commit is contained in:
焦龙言
@@ -0,0 +1,785 @@
|
||||
import unittest
|
||||
|
||||
from pydantic import ValidationError
|
||||
from pydantic.alias_generators import to_camel
|
||||
|
||||
from module_business.entity.vo.master_vo import (
|
||||
CustomerModel,
|
||||
CustomerPageQueryModel,
|
||||
DeleteCustomerModel,
|
||||
DeleteLineModel,
|
||||
DeletePalletModel,
|
||||
DeleteSiteModel,
|
||||
DeleteStrainModel,
|
||||
DeleteWarehouseModel,
|
||||
LineModel,
|
||||
LinePageQueryModel,
|
||||
PalletCreateModel,
|
||||
PalletEditModel,
|
||||
PalletModel,
|
||||
PalletPageQueryModel,
|
||||
SiteModel,
|
||||
SitePageQueryModel,
|
||||
StrainModel,
|
||||
StrainPageQueryModel,
|
||||
UserAssignModel,
|
||||
WarehouseModel,
|
||||
WarehousePageQueryModel,
|
||||
)
|
||||
from module_business.entity.vo.operation_vo import (
|
||||
BusinessOptionModel,
|
||||
ConfirmCommandModel,
|
||||
FillingCreateModel,
|
||||
FillingModel,
|
||||
FillingPageQueryModel,
|
||||
InboundCreateModel,
|
||||
InboundItemModel,
|
||||
InboundModel,
|
||||
InboundPageQueryModel,
|
||||
InboundScanQueryModel,
|
||||
InboundUpdateModel,
|
||||
PalletScanResponseModel,
|
||||
ReturnConfirmModel,
|
||||
ReturnItemModel,
|
||||
ReturnModel,
|
||||
ReturnPageQueryModel,
|
||||
ReturnRequestModel,
|
||||
ReturnScanQueryModel,
|
||||
ShipmentCreateModel,
|
||||
ShipmentItemModel,
|
||||
ShipmentModel,
|
||||
ShipmentPageQueryModel,
|
||||
ShipmentScanQueryModel,
|
||||
ShipmentUpdateModel,
|
||||
UsageCreateModel,
|
||||
UsageModel,
|
||||
UsagePageQueryModel,
|
||||
)
|
||||
from module_business.entity.vo.report_vo import (
|
||||
ConsumptionReportModel,
|
||||
ConsumptionReportQueryModel,
|
||||
FillingReportModel,
|
||||
FillingReportQueryModel,
|
||||
PalletInventoryReportModel,
|
||||
PalletInventoryReportQueryModel,
|
||||
PalletTraceReportModel,
|
||||
PalletTraceReportQueryModel,
|
||||
ReturnReportModel,
|
||||
ReturnReportQueryModel,
|
||||
ShipmentReportModel,
|
||||
ShipmentReportQueryModel,
|
||||
)
|
||||
|
||||
MASTER_MODEL_FIELDS = {
|
||||
StrainModel: {
|
||||
'strain_id',
|
||||
'strain_code',
|
||||
'strain_name',
|
||||
'status',
|
||||
'create_by',
|
||||
'create_time',
|
||||
'update_by',
|
||||
'update_time',
|
||||
'remark',
|
||||
},
|
||||
StrainPageQueryModel: {'page_num', 'page_size', 'strain_code', 'strain_name', 'status'},
|
||||
DeleteStrainModel: {'strain_ids'},
|
||||
CustomerModel: {
|
||||
'customer_id',
|
||||
'customer_code',
|
||||
'customer_name',
|
||||
'status',
|
||||
'create_by',
|
||||
'create_time',
|
||||
'update_by',
|
||||
'update_time',
|
||||
'remark',
|
||||
},
|
||||
CustomerPageQueryModel: {'page_num', 'page_size', 'customer_code', 'customer_name', 'status'},
|
||||
DeleteCustomerModel: {'customer_ids'},
|
||||
SiteModel: {
|
||||
'site_id',
|
||||
'customer_id',
|
||||
'site_code',
|
||||
'site_name',
|
||||
'address',
|
||||
'status',
|
||||
'create_by',
|
||||
'create_time',
|
||||
'update_by',
|
||||
'update_time',
|
||||
'remark',
|
||||
},
|
||||
SitePageQueryModel: {'page_num', 'page_size', 'customer_id', 'site_code', 'site_name', 'status'},
|
||||
DeleteSiteModel: {'site_ids'},
|
||||
LineModel: {
|
||||
'line_id',
|
||||
'site_id',
|
||||
'line_code',
|
||||
'line_name',
|
||||
'status',
|
||||
'create_by',
|
||||
'create_time',
|
||||
'update_by',
|
||||
'update_time',
|
||||
'remark',
|
||||
},
|
||||
LinePageQueryModel: {'page_num', 'page_size', 'site_id', 'line_code', 'line_name', 'status'},
|
||||
DeleteLineModel: {'line_ids'},
|
||||
WarehouseModel: {
|
||||
'warehouse_id',
|
||||
'warehouse_code',
|
||||
'warehouse_name',
|
||||
'address',
|
||||
'status',
|
||||
'create_by',
|
||||
'create_time',
|
||||
'update_by',
|
||||
'update_time',
|
||||
'remark',
|
||||
},
|
||||
WarehousePageQueryModel: {'page_num', 'page_size', 'warehouse_code', 'warehouse_name', 'status'},
|
||||
DeleteWarehouseModel: {'warehouse_ids'},
|
||||
PalletModel: {
|
||||
'pallet_id',
|
||||
'pallet_code',
|
||||
'qr_token',
|
||||
'capacity_packages',
|
||||
'asset_status',
|
||||
'remark',
|
||||
'create_by',
|
||||
'create_time',
|
||||
'update_by',
|
||||
'update_time',
|
||||
'can_remove',
|
||||
},
|
||||
PalletCreateModel: {'pallet_code', 'capacity_packages', 'remark'},
|
||||
PalletEditModel: {'pallet_id', 'capacity_packages', 'asset_status', 'remark'},
|
||||
PalletPageQueryModel: {'page_num', 'page_size', 'pallet_code', 'asset_status'},
|
||||
DeletePalletModel: {'pallet_ids'},
|
||||
UserAssignModel: {'user_ids'},
|
||||
}
|
||||
|
||||
OPERATION_MODEL_FIELDS = {
|
||||
InboundCreateModel: {'warehouse_id', 'pallet_ids'},
|
||||
InboundUpdateModel: {'inbound_id', 'warehouse_id', 'pallet_ids'},
|
||||
ConfirmCommandModel: {'idempotency_key'},
|
||||
FillingCreateModel: {
|
||||
'qr_token',
|
||||
'warehouse_id',
|
||||
'strain_id',
|
||||
'batch_no',
|
||||
'package_spec',
|
||||
'initial_packages',
|
||||
'idempotency_key',
|
||||
},
|
||||
ShipmentCreateModel: {'source_warehouse_id', 'target_site_id', 'pallet_ids'},
|
||||
ShipmentUpdateModel: {'shipment_id', 'source_warehouse_id', 'target_site_id', 'pallet_ids'},
|
||||
UsageCreateModel: {'qr_token', 'line_id', 'used_packages', 'idempotency_key'},
|
||||
ReturnRequestModel: {'source_site_id', 'target_warehouse_id', 'pallet_ids', 'idempotency_key'},
|
||||
ReturnConfirmModel: {'scanned_pallet_ids', 'idempotency_key'},
|
||||
InboundItemModel: {'inbound_item_id', 'inbound_id', 'pallet_id', 'create_by', 'create_time'},
|
||||
InboundModel: {
|
||||
'inbound_id',
|
||||
'inbound_no',
|
||||
'warehouse_id',
|
||||
'status',
|
||||
'confirm_idempotency_key',
|
||||
'confirm_request_hash',
|
||||
'confirmed_by',
|
||||
'confirmed_at',
|
||||
'create_by',
|
||||
'create_time',
|
||||
'update_by',
|
||||
'update_time',
|
||||
'remark',
|
||||
'items',
|
||||
},
|
||||
FillingModel: {
|
||||
'filling_id',
|
||||
'filling_no',
|
||||
'pallet_id',
|
||||
'warehouse_id',
|
||||
'strain_id',
|
||||
'strain_code_snapshot',
|
||||
'strain_name_snapshot',
|
||||
'batch_no',
|
||||
'package_spec',
|
||||
'initial_packages',
|
||||
'active_slot',
|
||||
'closed_at',
|
||||
'idempotency_key',
|
||||
'request_hash',
|
||||
'operator_id',
|
||||
'operated_at',
|
||||
'create_by',
|
||||
'create_time',
|
||||
'update_by',
|
||||
'update_time',
|
||||
'remark',
|
||||
},
|
||||
ShipmentItemModel: {
|
||||
'shipment_item_id',
|
||||
'shipment_id',
|
||||
'pallet_id',
|
||||
'active_filling_id',
|
||||
'package_qty',
|
||||
'create_by',
|
||||
'create_time',
|
||||
},
|
||||
ShipmentModel: {
|
||||
'shipment_id',
|
||||
'shipment_no',
|
||||
'source_warehouse_id',
|
||||
'target_site_id',
|
||||
'status',
|
||||
'confirm_idempotency_key',
|
||||
'confirm_request_hash',
|
||||
'confirmed_by',
|
||||
'confirmed_at',
|
||||
'create_by',
|
||||
'create_time',
|
||||
'update_by',
|
||||
'update_time',
|
||||
'remark',
|
||||
'items',
|
||||
},
|
||||
UsageModel: {
|
||||
'usage_id',
|
||||
'usage_no',
|
||||
'pallet_id',
|
||||
'active_filling_id',
|
||||
'site_id',
|
||||
'line_id',
|
||||
'used_packages',
|
||||
'remaining_packages_after',
|
||||
'idempotency_key',
|
||||
'request_hash',
|
||||
'operator_id',
|
||||
'occurred_at',
|
||||
},
|
||||
ReturnItemModel: {
|
||||
'return_item_id',
|
||||
'return_id',
|
||||
'pallet_id',
|
||||
'active_filling_id_snapshot',
|
||||
'remaining_packages_snapshot',
|
||||
'create_by',
|
||||
'create_time',
|
||||
},
|
||||
ReturnModel: {
|
||||
'return_id',
|
||||
'return_no',
|
||||
'source_site_id',
|
||||
'target_warehouse_id',
|
||||
'status',
|
||||
'request_idempotency_key',
|
||||
'request_hash',
|
||||
'confirm_idempotency_key',
|
||||
'confirm_request_hash',
|
||||
'requested_by',
|
||||
'requested_at',
|
||||
'confirmed_by',
|
||||
'confirmed_at',
|
||||
'create_by',
|
||||
'create_time',
|
||||
'update_by',
|
||||
'update_time',
|
||||
'remark',
|
||||
'items',
|
||||
},
|
||||
InboundPageQueryModel: {
|
||||
'page_num',
|
||||
'page_size',
|
||||
'inbound_no',
|
||||
'warehouse_id',
|
||||
'status',
|
||||
'begin_time',
|
||||
'end_time',
|
||||
},
|
||||
FillingPageQueryModel: {
|
||||
'page_num',
|
||||
'page_size',
|
||||
'filling_no',
|
||||
'warehouse_id',
|
||||
'strain_id',
|
||||
'batch_no',
|
||||
'pallet_code',
|
||||
'begin_time',
|
||||
'end_time',
|
||||
},
|
||||
ShipmentPageQueryModel: {
|
||||
'page_num',
|
||||
'page_size',
|
||||
'shipment_no',
|
||||
'source_warehouse_id',
|
||||
'target_site_id',
|
||||
'status',
|
||||
'begin_time',
|
||||
'end_time',
|
||||
},
|
||||
UsagePageQueryModel: {
|
||||
'page_num',
|
||||
'page_size',
|
||||
'usage_no',
|
||||
'site_id',
|
||||
'line_id',
|
||||
'pallet_code',
|
||||
'begin_time',
|
||||
'end_time',
|
||||
},
|
||||
ReturnPageQueryModel: {
|
||||
'page_num',
|
||||
'page_size',
|
||||
'return_no',
|
||||
'source_site_id',
|
||||
'target_warehouse_id',
|
||||
'status',
|
||||
'begin_time',
|
||||
'end_time',
|
||||
},
|
||||
InboundScanQueryModel: {'warehouse_id'},
|
||||
ShipmentScanQueryModel: {'source_warehouse_id', 'target_site_id'},
|
||||
ReturnScanQueryModel: {'mode', 'return_id'},
|
||||
BusinessOptionModel: {'id', 'code', 'name'},
|
||||
PalletScanResponseModel: {
|
||||
'pallet_id',
|
||||
'pallet_code',
|
||||
'flow_state',
|
||||
'warehouse_id',
|
||||
'site_id',
|
||||
'current_line_id',
|
||||
'active_filling_id',
|
||||
'strain_id',
|
||||
'strain_code',
|
||||
'strain_name',
|
||||
'batch_no',
|
||||
'package_spec',
|
||||
'capacity_packages',
|
||||
'remaining_packages',
|
||||
'allowed_actions',
|
||||
},
|
||||
}
|
||||
|
||||
REPORT_MODEL_FIELDS = {
|
||||
PalletInventoryReportQueryModel: {
|
||||
'page_num',
|
||||
'page_size',
|
||||
'begin_time',
|
||||
'end_time',
|
||||
'pallet_code',
|
||||
'flow_state',
|
||||
'warehouse_id',
|
||||
'site_id',
|
||||
'line_id',
|
||||
'strain_id',
|
||||
'batch_no',
|
||||
},
|
||||
PalletInventoryReportModel: {
|
||||
'pallet_id',
|
||||
'pallet_code',
|
||||
'flow_state',
|
||||
'warehouse_id',
|
||||
'warehouse_name',
|
||||
'site_id',
|
||||
'site_name',
|
||||
'line_id',
|
||||
'line_name',
|
||||
'strain_id',
|
||||
'strain_code',
|
||||
'strain_name',
|
||||
'batch_no',
|
||||
'initial_packages',
|
||||
'remaining_packages',
|
||||
'updated_at',
|
||||
},
|
||||
FillingReportQueryModel: {
|
||||
'page_num',
|
||||
'page_size',
|
||||
'begin_time',
|
||||
'end_time',
|
||||
'warehouse_id',
|
||||
'pallet_code',
|
||||
'strain_id',
|
||||
'batch_no',
|
||||
},
|
||||
FillingReportModel: {
|
||||
'filling_id',
|
||||
'filling_no',
|
||||
'operated_at',
|
||||
'warehouse_id',
|
||||
'warehouse_name',
|
||||
'pallet_id',
|
||||
'pallet_code',
|
||||
'strain_id',
|
||||
'strain_code_snapshot',
|
||||
'strain_name_snapshot',
|
||||
'batch_no',
|
||||
'package_spec',
|
||||
'initial_packages',
|
||||
'operator_id',
|
||||
'operator_name',
|
||||
},
|
||||
ShipmentReportQueryModel: {
|
||||
'page_num',
|
||||
'page_size',
|
||||
'begin_time',
|
||||
'end_time',
|
||||
'shipment_no',
|
||||
'source_warehouse_id',
|
||||
'target_site_id',
|
||||
'pallet_code',
|
||||
'status',
|
||||
},
|
||||
ShipmentReportModel: {
|
||||
'shipment_id',
|
||||
'shipment_no',
|
||||
'confirmed_at',
|
||||
'source_warehouse_id',
|
||||
'source_warehouse_name',
|
||||
'target_site_id',
|
||||
'target_site_name',
|
||||
'customer_id',
|
||||
'customer_name',
|
||||
'pallet_id',
|
||||
'pallet_code',
|
||||
'active_filling_id',
|
||||
'package_qty',
|
||||
},
|
||||
ConsumptionReportQueryModel: {
|
||||
'page_num',
|
||||
'page_size',
|
||||
'begin_time',
|
||||
'end_time',
|
||||
'customer_id',
|
||||
'site_id',
|
||||
'line_id',
|
||||
'pallet_code',
|
||||
'strain_id',
|
||||
'batch_no',
|
||||
},
|
||||
ConsumptionReportModel: {
|
||||
'usage_id',
|
||||
'usage_no',
|
||||
'occurred_at',
|
||||
'customer_id',
|
||||
'customer_name',
|
||||
'site_id',
|
||||
'site_name',
|
||||
'line_id',
|
||||
'line_name',
|
||||
'pallet_id',
|
||||
'pallet_code',
|
||||
'strain_id',
|
||||
'strain_code',
|
||||
'strain_name',
|
||||
'batch_no',
|
||||
'used_packages',
|
||||
'remaining_packages_after',
|
||||
'operator_id',
|
||||
'operator_name',
|
||||
},
|
||||
ReturnReportQueryModel: {
|
||||
'page_num',
|
||||
'page_size',
|
||||
'begin_time',
|
||||
'end_time',
|
||||
'return_no',
|
||||
'source_site_id',
|
||||
'target_warehouse_id',
|
||||
'pallet_code',
|
||||
'status',
|
||||
},
|
||||
ReturnReportModel: {
|
||||
'return_id',
|
||||
'return_no',
|
||||
'status',
|
||||
'requested_at',
|
||||
'confirmed_at',
|
||||
'source_site_id',
|
||||
'source_site_name',
|
||||
'target_warehouse_id',
|
||||
'target_warehouse_name',
|
||||
'pallet_id',
|
||||
'pallet_code',
|
||||
'active_filling_id_snapshot',
|
||||
'remaining_packages_snapshot',
|
||||
},
|
||||
PalletTraceReportQueryModel: {
|
||||
'page_num',
|
||||
'page_size',
|
||||
'begin_time',
|
||||
'end_time',
|
||||
'pallet_code',
|
||||
'event_type',
|
||||
},
|
||||
PalletTraceReportModel: {
|
||||
'event_id',
|
||||
'event_seq',
|
||||
'event_type',
|
||||
'source_type',
|
||||
'source_id',
|
||||
'pallet_id',
|
||||
'pallet_code',
|
||||
'before_state',
|
||||
'after_state',
|
||||
'before_warehouse_id',
|
||||
'before_warehouse_name',
|
||||
'before_site_id',
|
||||
'before_site_name',
|
||||
'before_line_id',
|
||||
'before_line_name',
|
||||
'after_warehouse_id',
|
||||
'after_warehouse_name',
|
||||
'after_site_id',
|
||||
'after_site_name',
|
||||
'after_line_id',
|
||||
'after_line_name',
|
||||
'before_remaining_packages',
|
||||
'after_remaining_packages',
|
||||
'operator_id',
|
||||
'operator_name',
|
||||
'occurred_at',
|
||||
},
|
||||
}
|
||||
|
||||
WRITE_MODELS = (
|
||||
StrainModel,
|
||||
DeleteStrainModel,
|
||||
CustomerModel,
|
||||
DeleteCustomerModel,
|
||||
SiteModel,
|
||||
DeleteSiteModel,
|
||||
LineModel,
|
||||
DeleteLineModel,
|
||||
WarehouseModel,
|
||||
DeleteWarehouseModel,
|
||||
PalletCreateModel,
|
||||
PalletEditModel,
|
||||
DeletePalletModel,
|
||||
UserAssignModel,
|
||||
InboundCreateModel,
|
||||
InboundUpdateModel,
|
||||
ConfirmCommandModel,
|
||||
FillingCreateModel,
|
||||
ShipmentCreateModel,
|
||||
ShipmentUpdateModel,
|
||||
UsageCreateModel,
|
||||
ReturnRequestModel,
|
||||
ReturnConfirmModel,
|
||||
)
|
||||
|
||||
PAGE_QUERY_MODELS = (
|
||||
StrainPageQueryModel,
|
||||
CustomerPageQueryModel,
|
||||
SitePageQueryModel,
|
||||
LinePageQueryModel,
|
||||
WarehousePageQueryModel,
|
||||
PalletPageQueryModel,
|
||||
InboundPageQueryModel,
|
||||
FillingPageQueryModel,
|
||||
ShipmentPageQueryModel,
|
||||
UsagePageQueryModel,
|
||||
ReturnPageQueryModel,
|
||||
PalletInventoryReportQueryModel,
|
||||
FillingReportQueryModel,
|
||||
ShipmentReportQueryModel,
|
||||
ConsumptionReportQueryModel,
|
||||
ReturnReportQueryModel,
|
||||
PalletTraceReportQueryModel,
|
||||
)
|
||||
|
||||
|
||||
class BusinessVoContractTest(unittest.TestCase):
|
||||
def test_all_models_have_exact_fields_and_camel_case_aliases(self) -> None:
|
||||
expected_models = MASTER_MODEL_FIELDS | OPERATION_MODEL_FIELDS | REPORT_MODEL_FIELDS
|
||||
|
||||
for model_class, expected_fields in expected_models.items():
|
||||
with self.subTest(model=model_class.__name__):
|
||||
self.assertEqual(expected_fields, set(model_class.model_fields))
|
||||
self.assertTrue(model_class.model_config.get('from_attributes'))
|
||||
self.assertTrue(model_class.model_config.get('populate_by_name'))
|
||||
self.assertIs(to_camel, model_class.model_config.get('alias_generator'))
|
||||
aliases = {name: field.alias for name, field in model_class.model_fields.items()}
|
||||
self.assertEqual({name: to_camel(name) for name in expected_fields}, aliases)
|
||||
|
||||
def test_write_models_forbid_extra_fields(self) -> None:
|
||||
for model_class in WRITE_MODELS:
|
||||
with self.subTest(model=model_class.__name__):
|
||||
self.assertEqual('forbid', model_class.model_config.get('extra'))
|
||||
|
||||
def test_page_queries_have_positive_defaults(self) -> None:
|
||||
for model_class in PAGE_QUERY_MODELS:
|
||||
with self.subTest(model=model_class.__name__):
|
||||
model = model_class.model_validate({})
|
||||
self.assertEqual((1, 10), (model.page_num, model.page_size))
|
||||
with self.assertRaises(ValidationError):
|
||||
model_class.model_validate({'pageNum': 0})
|
||||
with self.assertRaises(ValidationError):
|
||||
model_class.model_validate({'pageSize': 0})
|
||||
|
||||
def test_filling_accepts_camel_case_and_positive_packages(self) -> None:
|
||||
model = FillingCreateModel.model_validate(
|
||||
{
|
||||
'qrToken': 'qr-12345678',
|
||||
'warehouseId': 1,
|
||||
'strainId': 2,
|
||||
'batchNo': 'B20260724',
|
||||
'packageSpec': '20kg/包',
|
||||
'initialPackages': 10,
|
||||
'idempotencyKey': 'fill-key-1',
|
||||
}
|
||||
)
|
||||
|
||||
self.assertEqual(10, model.initial_packages)
|
||||
self.assertEqual(10, model.model_dump(by_alias=True)['initialPackages'])
|
||||
|
||||
def test_usage_rejects_zero_packages(self) -> None:
|
||||
with self.assertRaises(ValidationError):
|
||||
UsageCreateModel.model_validate(
|
||||
{
|
||||
'qrToken': 'qr-12345678',
|
||||
'lineId': 3,
|
||||
'usedPackages': 0,
|
||||
'idempotencyKey': 'usage-key-1',
|
||||
}
|
||||
)
|
||||
|
||||
def test_usage_rejects_non_integer_packages(self) -> None:
|
||||
for used_packages in (1.5, '1', True):
|
||||
with self.subTest(used_packages=used_packages), self.assertRaises(ValidationError):
|
||||
UsageCreateModel.model_validate(
|
||||
{
|
||||
'qrToken': 'qr-12345678',
|
||||
'lineId': 3,
|
||||
'usedPackages': used_packages,
|
||||
'idempotencyKey': 'usage-key-1',
|
||||
}
|
||||
)
|
||||
|
||||
def test_other_package_inputs_are_strict_positive_integers(self) -> None:
|
||||
cases = (
|
||||
(
|
||||
PalletCreateModel,
|
||||
{'palletCode': 'PLT-001', 'capacityPackages': 10, 'remark': None},
|
||||
'capacityPackages',
|
||||
),
|
||||
(
|
||||
FillingCreateModel,
|
||||
{
|
||||
'qrToken': 'qr-12345678',
|
||||
'warehouseId': 1,
|
||||
'strainId': 2,
|
||||
'batchNo': 'B20260724',
|
||||
'packageSpec': '20kg/包',
|
||||
'initialPackages': 10,
|
||||
'idempotencyKey': 'fill-key-1',
|
||||
},
|
||||
'initialPackages',
|
||||
),
|
||||
)
|
||||
|
||||
for model_class, payload, field_name in cases:
|
||||
for invalid_value in (0, 1.5, '1', True):
|
||||
with self.subTest(model=model_class.__name__, value=invalid_value), self.assertRaises(ValidationError):
|
||||
model_class.model_validate(payload | {field_name: invalid_value})
|
||||
|
||||
def test_id_lists_reject_empty_non_positive_and_duplicate_values(self) -> None:
|
||||
cases = (
|
||||
(DeleteStrainModel, {}, 'strainIds'),
|
||||
(DeleteCustomerModel, {}, 'customerIds'),
|
||||
(DeleteSiteModel, {}, 'siteIds'),
|
||||
(DeleteLineModel, {}, 'lineIds'),
|
||||
(DeleteWarehouseModel, {}, 'warehouseIds'),
|
||||
(DeletePalletModel, {}, 'palletIds'),
|
||||
(UserAssignModel, {}, 'userIds'),
|
||||
(InboundCreateModel, {'warehouseId': 1}, 'palletIds'),
|
||||
(InboundUpdateModel, {'inboundId': 1, 'warehouseId': 1}, 'palletIds'),
|
||||
(ShipmentCreateModel, {'sourceWarehouseId': 1, 'targetSiteId': 2}, 'palletIds'),
|
||||
(
|
||||
ShipmentUpdateModel,
|
||||
{'shipmentId': 1, 'sourceWarehouseId': 1, 'targetSiteId': 2},
|
||||
'palletIds',
|
||||
),
|
||||
(
|
||||
ReturnRequestModel,
|
||||
{'sourceSiteId': 1, 'targetWarehouseId': 2, 'idempotencyKey': 'return-key-1'},
|
||||
'palletIds',
|
||||
),
|
||||
(ReturnConfirmModel, {'idempotencyKey': 'return-confirm-key-1'}, 'scannedPalletIds'),
|
||||
)
|
||||
|
||||
for model_class, payload, field_name in cases:
|
||||
for invalid_value in ([], [0], [1, 1]):
|
||||
with self.subTest(model=model_class.__name__, value=invalid_value), self.assertRaises(ValidationError):
|
||||
model_class.model_validate(payload | {field_name: invalid_value})
|
||||
|
||||
def test_pallet_edit_forbids_identity_fields(self) -> None:
|
||||
valid_payload = {
|
||||
'palletId': 1,
|
||||
'capacityPackages': 20,
|
||||
'assetStatus': '0',
|
||||
'remark': None,
|
||||
}
|
||||
|
||||
for field_name in ('palletCode', 'qrToken'):
|
||||
with self.subTest(field=field_name), self.assertRaises(ValidationError):
|
||||
PalletEditModel.model_validate(valid_payload | {field_name: 'immutable'})
|
||||
|
||||
def test_return_scan_mode_controls_return_id(self) -> None:
|
||||
request_model = ReturnScanQueryModel.model_validate({'mode': 'request'})
|
||||
confirm_model = ReturnScanQueryModel.model_validate({'mode': 'confirm', 'returnId': 9})
|
||||
|
||||
self.assertIsNone(request_model.return_id)
|
||||
self.assertEqual(9, confirm_model.return_id)
|
||||
for payload in (
|
||||
{'mode': 'confirm'},
|
||||
{'mode': 'request', 'returnId': 9},
|
||||
{'mode': 'other'},
|
||||
):
|
||||
with self.subTest(payload=payload), self.assertRaises(ValidationError):
|
||||
ReturnScanQueryModel.model_validate(payload)
|
||||
|
||||
def test_required_business_strings_reject_whitespace(self) -> None:
|
||||
with self.assertRaises(ValidationError):
|
||||
ConfirmCommandModel.model_validate({'idempotencyKey': ' '})
|
||||
with self.assertRaises(ValidationError):
|
||||
StrainModel.model_validate({'strainCode': '\t', 'strainName': '菌种 A', 'status': '0'})
|
||||
with self.assertRaises(ValidationError):
|
||||
FillingCreateModel.model_validate(
|
||||
{
|
||||
'qrToken': 'qr-12345678',
|
||||
'warehouseId': 1,
|
||||
'strainId': 2,
|
||||
'batchNo': ' ',
|
||||
'packageSpec': '20kg/包',
|
||||
'initialPackages': 10,
|
||||
'idempotencyKey': 'fill-key-1',
|
||||
}
|
||||
)
|
||||
|
||||
def test_business_ids_must_be_positive(self) -> None:
|
||||
for model_class, payload in (
|
||||
(InboundScanQueryModel, {'warehouseId': 0}),
|
||||
(ShipmentScanQueryModel, {'sourceWarehouseId': 1, 'targetSiteId': -1}),
|
||||
(BusinessOptionModel, {'id': 0, 'code': 'WH-01', 'name': '一号仓'}),
|
||||
):
|
||||
with self.subTest(model=model_class.__name__), self.assertRaises(ValidationError):
|
||||
model_class.model_validate(payload)
|
||||
|
||||
def test_non_negative_output_quantities_are_validated(self) -> None:
|
||||
with self.assertRaises(ValidationError):
|
||||
PalletScanResponseModel.model_validate(
|
||||
{
|
||||
'palletId': 1,
|
||||
'palletCode': 'PLT-001',
|
||||
'flowState': 'AT_SITE_PARTIAL',
|
||||
'capacityPackages': 20,
|
||||
'remainingPackages': -1,
|
||||
'allowedActions': [],
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user