Files
cgrates/dispatchers/dispatchers_test.go
2026-01-14 11:07:12 +01:00

1848 lines
55 KiB
Go

/*
Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
Copyright (C) ITsysCOM GmbH
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>
*/
package dispatchers
import (
"reflect"
"testing"
"time"
"github.com/cgrates/birpc"
"github.com/cgrates/birpc/context"
"github.com/cgrates/cgrates/config"
"github.com/cgrates/cgrates/engine"
"github.com/cgrates/cgrates/utils"
"github.com/cgrates/rpcclient"
)
func TestDispatcherServiceDispatcherProfileForEventGetDispatcherProfileNF(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dm := engine.NewDataManager(&engine.DataDBMock{
GetKeysForPrefixF: func(string, string) ([]string, error) {
return []string{"dpp_cgrates.org:123"}, nil
},
}, nil, connMng)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
ID: "321",
Subsystems: []string{utils.MetaAccounts},
FilterIDs: []string{"filter"},
ActivationInterval: &utils.ActivationInterval{},
Strategy: "",
StrategyParams: nil,
Weight: 0,
Hosts: nil,
}
err := dm.SetDispatcherProfile(dsp, false)
if err == nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", utils.ErrNotImplemented, err)
}
fltr := &engine.Filter{
Tenant: "cgrates.org",
ID: "filter",
Rules: nil,
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(1999, 2, 3, 4, 5, 6, 700000000, time.UTC),
ExpiryTime: time.Date(2000, 2, 3, 4, 5, 6, 700000000, time.UTC),
},
}
err = dm.SetFilter(fltr, false)
if err == nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", utils.ErrNotImplemented, err)
}
fltrs := engine.NewFilterS(cfg, connMng, dm)
dss := NewDispatcherService(dm, cfg, fltrs, connMng)
ev := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "321",
Event: map[string]any{
utils.AccountField: "1001",
"Password": "CGRateS.org",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]any{
utils.MetaSubsys: utils.MetaAccounts,
},
}
tnt := ev.Tenant
subsys := utils.IfaceAsString(ev.APIOpts[utils.MetaSubsys])
_, err = dss.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys)
expected := utils.ErrNotImplemented
if err == nil || err != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
}
func TestDispatcherServiceDispatcherProfileForEventMIIDENotFound(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, nil, connMng)
dss := NewDispatcherService(dm, cfg, nil, connMng)
ev := &utils.CGREvent{}
tnt := ""
subsys := utils.IfaceAsString(ev.APIOpts[utils.MetaSubsys])
_, err := dss.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys)
if err == nil || err != utils.ErrNotFound {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", utils.ErrNotFound, err)
}
}
func (dS *DispatcherService) DispatcherServicePing(ev *utils.CGREvent, reply *string) error {
*reply = utils.Pong
return nil
}
func TestDispatcherCall1(t *testing.T) {
dS := &DispatcherService{}
var reply string
if err := dS.Call(context.Background(), utils.DispatcherServicePing, &utils.CGREvent{}, &reply); err != nil {
t.Error(err)
} else if reply != utils.Pong {
t.Errorf("Expected: %s , received: %s", utils.Pong, reply)
}
}
func TestDispatcherCall2(t *testing.T) {
dS := &DispatcherService{}
var reply string
if err := dS.Call(context.Background(), "DispatcherServicePing", &utils.CGREvent{}, &reply); err == nil || err.Error() != rpcclient.ErrUnsupporteServiceMethod.Error() {
t.Error(err)
}
if err := dS.Call(context.Background(), "DispatcherService.Pong", &utils.CGREvent{}, &reply); err == nil || err.Error() != rpcclient.ErrUnsupporteServiceMethod.Error() {
t.Error(err)
}
dS.Shutdown()
}
func TestDispatcherauthorizeEvent(t *testing.T) {
dm := &engine.DataManager{}
cfg := config.NewDefaultCGRConfig()
fltr := &engine.FilterS{}
connMgr := &engine.ConnManager{}
dsp := NewDispatcherService(dm, cfg, fltr, connMgr)
ev := &utils.CGREvent{
APIOpts: make(map[string]any),
}
reply := &engine.AttrSProcessEventReply{}
err := dsp.authorizeEvent(ev, reply)
expected := "MANDATORY_IE_MISSING: [connIDs]"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
}
func TestDispatcherAuthorizeEventErr(t *testing.T) {
dm := &engine.DataManager{}
cfg := config.NewDefaultCGRConfig()
fltr := &engine.FilterS{}
connMgr := &engine.ConnManager{}
dsp := NewDispatcherService(dm, cfg, fltr, connMgr)
ev := &utils.CGREvent{
APIOpts: make(map[string]any),
}
reply := &engine.AttrSProcessEventReply{}
err := dsp.authorizeEvent(ev, reply)
expected := "MANDATORY_IE_MISSING: [connIDs]"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
}
func TestDispatcherV1GetProfileForEventErr(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.GeneralCfg().DefaultTenant = utils.EmptyString
dsp := NewDispatcherService(nil, cfg, nil, nil)
ev := &utils.CGREvent{}
dPfl := &engine.DispatcherProfiles{}
err := dsp.DispatcherSv1GetProfilesForEvent(context.Background(), ev, dPfl)
expected := "DISPATCHER_ERROR:NO_DATABASE_CONNECTION"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
}
func TestDispatcherV1GetProfileForEvent(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.GeneralCfg().DefaultTenant = utils.EmptyString
dsp := NewDispatcherService(nil, cfg, nil, nil)
ev := &utils.CGREvent{}
dPfl := &engine.DispatcherProfiles{}
err := dsp.DispatcherSv1GetProfilesForEvent(context.Background(), ev, dPfl)
expected := "DISPATCHER_ERROR:NO_DATABASE_CONNECTION"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
}
func TestDispatcherDispatch(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.GeneralCfg().DefaultTenant = utils.EmptyString
dsp := NewDispatcherService(nil, cfg, nil, nil)
ev := &utils.CGREvent{}
err := dsp.Dispatch(ev, "", "", "", "")
expected := "DISPATCHER_ERROR:NO_DATABASE_CONNECTION"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
}
func TestDispatcherAuthorizeError(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().AttributeSConns = []string{"connID"}
cfg.RPCConns()["connID"] = &config.RPCConn{
Strategy: rpcclient.PoolFirst,
PoolSize: 0,
Conns: []*config.RemoteHost{
{
ID: "",
Address: "error",
Transport: "",
TLS: false,
},
},
}
connMng := engine.NewConnManager(cfg, nil)
dsp := NewDispatcherService(nil, cfg, nil, connMng)
err := dsp.authorize("", "cgrates.org", utils.APIMethods, nil)
expected := "dial tcp: address error: missing port in address"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
}
func TestDispatcherAuthorizeError2(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().AttributeSConns = []string{utils.APIMethods}
cfg.RPCConns()[utils.APIMethods] = &config.RPCConn{
Strategy: rpcclient.PoolFirst,
PoolSize: 0,
Conns: []*config.RemoteHost{
{
ID: "",
Address: "error",
Transport: "",
TLS: false,
},
},
}
connMng := engine.NewConnManager(cfg, nil)
dsp := NewDispatcherService(nil, cfg, nil, connMng)
err := dsp.authorize("", "cgrates.org", utils.APIMethods, nil)
expected := "dial tcp: address error: missing port in address"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
}
func TestDispatcherServiceAuthorizeEvenError1(t *testing.T) {
cacheInit := engine.Cache
cfg := config.NewDefaultCGRConfig()
dm := engine.NewDataManager(nil, nil, nil)
newCache := engine.NewCacheS(cfg, dm, nil)
engine.Cache = newCache
fltr := &engine.FilterS{}
connMgr := &engine.ConnManager{}
dsp := NewDispatcherService(dm, cfg, fltr, connMgr)
cfg.DispatcherSCfg().AttributeSConns = []string{"connID"}
ev := &utils.CGREvent{
APIOpts: make(map[string]any),
}
reply := &engine.AttrSProcessEventReply{}
engine.Cache.SetWithoutReplicate(utils.CacheRPCConnections, "connID",
nil, nil, true, utils.NonTransactional)
err := dsp.authorizeEvent(ev, reply)
expected := "UNKNOWN_API_KEY"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
engine.Cache = cacheInit
}
func TestDispatcherServiceAuthorizeEventError2(t *testing.T) {
cacheInit := engine.Cache
cfg := config.NewDefaultCGRConfig()
dm := engine.NewDataManager(nil, nil, nil)
newCache := engine.NewCacheS(cfg, dm, nil)
engine.Cache = newCache
fltr := &engine.FilterS{}
connMgr := &engine.ConnManager{}
dsp := NewDispatcherService(dm, cfg, fltr, connMgr)
cfg.DispatcherSCfg().AttributeSConns = []string{"connID"}
ev := &utils.CGREvent{
APIOpts: make(map[string]any),
}
reply := &engine.AttrSProcessEventReply{}
value := &engine.DispatcherHost{
Tenant: "testTenant",
RemoteHost: &config.RemoteHost{
ID: "testID",
Address: "",
Transport: "",
TLS: false,
},
}
engine.Cache.SetWithoutReplicate(utils.CacheRPCConnections, "connID",
value, nil, true, utils.NonTransactional)
err := dsp.authorizeEvent(ev, reply)
expected := "dial tcp: missing address"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
engine.Cache = cacheInit
}
type mockTypeCon2 struct{}
func (*mockTypeCon2) Call(ctx *context.Context, serviceMethod string, args, reply any) error {
return nil
}
func TestDispatcherServiceAuthorizeEventError3(t *testing.T) {
cacheInit := engine.Cache
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().AttributeSConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes)}
dm := engine.NewDataManager(nil, nil, nil)
chanRPC := make(chan birpc.ClientConnector, 1)
chanRPC <- new(mockTypeCon2)
rpcInt := map[string]chan birpc.ClientConnector{
utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes): chanRPC,
}
connMgr := engine.NewConnManager(cfg, rpcInt)
dsp := NewDispatcherService(dm, cfg, nil, connMgr)
ev := &utils.CGREvent{
Tenant: "testTenant",
ID: "testID",
Time: nil,
Event: map[string]any{},
APIOpts: make(map[string]any),
}
value := &engine.DispatcherHost{
Tenant: "testTenant",
RemoteHost: &config.RemoteHost{
ID: "testID",
Address: rpcclient.InternalRPC,
Transport: utils.MetaInternal,
TLS: false,
},
}
newCache := engine.NewCacheS(cfg, dm, nil)
engine.Cache = newCache
engine.Cache.SetWithoutReplicate(utils.CacheRPCConnections, "testID",
value, nil, true, utils.NonTransactional)
rply := &engine.AttrSProcessEventReply{}
err := dsp.authorizeEvent(ev, rply)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
engine.Cache = cacheInit
}
type mockTypeCon3 struct{}
func (*mockTypeCon3) Call(_ *context.Context, serviceMethod string, args, reply any) error {
eVreply := &engine.AttrSProcessEventReply{
CGREvent: &utils.CGREvent{
Tenant: "testTenant",
ID: "testID",
Time: nil,
Event: map[string]any{
utils.APIMethods: "yes",
},
APIOpts: nil,
},
}
*reply.(*engine.AttrSProcessEventReply) = *eVreply
return nil
}
func TestDispatcherServiceAuthorizeError(t *testing.T) {
cacheInit := engine.Cache
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().AttributeSConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes)}
dm := engine.NewDataManager(nil, nil, nil)
chanRPC := make(chan birpc.ClientConnector, 1)
chanRPC <- new(mockTypeCon3)
rpcInt := map[string]chan birpc.ClientConnector{
utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes): chanRPC,
}
connMgr := engine.NewConnManager(cfg, rpcInt)
dsp := NewDispatcherService(dm, cfg, nil, connMgr)
value := &engine.DispatcherHost{
Tenant: "testTenant",
RemoteHost: &config.RemoteHost{
ID: "testID",
Address: rpcclient.InternalRPC,
Transport: utils.MetaInternal,
TLS: false,
},
}
newCache := engine.NewCacheS(cfg, dm, nil)
engine.Cache = newCache
engine.Cache.SetWithoutReplicate(utils.CacheRPCConnections, "testID",
value, nil, true, utils.NonTransactional)
err := dsp.authorize(utils.APIMethods, "testTenant", "apikey", &time.Time{})
expected := "UNAUTHORIZED_API"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
engine.Cache = cacheInit
}
type mockTypeCon4 struct{}
func (*mockTypeCon4) Call(ctx *context.Context, serviceMethod string, args, reply any) error {
eVreply := &engine.AttrSProcessEventReply{
CGREvent: &utils.CGREvent{
Tenant: "testTenant",
ID: "testID",
Time: nil,
Event: map[string]any{},
APIOpts: nil,
},
}
*reply.(*engine.AttrSProcessEventReply) = *eVreply
return nil
}
func TestDispatcherServiceAuthorizeError2(t *testing.T) {
cacheInit := engine.Cache
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().AttributeSConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes)}
dm := engine.NewDataManager(nil, nil, nil)
chanRPC := make(chan birpc.ClientConnector, 1)
chanRPC <- new(mockTypeCon4)
rpcInt := map[string]chan birpc.ClientConnector{
utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes): chanRPC,
}
connMgr := engine.NewConnManager(cfg, rpcInt)
dsp := NewDispatcherService(dm, cfg, nil, connMgr)
value := &engine.DispatcherHost{
Tenant: "testTenant",
RemoteHost: &config.RemoteHost{
ID: "testID",
Address: rpcclient.InternalRPC,
Transport: utils.MetaInternal,
TLS: false,
},
}
newCache := engine.NewCacheS(cfg, dm, nil)
engine.Cache = newCache
engine.Cache.SetWithoutReplicate(utils.CacheRPCConnections, "testID",
value, nil, true, utils.NonTransactional)
err := dsp.authorize(utils.APIMethods, "testTenant", "apikey", &time.Time{})
expected := "NOT_FOUND"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
engine.Cache = cacheInit
}
type mockTypeCon5 struct{}
func (*mockTypeCon5) Call(ctx *context.Context, serviceMethod string, args, reply any) error {
eVreply := &engine.AttrSProcessEventReply{
CGREvent: &utils.CGREvent{
Tenant: "testTenant",
ID: "testID",
Time: nil,
Event: map[string]any{
utils.APIMethods: "testMethod",
},
APIOpts: nil,
},
}
*reply.(*engine.AttrSProcessEventReply) = *eVreply
return nil
}
func TestDispatcherServiceAuthorizeError3(t *testing.T) {
cacheInit := engine.Cache
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().AttributeSConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes)}
dm := engine.NewDataManager(nil, nil, nil)
chanRPC := make(chan birpc.ClientConnector, 1)
chanRPC <- new(mockTypeCon5)
rpcInt := map[string]chan birpc.ClientConnector{
utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAttributes): chanRPC,
}
connMgr := engine.NewConnManager(cfg, rpcInt)
dsp := NewDispatcherService(dm, cfg, nil, connMgr)
value := &engine.DispatcherHost{
Tenant: "testTenant",
RemoteHost: &config.RemoteHost{
ID: "testID",
Address: rpcclient.InternalRPC,
Transport: utils.MetaInternal,
TLS: false,
},
}
newCache := engine.NewCacheS(cfg, dm, nil)
engine.Cache = newCache
engine.Cache.SetWithoutReplicate(utils.CacheRPCConnections, "testID",
value, nil, true, utils.NonTransactional)
err := dsp.authorize("testMethod", "testTenant", "apikey", &time.Time{})
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
engine.Cache = cacheInit
}
func TestDispatcherServiceCall1(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dm := engine.NewDataManager(nil, nil, nil)
dsp := NewDispatcherService(dm, cfg, nil, connMng)
reply := "reply"
args := &utils.CGREvent{
Tenant: "tenantTest",
ID: "tenantID",
Time: nil,
Event: map[string]any{
"event": "value",
},
APIOpts: nil,
}
err := dsp.Call(context.Background(), utils.DispatcherServicePing, args, &reply)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
}
func TestDispatcherServiceDispatcherProfileForEventErrNil(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, nil, connMng)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
ID: "123",
Subsystems: []string{utils.MetaAccounts},
FilterIDs: nil,
ActivationInterval: nil,
Strategy: "",
StrategyParams: nil,
Weight: 0,
Hosts: nil,
}
err := dm.SetDispatcherProfile(dsp, false)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
dss := NewDispatcherService(dm, cfg, nil, connMng)
ev := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "123",
Time: nil,
Event: map[string]any{
utils.AccountField: "1001",
"Password": "CGRateS.org",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]any{
utils.MetaSubsys: utils.MetaAccounts,
},
}
tnt := ev.Tenant
subsys := utils.IfaceAsString(ev.APIOpts[utils.MetaSubsys])
_, err = dss.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
}
func TestDispatcherV1GetProfileForEventReturn(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, nil, connMng)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
ID: "123",
Subsystems: []string{utils.MetaAccounts},
FilterIDs: nil,
ActivationInterval: nil,
Strategy: "",
StrategyParams: nil,
Weight: 0,
Hosts: nil,
}
err := dm.SetDispatcherProfile(dsp, false)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
dss := NewDispatcherService(dm, cfg, nil, connMng)
ev := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "123",
Time: nil,
Event: map[string]any{
utils.AccountField: "1001",
"Password": "CGRateS.org",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]any{
utils.MetaSubsys: utils.MetaAccounts,
},
}
tnt := ev.Tenant
subsys := utils.IfaceAsString(ev.APIOpts[utils.MetaSubsys])
_, err = dss.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
dPfl := &engine.DispatcherProfiles{}
err = dss.DispatcherSv1GetProfilesForEvent(context.Background(), ev, dPfl)
expected := "DISPATCHER_ERROR:NO_DATABASE_CONNECTION"
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
}
func TestDispatcherServiceDispatcherProfileForEventErrNotFound(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, nil, connMng)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
ID: "123",
Subsystems: []string{utils.MetaAccounts},
FilterIDs: nil,
ActivationInterval: nil,
Strategy: "",
StrategyParams: nil,
Weight: 0,
Hosts: nil,
}
err := dm.SetDispatcherProfile(dsp, false)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
dss := NewDispatcherService(dm, cfg, nil, connMng)
ev := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "123",
Time: nil,
Event: map[string]any{
utils.AccountField: "1001",
"Password": "CGRateS.org",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]any{
utils.MetaSubsys: utils.MetaAny,
},
}
tnt := ev.Tenant
subsys := utils.IfaceAsString(ev.APIOpts[utils.MetaSubsys])
_, err = dss.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys)
if err == nil || err != utils.ErrNotFound {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", utils.ErrNotFound, err)
}
}
func TestDispatcherServiceDispatcherProfileForEventErrNotFound2(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, nil, connMng)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
ID: "123",
Subsystems: []string{utils.MetaAccounts},
FilterIDs: nil,
ActivationInterval: nil,
Strategy: "",
StrategyParams: nil,
Weight: 0,
Hosts: nil,
}
err := dm.SetDispatcherProfile(dsp, false)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
dss := NewDispatcherService(dm, cfg, nil, connMng)
ev := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "123",
Time: nil,
Event: map[string]any{
utils.AccountField: "1001",
"Password": "CGRateS.org",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]any{
utils.MetaSubsys: utils.MetaAccounts,
},
}
tnt := ""
subsys := utils.IfaceAsString(ev.APIOpts[utils.MetaSubsys])
_, err = dss.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys)
if err == nil || err != utils.ErrNotFound {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", utils.ErrNotFound, err)
}
}
func TestDispatcherServiceDispatcherProfileForEventErrNotFoundTime(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, nil, connMng)
engine.Cache.Clear(nil)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
ID: "123",
Subsystems: []string{utils.MetaAccounts},
FilterIDs: nil,
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(1999, 2, 3, 4, 5, 6, 700000000, time.UTC),
ExpiryTime: time.Date(2000, 2, 3, 4, 5, 6, 700000000, time.UTC),
},
Strategy: "",
StrategyParams: nil,
Weight: 0,
Hosts: nil,
}
err := dm.SetDispatcherProfile(dsp, false)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
fltrs := engine.NewFilterS(cfg, connMng, dm)
dss := NewDispatcherService(dm, cfg, fltrs, connMng)
ev := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "123",
Time: utils.TimePointer(time.Now()),
Event: map[string]any{
utils.AccountField: "1001",
"Password": "CGRateS.org",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]any{
utils.MetaSubsys: utils.MetaAccounts,
},
}
tnt := ev.Tenant
subsys := utils.IfaceAsString(ev.APIOpts[utils.MetaSubsys])
_, err = dss.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys)
if err == nil || err != utils.ErrNotFound {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", utils.ErrNotFound, err)
}
}
func TestDispatcherServiceDispatcherProfileForEventErrNotFoundFilter(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, nil, connMng)
engine.Cache.Clear(nil)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
ID: "123",
Subsystems: []string{utils.MetaAccounts},
FilterIDs: []string{"filter"},
ActivationInterval: &utils.ActivationInterval{},
Strategy: "",
StrategyParams: nil,
Weight: 0,
Hosts: nil,
}
err := dm.SetDispatcherProfile(dsp, false)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
fltrs := engine.NewFilterS(cfg, connMng, dm)
dss := NewDispatcherService(dm, cfg, fltrs, connMng)
ev := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "123",
Event: map[string]any{
utils.AccountField: "1001",
"Password": "CGRateS.org",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]any{
utils.MetaSubsys: utils.MetaAccounts,
},
}
tnt := ev.Tenant
subsys := utils.IfaceAsString(ev.APIOpts[utils.MetaSubsys])
_, err = dss.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys)
if err == nil || err.Error() != "NOT_FOUND:filter" {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "NOT_FOUND:filter", err)
}
}
func TestDispatcherServiceDispatchDspErr(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
engine.Cache.Clear(nil)
dm := engine.NewDataManager(dataDB, nil, connMng)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
ID: "123",
Subsystems: []string{utils.MetaAccounts},
FilterIDs: nil,
ActivationInterval: nil,
Strategy: "",
StrategyParams: nil,
Weight: 0,
Hosts: nil,
}
err := dm.SetDispatcherProfile(dsp, false)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
dss := NewDispatcherService(dm, cfg, nil, connMng)
ev := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "123",
Time: nil,
Event: map[string]any{
utils.AccountField: "1001",
"Password": "CGRateS.org",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]any{
utils.MetaSubsys: utils.MetaAccounts,
},
}
subsys := utils.IfaceAsString(ev.APIOpts[utils.MetaSubsys])
err = dss.Dispatch(ev, subsys, "", "", "")
expected := "unsupported dispatch strategy: <>"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
}
func TestDispatcherServiceDispatchDspErrHostNotFound(t *testing.T) {
cacheInit := engine.Cache
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, nil, connMng)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
ID: "123",
Subsystems: []string{utils.MetaAccounts},
FilterIDs: nil,
ActivationInterval: nil,
StrategyParams: make(map[string]any),
Strategy: utils.MetaWeight,
Weight: 0,
Hosts: nil,
}
newCache := engine.NewCacheS(cfg, dm, nil)
value, errDsp := newDispatcher(dsp)
if errDsp != nil {
t.Fatal(errDsp)
}
engine.Cache = newCache
engine.Cache.Set(utils.CacheDispatchers, dsp.TenantID(), value, nil, true, utils.EmptyString)
err := dm.SetDispatcherProfile(dsp, false)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
dss := NewDispatcherService(dm, cfg, nil, connMng)
ev := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "123",
Time: nil,
Event: map[string]any{
utils.AccountField: "1001",
"Password": "CGRateS.org",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]any{
utils.MetaSubsys: utils.MetaAccounts,
},
}
subsys := utils.IfaceAsString(ev.APIOpts[utils.MetaSubsys])
err = dss.Dispatch(ev, subsys, "", "", "")
expected := "DSP_HOST_NOT_FOUND"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
engine.Cache = cacheInit
}
func TestDispatcherServiceDispatcherProfileForEventFoundFilter(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, nil, connMng)
engine.Cache.Clear(nil)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
ID: "123",
Subsystems: []string{utils.MetaAccounts},
FilterIDs: []string{"filter"},
ActivationInterval: &utils.ActivationInterval{},
Strategy: "",
StrategyParams: nil,
Weight: 0,
Hosts: nil,
}
err := dm.SetDispatcherProfile(dsp, false)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
fltr := &engine.Filter{
Tenant: "cgrates.org",
ID: "filter",
Rules: nil,
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(1999, 2, 3, 4, 5, 6, 700000000, time.UTC),
ExpiryTime: time.Date(2000, 2, 3, 4, 5, 6, 700000000, time.UTC),
},
}
err = dm.SetFilter(fltr, false)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
fltrs := engine.NewFilterS(cfg, connMng, dm)
dss := NewDispatcherService(dm, cfg, fltrs, connMng)
ev := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "123",
Event: map[string]any{
utils.AccountField: "1001",
"Password": "CGRateS.org",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]any{
utils.MetaSubsys: utils.MetaAccounts,
},
}
tnt := ev.Tenant
subsys := utils.IfaceAsString(ev.APIOpts[utils.MetaSubsys])
_, err = dss.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys)
if err == nil || err.Error() != "NOT_FOUND" {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "NOT_FOUND:filter", err)
}
}
func TestDispatcherServiceDispatcherProfileForEventNotNotFound(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().IndexedSelects = true
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
var cnt int
dm := engine.NewDataManager(&engine.DataDBMock{
GetIndexesDrvF: func(idxItmType, tntCtx string, idxKeys ...string) (indexes map[string]utils.StringSet, err error) {
if cnt == 0 {
cnt++
m := make(map[string]utils.StringSet)
for _, idxKey := range idxKeys {
itemID := "cgrates.org:dsp1"
m[idxKey] = utils.StringSet{
itemID: {},
}
}
return m, nil
}
return nil, utils.ErrNotImplemented
},
}, nil, connMng)
fltrs := engine.NewFilterS(cfg, connMng, dm)
dss := NewDispatcherService(dm, cfg, fltrs, connMng)
ev := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "123",
Event: map[string]any{
utils.AccountField: "1001",
"Password": "CGRateS.org",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]any{
utils.MetaSubsys: utils.MetaAccounts,
},
}
tnt := ev.Tenant
subsys := utils.IfaceAsString(ev.APIOpts[utils.MetaSubsys])
_, err := dss.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys)
expected := utils.ErrNotImplemented
if err == nil || err != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
}
func TestDispatcherServiceDispatcherProfileForEventGetDispatcherError(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, nil, connMng)
engine.Cache.Clear(nil)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
ID: "123",
Subsystems: []string{utils.MetaAccounts},
FilterIDs: []string{"filter"},
ActivationInterval: &utils.ActivationInterval{},
Strategy: "",
StrategyParams: nil,
Weight: 0,
Hosts: nil,
}
err := dm.SetDispatcherProfile(dsp, false)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
fltr := &engine.Filter{
Tenant: "cgrates.org",
ID: "filter",
Rules: nil,
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(1999, 2, 3, 4, 5, 6, 700000000, time.UTC),
ExpiryTime: time.Date(2000, 2, 3, 4, 5, 6, 700000000, time.UTC),
},
}
err = dm.SetFilter(fltr, false)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
fltrs := engine.NewFilterS(cfg, connMng, dm)
dss := NewDispatcherService(dm, cfg, fltrs, connMng)
ev := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "123",
Event: map[string]any{
utils.AccountField: "1001",
"Password": "CGRateS.org",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]any{
utils.MetaSubsys: utils.MetaAccounts,
},
}
tnt := ev.Tenant
subsys := utils.IfaceAsString(ev.APIOpts[utils.MetaSubsys])
_, err = dss.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys)
if err == nil || err.Error() != "NOT_FOUND" {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "NOT_FOUND:filter", err)
}
}
func TestDispatcherServiceDispatchDspErrHostNotFound2(t *testing.T) {
cacheInit := engine.Cache
cfg := config.NewDefaultCGRConfig()
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, nil, connMng)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
ID: "123",
Subsystems: []string{utils.MetaAccounts},
FilterIDs: nil,
ActivationInterval: nil,
StrategyParams: make(map[string]any),
Strategy: utils.MetaWeight,
Weight: 0,
Hosts: nil,
}
newCache := engine.NewCacheS(cfg, dm, nil)
value, errDsp := newDispatcher(dsp)
if errDsp != nil {
t.Fatal(errDsp)
}
engine.Cache = newCache
engine.Cache.Set(utils.CacheDispatchers, dsp.TenantID(), value, nil, true, utils.EmptyString)
err := dm.SetDispatcherProfile(dsp, false)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
dss := NewDispatcherService(dm, cfg, nil, connMng)
ev := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "123",
Time: nil,
Event: map[string]any{
utils.AccountField: "1001",
"Password": "CGRateS.org",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]any{
utils.MetaSubsys: utils.MetaAccounts,
},
}
subsys := utils.IfaceAsString(ev.APIOpts[utils.MetaSubsys])
err = dss.Dispatch(ev, subsys, "", "", "")
expected := "DSP_HOST_NOT_FOUND"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
engine.Cache = cacheInit
}
type mockTypeConSetCache struct{}
func (*mockTypeConSetCache) Call(ctx *context.Context, serviceMethod string, args, reply any) error {
return utils.ErrNotImplemented
}
func TestDispatcherServiceDispatchDspErrHostNotFound3(t *testing.T) {
cacheInit := engine.Cache
cfg := config.NewDefaultCGRConfig()
cfg.CacheCfg().ReplicationConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaReplicator)}
cfg.CacheCfg().Partitions[utils.CacheDispatchers] = &config.CacheParamCfg{
Replicate: true,
}
cfg.DispatcherSCfg().IndexedSelects = false
chanRPC := make(chan birpc.ClientConnector, 1)
chanRPC <- new(mockTypeConSetCache)
rpcInt := map[string]chan birpc.ClientConnector{
utils.ConcatenatedKey(utils.MetaInternal, utils.MetaReplicator): chanRPC,
}
connMgr := engine.NewConnManager(cfg, rpcInt)
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, nil, connMgr)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
ID: "123",
Subsystems: []string{utils.MetaAccounts},
FilterIDs: nil,
ActivationInterval: nil,
StrategyParams: make(map[string]any),
Strategy: utils.MetaWeight,
Weight: 0,
Hosts: nil,
}
newCache := engine.NewCacheS(cfg, dm, nil)
engine.Cache = newCache
err := dm.SetDispatcherProfile(dsp, false)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
dss := NewDispatcherService(dm, cfg, nil, connMgr)
ev := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "123",
Time: nil,
Event: map[string]any{
utils.AccountField: "1001",
"Password": "CGRateS.org",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]any{
utils.MetaSubsys: utils.MetaAccounts,
},
}
subsys := utils.IfaceAsString(ev.APIOpts[utils.MetaSubsys])
err = dss.Dispatch(ev, subsys, "", "", "")
expected := "NOT_IMPLEMENTED"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
engine.Cache = cacheInit
}
func (dS *DispatcherService) DispatcherServiceTest(ev *utils.CGREvent, reply *string) (any, error) {
*reply = utils.Pong
return nil, nil
}
func TestDispatcherServiceCall2(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dm := engine.NewDataManager(nil, nil, nil)
dsp := NewDispatcherService(dm, cfg, nil, connMng)
reply := "reply"
args := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "123",
Time: nil,
Event: map[string]any{
utils.AccountField: "1001",
"Password": "CGRateS.org",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]any{
utils.MetaSubsys: utils.MetaDispatchers,
},
}
err := dsp.Call(context.Background(), "DispatcherService.Test", args, &reply)
expected := "SERVER_ERROR"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
}
func (dS *DispatcherService) DispatcherServiceTest2(ev *utils.CGREvent, reply *string) any {
*reply = utils.Pong
return utils.ErrNotImplemented
}
func TestDispatcherServiceCall3(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dm := engine.NewDataManager(nil, nil, nil)
dsp := NewDispatcherService(dm, cfg, nil, connMng)
reply := "reply"
args := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "123",
Time: nil,
Event: map[string]any{
utils.AccountField: "1001",
"Password": "CGRateS.org",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]any{
utils.MetaSubsys: utils.MetaDispatchers,
},
}
err := dsp.Call(context.Background(), "DispatcherService.Test2", args, &reply)
expected := utils.ErrNotImplemented
if err == nil || err != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
}
func (dS *DispatcherService) DispatcherServiceTest3(ev *utils.CGREvent, reply *string) int {
*reply = utils.Pong
return 1
}
func TestDispatcherServiceCall4(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dm := engine.NewDataManager(nil, nil, nil)
dsp := NewDispatcherService(dm, cfg, nil, connMng)
reply := "reply"
args := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "123",
Time: nil,
Event: map[string]any{
utils.AccountField: "1001",
"Password": "CGRateS.org",
"RunID": utils.MetaDefault,
},
APIOpts: map[string]any{
utils.MetaSubsys: utils.MetaDispatchers,
},
}
err := dsp.Call(context.Background(), "DispatcherService.Test3", args, &reply)
expected := "SERVER_ERROR"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
}
func TestDispatchersdispatcherProfileForEventAnySSfalses(t *testing.T) {
tmp := engine.Cache
defer func() {
engine.Cache = tmp
}()
cfg := config.NewDefaultCGRConfig()
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, cfg.CacheCfg(), nil)
engine.Cache = engine.NewCacheS(cfg, dm, nil)
dS := &DispatcherService{
cfg: cfg,
dm: dm,
fltrS: engine.NewFilterS(cfg, nil, dm),
}
dS.cfg.DispatcherSCfg().AnySubsystem = false
dsp1 := &engine.DispatcherProfile{
Tenant: "cgrates.org",
FilterIDs: []string{"*string:~*req.Account:1002"},
Subsystems: []string{utils.MetaSessionS},
ID: "DSP_1",
Strategy: "*weight",
Weight: 10,
}
err := dS.dm.SetDispatcherProfile(dsp1, true)
if err != nil {
t.Error(err)
}
dsp2 := &engine.DispatcherProfile{
Tenant: "cgrates.org",
FilterIDs: []string{"*string:~*req.Account:1001"},
Subsystems: []string{utils.MetaAny},
ID: "DSP_2",
Strategy: "*weight",
Weight: 20,
}
err = dS.dm.SetDispatcherProfile(dsp2, true)
if err != nil {
t.Error(err)
}
tnt := "cgrates.org"
ev := &utils.CGREvent{
Tenant: tnt,
Event: map[string]any{
utils.AccountField: "1001",
},
APIOpts: map[string]any{
utils.OptsDispatchersProfilesCount: 1,
},
}
subsys := utils.MetaSessionS
if rcv, err := dS.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys); err != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, err)
} else if len(rcv) != 1 {
t.Errorf("Unexpected number of profiles:%v", len(rcv))
} else if !reflect.DeepEqual(rcv[0], dsp2) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", dsp2, rcv[0])
}
dataDB.Flush("")
engine.Cache.Clear(nil)
dsp1.FilterIDs = []string{"*string:~*req.Account:1001"}
err = dS.dm.SetDispatcherProfile(dsp1, true)
if err != nil {
t.Error(err)
}
err = dS.dm.SetDispatcherProfile(dsp2, true)
if err != nil {
t.Error(err)
}
if rcv, err := dS.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys); err != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, err)
} else if len(rcv) != 1 {
t.Errorf("Unexpected number of profiles:%v", len(rcv))
} else if !reflect.DeepEqual(rcv[0], dsp1) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", utils.ToJSON(dsp1), utils.ToJSON(rcv[0]))
}
}
func TestDispatchersdispatcherProfileForEventAnySSfalseFirstNotFound(t *testing.T) {
tmp := engine.Cache
defer func() {
engine.Cache = tmp
}()
cfg := config.NewDefaultCGRConfig()
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, cfg.CacheCfg(), nil)
engine.Cache = engine.NewCacheS(cfg, dm, nil)
dS := &DispatcherService{
cfg: cfg,
dm: dm,
fltrS: engine.NewFilterS(cfg, nil, dm),
}
dS.cfg.DispatcherSCfg().AnySubsystem = false
dsp1 := &engine.DispatcherProfile{
Tenant: "cgrates.org",
FilterIDs: []string{"*string:~*req.Account:1002"},
Subsystems: []string{utils.MetaSessionS},
ID: "DSP_1",
Strategy: "*weight",
Weight: 10,
}
err := dS.dm.SetDispatcherProfile(dsp1, true)
if err != nil {
t.Error(err)
}
dsp2 := &engine.DispatcherProfile{
Tenant: "cgrates.org",
FilterIDs: []string{"*string:~*req.Account:1001"},
Subsystems: []string{utils.MetaAny},
ID: "DSP_2",
Strategy: "*weight",
Weight: 20,
}
err = dS.dm.SetDispatcherProfile(dsp2, true)
if err != nil {
t.Error(err)
}
tnt := "cgrates.org"
ev := &utils.CGREvent{
Tenant: tnt,
Event: map[string]any{
utils.AccountField: "1001",
},
APIOpts: map[string]any{
utils.OptsDispatchersProfilesCount: 1,
},
}
subsys := utils.MetaSessionS
if rcv, err := dS.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys); err != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, err)
} else if len(rcv) != 1 {
t.Errorf("Unexpected number of profiles:%v", len(rcv))
} else if !reflect.DeepEqual(rcv[0], dsp2) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", dsp2, rcv[0])
}
}
func TestDispatchersdispatcherProfileForEventAnySSfalseFound(t *testing.T) {
tmp := engine.Cache
defer func() {
engine.Cache = tmp
}()
cfg := config.NewDefaultCGRConfig()
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, cfg.CacheCfg(), nil)
engine.Cache = engine.NewCacheS(cfg, dm, nil)
dS := &DispatcherService{
cfg: cfg,
dm: dm,
fltrS: engine.NewFilterS(cfg, nil, dm),
}
dS.cfg.DispatcherSCfg().AnySubsystem = false
dsp1 := &engine.DispatcherProfile{
Tenant: "cgrates.org",
FilterIDs: []string{"*string:~*req.Account:1001"},
Subsystems: []string{utils.MetaSessionS},
ID: "DSP_1",
Strategy: "*weight",
Weight: 20,
}
err := dS.dm.SetDispatcherProfile(dsp1, true)
if err != nil {
t.Error(err)
}
dsp2 := &engine.DispatcherProfile{
Tenant: "cgrates.org",
FilterIDs: []string{"*string:~*req.Account:1001"},
Subsystems: []string{utils.MetaAny},
ID: "DSP_2",
Strategy: "*weight",
Weight: 10,
}
err = dS.dm.SetDispatcherProfile(dsp2, true)
if err != nil {
t.Error(err)
}
tnt := "cgrates.org"
ev := &utils.CGREvent{
Tenant: tnt,
Event: map[string]any{
utils.AccountField: "1001",
},
APIOpts: map[string]any{
utils.OptsDispatchersProfilesCount: 1,
},
}
subsys := utils.MetaSessionS
if rcv, err := dS.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys); err != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, err)
} else if len(rcv) != 1 {
t.Errorf("Unexpected number of profiles:%v", len(rcv))
} else if !reflect.DeepEqual(rcv[0], dsp1) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", dsp1, rcv)
}
}
func TestDispatchersdispatcherProfileForEventAnySSfalseNotFound(t *testing.T) {
tmp := engine.Cache
defer func() {
engine.Cache = tmp
}()
cfg := config.NewDefaultCGRConfig()
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, cfg.CacheCfg(), nil)
engine.Cache = engine.NewCacheS(cfg, dm, nil)
dS := &DispatcherService{
cfg: cfg,
dm: dm,
fltrS: engine.NewFilterS(cfg, nil, dm),
}
dS.cfg.DispatcherSCfg().AnySubsystem = false
dsp1 := &engine.DispatcherProfile{
Tenant: "cgrates.org",
FilterIDs: []string{"*string:~*req.Account:1002"},
Subsystems: []string{utils.MetaSessionS},
ID: "DSP_1",
Strategy: "*weight",
Weight: 20,
}
err := dS.dm.SetDispatcherProfile(dsp1, true)
if err != nil {
t.Error(err)
}
dsp2 := &engine.DispatcherProfile{
Tenant: "cgrates.org",
FilterIDs: []string{"*string:~*req.Account:1002"},
Subsystems: []string{utils.MetaAny},
ID: "DSP_2",
Strategy: "*weight",
Weight: 10,
}
err = dS.dm.SetDispatcherProfile(dsp2, true)
if err != nil {
t.Error(err)
}
tnt := "cgrates.org"
ev := &utils.CGREvent{
Tenant: tnt,
Event: map[string]any{
utils.AccountField: "1001",
},
APIOpts: map[string]any{
utils.OptsDispatchersProfilesCount: 1,
},
}
subsys := utils.MetaSessionS
if rcv, err := dS.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys); err == nil || err != utils.ErrNotFound {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", utils.ErrNotFound, err)
} else if rcv != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, rcv)
}
}
func TestDispatchersdispatcherProfileForEventAnySStrueNotFound(t *testing.T) {
tmp := engine.Cache
defer func() {
engine.Cache = tmp
}()
cfg := config.NewDefaultCGRConfig()
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, cfg.CacheCfg(), nil)
engine.Cache = engine.NewCacheS(cfg, dm, nil)
dS := &DispatcherService{
cfg: cfg,
dm: dm,
fltrS: engine.NewFilterS(cfg, nil, dm),
}
dsp1 := &engine.DispatcherProfile{
Tenant: "cgrates.org",
FilterIDs: []string{"*string:~*req.Account:1002"},
Subsystems: []string{utils.MetaSessionS},
ID: "DSP_1",
Strategy: "*weight",
Weight: 20,
}
err := dS.dm.SetDispatcherProfile(dsp1, true)
if err != nil {
t.Error(err)
}
dsp2 := &engine.DispatcherProfile{
Tenant: "cgrates.org",
FilterIDs: []string{"*string:~*req.Account:1002"},
Subsystems: []string{utils.MetaAny},
ID: "DSP_2",
Strategy: "*weight",
Weight: 10,
}
err = dS.dm.SetDispatcherProfile(dsp2, true)
if err != nil {
t.Error(err)
}
tnt := "cgrates.org"
ev := &utils.CGREvent{
Tenant: tnt,
Event: map[string]any{
utils.AccountField: "1001",
},
APIOpts: map[string]any{
utils.OptsDispatchersProfilesCount: 1,
},
}
subsys := utils.MetaSessionS
if rcv, err := dS.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys); err == nil || err != utils.ErrNotFound {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", utils.ErrNotFound, err)
} else if rcv != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, rcv)
}
}
func TestDispatchersdispatcherProfileForEventAnySStrueBothFound(t *testing.T) {
tmp := engine.Cache
defer func() {
engine.Cache = tmp
}()
cfg := config.NewDefaultCGRConfig()
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
dm := engine.NewDataManager(dataDB, cfg.CacheCfg(), nil)
engine.Cache = engine.NewCacheS(cfg, dm, nil)
dS := &DispatcherService{
cfg: cfg,
dm: dm,
fltrS: engine.NewFilterS(cfg, nil, dm),
}
dsp1 := &engine.DispatcherProfile{
Tenant: "cgrates.org",
FilterIDs: []string{"*string:~*req.Account:1001"},
Subsystems: []string{utils.MetaSessionS},
ID: "DSP_1",
Strategy: "*weight",
Weight: 10,
}
err := dS.dm.SetDispatcherProfile(dsp1, true)
if err != nil {
t.Error(err)
}
engine.Cache.Clear(nil)
dsp2 := &engine.DispatcherProfile{
Tenant: "cgrates.org",
FilterIDs: []string{"*string:~*req.Account:1001"},
Subsystems: []string{utils.MetaAny},
ID: "DSP_2",
Strategy: "*weight",
Weight: 20,
}
err = dS.dm.SetDispatcherProfile(dsp2, true)
if err != nil {
t.Error(err)
}
engine.Cache.Clear(nil)
tnt := "cgrates.org"
ev := &utils.CGREvent{
Tenant: tnt,
Event: map[string]any{
utils.AccountField: "1001",
},
APIOpts: map[string]any{
utils.OptsDispatchersProfilesCount: 1,
},
}
subsys := utils.MetaSessionS
if rcv, err := dS.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys); err != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, err)
} else if len(rcv) != 1 {
t.Errorf("Unexpected number of profiles:%v", len(rcv))
} else if !reflect.DeepEqual(rcv[0], dsp2) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", dsp2, rcv[0])
}
dsp1.Weight = 30
err = dS.dm.SetDispatcherProfile(dsp1, true)
if err != nil {
t.Error(err)
}
engine.Cache.Clear(nil)
if rcv, err := dS.dispatcherProfilesForEvent(tnt, ev, utils.MapStorage{
utils.MetaReq: ev.Event,
utils.MetaOpts: ev.APIOpts,
}, subsys); err != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, err)
} else if len(rcv) != 1 {
t.Errorf("Unexpected number of profiles:%v", len(rcv))
} else if !reflect.DeepEqual(rcv[0], dsp1) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", utils.ToJSON(dsp1), utils.ToJSON(rcv))
}
}
func TestDispatcherServiceDispatcherProfilesForEventBoolOptsErr(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
rpcCl := map[string]chan birpc.ClientConnector{}
dataDB, derr := engine.NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
if derr != nil {
t.Error(derr)
}
connMng := engine.NewConnManager(cfg, rpcCl)
dm := engine.NewDataManager(dataDB, nil, connMng)
fltrs := engine.NewFilterS(cfg, connMng, dm)
dss := NewDispatcherService(dm, cfg, fltrs, connMng)
ev := &utils.CGREvent{
APIOpts: map[string]any{
utils.MetaDispatchers: []byte{},
},
}
expected := "cannot convert field: [] to bool"
_, err := dss.dispatcherProfilesForEvent("cgrates.org", ev, utils.MapStorage{}, "test substring")
if err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
}