uppal/cgrates
1
0
Fork 1
mirror of https://github.com/cgrates/cgrates.git synced 2026-02-11 18:16:24 +05:00
Code Issues Packages Projects Releases Wiki Activity

Test functions from stats.go

Browse Source
This commit is contained in:
ionutboangiu
2021-08-02 17:01:28 +03:00
committed by Dan Christian Bogos
parent e8a0d39473
commit b5bba6fb1d
1 changed files with 151 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

152
engine/stats_test.go
View File

@@ -1834,7 +1834,7 @@ func TestStatQueueSetCloneable(t *testing.T) {
}
}
func TestStatQueueRPCClone(t *testing.T) {
func TestStatQueueRPCCloneOK(t *testing.T) {
args := &StatsArgsProcessEvent{
StatIDs: []string{"SQ_ID"},
CGREvent: &utils.CGREvent{
@@ -1863,3 +1863,153 @@ func TestStatQueueRPCClone(t *testing.T) {
args, exp)
}
}
func TestStatQueueRPCCloneNotCloneable(t *testing.T) {
args := &StatsArgsProcessEvent{
StatIDs: []string{"SQ_ID"},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "EventTest",
Event: make(map[string]interface{}),
APIOpts: make(map[string]interface{}),
},
clnb: false,
}
exp := &StatsArgsProcessEvent{
StatIDs: []string{"SQ_ID"},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "EventTest",
Event: make(map[string]interface{}),
APIOpts: make(map[string]interface{}),
},
}
if out, err := args.RPCClone(); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(out.(*StatsArgsProcessEvent), exp) {
t.Errorf("expected: <%T>, \nreceived: <%T>",
args, exp)
}
}
func TestStatQueueProcessEventOK(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
data := NewInternalDB(nil, nil, true)
dm := NewDataManager(data, cfg.CacheCfg(), nil)
filterS := NewFilterS(cfg, nil, dm)
sS := NewStatService(dm, cfg, filterS, nil)
sqPrf := &StatQueueProfile{
Tenant: "cgrates.org",
ID: "SQ1",
FilterIDs: []string{"*string:~*req.Account:1001"},
ActivationInterval: &utils.ActivationInterval{
ExpiryTime: time.Date(2021, 6, 1, 12, 0, 0, 0, time.UTC),
},
Weight: 10,
Blocker: true,
QueueLength: 10,
ThresholdIDs: []string{"*none"},
MinItems: 5,
Metrics: []*MetricWithFilters{
{
MetricID: utils.MetaASR,
},
{
MetricID: utils.MetaTCD,
},
},
}
sq := &StatQueue{
sqPrfl: sqPrf,
Tenant: "cgrates.org",
ID: "SQ1",
SQItems: []SQItem{
{
EventID: "SqProcessEvent",
},
},
}
if err := dm.SetStatQueueProfile(sqPrf, true); err != nil {
t.Error(err)
}
if err := dm.SetStatQueue(sq); err != nil {
t.Error(err)
}
args := &StatsArgsProcessEvent{
StatIDs: []string{"SQ1"},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "SqProcessEvent",
Event: map[string]interface{}{
utils.AccountField: "1001",
},
},
}
expIDs := []string{"SQ1"}
if rcvIDs, err := sS.processEvent(args.Tenant, args); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(rcvIDs, expIDs) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", expIDs, rcvIDs)
}
}
func TestStatQueueProcessEventProcessThPartExec(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
data := NewInternalDB(nil, nil, true)
dm := NewDataManager(data, cfg.CacheCfg(), nil)
filterS := NewFilterS(cfg, nil, dm)
sS := NewStatService(dm, cfg, filterS, nil)
sqPrf := &StatQueueProfile{
Tenant: "cgrates.org",
ID: "SQ1",
FilterIDs: []string{"*string:~*req.Account:1001"},
ActivationInterval: &utils.ActivationInterval{
ExpiryTime: time.Date(2021, 6, 1, 12, 0, 0, 0, time.UTC),
},
Weight: 10,
Blocker: true,
QueueLength: 10,
ThresholdIDs: []string{"*none"},
MinItems: 5,
}
sq := &StatQueue{
sqPrfl: sqPrf,
Tenant: "cgrates.org",
ID: "SQ1",
SQItems: []SQItem{
{
EventID: "SqProcessEvent",
},
},
SQMetrics: make(map[string]StatMetric),
}
if err := dm.SetStatQueueProfile(sqPrf, true); err != nil {
t.Error(err)
}
if err := dm.SetStatQueue(sq); err != nil {
t.Error(err)
}
args := &StatsArgsProcessEvent{
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "SqProcessEvent",
Event: map[string]interface{}{
utils.AccountField: "1002",
},
},
}
if _, err := sS.processEvent(args.Tenant, args); err == nil ||
err != utils.ErrNotFound {
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ErrNotFound, err)
}
}