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

Cover the remainder of engine/units_counter.go

Browse Source
This commit is contained in:
ionutboangiu
2021-04-09 14:46:51 +03:00
committed by Dan Christian Bogos
parent 403df524bd
commit 65bf6eeb2c
1 changed files with 209 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

209
engine/units_counter_test.go
View File

@@ -589,3 +589,212 @@ func TestUnitCountersResetCounterById(t *testing.T) {
t.Errorf("Error Initializing adding unit counters: %v", len(a.UnitCounters))
}
}
func TestUnitCounterHasCounter(t *testing.T) {
cfs := CounterFilters{
{
Value: 15,
Filter: &BalanceFilter{
ID: utils.StringPointer("testID1"),
Type: utils.StringPointer("testType1"),
Weight: utils.Float64Pointer(15),
},
},
{
Value: 10,
Filter: &BalanceFilter{
ID: utils.StringPointer("testID2"),
Type: utils.StringPointer("testType2"),
Weight: utils.Float64Pointer(10),
},
},
}
cf := &CounterFilter{
Value: 10,
Filter: &BalanceFilter{
ID: utils.StringPointer("testID2"),
Type: utils.StringPointer("testType2"),
Weight: utils.Float64Pointer(10),
},
}
rcv := cfs.HasCounter(cf)
if rcv != true {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", true, rcv)
}
}
func TestUnitsCounteraddUnits(t *testing.T) {
var uc *UnitCounter
ucs := UnitCounters{
"kind": []*UnitCounter{
{
Counters: CounterFilters{
{
Value: 10,
},
{
Value: 15,
},
},
CounterType: "",
},
uc,
{
Counters: CounterFilters{
{
Value: 20,
},
{
Value: 25,
},
},
CounterType: utils.MetaBalance,
},
},
}
cc := &CallCost{}
b := &Balance{}
exp := []*UnitCounter{
{
Counters: CounterFilters{
{
Value: 15,
},
{
Value: 20,
},
},
CounterType: utils.MetaCounterEvent,
},
uc,
{
Counters: CounterFilters{
{
Value: 25,
},
{
Value: 30,
},
},
CounterType: utils.MetaBalance,
},
}
ucs.addUnits(5, "kind", cc, b)
if !reflect.DeepEqual(ucs["kind"], exp) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", utils.ToJSON(exp), utils.ToJSON(ucs["kind"]))
}
}
func TestUnitsCounterresetCounters(t *testing.T) {
var emptyctr *UnitCounter
ucs := UnitCounters{
"kind": []*UnitCounter{
{
Counters: CounterFilters{
{
Value: 10,
Filter: &BalanceFilter{
ID: utils.StringPointer("testID1"),
Type: utils.StringPointer("kind"),
Weight: utils.Float64Pointer(10),
},
},
{
Value: 15,
Filter: &BalanceFilter{
ID: utils.StringPointer("testID2"),
Type: utils.StringPointer("kind"),
Weight: utils.Float64Pointer(15),
},
},
},
CounterType: "",
},
emptyctr,
{
Counters: CounterFilters{
{
Value: 20,
Filter: &BalanceFilter{
ID: utils.StringPointer("testID2"),
Type: utils.StringPointer("kind"),
Weight: utils.Float64Pointer(15),
},
},
{
Value: 25,
Filter: &BalanceFilter{
ID: utils.StringPointer("testID1"),
Type: utils.StringPointer("kind"),
Weight: utils.Float64Pointer(10),
},
},
},
CounterType: utils.MetaBalance,
},
},
}
a := &Action{
Balance: &BalanceFilter{
ID: utils.StringPointer("testID1"),
Type: utils.StringPointer("kind"),
Weight: utils.Float64Pointer(10),
},
}
exp := []*UnitCounter{
{
Counters: CounterFilters{
{
Value: 0,
Filter: &BalanceFilter{
ID: utils.StringPointer("testID1"),
Type: utils.StringPointer("kind"),
Weight: utils.Float64Pointer(10),
},
},
{
Value: 15,
Filter: &BalanceFilter{
ID: utils.StringPointer("testID2"),
Type: utils.StringPointer("kind"),
Weight: utils.Float64Pointer(15),
},
},
},
CounterType: "",
},
emptyctr,
{
Counters: CounterFilters{
{
Value: 20,
Filter: &BalanceFilter{
ID: utils.StringPointer("testID2"),
Type: utils.StringPointer("kind"),
Weight: utils.Float64Pointer(15),
},
},
{
Value: 0,
Filter: &BalanceFilter{
ID: utils.StringPointer("testID1"),
Type: utils.StringPointer("kind"),
Weight: utils.Float64Pointer(10),
},
},
},
CounterType: utils.MetaBalance,
},
}
ucs.resetCounters(a)
if !reflect.DeepEqual(ucs["kind"], exp) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", utils.ToJSON(exp), utils.ToJSON(ucs["kind"]))
}
}