Files
cgrates/engine/callcost_test.go
2021-09-28 12:17:04 +02:00

817 lines
21 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 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 General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>
*/
package engine
import (
"reflect"
"testing"
"time"
"github.com/cgrates/cgrates/utils"
)
func TestSingleResultMerge(t *testing.T) {
t1 := time.Date(2012, time.February, 2, 17, 0, 0, 0, time.UTC)
t2 := time.Date(2012, time.February, 2, 17, 1, 0, 0, time.UTC)
cd := &CallDescriptor{Category: "0",
Tenant: "vdf", Subject: "rif",
Destination: "0256", TimeStart: t1, TimeEnd: t2}
cc1, _ := cd.getCost()
if cc1.Cost != 61 {
t.Errorf("expected 61 was %v", cc1.Cost)
}
t1 = time.Date(2012, time.February, 2, 17, 1, 0, 0, time.UTC)
t2 = time.Date(2012, time.February, 2, 17, 2, 0, 0, time.UTC)
cd = &CallDescriptor{Category: "0", Tenant: "vdf",
Subject: "rif", Destination: "0256", TimeStart: t1, TimeEnd: t2}
cc2, _ := cd.GetCost()
if cc2.Cost != 61 {
t.Errorf("expected 60 was %v", cc2.Cost)
}
cc1.Merge(cc2)
if len(cc1.Timespans) != 2 || cc1.Timespans[0].GetDuration().Seconds() != 60 || cc1.Timespans[1].GetDuration().Seconds() != 60 {
for _, ts := range cc1.Timespans {
t.Logf("TS: %+v", ts)
}
t.Error("wrong resulted timespan: ", len(cc1.Timespans), cc1.Timespans[0].GetDuration().Seconds())
}
if cc1.Cost != 122 {
t.Errorf("Exdpected 120 was %v", cc1.Cost)
}
d := cc1.UpdateRatedUsage()
if d != 2*time.Minute || cc1.RatedUsage != 120.0*float64(time.Second) {
t.Errorf("error updating rating usage: %v, %v", d, cc1.RatedUsage)
}
}
func TestMultipleResultMerge(t *testing.T) {
t1 := time.Date(2012, time.February, 2, 17, 59, 0, 0, time.UTC)
t2 := time.Date(2012, time.February, 2, 18, 0, 0, 0, time.UTC)
cd := &CallDescriptor{Category: "0", Tenant: "vdf",
Subject: "rif", Destination: "0256", TimeStart: t1, TimeEnd: t2}
cc1, _ := cd.getCost()
if cc1.Cost != 61 {
//ils.LogFull(cc1)
t.Errorf("expected 61 was %v", cc1.Cost)
for _, ts := range cc1.Timespans {
t.Log(ts.RateInterval)
}
}
t1 = time.Date(2012, time.February, 2, 18, 00, 0, 0, time.UTC)
t2 = time.Date(2012, time.February, 2, 18, 01, 0, 0, time.UTC)
cd = &CallDescriptor{Category: "0", Tenant: "vdf", Subject: "rif",
Destination: "0256", TimeStart: t1, TimeEnd: t2}
cc2, _ := cd.getCost()
if cc2.Cost != 30 {
t.Errorf("expected 30 was %v", cc2.Cost)
for _, ts := range cc1.Timespans {
t.Log(ts.RateInterval)
}
}
cc1.Merge(cc2)
if len(cc1.Timespans) != 2 || cc1.Timespans[0].GetDuration().Seconds() != 60 {
t.Error("wrong resulted timespans: ", len(cc1.Timespans))
}
if cc1.Cost != 91 {
t.Errorf("Exdpected 91 was %v", cc1.Cost)
}
}
func TestMultipleInputLeftMerge(t *testing.T) {
t1 := time.Date(2012, time.February, 2, 17, 59, 0, 0, time.UTC)
t2 := time.Date(2012, time.February, 2, 18, 01, 0, 0, time.UTC)
cd := &CallDescriptor{Category: "0", Tenant: "vdf",
Subject: "rif", Destination: "0256", TimeStart: t1, TimeEnd: t2}
cc1, _ := cd.getCost()
//log.Printf("Timing: %+v", cc1.Timespans[1].RateInterval.Timing)
//log.Printf("Rating: %+v", cc1.Timespans[1].RateInterval.Rating)
if cc1.Cost != 91 {
t.Errorf("expected 91 was %v", cc1.Cost)
}
/*t1 = time.Date(2012, time.February, 2, 18, 01, 0, 0, time.UTC)
t2 = time.Date(2012, time.February, 2, 18, 02, 0, 0, time.UTC)
cd = &CallDescriptor{ToR: "0", Tenant: "vdf", Subject: "rif", Destination: "0256", TimeStart: t1, TimeEnd: t2}
cc2, _ := cd.getCost()
if cc2.Cost != 30 {
t.Errorf("expected 30 was %v", cc2.Cost)
}
cc1.Merge(cc2)
if len(cc1.Timespans) != 2 || cc1.Timespans[1].GetDuration().Seconds() != 120 {
t.Error("wrong resulted timespan: ", len(cc1.Timespans))
}
if cc1.Cost != 120 {
t.Errorf("Exdpected 120 was %v", cc1.Cost)
}*/
}
func TestMultipleInputRightMerge(t *testing.T) {
t1 := time.Date(2012, time.February, 2, 17, 58, 0, 0, time.UTC)
t2 := time.Date(2012, time.February, 2, 17, 59, 0, 0, time.UTC)
cd := &CallDescriptor{Category: "0", Tenant: "vdf", Subject: "rif",
Destination: "0256", TimeStart: t1, TimeEnd: t2}
cc1, _ := cd.getCost()
if cc1.Cost != 61 {
t.Errorf("expected 61 was %v", cc1.Cost)
}
t1 = time.Date(2012, time.February, 2, 17, 59, 0, 0, time.UTC)
t2 = time.Date(2012, time.February, 2, 18, 01, 0, 0, time.UTC)
cd = &CallDescriptor{Category: "0", Tenant: "vdf",
Subject: "rif", Destination: "0256", TimeStart: t1, TimeEnd: t2}
cc2, _ := cd.getCost()
if cc2.Cost != 91 {
t.Errorf("expected 91 was %v", cc2.Cost)
}
cc1.Merge(cc2)
if len(cc1.Timespans) != 3 || cc1.Timespans[0].GetDuration().Seconds() != 60 {
t.Error("wrong resulted timespan: ", len(cc1.Timespans), cc1.Timespans[0].GetDuration().Seconds())
}
if cc1.Cost != 152 {
t.Errorf("Exdpected 152 was %v", cc1.Cost)
}
}
func TestCallCostMergeEmpty(t *testing.T) {
t1 := time.Date(2012, time.February, 2, 17, 58, 0, 0, time.UTC)
t2 := time.Date(2012, time.February, 2, 17, 59, 0, 0, time.UTC)
cd := &CallDescriptor{Category: "0", Tenant: "vdf",
Subject: "rif", Destination: "0256", TimeStart: t1, TimeEnd: t2}
cc1, _ := cd.getCost()
cc2 := &CallCost{}
cc1.Merge(cc2)
if len(cc1.Timespans) != 1 {
t.Error("Error mergin empty call cost: ", len(cc1.Timespans))
}
}
func TestCallCostGetDurationZero(t *testing.T) {
cc := &CallCost{}
if cc.GetDuration().Seconds() != 0 {
t.Error("Wrong call cost duration for zero timespans: ", cc.GetDuration())
}
}
func TestCallCostGetDuration(t *testing.T) {
cc := &CallCost{
Timespans: []*TimeSpan{
{
TimeStart: time.Date(2013, 9, 10, 13, 40, 0, 0, time.UTC),
TimeEnd: time.Date(2013, 9, 10, 13, 41, 0, 0, time.UTC),
},
{
TimeStart: time.Date(2013, 9, 10, 13, 41, 0, 0, time.UTC),
TimeEnd: time.Date(2013, 9, 10, 13, 41, 30, 0, time.UTC),
},
},
}
if cc.GetDuration().Seconds() != 90 {
t.Error("Wrong call cost duration: ", cc.GetDuration())
}
}
func TestCallCostToDataCostError(t *testing.T) {
cd := &CallDescriptor{
Category: "data",
Tenant: "cgrates.org",
Subject: "rif",
Destination: utils.MetaAny,
TimeStart: time.Date(2014, 3, 4, 6, 0, 0, 0, time.UTC),
TimeEnd: time.Date(2014, 3, 4, 6, 1, 5, 0, time.UTC),
ToR: utils.MetaVoice,
}
cc, _ := cd.getCost()
_, err := cc.ToDataCost()
if err == nil {
t.Error("Failed to throw error on call to datacost!")
}
}
func TestCallcostCallCostToDataCost(t *testing.T) {
cc := &CallCost{
Category: "call",
Tenant: "cgrates.org",
Subject: "1001",
Account: "1001",
Destination: "1002",
ToR: "*sms",
Cost: 10,
deductConnectFee: true,
Timespans: TimeSpans{
{
DurationIndex: 10,
TimeStart: time.Date(2021, 1, 1, 10, 25, 0, 0, time.UTC),
TimeEnd: time.Date(2021, 1, 1, 10, 30, 0, 0, time.UTC),
ratingInfo: &RatingInfo{
MatchedSubject: "1001",
RatingPlanId: "RP_1001",
MatchedPrefix: "1001",
MatchedDestId: "1002",
ActivationTime: time.Date(2021, 1, 1, 10, 30, 0, 0, time.UTC),
RateIntervals: RateIntervalList{
{
Weight: 10,
Rating: &RIRate{
tag: "RT_20CNT",
ConnectFee: 15,
RoundingMethod: "*up",
RoundingDecimals: 1,
MaxCost: 100,
MaxCostStrategy: "*disconnect",
Rates: RateGroups{
{
GroupIntervalStart: 1 * time.Second,
Value: 5,
RateIncrement: 60 * time.Second,
RateUnit: 60 * time.Second,
},
{
GroupIntervalStart: 60 * time.Second,
Value: 2,
RateIncrement: 1 * time.Second,
RateUnit: 60 * time.Second,
},
},
},
},
},
FallbackKeys: []string{"key1", "key2"},
},
Cost: 15,
RateInterval: &RateInterval{
Timing: &RITiming{
ID: "ritTimingID",
tag: "ritTimingTag",
},
Rating: &RIRate{
tag: "RT_20CNT",
ConnectFee: 15,
RoundingMethod: "*up",
RoundingDecimals: 1,
MaxCost: 100,
MaxCostStrategy: "*disconnect",
Rates: RateGroups{
{
GroupIntervalStart: 30 * time.Second,
Value: 5,
RateIncrement: 60 * time.Second,
RateUnit: 60 * time.Second,
},
{
GroupIntervalStart: 60 * time.Second,
Value: 5,
RateIncrement: 1 * time.Second,
RateUnit: 60 * time.Second,
},
},
},
Weight: 10,
},
RatingPlanId: "RP_1001",
MatchedDestId: "1002",
MatchedSubject: "1001",
MatchedPrefix: "1001",
CompressFactor: 5,
RoundIncrement: &Increment{
Cost: 15,
BalanceInfo: &DebitInfo{
Unit: &UnitInfo{
ID: "1001",
UUID: "unitUUID1",
Value: 10,
DestinationID: "1002",
Consumed: 20,
ToR: "*sms",
RateInterval: &RateInterval{
Weight: 10,
Rating: &RIRate{},
Timing: &RITiming{},
},
},
},
Duration: 12 * time.Second,
CompressFactor: 2,
},
Increments: Increments{
&Increment{
Cost: 20,
BalanceInfo: &DebitInfo{
Unit: &UnitInfo{
ID: "1001",
UUID: "unitUUID2",
Value: 10,
DestinationID: "1002",
Consumed: 20,
ToR: "*sms",
RateInterval: &RateInterval{
Weight: 15,
Rating: &RIRate{},
Timing: &RITiming{},
},
},
},
Duration: 24 * time.Second,
CompressFactor: 2,
},
},
},
},
}
exp := &DataCost{
Category: "call",
Tenant: "cgrates.org",
Subject: "1001",
Account: "1001",
Destination: "1002",
ToR: "*sms",
Cost: 10,
deductConnectFee: true,
DataSpans: []*DataSpan{
{
DataStart: -299999999990,
DataEnd: 10,
Cost: 15,
RateInterval: &RateInterval{
Timing: &RITiming{
ID: "ritTimingID",
tag: "ritTimingTag",
},
Rating: &RIRate{
tag: "RT_20CNT",
ConnectFee: 15,
RoundingMethod: "*up",
RoundingDecimals: 1,
MaxCost: 100,
MaxCostStrategy: "*disconnect",
Rates: RateGroups{
{
GroupIntervalStart: 30 * time.Second,
Value: 5,
RateIncrement: 60 * time.Second,
RateUnit: 60 * time.Second,
},
{
GroupIntervalStart: 60 * time.Second,
Value: 5,
RateIncrement: 1 * time.Second,
RateUnit: 60 * time.Second,
},
},
},
Weight: 10,
},
DataIndex: 10,
Increments: []*DataIncrement{
{
Amount: 24000000000,
Cost: 20,
BalanceInfo: &DebitInfo{
Unit: &UnitInfo{
ID: "1001",
UUID: "unitUUID2",
Value: 10,
DestinationID: "1002",
Consumed: 20,
ToR: "*sms",
RateInterval: &RateInterval{
Weight: 15,
Rating: &RIRate{},
Timing: &RITiming{},
},
},
},
CompressFactor: 2,
},
},
MatchedSubject: "1001",
MatchedPrefix: "1001",
MatchedDestId: "1002",
RatingPlanId: "RP_1001",
},
},
}
rcv, err := cc.ToDataCost()
exp.DataSpans = rcv.DataSpans
if err != nil {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", nil, err)
}
if !reflect.DeepEqual(rcv, exp) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", utils.ToJSON(exp), utils.ToJSON(rcv))
}
}
func TestCallcostUpdateRatedUsage(t *testing.T) {
var cc *CallCost
var exp time.Duration
rcv := cc.UpdateRatedUsage()
if rcv != exp {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", exp, rcv)
}
}
func TestCallcostUpdateCost(t *testing.T) {
cc := &CallCost{
deductConnectFee: false,
Timespans: TimeSpans{
{
DurationIndex: 10,
TimeStart: time.Date(2021, 1, 1, 10, 25, 0, 0, time.UTC),
TimeEnd: time.Date(2021, 1, 1, 10, 30, 0, 0, time.UTC),
ratingInfo: &RatingInfo{
MatchedSubject: "1001",
RatingPlanId: "RP_1001",
MatchedPrefix: "1001",
MatchedDestId: "1002",
ActivationTime: time.Date(2021, 1, 1, 10, 30, 0, 0, time.UTC),
RateIntervals: RateIntervalList{
{
Weight: 10,
Rating: &RIRate{
tag: "RT_20CNT",
ConnectFee: 15,
RoundingMethod: "*up",
RoundingDecimals: 1,
MaxCost: 100,
MaxCostStrategy: "*disconnect",
Rates: RateGroups{
{
GroupIntervalStart: 1 * time.Second,
Value: 5,
RateIncrement: 60 * time.Second,
RateUnit: 60 * time.Second,
},
{
GroupIntervalStart: 60 * time.Second,
Value: 2,
RateIncrement: 1 * time.Second,
RateUnit: 60 * time.Second,
},
},
},
},
},
FallbackKeys: []string{"key1", "key2"},
},
Cost: 15,
RateInterval: &RateInterval{
Timing: &RITiming{
ID: "ritTimingID",
tag: "ritTimingTag",
},
Rating: &RIRate{
tag: "RT_20CNT",
ConnectFee: 15,
RoundingMethod: "*up",
RoundingDecimals: 1,
MaxCost: 100,
MaxCostStrategy: "*disconnect",
Rates: RateGroups{
{
GroupIntervalStart: 30 * time.Second,
Value: 5,
RateIncrement: 60 * time.Second,
RateUnit: 60 * time.Second,
},
{
GroupIntervalStart: 60 * time.Second,
Value: 5,
RateIncrement: 1 * time.Second,
RateUnit: 60 * time.Second,
},
},
},
Weight: 10,
},
RatingPlanId: "RP_1001",
MatchedDestId: "1002",
MatchedSubject: "1001",
MatchedPrefix: "1001",
CompressFactor: 5,
RoundIncrement: &Increment{
Cost: 15,
BalanceInfo: &DebitInfo{
Unit: &UnitInfo{
ID: "1001",
UUID: "unitUUID1",
Value: 10,
DestinationID: "1002",
Consumed: 20,
ToR: utils.MetaVoice,
RateInterval: &RateInterval{
Weight: 10,
Rating: &RIRate{},
Timing: &RITiming{},
},
},
},
Duration: 12 * time.Second,
CompressFactor: 2,
},
Increments: Increments{
&Increment{
Cost: 20,
BalanceInfo: &DebitInfo{
Unit: &UnitInfo{
ID: "1001",
UUID: "unitUUID2",
Value: 10,
DestinationID: "1002",
Consumed: 20,
ToR: utils.MetaVoice,
RateInterval: &RateInterval{
Weight: 15,
Rating: &RIRate{},
Timing: &RITiming{},
},
},
},
Duration: 24 * time.Second,
CompressFactor: 2,
},
},
},
},
}
exp := &CallCost{
Cost: 200,
deductConnectFee: true,
Timespans: cc.Timespans,
}
cc.UpdateCost()
if !reflect.DeepEqual(cc, exp) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", exp, cc)
}
}
func TestCallcostGetStartTimeNil(t *testing.T) {
cc := &CallCost{}
exp := time.Now()
rcv := cc.GetStartTime()
if timeRcv := exp.Sub(rcv); timeRcv > time.Second {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", exp, rcv)
}
}
func TestCallcostGetStartTime(t *testing.T) {
cc := &CallCost{
Timespans: TimeSpans{
{
DurationIndex: 10,
TimeStart: time.Date(2021, 1, 1, 10, 25, 0, 0, time.UTC),
TimeEnd: time.Date(2021, 1, 1, 10, 30, 0, 0, time.UTC),
ratingInfo: &RatingInfo{
MatchedSubject: "1001",
RatingPlanId: "RP_1001",
MatchedPrefix: "1001",
MatchedDestId: "1002",
ActivationTime: time.Date(2021, 1, 1, 10, 30, 0, 0, time.UTC),
RateIntervals: RateIntervalList{
{
Weight: 10,
Rating: &RIRate{
tag: "RT_20CNT",
ConnectFee: 15,
RoundingMethod: "*up",
RoundingDecimals: 1,
MaxCost: 100,
MaxCostStrategy: "*disconnect",
Rates: RateGroups{
{
GroupIntervalStart: 1 * time.Second,
Value: 5,
RateIncrement: 60 * time.Second,
RateUnit: 60 * time.Second,
},
{
GroupIntervalStart: 60 * time.Second,
Value: 2,
RateIncrement: 1 * time.Second,
RateUnit: 60 * time.Second,
},
},
},
},
},
FallbackKeys: []string{"key1", "key2"},
},
Cost: 15,
RateInterval: &RateInterval{
Timing: &RITiming{
ID: "ritTimingID",
tag: "ritTimingTag",
},
Rating: &RIRate{
tag: "RT_20CNT",
ConnectFee: 15,
RoundingMethod: "*up",
RoundingDecimals: 1,
MaxCost: 100,
MaxCostStrategy: "*disconnect",
Rates: RateGroups{
{
GroupIntervalStart: 30 * time.Second,
Value: 5,
RateIncrement: 60 * time.Second,
RateUnit: 60 * time.Second,
},
{
GroupIntervalStart: 60 * time.Second,
Value: 5,
RateIncrement: 1 * time.Second,
RateUnit: 60 * time.Second,
},
},
},
Weight: 10,
},
RatingPlanId: "RP_1001",
MatchedDestId: "1002",
MatchedSubject: "1001",
MatchedPrefix: "1001",
CompressFactor: 5,
RoundIncrement: &Increment{
Cost: 15,
BalanceInfo: &DebitInfo{
Unit: &UnitInfo{
ID: "1001",
UUID: "unitUUID1",
Value: 10,
DestinationID: "1002",
Consumed: 20,
ToR: utils.MetaVoice,
RateInterval: &RateInterval{
Weight: 10,
Rating: &RIRate{},
Timing: &RITiming{},
},
},
},
Duration: 12 * time.Second,
CompressFactor: 2,
},
Increments: Increments{
&Increment{
Cost: 20,
BalanceInfo: &DebitInfo{
Unit: &UnitInfo{
ID: "1001",
UUID: "unitUUID2",
Value: 10,
DestinationID: "1002",
Consumed: 20,
ToR: utils.MetaVoice,
RateInterval: &RateInterval{
Weight: 15,
Rating: &RIRate{},
Timing: &RITiming{},
},
},
},
Duration: 24 * time.Second,
CompressFactor: 2,
},
},
},
},
}
exp := time.Date(2021, 1, 1, 10, 25, 0, 0, time.UTC)
rcv := cc.GetStartTime()
if rcv != exp {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", exp, rcv)
}
}
func TestCallcostGetEndTime(t *testing.T) {
cc := &CallCost{}
exp := time.Now()
rcv := cc.GetEndTime()
if timeRcv := exp.Sub(rcv); timeRcv > time.Second {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", exp, rcv)
}
}
func TestCallcostRoundTimeSpansNil(t *testing.T) {
cc := &CallCost{
deductConnectFee: true,
Tenant: "tenant",
Account: "acc",
}
exp := &CallCost{
deductConnectFee: true,
Tenant: "tenant",
Account: "acc",
}
cc.Round()
if !reflect.DeepEqual(cc, exp) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", exp, cc)
}
}
func TestCallcostRoundNoIncrements(t *testing.T) {
cc := &CallCost{
Subject: "1001",
Destination: "1002",
Timespans: TimeSpans{
{
Cost: 10,
},
},
}
exp := &CallCost{
Subject: "1001",
Destination: "1002",
Timespans: TimeSpans{
{
Cost: 10,
},
},
}
cc.Round()
if !reflect.DeepEqual(cc, exp) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", exp, cc)
}
}
func TestCallcostRoundNoBalanceInfo(t *testing.T) {
cc := &CallCost{
Subject: "1001",
Destination: "1002",
Timespans: TimeSpans{
{
Cost: 10,
Increments: Increments{
{
Cost: 5,
},
},
},
},
}
exp := &CallCost{
Subject: "1001",
Destination: "1002",
Timespans: TimeSpans{
{
Cost: 10,
Increments: Increments{
{
Cost: 5,
},
},
},
},
}
cc.Round()
if !reflect.DeepEqual(cc, exp) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", exp, cc)
}
}
func TestCallcostMatchCCFilter(t *testing.T) {
cc := &CallCost{
Category: "call",
}
bf := &BalanceFilter{
Categories: &utils.StringMap{
"call": false,
},
}
rcv := cc.MatchCCFilter(bf)
if rcv != false {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", false, rcv)
}
}