mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 10:06:24 +05:00
add test for resource & route
This commit is contained in:
armirveliaj
committed by
Dan Christian Bogos
Dan Christian Bogos
parent
3af7df0e84
commit
1c2aac5547
@@ -272,3 +272,304 @@ func TestPopulateResourcesForRoutesLazyPassErr(t *testing.T) {
|
||||
t.Errorf("Expected error <%v>, received <%v>", expErr, err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestResourceDescendentSorterSortRoutesNoResourceSConns(t *testing.T) {
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cM := engine.NewConnManager(cfg)
|
||||
rds := NewResourceDescendentSorter(cfg, cM)
|
||||
|
||||
routes := map[string]*RouteWithWeight{}
|
||||
ev := &utils.CGREvent{
|
||||
Tenant: "cgrates.org",
|
||||
Event: map[string]any{},
|
||||
APIOpts: map[string]any{},
|
||||
}
|
||||
extraOpts := &optsGetRoutes{}
|
||||
|
||||
_, err := rds.SortRoutes(context.Background(), "PROFILE1", routes, ev, extraOpts)
|
||||
errExpect := "MANDATORY_IE_MISSING: [connIDs]"
|
||||
if err == nil || err.Error() != errExpect {
|
||||
t.Errorf("Expected %v\n but received %v", errExpect, err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestResourceDescendentSorterSortRoutesNoResourceIDs(t *testing.T) {
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.RouteSCfg().ResourceSConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaResources)}
|
||||
cM := engine.NewConnManager(cfg)
|
||||
rds := NewResourceDescendentSorter(cfg, cM)
|
||||
|
||||
routes := map[string]*RouteWithWeight{
|
||||
"RW": {
|
||||
Route: &utils.Route{},
|
||||
},
|
||||
}
|
||||
ev := &utils.CGREvent{
|
||||
Tenant: "cgrates.org",
|
||||
Event: map[string]any{},
|
||||
APIOpts: map[string]any{},
|
||||
}
|
||||
extraOpts := &optsGetRoutes{}
|
||||
|
||||
_, err := rds.SortRoutes(context.Background(), "PROFILE1", routes, ev, extraOpts)
|
||||
errExpect := "MANDATORY_IE_MISSING: [ResourceIDs]"
|
||||
if err == nil || err.Error() != errExpect {
|
||||
t.Errorf("Expected %v\n but received %v", errExpect, err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestResourceDescendentSorterSortRoutesOK(t *testing.T) {
|
||||
|
||||
defer func() {
|
||||
engine.Cache = engine.NewCacheS(config.NewDefaultCGRConfig(), nil, nil, nil)
|
||||
}()
|
||||
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.RouteSCfg().ResourceSConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaResources)}
|
||||
|
||||
res1 := &utils.Resource{
|
||||
Tenant: "cgrates.org",
|
||||
ID: "RSC1",
|
||||
Usages: map[string]*utils.ResourceUsage{
|
||||
"RU1": {
|
||||
ID: "RU1",
|
||||
Units: 10,
|
||||
},
|
||||
},
|
||||
}
|
||||
res2 := &utils.Resource{
|
||||
Tenant: "cgrates.org",
|
||||
ID: "RSC2",
|
||||
Usages: map[string]*utils.ResourceUsage{
|
||||
"RU2": {
|
||||
ID: "RU2",
|
||||
Units: 5,
|
||||
},
|
||||
},
|
||||
}
|
||||
res3 := &utils.Resource{
|
||||
Tenant: "cgrates.org",
|
||||
ID: "RSC3",
|
||||
Usages: map[string]*utils.ResourceUsage{
|
||||
"RU3": {
|
||||
ID: "RU3",
|
||||
Units: 15,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
cc := make(chan birpc.ClientConnector, 1)
|
||||
cc <- &ccMock{
|
||||
calls: map[string]func(ctx *context.Context, args any, reply any) error{
|
||||
utils.ResourceSv1GetResource: func(ctx *context.Context, args, reply any) error {
|
||||
rplCast, canCast := reply.(*utils.Resource)
|
||||
if !canCast {
|
||||
t.Errorf("Wrong argument type : %T", reply)
|
||||
return nil
|
||||
}
|
||||
argsCast := args.(*utils.TenantIDWithAPIOpts)
|
||||
switch argsCast.ID {
|
||||
case "RSC1":
|
||||
*rplCast = *res1
|
||||
case "RSC2":
|
||||
*rplCast = *res2
|
||||
case "RSC3":
|
||||
*rplCast = *res3
|
||||
}
|
||||
return nil
|
||||
},
|
||||
},
|
||||
}
|
||||
cM := engine.NewConnManager(cfg)
|
||||
cM.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal, utils.MetaResources), utils.ResourceSv1, cc)
|
||||
|
||||
rds := NewResourceDescendentSorter(cfg, cM)
|
||||
|
||||
routes := map[string]*RouteWithWeight{
|
||||
"RW1": {
|
||||
Route: &utils.Route{
|
||||
ID: "Route1",
|
||||
ResourceIDs: []string{"RSC1"},
|
||||
RouteParameters: "param1",
|
||||
},
|
||||
Weight: 10,
|
||||
},
|
||||
"RW2": {
|
||||
Route: &utils.Route{
|
||||
ID: "Route2",
|
||||
ResourceIDs: []string{"RSC2"},
|
||||
RouteParameters: "param2",
|
||||
},
|
||||
Weight: 20,
|
||||
},
|
||||
"RW3": {
|
||||
Route: &utils.Route{
|
||||
ID: "Route3",
|
||||
ResourceIDs: []string{"RSC3"},
|
||||
RouteParameters: "param3",
|
||||
},
|
||||
Weight: 15,
|
||||
blocker: true,
|
||||
},
|
||||
}
|
||||
ev := &utils.CGREvent{
|
||||
Tenant: "cgrates.org",
|
||||
Event: map[string]any{},
|
||||
APIOpts: map[string]any{},
|
||||
}
|
||||
extraOpts := &optsGetRoutes{}
|
||||
|
||||
rcv, err := rds.SortRoutes(context.Background(), "PROFILE1", routes, ev, extraOpts)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
if rcv.ProfileID != "PROFILE1" {
|
||||
t.Errorf("Expected ProfileID <%s>, received <%s>", "PROFILE1", rcv.ProfileID)
|
||||
}
|
||||
|
||||
if rcv.Sorting != utils.MetaReds {
|
||||
t.Errorf("Expected Sorting <%s>, received <%s>", utils.MetaReds, rcv.Sorting)
|
||||
}
|
||||
|
||||
if len(rcv.Routes) != 3 {
|
||||
t.Fatalf("Expected 3 routes, received %d", len(rcv.Routes))
|
||||
}
|
||||
|
||||
if rcv.Routes[0].RouteID != "Route3" {
|
||||
t.Errorf("Expected first route to be Route3, got %s", rcv.Routes[0].RouteID)
|
||||
}
|
||||
if rcv.Routes[1].RouteID != "Route1" {
|
||||
t.Errorf("Expected second route to be Route1, got %s", rcv.Routes[1].RouteID)
|
||||
}
|
||||
if rcv.Routes[2].RouteID != "Route2" {
|
||||
t.Errorf("Expected third route to be Route2, got %s", rcv.Routes[2].RouteID)
|
||||
}
|
||||
|
||||
if _, hasBlocker := rcv.Routes[0].SortingData[utils.Blocker]; !hasBlocker {
|
||||
t.Errorf("Expected Route3 to have blocker flag")
|
||||
}
|
||||
}
|
||||
|
||||
func TestResourceDescendentSorterSortRoutesEmptyRoutes(t *testing.T) {
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.RouteSCfg().ResourceSConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaResources)}
|
||||
|
||||
cc := make(chan birpc.ClientConnector, 1)
|
||||
cc <- &ccMock{
|
||||
calls: map[string]func(ctx *context.Context, args any, reply any) error{
|
||||
utils.ResourceSv1GetResource: func(ctx *context.Context, args, reply any) error {
|
||||
return nil
|
||||
},
|
||||
},
|
||||
}
|
||||
cM := engine.NewConnManager(cfg)
|
||||
cM.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal, utils.MetaResources), utils.ResourceSv1, cc)
|
||||
|
||||
rds := NewResourceDescendentSorter(cfg, cM)
|
||||
|
||||
routes := map[string]*RouteWithWeight{}
|
||||
ev := &utils.CGREvent{
|
||||
Tenant: "cgrates.org",
|
||||
Event: map[string]any{},
|
||||
APIOpts: map[string]any{},
|
||||
}
|
||||
extraOpts := &optsGetRoutes{}
|
||||
|
||||
rcv, err := rds.SortRoutes(context.Background(), "PROFILE1", routes, ev, extraOpts)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
if rcv.ProfileID != "PROFILE1" {
|
||||
t.Errorf("Expected ProfileID <%s>, received <%s>", "PROFILE1", rcv.ProfileID)
|
||||
}
|
||||
|
||||
if rcv.Sorting != utils.MetaReds {
|
||||
t.Errorf("Expected Sorting <%s>, received <%s>", utils.MetaReds, rcv.Sorting)
|
||||
}
|
||||
|
||||
if len(rcv.Routes) != 0 {
|
||||
t.Errorf("Expected 0 routes, received %d", len(rcv.Routes))
|
||||
}
|
||||
}
|
||||
|
||||
func TestResourceDescendentSorterSortRoutesSingleRoute(t *testing.T) {
|
||||
|
||||
defer func() {
|
||||
engine.Cache = engine.NewCacheS(config.NewDefaultCGRConfig(), nil, nil, nil)
|
||||
}()
|
||||
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.RouteSCfg().ResourceSConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaResources)}
|
||||
|
||||
res := &utils.Resource{
|
||||
Tenant: "cgrates.org",
|
||||
ID: "RSC1",
|
||||
Usages: map[string]*utils.ResourceUsage{
|
||||
"RU1": {
|
||||
ID: "RU1",
|
||||
Units: 25,
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
cc := make(chan birpc.ClientConnector, 1)
|
||||
cc <- &ccMock{
|
||||
calls: map[string]func(ctx *context.Context, args any, reply any) error{
|
||||
utils.ResourceSv1GetResource: func(ctx *context.Context, args, reply any) error {
|
||||
rplCast, canCast := reply.(*utils.Resource)
|
||||
if !canCast {
|
||||
t.Errorf("Wrong argument type : %T", reply)
|
||||
return nil
|
||||
}
|
||||
*rplCast = *res
|
||||
return nil
|
||||
},
|
||||
},
|
||||
}
|
||||
cM := engine.NewConnManager(cfg)
|
||||
cM.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal, utils.MetaResources), utils.ResourceSv1, cc)
|
||||
|
||||
rds := NewResourceDescendentSorter(cfg, cM)
|
||||
|
||||
routes := map[string]*RouteWithWeight{
|
||||
"RW": {
|
||||
Route: &utils.Route{
|
||||
ID: "Route1",
|
||||
ResourceIDs: []string{"RSC1"},
|
||||
RouteParameters: "param1",
|
||||
},
|
||||
Weight: 50,
|
||||
},
|
||||
}
|
||||
ev := &utils.CGREvent{
|
||||
Tenant: "cgrates.org",
|
||||
Event: map[string]any{},
|
||||
APIOpts: map[string]any{},
|
||||
}
|
||||
|
||||
exp := &SortedRoutes{
|
||||
ProfileID: "PROFILE1",
|
||||
Sorting: utils.MetaReds,
|
||||
Routes: []*SortedRoute{
|
||||
{
|
||||
RouteID: "Route1",
|
||||
RouteParameters: "param1",
|
||||
SortingData: map[string]any{
|
||||
utils.ResourceUsageStr: 25.0,
|
||||
utils.Weight: 50.0,
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
rcv, err := rds.SortRoutes(context.Background(), "PROFILE1", routes, ev, nil)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
if !reflect.DeepEqual(utils.ToJSON(exp), utils.ToJSON(rcv)) {
|
||||
t.Errorf("Expected \n<%+v>,\n received \n<%+v>", utils.ToJSON(exp), utils.ToJSON(rcv))
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user