152 lines
7.6 KiB
Go
152 lines
7.6 KiB
Go
package service
|
||
|
||
import (
|
||
"errors"
|
||
"testing"
|
||
"time"
|
||
|
||
"github.com/DATA-DOG/go-sqlmock"
|
||
"github.com/wmms/wmms/internal/model"
|
||
"github.com/wmms/wmms/pkg/database"
|
||
)
|
||
|
||
func outboundMock(t *testing.T) sqlmock.Sqlmock {
|
||
t.Helper()
|
||
db, mock, err := sqlmock.New()
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
previous := database.DB
|
||
database.DB = db
|
||
t.Cleanup(func() { database.DB = previous; db.Close() })
|
||
return mock
|
||
}
|
||
|
||
func expectActiveMaterial(mock sqlmock.Sqlmock, id int64) {
|
||
for i := 0; i < 2; i++ {
|
||
mock.ExpectQuery("SELECT id, organization_id, material_code, name, material_image").WithArgs(int64(1), id).
|
||
WillReturnRows(sqlmock.NewRows([]string{"id", "organization_id", "material_code", "name", "material_image", "spec", "manufacturer_model", "category_id", "base_unit_id", "unit", "status", "version", "created_at", "updated_at"}).
|
||
AddRow(id, 1, "M-1", "Material", nil, "", nil, 1, 1, "pc", "active", 1, time.Now(), time.Now()))
|
||
}
|
||
}
|
||
|
||
// expectStockScope 让 ResolveInventoryScope 解析出公司 1 的仓库 10、库位 20。
|
||
func expectStockScope(mock sqlmock.Sqlmock, orgID int64) {
|
||
mock.ExpectQuery("SELECT w.id, w.organization_id").WithArgs(orgID).WillReturnRows(
|
||
sqlmock.NewRows([]string{"id", "organization_id", "code", "name", "status", "location_id", "location_code", "location_name", "location_status"}).
|
||
AddRow(10, orgID, "W", "仓库", "active", 20, "L", "库位", "active"))
|
||
}
|
||
|
||
func createOutboundRequest() *model.CreateOutboundRequest {
|
||
req := &model.CreateOutboundRequest{Receiver: "receiver"}
|
||
req.Items = append(req.Items, struct {
|
||
MaterialID int64 `json:"material_id" binding:"required"`
|
||
Quantity int `json:"quantity" binding:"required,gt=0"`
|
||
}{MaterialID: 7, Quantity: 2})
|
||
return req
|
||
}
|
||
|
||
func TestOutboundCreateRollsBackWhenLockedStockRunsOut(t *testing.T) {
|
||
mock := outboundMock(t)
|
||
expectStockScope(mock, 1)
|
||
expectActiveMaterial(mock, 7)
|
||
// 事务前预检看到 2，够本次出库。
|
||
mock.ExpectQuery("SELECT COALESCE\\(SUM\\(on_hand\\), 0\\) FROM inventory_balances").WithArgs(int64(1), int64(10), int64(20), int64(7)).WillReturnRows(sqlmock.NewRows([]string{"on_hand"}).AddRow("2"))
|
||
mock.ExpectBegin()
|
||
// 事务内条件扣减影响 0 行：并发下余额已被别人扣走，必须整体回滚，不能透支。
|
||
mock.ExpectExec("UPDATE inventory_balances SET on_hand = on_hand").WithArgs("2", int64(1), int64(10), int64(20), int64(7), "2").WillReturnResult(sqlmock.NewResult(0, 0))
|
||
mock.ExpectRollback()
|
||
_, err := NewOutboundService().Create(1, createOutboundRequest(), "operator")
|
||
if !errors.Is(err, ErrInsufficientStock) {
|
||
t.Fatalf("got %v, want insufficient stock", err)
|
||
}
|
||
if err := mock.ExpectationsWereMet(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
}
|
||
|
||
func TestOutboundCreateChecksStockBeforeTransaction(t *testing.T) {
|
||
mock := outboundMock(t)
|
||
expectStockScope(mock, 1)
|
||
expectActiveMaterial(mock, 7)
|
||
mock.ExpectQuery("SELECT COALESCE\\(SUM\\(on_hand\\), 0\\) FROM inventory_balances").WithArgs(int64(1), int64(10), int64(20), int64(7)).WillReturnRows(sqlmock.NewRows([]string{"on_hand"}).AddRow("1"))
|
||
_, err := NewOutboundService().Create(1, createOutboundRequest(), "operator")
|
||
if !errors.Is(err, ErrInsufficientStock) {
|
||
t.Fatalf("got %v, want insufficient stock", err)
|
||
}
|
||
if err := mock.ExpectationsWereMet(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
}
|
||
|
||
func TestOutboundCreateRollsBackWhenMaterialReadFailsAfterOrderInsert(t *testing.T) {
|
||
mock := outboundMock(t)
|
||
expectStockScope(mock, 1)
|
||
expectActiveMaterial(mock, 7)
|
||
mock.ExpectQuery("SELECT COALESCE\\(SUM\\(on_hand\\), 0\\) FROM inventory_balances").WithArgs(int64(1), int64(10), int64(20), int64(7)).WillReturnRows(sqlmock.NewRows([]string{"on_hand"}).AddRow("2"))
|
||
mock.ExpectBegin()
|
||
mock.ExpectExec("UPDATE inventory_balances SET on_hand = on_hand").WithArgs("2", int64(1), int64(10), int64(20), int64(7), "2").WillReturnResult(sqlmock.NewResult(0, 1))
|
||
mock.ExpectExec("INSERT INTO outbound_orders").WillReturnResult(sqlmock.NewResult(11, 1))
|
||
mock.ExpectExec("UPDATE outbound_orders").WillReturnResult(sqlmock.NewResult(0, 1))
|
||
mock.ExpectExec("INSERT INTO outbound_items").WillReturnResult(sqlmock.NewResult(12, 1))
|
||
mock.ExpectQuery("SELECT id, organization_id, material_code, name, material_image").WithArgs(int64(1), int64(7)).WillReturnError(errors.New("material read failed"))
|
||
mock.ExpectRollback()
|
||
_, err := NewOutboundService().Create(1, createOutboundRequest(), "operator")
|
||
if err == nil || err.Error() != "material read failed" {
|
||
t.Fatalf("got %v, want material read failure", err)
|
||
}
|
||
if err := mock.ExpectationsWereMet(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
}
|
||
|
||
func TestOutboundUpdateDoesNotCountRestoredStockTwice(t *testing.T) {
|
||
mock := outboundMock(t)
|
||
now := time.Now()
|
||
mock.ExpectQuery("SELECT id, COALESCE\\(outbound_no").WithArgs(int64(11)).WillReturnRows(
|
||
sqlmock.NewRows([]string{"id", "outbound_no", "operator", "outbound_date", "handler", "receiver", "remark", "created_at", "updated_at"}).AddRow(11, "OUT-11", "operator", now, "operator", "receiver", "", now, now))
|
||
mock.ExpectQuery("SELECT oi.id, oi.outbound_order_id").WithArgs(int64(11)).WillReturnRows(
|
||
sqlmock.NewRows([]string{"id", "outbound_order_id", "material_id", "quantity", "material_code", "name", "status"}).AddRow(12, 11, 7, 2, "M-1", "Material", "active"))
|
||
expectStockScope(mock, 1)
|
||
expectActiveMaterial(mock, 7)
|
||
// 预检：当前 2 + 旧单已扣 2 = 4，够新单的 4。
|
||
mock.ExpectQuery("SELECT COALESCE\\(SUM\\(on_hand\\), 0\\) FROM inventory_balances").WithArgs(int64(1), int64(10), int64(20), int64(7)).WillReturnRows(sqlmock.NewRows([]string{"on_hand"}).AddRow("2"))
|
||
mock.ExpectBegin()
|
||
mock.ExpectExec("DELETE FROM stock_logs").WillReturnResult(sqlmock.NewResult(0, 1))
|
||
mock.ExpectExec("UPDATE outbound_orders").WillReturnResult(sqlmock.NewResult(0, 1))
|
||
mock.ExpectExec("DELETE FROM outbound_items").WillReturnResult(sqlmock.NewResult(0, 1))
|
||
mock.ExpectExec("INSERT INTO outbound_items").WillReturnResult(sqlmock.NewResult(13, 1))
|
||
// 先冲销旧单扣的 2（按物料 ID 排序后，冲销与扣减在同一物料上连续进行）。
|
||
mock.ExpectExec("INSERT INTO inventory_balances").WithArgs(int64(1), int64(10), int64(20), int64(7), "2").WillReturnResult(sqlmock.NewResult(0, 1))
|
||
// 新单扣 4：冲销后余额仍不足，条件扣减影响 0 行 → 回滚。
|
||
mock.ExpectExec("UPDATE inventory_balances SET on_hand = on_hand").WithArgs("4", int64(1), int64(10), int64(20), int64(7), "4").WillReturnResult(sqlmock.NewResult(0, 0))
|
||
mock.ExpectRollback()
|
||
req := &model.UpdateOutboundRequest{Receiver: "receiver"}
|
||
req.Items = append(req.Items, struct {
|
||
MaterialID int64 `json:"material_id" binding:"required"`
|
||
Quantity int `json:"quantity" binding:"required,gt=0"`
|
||
}{MaterialID: 7, Quantity: 4})
|
||
_, err := NewOutboundService().Update(1, 11, req, "operator")
|
||
if !errors.Is(err, ErrInsufficientStock) {
|
||
t.Fatalf("got %v, want insufficient stock", err)
|
||
}
|
||
if err := mock.ExpectationsWereMet(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
}
|
||
|
||
// 公司未配置启用仓库/库位时，改变库存的出库操作必须直接失败而不是回退默认公司。
|
||
func TestOutboundCreateRejectsWhenInventoryScopeUnavailable(t *testing.T) {
|
||
mock := outboundMock(t)
|
||
// List 查不到仓库：Summary 返回 nil。
|
||
mock.ExpectQuery("SELECT w.id, w.organization_id").WithArgs(int64(1)).WillReturnRows(
|
||
sqlmock.NewRows([]string{"id", "organization_id", "code", "name", "status", "location_id", "location_code", "location_name", "location_status"}))
|
||
_, err := NewOutboundService().Create(1, createOutboundRequest(), "operator")
|
||
if !errors.Is(err, ErrInventoryScopeUnavailable) {
|
||
t.Fatalf("got %v, want ErrInventoryScopeUnavailable", err)
|
||
}
|
||
if err := mock.ExpectationsWereMet(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
}
|