uppal/cgrates
1
0
Fork 1
mirror of https://github.com/cgrates/cgrates.git synced 2026-02-20 06:38:45 +05:00
Code Issues Packages Projects Releases Wiki Activity

Improving coverage tests at engine

Browse Source
This commit is contained in:
gezimbll
2023-05-15 10:55:11 -04:00
committed by Dan Christian Bogos
parent 4bfdebd48f
commit 021c860a81
2 changed files with 216 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
engine/actions_test.go
View File

@@ -2734,6 +2734,64 @@ func TestActionSetDestinations(t *testing.T) {
}
}
func TestActionRemBalanceActionErr(t *testing.T) {
testCases := []struct {
name string
ub *Account
a *Action
acs Actions
}{
{
name: "Nil Account Error",
},
{
name: "Balance Not Found",
ub: &Account{
ID: "cgrates.org:1001",
BalanceMap: map[string]Balances{
utils.VOICE: {
&Balance{Value: 20 * float64(time.Second),
DestinationIDs: utils.NewStringMap("1002"),
Weight: 10, RatingSubject: "rif"},
}},
},
a: &Action{
ActionType: "*topup",
Balance: &BalanceFilter{Type: utils.StringPointer(utils.MONETARY),
Value: &utils.ValueFormula{Static: 10}},
},
},
{
name: "Unmatched Balance",
ub: &Account{
ID: "cgrates.org:1001",
BalanceMap: map[string]Balances{
utils.VOICE: {
&Balance{
ID: "balance_id",
Value: 20 * float64(time.Second),
DestinationIDs: utils.NewStringMap("1002"),
Weight: 10, RatingSubject: "rif"},
}},
},
a: &Action{
ActionType: "*topup",
Balance: &BalanceFilter{
ID: utils.StringPointer("balance_id2"),
Type: utils.StringPointer(utils.VOICE),
Value: &utils.ValueFormula{Static: 10}},
},
},
}
for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
if err := removeBalanceAction(tc.ub, tc.a, tc.acs, nil); err == nil {
t.Errorf("Expected error received nil")
}
})
}
}
// TestCdrLogAction
type RPCMock struct {
args *ArgV1ProcessEvent

158
engine/datamanager_test.go
View File

@@ -2145,3 +2145,161 @@ func TestDmRemoveThresholdProfileErrs(t *testing.T) {
})
}
}
func TestDmGetAccountActionPlans(t *testing.T) {
cfg, _ := config.NewDefaultCGRConfig()
defer func() {
cfg2, _ := config.NewDefaultCGRConfig()
config.SetCgrConfig(cfg2)
}()
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
cfg.DataDbCfg().RmtConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.ReplicatorSv1)}
cfg.DataDbCfg().Items[utils.MetaAccountActionPlans].Remote = true
clientConn := make(chan birpc.ClientConnector, 1)
clientConn <- clMock(func(_ *context.Context, serviceMethod string, _, _ interface{}) error {
if serviceMethod == utils.ReplicatorSv1GetAccountActionPlans {
return nil
}
return utils.ErrNotImplemented
})
dm := NewDataManager(db, cfg.CacheCfg(), NewConnManager(
cfg, map[string]chan birpc.ClientConnector{
utils.ConcatenatedKey(utils.MetaInternal, utils.ReplicatorSv1): clientConn,
},
))
Cache.Clear(nil)
testCases := []struct {
name string
acntId string
cacheRead bool
error bool
}{
{
name: "Not Found in Cache",
acntId: "Actions1",
cacheRead: true,
error: true,
},
{
name: "Get AccountActions Remote",
acntId: "Actions2",
cacheRead: false,
error: false,
},
}
Cache.Set(utils.CacheAccountActionPlans, "Actions1", nil, []string{}, true, "")
config.SetCgrConfig(cfg)
for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
if _, err := dm.GetAccountActionPlans(tc.acntId, tc.cacheRead, false, utils.NonTransactional); (err != nil) != tc.error {
t.Error(err)
}
})
}
}
func TestDmSetFilter(t *testing.T) {
cfg, _ := config.NewDefaultCGRConfig()
defer func() {
cfg2, _ := config.NewDefaultCGRConfig()
config.SetCgrConfig(cfg2)
}()
Cache.Clear(nil)
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
cfg.DataDbCfg().Items[utils.MetaFilters].Replicate = true
cfg.DataDbCfg().RplConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.ReplicatorSv1)}
clientConn := make(chan birpc.ClientConnector, 1)
clientConn <- clMock(func(_ *context.Context, serviceMethod string, _, _ interface{}) error {
if serviceMethod == utils.ReplicatorSv1SetFilter {
return nil
}
return utils.ErrNotImplemented
})
dm := NewDataManager(db, cfg.CacheCfg(), NewConnManager(cfg, map[string]chan context.ClientConnector{
utils.ConcatenatedKey(utils.MetaInternal, utils.ReplicatorSv1): clientConn,
}))
fltr := &Filter{
Tenant: "cgrates.org",
ID: "FLT_1",
Rules: []*FilterRule{
{
Type: utils.MetaString,
Element: "~*req.Account",
Values: []string{"1001"},
},
},
}
testCases := []struct {
name string
fltr *Filter
expErr bool
}{
{
name: "GetFilter Remote",
fltr: fltr,
expErr: false,
},
}
config.SetCgrConfig(cfg)
for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
if err := dm.SetFilter(tc.fltr); (err != nil) != tc.expErr {
t.Error(err)
}
})
}
}
func TestDmRemAccountActionPlans(t *testing.T) {
cfg, _ := config.NewDefaultCGRConfig()
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
cfg.DataDbCfg().Items[utils.MetaAccountActionPlans].Replicate = true
cfg.DataDbCfg().RplConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.ReplicatorSv1)}
clientConn := make(chan birpc.ClientConnector, 1)
clientConn <- clMock(func(_ *context.Context, serviceMethod string, _, _ interface{}) error {
if serviceMethod == utils.ReplicatorSv1RemAccountActionPlans {
return nil
}
return utils.ErrNotImplemented
})
dm := NewDataManager(db, cfg.CacheCfg(), NewConnManager(cfg, map[string]chan birpc.ClientConnector{
utils.ConcatenatedKey(utils.MetaInternal, utils.ReplicatorSv1): clientConn,
}))
defer func() {
cfg2, _ := config.NewDefaultCGRConfig()
config.SetCgrConfig(cfg2)
}()
testCases := []struct {
name string
acnId string
apIDs []string
expErr bool
}{
{
name: "DataManager.RemAccountActionPlans NotFound",
acnId: "cgrates.org:1001",
apIDs: []string{"Actions1"},
expErr: true,
},
{
name: "DataManager.RemAccountActionPlans Replicate",
acnId: "cgrates.org:1001",
apIDs: []string{"ACTIONS1"},
expErr: false,
},
}
config.SetCgrConfig(cfg)
dm.SetAccountActionPlans(testCases[1].acnId, testCases[1].apIDs, true)
for _, tc := range testCases {
t.Run(tc.name, func(t *testing.T) {
if err := dm.RemAccountActionPlans(tc.acnId, tc.apIDs); (err != nil) != tc.expErr {
t.Error("Expected error ,received nil")
}
})
}
}