chore: 并入 iotlcm-worker-svr 历史到 worker-svr/(原提交 96f0f4e)
This commit is contained in:
@@ -0,0 +1,85 @@
|
||||
package renewchange
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
|
||||
rc "git.5iot.com/iotlcm-public/iotlcm-worker-svr/internal/repository/renewchange"
|
||||
)
|
||||
|
||||
type fakeRenewEffectivePriceStore struct {
|
||||
depts map[int64]*rc.SysDeptRow
|
||||
prices map[renewEffectivePriceKey]*rc.FlowPackagePrice
|
||||
}
|
||||
|
||||
type renewEffectivePriceKey struct {
|
||||
deptID int64
|
||||
targetAgentID int64
|
||||
scopeType int
|
||||
packageCode string
|
||||
}
|
||||
|
||||
func (s fakeRenewEffectivePriceStore) LoadDept(ctx context.Context, tenantID string, deptID int64) (*rc.SysDeptRow, error) {
|
||||
return s.depts[deptID], nil
|
||||
}
|
||||
|
||||
func (s fakeRenewEffectivePriceStore) LoadPriceOverride(ctx context.Context, tenantID string, deptID, targetAgentID int64, scopeType int, packageCode string) (*rc.FlowPackagePrice, error) {
|
||||
return s.prices[renewEffectivePriceKey{
|
||||
deptID: deptID,
|
||||
targetAgentID: targetAgentID,
|
||||
scopeType: scopeType,
|
||||
packageCode: packageCode,
|
||||
}], nil
|
||||
}
|
||||
|
||||
func TestResolveRenewEffectiveSalePriceUsesRecursiveParentCost(t *testing.T) {
|
||||
store := fakeRenewEffectivePriceStore{
|
||||
depts: map[int64]*rc.SysDeptRow{
|
||||
100: {DeptId: 100, ParentId: 0},
|
||||
117: {DeptId: 117, ParentId: 100},
|
||||
131: {DeptId: 131, ParentId: 117},
|
||||
},
|
||||
prices: map[renewEffectivePriceKey]*rc.FlowPackagePrice{
|
||||
{deptID: 100, targetAgentID: 117, scopeType: PriceScopeTargetAgent, packageCode: "yd-100m-3y"}: {
|
||||
Id: 86,
|
||||
AgentCostRatio: 1,
|
||||
AgentCostProfit: 4,
|
||||
},
|
||||
{deptID: 117, targetAgentID: 0, scopeType: PriceScopeAgentDefault, packageCode: "yd-100m-3y"}: {
|
||||
Id: 91,
|
||||
AgentCostRatio: 1,
|
||||
AgentCostProfit: 1,
|
||||
},
|
||||
{deptID: 117, targetAgentID: 131, scopeType: PriceScopeTargetAgent, packageCode: "yd-100m-3y"}: {
|
||||
Id: 92,
|
||||
AgentCostRatio: 1,
|
||||
AgentCostProfit: 5,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
got, err := resolveRenewEffectiveSalePrice(
|
||||
context.Background(),
|
||||
store,
|
||||
"000000",
|
||||
117,
|
||||
131,
|
||||
&rc.FlowPackageInfo{
|
||||
PackageCode: "yd-100m-3y",
|
||||
CostPrice: 10,
|
||||
SysDefaultRatio: 1,
|
||||
},
|
||||
)
|
||||
if err != nil {
|
||||
t.Fatalf("resolveRenewEffectiveSalePrice() error = %v", err)
|
||||
}
|
||||
if got.BaseAgentCost != 14 {
|
||||
t.Fatalf("base agent cost = %.2f, want 14.00", got.BaseAgentCost)
|
||||
}
|
||||
if got.AgentShipmentPrice != 19 {
|
||||
t.Fatalf("agent shipment price = %.2f, want 19.00", got.AgentShipmentPrice)
|
||||
}
|
||||
if got.Rule.PriceSource != PriceSourceTarget || got.Rule.TargetPriceId != 92 {
|
||||
t.Fatalf("rule = %+v, want target price 92", got.Rule)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user