Files
cgrates/engine/responder_test.go
2024-07-21 19:05:17 +02:00

1313 lines
37 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 (
"bytes"
"encoding/gob"
"reflect"
"testing"
"time"
"github.com/cgrates/cgrates/config"
"github.com/cgrates/cgrates/utils"
)
var rsponder *Responder
func init() {
cfg, _ := config.NewDefaultCGRConfig()
config.SetCgrConfig(cfg)
rsponder = &Responder{MaxComputedUsage: cfg.RalsCfg().MaxComputedUsage}
}
func TestResponderGobSMCost(t *testing.T) {
cc := &CallCost{
Category: "generic",
Tenant: "cgrates.org",
Subject: "1001",
Account: "1001",
Destination: "data",
ToR: "*data",
Cost: 0,
Timespans: TimeSpans{&TimeSpan{
TimeStart: time.Date(2016, 1, 5, 12, 30, 10, 0, time.UTC),
TimeEnd: time.Date(2016, 1, 5, 12, 55, 46, 0, time.UTC),
Cost: 0,
RateInterval: &RateInterval{
Timing: nil,
Rating: &RIRate{
ConnectFee: 0,
RoundingMethod: "",
RoundingDecimals: 0,
MaxCost: 0,
MaxCostStrategy: "",
Rates: RateGroups{&Rate{
GroupIntervalStart: 0,
Value: 0,
RateIncrement: 1 * time.Second,
RateUnit: 1 * time.Second,
},
},
},
Weight: 0,
},
DurationIndex: 0,
Increments: Increments{&Increment{
Duration: 1 * time.Second,
Cost: 0,
BalanceInfo: &DebitInfo{
Unit: &UnitInfo{
UUID: "fa0aa280-2b76-4b5b-bb06-174f84b8c321",
ID: "",
Value: 100864,
DestinationID: "data",
Consumed: 1,
ToR: "*data",
RateInterval: nil,
},
Monetary: nil,
AccountID: "cgrates.org:1001",
},
CompressFactor: 1536,
},
},
RoundIncrement: nil,
MatchedSubject: "fa0aa280-2b76-4b5b-bb06-174f84b8c321",
MatchedPrefix: "data",
MatchedDestId: "*any",
RatingPlanId: "*none",
CompressFactor: 1,
},
},
RatedUsage: 1536,
}
attr := AttrCDRSStoreSMCost{
Cost: &SMCost{
CGRID: "b783a8bcaa356570436983cd8a0e6de4993f9ba6",
RunID: utils.MetaDefault,
OriginHost: "",
OriginID: "testdatagrp_grp1",
CostSource: "SMR",
Usage: 1536,
CostDetails: NewEventCostFromCallCost(cc, "b783a8bcaa356570436983cd8a0e6de4993f9ba6", utils.MetaDefault),
},
CheckDuplicate: false,
}
var network bytes.Buffer // Stand-in for a network connection
enc := gob.NewEncoder(&network) // Will write to network.
dec := gob.NewDecoder(&network) // Will read from network.
err := enc.Encode(attr)
if err != nil {
t.Error("encode error: ", err)
}
// Decode (receive) and print the values.
var q AttrCDRSStoreSMCost
err = dec.Decode(&q)
if err != nil {
t.Error("decode error: ", err)
}
q.Cost.CostDetails.initCache()
if !reflect.DeepEqual(attr, q) {
t.Error("wrong transmission")
}
}
func TestResponderUsageAllow(t *testing.T) {
rsp := &Responder{
MaxComputedUsage: map[string]time.Duration{
utils.ANY: time.Duration(10 * time.Second),
utils.VOICE: time.Duration(20 * time.Second),
},
}
if allow := rsp.usageAllowed(utils.VOICE, time.Duration(17*time.Second)); !allow {
t.Errorf("Expected true, received : %+v", allow)
}
if allow := rsp.usageAllowed(utils.VOICE, time.Duration(22*time.Second)); allow {
t.Errorf("Expected false, received : %+v", allow)
}
if allow := rsp.usageAllowed(utils.DATA, time.Duration(7*time.Second)); !allow {
t.Errorf("Expected true, received : %+v", allow)
}
if allow := rsp.usageAllowed(utils.DATA, time.Duration(12*time.Second)); allow {
t.Errorf("Expected false, received : %+v", allow)
}
}
func TestResponderGetCostMaxUsageANY(t *testing.T) {
rsponder.MaxComputedUsage = map[string]time.Duration{
utils.ANY: time.Duration(10 * time.Second),
utils.VOICE: time.Duration(20 * time.Second),
}
tStart, _ := utils.ParseTimeDetectLayout("2013-08-07T17:30:00Z", "")
tEnd, _ := utils.ParseTimeDetectLayout("2013-08-07T17:31:11Z", "")
cd := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "dan",
ToR: utils.ANY,
Account: "dan",
Destination: "+4917621621391",
DurationIndex: 9,
TimeStart: tStart,
TimeEnd: tEnd,
},
}
var cc CallCost
if err := rsponder.GetCost(cd, &cc); err == nil ||
err.Error() != utils.ErrMaxUsageExceeded.Error() {
t.Errorf("Expected %+v, received : %+v", utils.ErrMaxUsageExceeded, err)
}
}
func TestResponderGetCostMaxUsageVOICE(t *testing.T) {
rsponder.MaxComputedUsage = map[string]time.Duration{
utils.ANY: time.Duration(10 * time.Second),
utils.VOICE: time.Duration(20 * time.Second),
}
tStart, _ := utils.ParseTimeDetectLayout("2013-08-07T17:30:00Z", "")
tEnd, _ := utils.ParseTimeDetectLayout("2013-08-07T17:31:21Z", "")
cd := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "dan",
ToR: utils.VOICE,
Account: "dan",
Destination: "+4917621621391",
DurationIndex: 9,
TimeStart: tStart,
TimeEnd: tEnd,
},
}
var cc CallCost
if err := rsponder.GetCost(cd, &cc); err == nil ||
err.Error() != utils.ErrMaxUsageExceeded.Error() {
t.Errorf("Expected %+v, received : %+v", utils.ErrMaxUsageExceeded, err)
}
}
func TestResponderDebitMaxUsageANY(t *testing.T) {
rsponder.MaxComputedUsage = map[string]time.Duration{
utils.ANY: time.Duration(10 * time.Second),
utils.VOICE: time.Duration(20 * time.Second),
}
tStart, _ := utils.ParseTimeDetectLayout("2013-08-07T17:30:00Z", "")
tEnd, _ := utils.ParseTimeDetectLayout("2013-08-07T17:31:11Z", "")
cd := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "dan",
ToR: utils.ANY,
Account: "dan",
Destination: "+4917621621391",
DurationIndex: 9,
TimeStart: tStart,
TimeEnd: tEnd,
},
}
var cc CallCost
if err := rsponder.Debit(cd, &cc); err == nil ||
err.Error() != utils.ErrMaxUsageExceeded.Error() {
t.Errorf("Expected %+v, received : %+v", utils.ErrMaxUsageExceeded, err)
}
}
func TestResponderDebitMaxUsageVOICE(t *testing.T) {
rsponder.MaxComputedUsage = map[string]time.Duration{
utils.ANY: time.Duration(10 * time.Second),
utils.VOICE: time.Duration(20 * time.Second),
}
tStart, _ := utils.ParseTimeDetectLayout("2013-08-07T17:30:00Z", "")
tEnd, _ := utils.ParseTimeDetectLayout("2013-08-07T17:31:21Z", "")
cd := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "dan",
ToR: utils.VOICE,
Account: "dan",
Destination: "+4917621621391",
DurationIndex: 9,
TimeStart: tStart,
TimeEnd: tEnd,
},
}
var cc CallCost
if err := rsponder.Debit(cd, &cc); err == nil ||
err.Error() != utils.ErrMaxUsageExceeded.Error() {
t.Errorf("Expected %+v, received : %+v", utils.ErrMaxUsageExceeded, err)
}
}
func TestResponderMaxDebitMaxUsageANY(t *testing.T) {
rsponder.MaxComputedUsage = map[string]time.Duration{
utils.ANY: time.Duration(10 * time.Second),
utils.VOICE: time.Duration(20 * time.Second),
}
tStart, _ := utils.ParseTimeDetectLayout("2013-08-07T17:30:00Z", "")
tEnd, _ := utils.ParseTimeDetectLayout("2013-08-07T17:31:11Z", "")
cd := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "dan",
ToR: utils.ANY,
Account: "dan",
Destination: "+4917621621391",
DurationIndex: 9,
TimeStart: tStart,
TimeEnd: tEnd,
},
}
var cc CallCost
if err := rsponder.MaxDebit(cd, &cc); err == nil ||
err.Error() != utils.ErrMaxUsageExceeded.Error() {
t.Errorf("Expected %+v, received : %+v", utils.ErrMaxUsageExceeded, err)
}
}
func TestResponderMaxDebitMaxUsageVOICE(t *testing.T) {
rsponder.MaxComputedUsage = map[string]time.Duration{
utils.ANY: time.Duration(10 * time.Second),
utils.VOICE: time.Duration(20 * time.Second),
}
tStart, _ := utils.ParseTimeDetectLayout("2013-08-07T17:30:00Z", "")
tEnd, _ := utils.ParseTimeDetectLayout("2013-08-07T17:31:21Z", "")
cd := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "dan",
ToR: utils.VOICE,
Account: "dan",
Destination: "+4917621621391",
DurationIndex: 9,
TimeStart: tStart,
TimeEnd: tEnd,
},
}
var cc CallCost
if err := rsponder.MaxDebit(cd, &cc); err == nil ||
err.Error() != utils.ErrMaxUsageExceeded.Error() {
t.Errorf("Expected %+v, received : %+v", utils.ErrMaxUsageExceeded, err)
}
}
func TestResponderRefundIncrementsMaxUsageANY(t *testing.T) {
rsponder.MaxComputedUsage = map[string]time.Duration{
utils.ANY: time.Duration(10 * time.Second),
utils.VOICE: time.Duration(20 * time.Second),
}
tStart, _ := utils.ParseTimeDetectLayout("2013-08-07T17:30:00Z", "")
tEnd, _ := utils.ParseTimeDetectLayout("2013-08-07T17:31:11Z", "")
cd := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "dan",
ToR: utils.ANY,
Account: "dan",
Destination: "+4917621621391",
DurationIndex: 9,
TimeStart: tStart,
TimeEnd: tEnd,
},
}
var acc Account
if err := rsponder.RefundIncrements(cd, &acc); err == nil ||
err.Error() != utils.ErrMaxUsageExceeded.Error() {
t.Errorf("Expected %+v, received : %+v", utils.ErrMaxUsageExceeded, err)
}
}
func TestResponderRefundIncrementsMaxUsageVOICE(t *testing.T) {
rsponder.MaxComputedUsage = map[string]time.Duration{
utils.ANY: time.Duration(10 * time.Second),
utils.VOICE: time.Duration(20 * time.Second),
}
tStart, _ := utils.ParseTimeDetectLayout("2013-08-07T17:30:00Z", "")
tEnd, _ := utils.ParseTimeDetectLayout("2013-08-07T17:31:21Z", "")
cd := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "dan",
ToR: utils.VOICE,
Account: "dan",
Destination: "+4917621621391",
DurationIndex: 9,
TimeStart: tStart,
TimeEnd: tEnd,
},
}
var acc Account
if err := rsponder.RefundIncrements(cd, &acc); err == nil ||
err.Error() != utils.ErrMaxUsageExceeded.Error() {
t.Errorf("Expected %+v, received : %+v", utils.ErrMaxUsageExceeded, err)
}
}
func TestResponderRefundRoundingMaxUsageANY(t *testing.T) {
rsponder.MaxComputedUsage = map[string]time.Duration{
utils.ANY: time.Duration(10 * time.Second),
utils.VOICE: time.Duration(20 * time.Second),
}
tStart, _ := utils.ParseTimeDetectLayout("2013-08-07T17:30:00Z", "")
tEnd, _ := utils.ParseTimeDetectLayout("2013-08-07T17:31:11Z", "")
cd := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "dan",
ToR: utils.ANY,
Account: "dan",
Destination: "+4917621621391",
DurationIndex: 9,
TimeStart: tStart,
TimeEnd: tEnd,
},
}
var reply Account
if err := rsponder.RefundRounding(cd, &reply); err == nil ||
err.Error() != utils.ErrMaxUsageExceeded.Error() {
t.Errorf("Expected %+v, received : %+v", utils.ErrMaxUsageExceeded, err)
}
}
func TestResponderRefundRoundingMaxUsageVOICE(t *testing.T) {
rsponder.MaxComputedUsage = map[string]time.Duration{
utils.ANY: time.Duration(10 * time.Second),
utils.VOICE: time.Duration(20 * time.Second),
}
tStart, _ := utils.ParseTimeDetectLayout("2013-08-07T17:30:00Z", "")
tEnd, _ := utils.ParseTimeDetectLayout("2013-08-07T17:31:21Z", "")
cd := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "dan",
ToR: utils.VOICE,
Account: "dan",
Destination: "+4917621621391",
DurationIndex: 9,
TimeStart: tStart,
TimeEnd: tEnd,
},
}
var reply Account
if err := rsponder.RefundRounding(cd, &reply); err == nil ||
err.Error() != utils.ErrMaxUsageExceeded.Error() {
t.Errorf("Expected %+v, received : %+v", utils.ErrMaxUsageExceeded, err)
}
}
func TestResponderGetMaxSessionTimeMaxUsageANY(t *testing.T) {
rsponder.MaxComputedUsage = map[string]time.Duration{
utils.ANY: time.Duration(10 * time.Second),
utils.VOICE: time.Duration(20 * time.Second),
}
tStart, _ := utils.ParseTimeDetectLayout("2013-08-07T17:30:00Z", "")
tEnd, _ := utils.ParseTimeDetectLayout("2013-08-07T17:31:11Z", "")
cd := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "dan",
ToR: utils.ANY,
Account: "dan",
Destination: "+4917621621391",
DurationIndex: 9,
TimeStart: tStart,
TimeEnd: tEnd,
},
}
var reply time.Duration
if err := rsponder.GetMaxSessionTime(cd, &reply); err == nil ||
err.Error() != utils.ErrMaxUsageExceeded.Error() {
t.Errorf("Expected %+v, received : %+v", utils.ErrMaxUsageExceeded, err)
}
}
func TestResponderGetMaxSessionTimeMaxUsageVOICE(t *testing.T) {
rsponder.MaxComputedUsage = map[string]time.Duration{
utils.ANY: time.Duration(10 * time.Second),
utils.VOICE: time.Duration(20 * time.Second),
}
tStart, _ := utils.ParseTimeDetectLayout("2013-08-07T17:30:00Z", "")
tEnd, _ := utils.ParseTimeDetectLayout("2013-08-07T17:31:21Z", "")
cd := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "dan",
ToR: utils.VOICE,
Account: "dan",
Destination: "+4917621621391",
DurationIndex: 9,
TimeStart: tStart,
TimeEnd: tEnd,
},
}
var reply time.Duration
if err := rsponder.GetMaxSessionTime(cd, &reply); err == nil ||
err.Error() != utils.ErrMaxUsageExceeded.Error() {
t.Errorf("Expected %+v, received : %+v", utils.ErrMaxUsageExceeded, err)
}
}
func TestResponderGetMaxSessionTimePass(t *testing.T) {
cfg, _ := config.NewDefaultCGRConfig()
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
tmpDM := dm
dm := NewDataManager(db, cfg.CacheCfg(), nil)
defer func() {
cfg2, _ := config.NewDefaultCGRConfig()
config.SetCgrConfig(cfg2)
SetDataStorage(tmpDM)
}()
arg := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "1001",
Account: "1001",
Destination: "1002",
TimeStart: time.Date(2023, 3, 4, 15, 54, 00, 0, time.UTC),
TimeEnd: time.Date(2023, 3, 4, 15, 55, 40, 0, time.UTC),
MaxCostSoFar: 0,
ToR: utils.MONETARY,
},
}
rsponder.MaxComputedUsage[utils.MONETARY] = 2 * time.Minute
dm.SetAccount(&Account{
ID: "cgrates.org:1001",
BalanceMap: map[string]Balances{
utils.MONETARY: {
&Balance{
Weight: 30,
Value: 12,
DestinationIDs: utils.NewStringMap("DEST"),
},
},
},
})
dest := &Destination{
Id: "DEST",
Prefixes: []string{
"1002", "1003", "1004",
},
}
if err := dm.SetReverseDestination(dest, utils.NonTransactional); err != nil {
t.Error(err)
}
rp := &RatingPlan{
Id: "RP_1001",
Timings: map[string]*RITiming{
"59a981b9": {
Years: utils.Years{},
Months: utils.Months{},
MonthDays: utils.MonthDays{},
WeekDays: utils.WeekDays{1, 2, 3, 4, 5},
StartTime: "00:00:00",
},
},
Ratings: map[string]*RIRate{
"efwqpqq32": {
ConnectFee: 0.1,
Rates: RateGroups{
&Rate{
GroupIntervalStart: 0,
Value: 0.7,
RateIncrement: time.Second,
RateUnit: time.Minute,
},
&Rate{
GroupIntervalStart: 60,
Value: 1,
RateIncrement: time.Second,
RateUnit: time.Second,
},
},
RoundingMethod: utils.ROUNDING_MIDDLE,
RoundingDecimals: 4,
},
},
DestinationRates: map[string]RPRateList{
"DEST": {
&RPRate{
Timing: "59a981b9",
Rating: "efwqpqq32",
Weight: 20,
},
},
},
}
dm.SetRatingPlan(rp, utils.NonTransactional)
rpP := &RatingProfile{
Id: utils.ConcatenatedKey(utils.META_OUT, "cgrates.org", "call", "1001"),
RatingPlanActivations: RatingPlanActivations{
&RatingPlanActivation{
RatingPlanId: "RP_1001",
},
},
}
dm.SetRatingProfile(rpP, utils.NonTransactional)
SetDataStorage(dm)
var reply time.Duration
if err := rsponder.GetMaxSessionTime(arg, &reply); err != nil {
t.Error(err)
} else if reply != 0 {
t.Errorf("Expected 0,Received %v", reply)
}
}
func TestMaxSessionTimeOnAccounts(t *testing.T) {
cfg, _ := config.NewDefaultCGRConfig()
tmpDm := dm
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dm := NewDataManager(db, cfg.CacheCfg(), nil)
defer func() {
SetDataStorage(tmpDm)
}()
args := &utils.GetMaxSessionTimeOnAccountsArgs{
Subject: "free",
Tenant: "cgrates.org",
AccountIDs: []string{"1001", "1002", "1003"},
Destination: "1004",
SetupTime: time.Date(2022, 12, 15, 8, 0, 0, 0, time.UTC),
Usage: time.Second * 30,
}
SetDataStorage(dm)
var reply map[string]any
for i, acc := range args.AccountIDs {
dm.SetAccount(&Account{ID: utils.ConcatenatedKey("cgrates.org", acc),
BalanceMap: map[string]Balances{utils.VOICE: {&Balance{
DestinationIDs: utils.NewStringMap("1004"),
Value: float64(10 * (i + 1)), Weight: 10}}},
})
}
if err := dm.SetReverseDestination(&Destination{
Id: "DEST", Prefixes: []string{"1004"}}, utils.NonTransactional); err != nil {
t.Error(err)
}
if err := rsponder.GetMaxSessionTimeOnAccounts(args, &reply); err != nil {
t.Error(err)
}
}
func TestResponderGetCostOnRatingPlans(t *testing.T) {
tmpCache := Cache
defer func() {
Cache = tmpCache
}()
arg := &utils.GetCostOnRatingPlansArgs{
Tenant: "cgrates.org",
Account: "test",
Subject: "1001",
Destination: "*any",
SetupTime: time.Date(2023, 3, 10, 8, 0, 0, 0, time.UTC),
Usage: 20 * time.Second,
RatingPlanIDs: []string{"RPL1"},
}
rp := &RatingPlan{
Id: "RPL1",
Timings: map[string]*RITiming{
"59a981b9": {
Years: utils.Years{},
Months: utils.Months{},
MonthDays: utils.MonthDays{},
WeekDays: utils.WeekDays{1, 2, 3, 4, 5},
StartTime: "00:00:00",
},
},
Ratings: map[string]*RIRate{
"ebefae11": {
ConnectFee: 0,
Rates: []*Rate{
{
GroupIntervalStart: 0,
Value: 0.2,
RateIncrement: 2 * time.Second,
RateUnit: time.Second,
},
},
RoundingMethod: utils.ROUNDING_MIDDLE,
RoundingDecimals: 1,
},
},
DestinationRates: map[string]RPRateList{
"*any": []*RPRate{
{
Timing: "59a981b9",
Rating: "ebefae11",
Weight: 10,
},
},
},
}
Cache.Set(utils.CacheRatingPlans, "RPL1", rp, []string{}, true, utils.NonTransactional)
var reply map[string]any
exp := map[string]any{
utils.Cost: 4.0,
utils.RatingPlanID: "RPL1",
}
if err := rsponder.GetCostOnRatingPlans(arg, &reply); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(exp, reply) {
t.Errorf("Expected %v,Received %v", utils.ToJSON(exp), utils.ToJSON(reply))
}
}
func TestResponderRefundIncrements22(t *testing.T) {
cfg, _ := config.NewDefaultCGRConfig()
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
tmpDm := dm
dm := NewDataManager(db, cfg.CacheCfg(), nil)
defer func() {
SetDataStorage(tmpDm)
}()
tStart, _ := utils.ParseTimeDetectLayout("2018-08-07T17:30:00Z", utils.EmptyString)
tEnd, _ := utils.ParseTimeDetectLayout("2018-08-07T17:31:30Z", utils.EmptyString)
acc := &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"},
}},
}
SetDataStorage(dm)
if err := dm.SetAccount(acc); err != nil {
t.Error(err)
}
rsponder.MaxComputedUsage[utils.VOICE] = 30 * time.Minute
cd := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "1001",
Account: "1001",
Destination: "1002",
ToR: utils.VOICE,
DurationIndex: 20,
TimeStart: tStart,
TimeEnd: tEnd,
Increments: Increments{
&Increment{Cost: 0, BalanceInfo: &DebitInfo{AccountID: acc.ID}},
&Increment{Cost: 0, Duration: 3 * time.Second, BalanceInfo: &DebitInfo{AccountID: acc.ID}},
},
},
}
var reply Account
if err := rsponder.RefundIncrements(cd, &reply); err != nil {
t.Error(err)
}
}
func TestResponderGetCost(t *testing.T) {
tmpCache := Cache
cfg, err := config.NewDefaultCGRConfig()
if err != nil {
t.Error(err)
}
tmpDm := dm
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dm := NewDataManager(db, cfg.CacheCfg(), nil)
defer func() {
Cache = tmpCache
SetDataStorage(tmpDm)
}()
rsponder.MaxComputedUsage = map[string]time.Duration{
utils.VOICE: time.Duration(20 * time.Second),
}
cd := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "1001",
ToR: utils.VOICE,
Account: "1001",
Destination: "1002",
DurationIndex: 90,
TimeStart: time.Date(2023, 3, 10, 9, 30, 0, 0, time.UTC),
TimeEnd: time.Date(2023, 3, 10, 9, 30, 2, 0, time.UTC),
},
}
rpDflt := &RatingPlan{
Id: "RP_DFLT",
Timings: map[string]*RITiming{
"30eab301": {
Years: utils.Years{},
Months: utils.Months{},
MonthDays: utils.MonthDays{},
WeekDays: utils.WeekDays{},
StartTime: "00:00:00",
},
},
Ratings: map[string]*RIRate{
"b457f861": {
Rates: []*Rate{
{
GroupIntervalStart: 0,
Value: 0.01,
RateIncrement: time.Second,
RateUnit: time.Second,
},
},
RoundingMethod: utils.ROUNDING_MIDDLE,
RoundingDecimals: 4,
},
},
DestinationRates: map[string]RPRateList{
"DEST": []*RPRate{
{
Timing: "30eab301",
Rating: "b457f861",
Weight: 10,
},
},
},
}
rpfTCDDBSWF := &RatingProfile{Id: utils.ConcatenatedKey(utils.META_OUT, cd.Tenant, "call", cd.Subject),
RatingPlanActivations: RatingPlanActivations{&RatingPlanActivation{
ActivationTime: time.Date(2015, 01, 01, 8, 0, 0, 0, time.UTC),
RatingPlanId: rpDflt.Id,
}},
}
dest := &Destination{
Id: "DEST",
Prefixes: []string{"1001", "1002", "1003"},
}
if err := dm.SetReverseDestination(dest, utils.NonTransactional); err != nil {
t.Error(err)
}
Cache.Set(utils.CacheRatingPlans, rpDflt.Id, rpDflt, []string{}, true, utils.NonTransactional)
Cache.Set(utils.CacheRatingProfilesTmp, rpfTCDDBSWF.Id, rpfTCDDBSWF, []string{}, true, utils.NonTransactional)
var reply CallCost
SetDataStorage(dm)
if err := rsponder.GetCost(cd, &reply); err != nil {
t.Error(err)
} else if reply.Cost != 0.02 {
t.Errorf("Received %v", reply.Cost)
}
}
func TestResponderDebit11(t *testing.T) {
tmpCache := Cache
tmpDm := *dm
defer func() {
Cache = tmpCache
SetDataStorage(&tmpDm)
}()
rsponder.MaxComputedUsage = map[string]time.Duration{
utils.VOICE: time.Duration(60 * time.Second),
}
arg := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "1001",
ToR: utils.VOICE,
Account: "1001",
Destination: "1002",
DurationIndex: 90,
TimeStart: time.Date(2023, 3, 10, 9, 30, 0, 0, time.UTC),
TimeEnd: time.Date(2023, 3, 10, 9, 30, 16, 0, time.UTC),
},
}
acc := &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"},
}},
}
dm.SetAccount(acc)
var reply CallCost
if err := rsponder.Debit(arg, &reply); err != nil {
t.Error(err)
}
}
func TestResponderMaxDebit11(t *testing.T) {
cfg, _ := config.NewDefaultCGRConfig()
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
tmpDm := dm
dm := NewDataManager(db, cfg.CacheCfg(), nil)
defer func() {
SetDataStorage(tmpDm)
}()
tStart, _ := utils.ParseTimeDetectLayout("2021-08-07T17:30:00Z", utils.EmptyString)
tEnd, _ := utils.ParseTimeDetectLayout("2021-08-07T17:31:30Z", utils.EmptyString)
cd := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "1001",
Account: "1001",
Destination: "1002",
DurationIndex: 30,
TimeStart: tStart,
TimeEnd: tEnd,
ToR: utils.MONETARY,
},
}
acc := &Account{
ID: "cgrates.org:1001",
BalanceMap: map[string]Balances{
utils.MONETARY: {
&Balance{Value: 11, Weight: 20, DestinationIDs: utils.NewStringMap("1002")},
}},
}
SetDataStorage(dm)
if err := dm.SetAccount(acc); err != nil {
t.Error(err)
}
dest := &Destination{
Id: "DEST",
Prefixes: []string{
"1002", "1003", "1004",
},
}
dm.SetReverseDestination(dest, utils.NonTransactional)
rp := &RatingPlan{
Id: "RP_1",
Ratings: map[string]*RIRate{
"b531174": {
Rates: RateGroups{
{
GroupIntervalStart: 0,
Value: 0.1,
RateIncrement: time.Second,
RateUnit: time.Second,
},
},
RoundingMethod: utils.ROUNDING_MIDDLE,
RoundingDecimals: 4,
},
},
Timings: map[string]*RITiming{
"30eab301": {
Years: utils.Years{},
Months: utils.Months{},
MonthDays: utils.MonthDays{},
WeekDays: utils.WeekDays{},
StartTime: "00:00:00",
},
},
DestinationRates: map[string]RPRateList{
"DEST": {
{
Rating: "b457f861",
Weight: 10,
},
},
},
}
rsponder.MaxComputedUsage = map[string]time.Duration{
utils.MONETARY: 90 * time.Second,
}
if err := dm.SetRatingPlan(rp, utils.NonTransactional); err != nil {
t.Error(err)
}
rP := &RatingProfile{
Id: utils.ConcatenatedKey(utils.META_OUT, cd.Tenant, "call", cd.Subject),
RatingPlanActivations: RatingPlanActivations{
{
RatingPlanId: rp.Id,
ActivationTime: time.Date(2022, 12, 20, 8, 30, 0, 0, time.UTC),
},
},
}
if err := dm.SetRatingProfile(rP, utils.NonTransactional); err != nil {
t.Error(err)
}
var reply CallCost
if err := rsponder.MaxDebit(cd, &reply); err != nil {
t.Error(err)
}
}
func TestResponderRefundRounding33(t *testing.T) {
cfg, _ := config.NewDefaultCGRConfig()
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
tmpDm := dm
dm := NewDataManager(db, cfg.CacheCfg(), nil)
defer func() {
SetDataStorage(tmpDm)
}()
tStart, _ := utils.ParseTimeDetectLayout("2019-08-07T17:30:00Z", utils.EmptyString)
tEnd, _ := utils.ParseTimeDetectLayout("2019-08-07T17:31:30Z", utils.EmptyString)
cd := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "1001",
Account: "1001",
Destination: "1002",
DurationIndex: 90,
TimeStart: tStart,
TimeEnd: tEnd,
Increments: Increments{
&Increment{
Duration: time.Minute,
Cost: 1,
BalanceInfo: &DebitInfo{
Unit: &UnitInfo{UUID: "1", DestinationID: "1", Consumed: 2.3, ToR: utils.VOICE, RateInterval: &RateInterval{Rating: &RIRate{Rates: RateGroups{&Rate{GroupIntervalStart: 0, Value: 100, RateIncrement: 10 * time.Second, RateUnit: time.Second}}}}},
Monetary: &MonetaryInfo{UUID: "2"},
AccountID: "cgrates.org:1001"},
},
&Increment{
Duration: time.Minute,
Cost: 1,
BalanceInfo: &DebitInfo{
Unit: &UnitInfo{UUID: "1", DestinationID: "1", Consumed: 2.3, ToR: utils.VOICE, RateInterval: &RateInterval{Rating: &RIRate{Rates: RateGroups{&Rate{GroupIntervalStart: 0, Value: 100, RateIncrement: 10 * time.Second, RateUnit: time.Second}}}}},
Monetary: &MonetaryInfo{UUID: "2"},
AccountID: "cgrates.org:1001"},
},
},
},
}
acc := &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"},
}},
}
if err := dm.SetAccount(acc); err != nil {
t.Error(err)
}
SetDataStorage(dm)
var reply Account
if rsponder.RefundRounding(cd, &reply); err != nil {
t.Error(err)
}
}
func TestResponderRounding(t *testing.T) {
cfg, _ := config.NewDefaultCGRConfig()
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
tmpDm := dm
dm := NewDataManager(db, cfg.CacheCfg(), nil)
defer func() {
SetDataStorage(tmpDm)
}()
tStart, _ := utils.ParseTimeDetectLayout("2020-08-07T17:30:00Z", utils.EmptyString)
tEnd, _ := utils.ParseTimeDetectLayout("2020-08-07T17:31:30Z", utils.EmptyString)
arg := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "1001",
Account: "1001",
Destination: "1002",
DurationIndex: 90,
TimeStart: tStart,
TimeEnd: tEnd,
ToR: utils.VOICE,
Increments: Increments{
&Increment{
Duration: time.Minute,
Cost: 10.4,
BalanceInfo: &DebitInfo{
Unit: &UnitInfo{UUID: "1", DestinationID: "1", Consumed: 2.3, ToR: utils.VOICE, RateInterval: &RateInterval{Rating: &RIRate{Rates: RateGroups{&Rate{GroupIntervalStart: 0, Value: 100, RateIncrement: 10 * time.Second, RateUnit: time.Second}}}}},
Monetary: &MonetaryInfo{UUID: "2"},
AccountID: "cgrates.org:1001"},
},
},
},
}
SetDataStorage(dm)
dm.SetAccount(&Account{
ID: "cgrates.org:1001",
BalanceMap: map[string]Balances{
utils.MONETARY: {
&Balance{
ID: utils.MetaDefault,
Value: 11, Weight: 20, DestinationIDs: utils.NewStringMap("1002")},
}},
})
rsponder.MaxComputedUsage = map[string]time.Duration{
utils.VOICE: 2 * time.Minute,
}
var reply Account
if err := rsponder.RefundRounding(arg, &reply); err != nil {
t.Error(err)
} else if reply.GetDefaultMoneyBalance().Value != 11 {
t.Error(utils.ToJSON(reply))
}
}
func TestResponderDebitCached(t *testing.T) {
cfg, _ := config.NewDefaultCGRConfig()
defer func() {
cfg2, _ := config.NewDefaultCGRConfig()
config.SetCgrConfig(cfg2)
}()
cfg.CacheCfg()[utils.CacheRPCResponses].Limit = 2
Cache.Clear(nil)
tStart, _ := utils.ParseTimeDetectLayout("2022-08-07T17:30:00Z", utils.EmptyString)
tEnd, _ := utils.ParseTimeDetectLayout("2022-08-07T17:31:30Z", utils.EmptyString)
cd := &CallDescriptorWithArgDispatcher{
CallDescriptor: &CallDescriptor{
CgrID: "test",
Category: "call",
Tenant: "cgrates.org",
Subject: "1001",
Account: "1001",
ToR: utils.VOICE,
Destination: "1002",
TimeStart: tStart,
DurationIndex: 10 * time.Second,
TimeEnd: tEnd,
},
}
cc := &CallCost{
Cost: 1.23,
Destination: "1002",
Timespans: []*TimeSpan{
{
TimeStart: time.Date(2022, 8, 7, 17, 30, 0, 0, time.UTC),
TimeEnd: time.Date(2022, 8, 7, 17, 31, 30, 0, time.UTC),
DurationIndex: 0,
RateInterval: &RateInterval{
Rating: &RIRate{
Rates: RateGroups{
&Rate{
GroupIntervalStart: 0,
Value: 100,
RateIncrement: 10 * time.Second,
RateUnit: time.Second,
},
},
},
},
},
},
ToR: utils.VOICE,
}
Cache.Set(utils.CacheRPCResponses, utils.ConcatenatedKey(utils.ResponderDebit, cd.CgrID),
&utils.CachedRPCResponse{Result: cc, Error: nil},
nil, true, utils.NonTransactional)
config.SetCgrConfig(cfg)
var reply CallCost
if err := rsponder.Debit(cd, &reply); err == nil {
t.Error(err)
}
}
// func TestResponderGetCostCache(t *testing.T) {
// cfg, _ := config.NewDefaultCGRConfig()
// cfg.CacheCfg()[utils.CacheRPCResponses].Limit = 1
// tmpCache := Cache
// defer func() {
// cfg2, _ := config.NewDefaultCGRConfig()
// config.SetCgrConfig(cfg2)
// Cache = tmpCache
// }()
// Cache.Clear(nil)
// config.SetCgrConfig(cfg)
// rs := &Responder{}
// arg := &CallDescriptorWithArgDispatcher{
// CallDescriptor: &CallDescriptor{
// CgrID: "CGRID",
// Tenant: "cgrates.org",
// Category: "call",
// Account: "1001",
// Subject: "1001",
// Destination: "1002",
// ToR: utils.VOICE,
// TimeStart: time.Now(),
// TimeEnd: time.Now().Add(10 * time.Minute),
// },
// }
// reply := &CallCost{}
// cacheKey := utils.ConcatenatedKey(utils.ResponderGetCost, arg.CgrID)
// Cache.Set(utils.CacheRPCResponses, cacheKey,
// &utils.CachedRPCResponse{Result: &CallCost{
// Tenant: "cgrates.org",
// Destination: "1002",
// ToR: utils.VOICE,
// Cost: 1.5,
// Timespans: TimeSpans{
// &TimeSpan{
// TimeStart: time.Now(),
// TimeEnd: time.Now().Add(10 * time.Minute),
// Cost: 1.5,
// RateInterval: &RateInterval{
// Timing: &RITiming{
// StartTime: "00:00:00",
// },
// Rating: &RIRate{
// ConnectFee: 0,
// Rates: RateGroups{
// &Rate{
// GroupIntervalStart: 0,
// Value: 0.15,
// RateIncrement: time.Minute,
// RateUnit: time.Minute,
// },
// },
// },
// },
// },
// },
// }, Error: nil},
// nil, true, utils.NonTransactional)
// err := rs.GetCost(arg, reply)
// if err != nil {
// t.Fatalf("Unexpected error: %v", err)
// }
// if rpl, has := Cache.Get(utils.CacheRPCResponses, cacheKey); !has {
// t.Fatalf("Expected cached result, but not found")
// } else if !reflect.DeepEqual(rpl, reply) {
// t.Errorf("Expected %+v,Received %+v", utils.ToJSON(reply), utils.ToJSON(rpl))
// }
// }
func TestResponderGetMaxSessionTimeOnAccounts(t *testing.T) {
cfg, _ := config.NewDefaultCGRConfig()
tmpDm := dm
defer func() {
dm = tmpDm
}()
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dm := NewDataManager(db, cfg.CacheCfg(), nil)
arg := &utils.GetMaxSessionTimeOnAccountsArgs{
Subject: "free",
Tenant: "cgrates.org",
AccountIDs: []string{"1001"},
Destination: "1004",
SetupTime: time.Date(2022, 12, 15, 8, 0, 0, 0, time.UTC),
Usage: time.Minute * 30,
}
SetDataStorage(dm)
var reply map[string]any
dm.SetAccount(&Account{ID: utils.ConcatenatedKey("cgrates.org", "1001"),
BalanceMap: map[string]Balances{utils.VOICE: {&Balance{
DestinationIDs: utils.NewStringMap("DEST"),
Value: float64(10), Weight: 10}}},
})
if err := dm.SetReverseDestination(&Destination{
Id: "DEST", Prefixes: []string{"1004"}}, utils.NonTransactional); err != nil {
t.Error(err)
}
dm.SetRatingPlan(&RatingPlan{
Id: "RP1",
Timings: map[string]*RITiming{
"78r22g8": {
Years: utils.Years{},
Months: utils.Months{},
MonthDays: utils.MonthDays{},
WeekDays: utils.WeekDays{},
StartTime: "00:00:00",
},
},
Ratings: map[string]*RIRate{
"29d929j": {
ConnectFee: 0.6,
RoundingMethod: utils.ROUNDING_UP,
RoundingDecimals: 4,
Rates: RateGroups{
{
GroupIntervalStart: 0,
Value: 0.02,
RateIncrement: 5 * time.Second,
RateUnit: 15 * time.Second,
},
},
},
},
DestinationRates: map[string]RPRateList{
"DEST": {
{
Timing: "78r22g8",
Rating: "29d929j",
},
},
},
}, utils.NonTransactional)
dm.SetRatingProfile(&RatingProfile{
Id: utils.ConcatenatedKey(utils.META_OUT, "cgrates.org", utils.MetaSuppliers, "free"),
RatingPlanActivations: RatingPlanActivations{
&RatingPlanActivation{
RatingPlanId: "RP1",
},
},
}, "")
if err := rsponder.GetMaxSessionTimeOnAccounts(arg, &reply); err != nil {
t.Error(err)
}
//
}
func TestResponderPing(t *testing.T) {
var r Responder
var reply string
err := r.Ping(nil, &reply)
if err != nil {
t.Fatalf("Expected no error, got %v", err)
}
if reply != utils.Pong {
t.Fatalf("Expected reply to be %s, got %s", utils.Pong, reply)
}
}
func TestResponderSetMaxComputedUsage(t *testing.T) {
r := &Responder{
MaxComputedUsage: make(map[string]time.Duration),
}
newMaxComputedUsage := map[string]time.Duration{
"user1": 5 * time.Hour,
"user2": 3 * time.Hour,
}
r.SetMaxComputedUsage(newMaxComputedUsage)
r.maxComputedUsageMutex.RLock()
defer r.maxComputedUsageMutex.RUnlock()
if !reflect.DeepEqual(r.MaxComputedUsage, newMaxComputedUsage) {
t.Fatalf("Expected MaxComputedUsage to be %v, got %v", newMaxComputedUsage, r.MaxComputedUsage)
}
}