uppal/cgrates
1
0
Fork 1
mirror of https://github.com/cgrates/cgrates.git synced 2026-02-21 23:28:44 +05:00
Code Issues Packages Projects Releases Wiki Activity

Cover test for cdrs.go

Browse Source
This commit is contained in:
andronache
2021-10-13 14:15:14 +03:00
committed by Dan Christian Bogos
parent 60d9627a3e
commit 3c82973ecd
1 changed files with 166 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

168
engine/cdrs_test.go
View File

@@ -1540,7 +1540,7 @@ func TestCDRsProcessEventMockChrgsErr(t *testing.T) {
}
_, err := newCDRSrv.processEvent(context.Background(), cgrEv)
if err == nil || err.Error() != "CHARGERS_ERROR:MANDATORY_IE_MISSING: [connIDs]" {
t.Errorf("\nExpected <%+v> \n, received <%+v>", "PARTIALLY_EXECUTED", err)
t.Errorf("\nExpected <%+v> \n, received <%+v>", "CHARGERS_ERROR:MANDATORY_IE_MISSING: [connIDs]", err)
}
}
@@ -2260,7 +2260,166 @@ func TestCDRsProcessEventMockStatsErrGetBoolOpts(t *testing.T) {
}
func TestCDRsV1ProcessEventMockStatsErr(t *testing.T) {
func TestCDRsV1ProcessEventMock(t *testing.T) {
testCache := Cache
tmpC := config.CgrConfig()
tmpCM := connMgr
defer func() {
Cache = testCache
config.SetCgrConfig(tmpC)
connMgr = tmpCM
}()
var sent StorDB
cfg := config.NewDefaultCGRConfig()
cfg.CdrsCfg().EEsConns = []string{utils.ConcatenatedKey(utils.MetaInternal,
utils.MetaEEs)}
storDBChan := make(chan StorDB, 1)
storDBChan <- sent
data := NewInternalDB(nil, nil, true)
connMng := NewConnManager(cfg)
dm := NewDataManager(data, cfg.CacheCfg(), nil)
fltrs := NewFilterS(cfg, nil, dm)
Cache = NewCacheS(cfg, dm, nil)
newCDRSrv := NewCDRServer(cfg, storDBChan, dm, fltrs, connMng)
ccM := &ccMock{
calls: map[string]func(ctx *context.Context, args interface{}, reply interface{}) error{
utils.EeSv1ProcessEvent: func(ctx *context.Context, args, reply interface{}) error {
*reply.(*map[string]map[string]interface{}) = map[string]map[string]interface{}{}
return utils.ErrNotFound
},
},
}
rpcInternal := make(chan birpc.ClientConnector, 1)
rpcInternal <- ccM
newCDRSrv.connMgr.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal,
utils.MetaEEs), utils.ThresholdSv1, rpcInternal)
cgrEv := &utils.CGREvent{
Event: map[string]interface{}{
"Resources": "ResourceProfile1",
utils.AnswerTime: time.Date(2014, 7, 14, 14, 30, 0, 0, time.UTC),
"UsageInterval": "1s",
"PddInterval": "1s",
utils.Weight: "20.0",
utils.Usage: 135 * time.Second,
utils.Cost: 123.0,
},
APIOpts: map[string]interface{}{
utils.OptsCDRsExport: true,
"*context": utils.MetaCDRs,
},
}
var rply string
err := newCDRSrv.V1ProcessEvent(context.Background(), cgrEv, &rply)
if err != nil {
t.Errorf("\nExpected <%+v> \n, received <%+v>", nil, err)
}
expected := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "testID",
Event: map[string]interface{}{
"Resources": "ResourceProfile1",
utils.AnswerTime: time.Date(2014, 7, 14, 14, 30, 0, 0, time.UTC),
"UsageInterval": "1s",
"PddInterval": "1s",
utils.Weight: "20.0",
utils.Usage: 135 * time.Second,
utils.Cost: 123.0,
},
APIOpts: map[string]interface{}{
utils.OptsCDRsExport: true,
"*context": utils.MetaCDRs,
},
}
cgrEv.ID = "testID"
if !reflect.DeepEqual(expected, cgrEv) {
t.Errorf("\nExpected <%+v> \n,received <%+v>", utils.ToJSON(expected), utils.ToJSON(cgrEv))
}
}
func TestCDRsV1ProcessEventMockErr(t *testing.T) {
testCache := Cache
tmpC := config.CgrConfig()
tmpCM := connMgr
defer func() {
Cache = testCache
config.SetCgrConfig(tmpC)
connMgr = tmpCM
}()
var sent StorDB
cfg := config.NewDefaultCGRConfig()
cfg.CdrsCfg().EEsConns = []string{utils.ConcatenatedKey(utils.MetaInternal,
utils.MetaEEs)}
storDBChan := make(chan StorDB, 1)
storDBChan <- sent
data := NewInternalDB(nil, nil, true)
connMng := NewConnManager(cfg)
dm := NewDataManager(data, cfg.CacheCfg(), nil)
fltrs := NewFilterS(cfg, nil, dm)
Cache = NewCacheS(cfg, dm, nil)
newCDRSrv := NewCDRServer(cfg, storDBChan, dm, fltrs, connMng)
ccM := &ccMock{
calls: map[string]func(ctx *context.Context, args interface{}, reply interface{}) error{
utils.EeSv1ProcessEvent: func(ctx *context.Context, args, reply interface{}) error {
*reply.(*map[string]map[string]interface{}) = map[string]map[string]interface{}{}
return utils.ErrNotFound
},
},
}
rpcInternal := make(chan birpc.ClientConnector, 1)
rpcInternal <- ccM
newCDRSrv.connMgr.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal,
utils.MetaEEs), utils.ThresholdSv1, rpcInternal)
cgrEv := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "testID",
Event: map[string]interface{}{
"Resources": "ResourceProfile1",
utils.AnswerTime: time.Date(2014, 7, 14, 14, 30, 0, 0, time.UTC),
"UsageInterval": "1s",
"PddInterval": "1s",
utils.Weight: "20.0",
utils.Usage: 135 * time.Second,
utils.Cost: 123.0,
},
APIOpts: map[string]interface{}{
utils.OptsStatS: true,
utils.OptsCDRsExport: true,
"*context": utils.MetaCDRs,
},
}
var rply string
err := newCDRSrv.V1ProcessEvent(context.Background(), cgrEv, &rply)
if err == nil || err.Error() != "PARTIALLY_EXECUTED" {
t.Errorf("\nExpected <%+v> \n, received <%+v>", "PARTIALLY_EXECUTED", err)
}
expected := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "testID",
Event: map[string]interface{}{
"Resources": "ResourceProfile1",
utils.AnswerTime: time.Date(2014, 7, 14, 14, 30, 0, 0, time.UTC),
"UsageInterval": "1s",
"PddInterval": "1s",
utils.Weight: "20.0",
utils.Usage: 135 * time.Second,
utils.Cost: 123.0,
},
APIOpts: map[string]interface{}{
utils.OptsStatS: true,
utils.OptsCDRsExport: true,
"*context": utils.MetaCDRs,
},
}
if !reflect.DeepEqual(expected, cgrEv) {
t.Errorf("\nExpected <%+v> \n,received <%+v>", expected, cgrEv)
}
}
func TestCDRsV1ProcessEventMockCache(t *testing.T) {
testCache := Cache
tmpC := config.CgrConfig()
tmpCM := connMgr
@@ -2312,6 +2471,11 @@ func TestCDRsV1ProcessEventMockStatsErr(t *testing.T) {
"*context": utils.MetaCDRs,
},
}
defaultConf := config.CgrConfig().CacheCfg().Partitions[utils.CacheRPCResponses]
config.CgrConfig().CacheCfg().Partitions[utils.CacheRPCResponses].Limit = 1
defer func() {
config.CgrConfig().CacheCfg().Partitions[utils.CacheRPCResponses] = defaultConf
}()
var rply string
err := newCDRSrv.V1ProcessEvent(context.Background(), cgrEv, &rply)
if err != nil {