ee569e7edb
Add benchmark documentation and middleware benchmark coverage, fix the settlement benchmark mock to satisfy the current SettlementStore interface, and add reusable domain test helper packages. Verified with fresh go test runs for ./internal/testutil/... and go test -tags=slow -run '^$' ./internal/benchmark/... before commit.
94 lines
3.5 KiB
Go
94 lines
3.5 KiB
Go
package assert
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"lijiaoqiao/supply-api/internal/domain"
|
|
)
|
|
|
|
// Account 相关的自定义断言
|
|
|
|
// AssertAccountEqual 断言两个账号相等（忽略指针差异）
|
|
func AssertAccountEqual(t *testing.T, expected, actual *domain.Account, msgAndArgs ...interface{}) {
|
|
assert.Equal(t, expected, actual, msgAndArgs...)
|
|
}
|
|
|
|
// AssertAccountStatus 断言账号状态
|
|
func AssertAccountStatus(t *testing.T, account *domain.Account, expectedStatus domain.AccountStatus, msgAndArgs ...interface{}) {
|
|
assert.Equal(t, expectedStatus, account.Status, msgAndArgs...)
|
|
}
|
|
|
|
// AssertAccountFields 断言账号字段
|
|
func AssertAccountFields(t *testing.T, account *domain.Account, supplierID int64, provider domain.Provider, accountType domain.AccountType, msgAndArgs ...interface{}) {
|
|
assert.Equal(t, supplierID, account.SupplierID, msgAndArgs...)
|
|
assert.Equal(t, provider, account.Provider, msgAndArgs...)
|
|
assert.Equal(t, accountType, account.AccountType, msgAndArgs...)
|
|
}
|
|
|
|
// Package 相关的自定义断言
|
|
|
|
// AssertPackageEqual 断言两个套餐相等
|
|
func AssertPackageEqual(t *testing.T, expected, actual *domain.Package, msgAndArgs ...interface{}) {
|
|
assert.Equal(t, expected, actual, msgAndArgs...)
|
|
}
|
|
|
|
// AssertPackageStatus 断言套餐状态
|
|
func AssertPackageStatus(t *testing.T, pkg *domain.Package, expectedStatus domain.PackageStatus, msgAndArgs ...interface{}) {
|
|
assert.Equal(t, expectedStatus, pkg.Status, msgAndArgs...)
|
|
}
|
|
|
|
// AssertPackageQuota 断言套餐配额
|
|
func AssertPackageQuota(t *testing.T, pkg *domain.Package, total, available, sold float64, msgAndArgs ...interface{}) {
|
|
assert.InDelta(t, total, pkg.TotalQuota, 0.001, msgAndArgs...)
|
|
assert.InDelta(t, available, pkg.AvailableQuota, 0.001, msgAndArgs...)
|
|
assert.InDelta(t, sold, pkg.SoldQuota, 0.001, msgAndArgs...)
|
|
}
|
|
|
|
// Settlement 相关的自定义断言
|
|
|
|
// AssertSettlementEqual 断言两个结算单相等
|
|
func AssertSettlementEqual(t *testing.T, expected, actual *domain.Settlement, msgAndArgs ...interface{}) {
|
|
assert.Equal(t, expected, actual, msgAndArgs...)
|
|
}
|
|
|
|
// AssertSettlementStatus 断言结算单状态
|
|
func AssertSettlementStatus(t *testing.T, s *domain.Settlement, expectedStatus domain.SettlementStatus, msgAndArgs ...interface{}) {
|
|
assert.Equal(t, expectedStatus, s.Status, msgAndArgs...)
|
|
}
|
|
|
|
// AssertSettlementAmount 断言结算单金额
|
|
func AssertSettlementAmount(t *testing.T, s *domain.Settlement, total, fee, net float64, msgAndArgs ...interface{}) {
|
|
assert.InDelta(t, total, s.TotalAmount, 0.001, msgAndArgs...)
|
|
assert.InDelta(t, fee, s.FeeAmount, 0.001, msgAndArgs...)
|
|
assert.InDelta(t, net, s.NetAmount, 0.001, msgAndArgs...)
|
|
}
|
|
|
|
// EarningRecord 相关的自定义断言
|
|
|
|
// AssertEarningRecordEqual 断言两条收益记录相等
|
|
func AssertEarningRecordEqual(t *testing.T, expected, actual *domain.EarningRecord, msgAndArgs ...interface{}) {
|
|
assert.Equal(t, expected, actual, msgAndArgs...)
|
|
}
|
|
|
|
// Error 相关的自定义断言
|
|
|
|
// AssertErrorCode 断言错误码包含特定字符串
|
|
func AssertErrorCode(t *testing.T, err error, code string, msgAndArgs ...interface{}) {
|
|
if err == nil {
|
|
assert.Fail(t, "Expected error but got nil", msgAndArgs...)
|
|
return
|
|
}
|
|
assert.Contains(t, err.Error(), code, msgAndArgs...)
|
|
}
|
|
|
|
// AssertErrorMessage 断言错误信息
|
|
func AssertErrorMessage(t *testing.T, err error, msg string, msgAndArgs ...interface{}) {
|
|
if err == nil {
|
|
assert.Fail(t, "Expected error but got nil", msgAndArgs...)
|
|
return
|
|
}
|
|
assert.Contains(t, err.Error(), msg, msgAndArgs...)
|
|
}
|