Files
cgrates/engine/dynamicdp_test.go
2025-10-13 09:57:41 +02:00

742 lines
20 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 Affero 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 Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>
*/
package engine
import (
"bytes"
"reflect"
"strings"
"testing"
"github.com/cgrates/birpc"
"github.com/cgrates/birpc/context"
"github.com/cgrates/cgrates/config"
"github.com/cgrates/cgrates/utils"
"github.com/ericlagergren/decimal"
"github.com/nyaruka/phonenumbers"
)
func TestDynamicDPnewDynamicDP(t *testing.T) {
expDDP := &DynamicDP{
resConns: []string{"conn1"},
stsConns: []string{"conn2"},
actsConns: []string{"conn3"},
tenant: "cgrates.org",
initialDP: utils.StringSet{
"test": struct{}{},
},
cache: utils.MapStorage{},
ctx: context.Background(),
}
if rcv := NewDynamicDP(context.Background(), []string{"conn1"}, []string{"conn2"},
[]string{"conn3"}, nil, nil, "cgrates.org",
utils.StringSet{"test": struct{}{}}); !reflect.DeepEqual(rcv, expDDP) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>",
utils.ToJSON(expDDP), utils.ToJSON(rcv))
}
}
func TestDynamicDPString(t *testing.T) {
rcv := &DynamicDP{
resConns: []string{"conn1"},
stsConns: []string{"conn2"},
actsConns: []string{"conn3"},
tenant: "cgrates.org",
initialDP: utils.StringSet{
"test": struct{}{},
},
cache: utils.MapStorage{},
ctx: context.Background(),
}
exp2 := "[\"test\"]"
rcv2 := rcv.String()
if !reflect.DeepEqual(rcv2, exp2) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>",
utils.ToJSON(exp2), utils.ToJSON(rcv2))
}
}
func TestDynamicDPFieldAsInterfaceErrFilename(t *testing.T) {
rcv := &DynamicDP{
resConns: []string{"conn1"},
stsConns: []string{"conn2"},
actsConns: []string{"conn3"},
tenant: "cgrates.org",
initialDP: utils.StringSet{
"test": struct{}{},
},
cache: utils.MapStorage{},
ctx: context.Background(),
}
_, err := rcv.FieldAsInterface([]string{""})
if err == nil || err.Error() != "invalid fieldname <[]>" {
t.Errorf("expected: <%+v>, \nreceived: <%+v>",
"invalid fieldname <[]>", err)
}
}
func TestDynamicDPFieldAsInterfaceErrLenFldPath(t *testing.T) {
rcv := &DynamicDP{
resConns: []string{"conn1"},
stsConns: []string{"conn2"},
actsConns: []string{"conn3"},
tenant: "cgrates.org",
initialDP: utils.StringSet{
"test": struct{}{},
},
cache: utils.MapStorage{},
ctx: context.Background(),
}
_, err := rcv.FieldAsInterface([]string{})
if err == nil || err != utils.ErrNotFound {
t.Errorf("expected: <%+v>, \nreceived: <%+v>",
utils.ErrNotFound, err)
}
}
func TestDynamicDPFieldAsInterface(t *testing.T) {
DDP := &DynamicDP{
resConns: []string{"conn1"},
stsConns: []string{"conn2"},
actsConns: []string{"conn3"},
tenant: "cgrates.org",
initialDP: utils.StringSet{
"test": struct{}{},
},
cache: utils.MapStorage{
"testField": "testValue",
},
ctx: context.Background(),
}
result, err := DDP.FieldAsInterface([]string{"testField"})
if err != nil {
t.Error(err)
}
exp := "testValue"
if !reflect.DeepEqual(result, exp) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>",
utils.ToJSON(exp), utils.ToJSON(result))
}
}
func TestLibphonenumberDPString(t *testing.T) {
var pInt int32 = 2
LDP := &libphonenumberDP{
pNumber: &phonenumbers.PhoneNumber{
CountryCode: &pInt,
},
}
exp2 := "country_code:2"
rcv2 := LDP.String()
if !reflect.DeepEqual(rcv2, exp2) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>",
utils.ToJSON(exp2), utils.ToJSON(rcv2))
}
}
func TestLibphonenumberDPFieldAsString(t *testing.T) {
var pInt int32 = 2
LDP := &libphonenumberDP{
pNumber: &phonenumbers.PhoneNumber{
CountryCode: &pInt,
},
cache: utils.MapStorage{
"testField": "testValue",
},
}
exp2 := "testValue"
rcv2, err := LDP.FieldAsString([]string{"testField"})
if err != nil {
t.Error(err)
}
if !reflect.DeepEqual(rcv2, exp2) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>",
utils.ToJSON(exp2), utils.ToJSON(rcv2))
}
}
func TestLibphonenumberDPFieldAsStringError(t *testing.T) {
var pInt int32 = 2
LDP := &libphonenumberDP{
pNumber: &phonenumbers.PhoneNumber{
CountryCode: &pInt,
},
cache: utils.MapStorage{
"testField": "testValue",
},
}
_, err := LDP.FieldAsString([]string{"testField", "testField2"})
if err == nil || err.Error() != "WRONG_PATH" {
t.Errorf("expected: <%+v>, \nreceived: <%+v>",
"WRONG_PATH", err)
}
}
func TestLibphonenumberDPFieldAsInterfaceLen0(t *testing.T) {
var pInt int32 = 2
LDP := &libphonenumberDP{
pNumber: &phonenumbers.PhoneNumber{
CountryCode: &pInt,
},
cache: utils.MapStorage{
"testField": "testValue",
},
}
exp2 := &libphonenumberDP{
pNumber: &phonenumbers.PhoneNumber{
CountryCode: &pInt,
},
cache: utils.MapStorage{
"testField": "testValue",
}}
exp2.setDefaultFields()
rcv2, err := LDP.FieldAsInterface([]string{})
if err != nil {
t.Error(err)
}
if !reflect.DeepEqual(rcv2, exp2.cache) {
t.Errorf("expected: <%+v>, \nreceived: <%+v>",
exp2.cache, rcv2)
}
}
func TestNewLibPhoneNumberDPErr(t *testing.T) {
number := "badNum"
if _, err := newLibPhoneNumberDP(number); err != phonenumbers.ErrNotANumber {
t.Error(err)
}
}
func TestNewLibPhoneNumberDP(t *testing.T) {
number := "+355123456789"
num, err := phonenumbers.ParseAndKeepRawInput(number, utils.EmptyString)
if err != nil {
t.Error(err)
}
exp := utils.DataProvider(&libphonenumberDP{pNumber: num, cache: make(utils.MapStorage)})
if err != nil {
t.Error(err)
}
if rcv, err := newLibPhoneNumberDP(number); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(rcv, exp) {
t.Errorf("Expected \n<%+v>, Received \n<%+v>", exp, rcv)
}
}
func TestFieldAsInterfacelibphonenumberDP(t *testing.T) {
dDP := &libphonenumberDP{
pNumber: &phonenumbers.PhoneNumber{},
cache: utils.MapStorage{
"testField": "testValue",
},
}
expErr := `invalid field path <[fld1 fld2 fld3]> for libphonenumberDP`
if _, err := dDP.FieldAsInterface([]string{"fld1", "fld2", "fld3"}); err.Error() != expErr {
t.Error(err)
}
}
func TestLibphonenumberDPfieldAsInterfaceGeoLocationErr(t *testing.T) {
tmp := utils.Logger
defer func() {
utils.Logger = tmp
}()
buf := new(bytes.Buffer)
utils.Logger = utils.NewStdLoggerWithWriter(buf, "", 7)
dDP := &libphonenumberDP{
pNumber: &phonenumbers.PhoneNumber{},
cache: utils.MapStorage{
"testField": "testValue",
},
}
if rcv, err := dDP.fieldAsInterface([]string{"GeoLocation"}); err != nil {
t.Error(rcv, err)
}
expErr := "Received error: <language: tag is not well-formed> when getting GeoLocation for number"
if rcvTxt := buf.String(); !strings.Contains(rcvTxt, expErr) {
t.Errorf("Expected <%v>, Received <%v>", expErr, rcvTxt)
}
buf.Reset()
}
func TestLibphonenumberDPfieldAsInterface(t *testing.T) {
var pInt int32 = 49
var nNum uint64 = 17222020
var leadingZeros int32 = 1
lDP, err := newLibPhoneNumberDP("+49 17222020")
if err != nil {
t.Error(err)
}
libphone, canCast := lDP.(*libphonenumberDP)
if !canCast {
t.Errorf("cant cast <%v> to a libphonenumberDP", lDP)
}
dDP := &libphonenumberDP{
pNumber: libphone.pNumber,
cache: utils.MapStorage{
"testField": "testValue",
},
}
dDP.pNumber.Extension = utils.StringPointer("+")
dDP.pNumber.PreferredDomesticCarrierCode = utils.StringPointer("49 172")
exp := any(pInt)
if rcv, err := dDP.fieldAsInterface([]string{"CountryCode"}); err != nil {
t.Error(err)
} else if rcv != exp {
t.Errorf("Expected <%+v>, Received <%v>", exp, rcv)
}
exp = any(nNum)
if rcv, err := dDP.fieldAsInterface([]string{"NationalNumber"}); err != nil {
t.Error(err)
} else if rcv != exp {
t.Errorf("Expected <%+v>, Received <%v>", exp, rcv)
}
exp = any("DE")
if rcv, err := dDP.fieldAsInterface([]string{"Region"}); err != nil {
t.Error(err)
} else if rcv != exp {
t.Errorf("Expected <%+v>, Received <%v>", exp, rcv)
}
exp = any(phonenumbers.PhoneNumberType(11))
if rcv, err := dDP.fieldAsInterface([]string{"NumberType"}); err != nil {
t.Error(err)
} else if rcv != exp {
t.Errorf("Expected <%+v>, Received <%v><%t>", exp, rcv, rcv)
}
exp = any("Deutschland")
if rcv, err := dDP.fieldAsInterface([]string{"GeoLocation"}); err != nil {
t.Error(err)
} else if rcv != exp {
t.Errorf("Expected <%+v>, Received <%v>", exp, rcv)
}
exp = any("Vodafone")
if rcv, err := dDP.fieldAsInterface([]string{"Carrier"}); err != nil {
t.Error(err)
} else if rcv != exp {
t.Errorf("Expected <%+v>, Received <%v><%T>", exp, rcv, rcv)
}
exp = any(0)
if rcv, err := dDP.fieldAsInterface([]string{"LengthOfNationalDestinationCode"}); err != nil {
t.Error(err)
} else if rcv != exp {
t.Errorf("Expected <%+v>, Received <%v>", exp, rcv)
}
exp = any("+49 17222020")
if rcv, err := dDP.fieldAsInterface([]string{"RawInput"}); err != nil {
t.Error(err)
} else if rcv != exp {
t.Errorf("Expected <%+v>, Received <%v>", exp, rcv)
}
exp = any("+")
if rcv, err := dDP.fieldAsInterface([]string{"Extension"}); err != nil {
t.Error(err)
} else if rcv != exp {
t.Errorf("Expected <%+v>, Received <%v>", exp, rcv)
}
exp = any(leadingZeros)
if rcv, err := dDP.fieldAsInterface([]string{"NumberOfLeadingZeros"}); err != nil {
t.Error(err)
} else if rcv != exp {
t.Errorf("Expected <%+v>, Received <%v>", exp, rcv)
}
exp = any(false)
if rcv, err := dDP.fieldAsInterface([]string{"ItalianLeadingZero"}); err != nil {
t.Error(err)
} else if rcv != exp {
t.Errorf("Expected <%+v>, Received <%v>", exp, rcv)
}
exp = any("49 172")
if rcv, err := dDP.fieldAsInterface([]string{"PreferredDomesticCarrierCode"}); err != nil {
t.Error(err)
} else if rcv != exp {
t.Errorf("Expected <%+v>, Received <%v>", exp, rcv)
}
exp = any(phonenumbers.PhoneNumber_FROM_NUMBER_WITH_PLUS_SIGN)
if rcv, err := dDP.fieldAsInterface([]string{"CountryCodeSource"}); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(rcv, exp) {
t.Errorf("Expected <%+v>, Received <%v>", exp, rcv)
}
}
func TestDynamicDPfieldAsInterfaceMetaLibPhoneNumber(t *testing.T) {
dDP := &DynamicDP{
resConns: []string{"conn1"},
stsConns: []string{"conn2"},
actsConns: []string{"conn3"},
tenant: "cgrates.org",
cache: utils.MapStorage{
"testField": "testValue",
},
ctx: context.Background(),
}
dp, err := newLibPhoneNumberDP("+4917642092123")
if err != nil {
t.Error(err)
}
expAsField, err := dp.FieldAsInterface([]string{})
if err != nil {
t.Error(err)
}
exp := any(expAsField)
if rcv, err := dDP.fieldAsInterface([]string{utils.MetaLibPhoneNumber, "+4917642092123"}); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(rcv, exp) {
t.Errorf("Expected \n<%v>, Received \n<%v>", exp, rcv)
}
}
func TestDynamicDPfieldAsInterfaceErrMetaLibPhoneNumber(t *testing.T) {
dDP := &DynamicDP{
resConns: []string{"conn1"},
stsConns: []string{"conn2"},
actsConns: []string{"conn3"},
tenant: "cgrates.org",
cache: utils.MapStorage{
"testField": "testValue",
},
ctx: context.Background(),
}
if _, err := dDP.fieldAsInterface([]string{utils.MetaLibPhoneNumber, "inexistentNum"}); err == nil || err != phonenumbers.ErrNotANumber {
t.Error(err)
}
}
func TestDynamicDPfieldAsInterfaceNotFound(t *testing.T) {
Cache.Clear(nil)
ms := utils.MapStorage{}
dDp := NewDynamicDP(context.Background(), []string{}, []string{}, []string{}, nil, nil, "cgrates.org", ms)
if _, err := dDp.fieldAsInterface([]string{"inexistentfld1", "inexistentfld2"}); err == nil || err != utils.ErrNotFound {
t.Error(err)
}
}
func TestDynamicDPfieldAsInterfaceErrMetaStats(t *testing.T) {
dDP := &DynamicDP{
stsConns: []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaStats)},
tenant: "cgrates.org",
cache: utils.MapStorage{
"testField": "testValue",
},
ctx: context.Background(),
}
cfg := config.NewDefaultCGRConfig()
cc := &ccMock{
calls: map[string]func(ctx *context.Context, args any, reply any) error{
utils.StatSv1GetQueueDecimalMetrics: func(ctx *context.Context, args, reply any) error {
return utils.ErrNotImplemented
},
},
}
rpcInternal := make(chan birpc.ClientConnector, 1)
rpcInternal <- cc
cM := NewConnManager(cfg)
cM.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal, utils.MetaStats), utils.StatSv1, rpcInternal)
if _, err := dDP.fieldAsInterface([]string{utils.MetaStats, "fld1"}); err == nil || err != utils.ErrNotImplemented {
t.Error(err)
}
}
func TestDynamicDPfieldAsInterfaceErrMetaResources(t *testing.T) {
dDP := &DynamicDP{
resConns: []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaResources)},
tenant: "cgrates.org",
cache: utils.MapStorage{
"testField": "testValue",
},
ctx: context.Background(),
}
cfg := config.NewDefaultCGRConfig()
cc := &ccMock{
calls: map[string]func(ctx *context.Context, args any, reply any) error{
utils.ResourceSv1GetResourceWithConfig: func(ctx *context.Context, args, reply any) error {
return utils.ErrNotImplemented
},
},
}
rpcInternal := make(chan birpc.ClientConnector, 1)
rpcInternal <- cc
cM := NewConnManager(cfg)
cM.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal, utils.MetaResources), utils.ResourceSv1, rpcInternal)
if _, err := dDP.fieldAsInterface([]string{utils.MetaResources, "fld1"}); err == nil || err != utils.ErrNotImplemented {
t.Error(err)
}
}
func TestDynamicDPfieldAsInterfaceErrMetaAccounts(t *testing.T) {
dDP := &DynamicDP{
actsConns: []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAccounts)},
tenant: "cgrates.org",
cache: utils.MapStorage{
"testField": "testValue",
},
ctx: context.Background(),
}
cfg := config.NewDefaultCGRConfig()
cc := &ccMock{
calls: map[string]func(ctx *context.Context, args any, reply any) error{
utils.AccountSv1GetAccount: func(ctx *context.Context, args, reply any) error {
return utils.ErrNotImplemented
},
},
}
rpcInternal := make(chan birpc.ClientConnector, 1)
rpcInternal <- cc
cM := NewConnManager(cfg)
cM.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAccounts), utils.AccountSv1, rpcInternal)
if _, err := dDP.fieldAsInterface([]string{utils.MetaAccounts, "fld1"}); err == nil || err != utils.ErrNotImplemented {
t.Error(err)
}
}
func TestDynamicDPfieldAsInterfaceMetaAccounts(t *testing.T) {
Cache.Clear(nil)
dDP := &DynamicDP{
actsConns: []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAccounts)},
tenant: "cgrates.org",
cache: utils.MapStorage{
"testField": "testValue",
},
ctx: context.Background(),
}
cfg := config.NewDefaultCGRConfig()
customRply := &utils.Account{
Tenant: "cgrates.org",
ID: "1004",
FilterIDs: []string{"*string:~*req.Account:1004"},
Balances: map[string]*utils.Balance{
"ConcreteBalance1": {
ID: "ConcreteBalance1",
Weights: utils.DynamicWeights{
{
Weight: 20,
},
},
Type: utils.MetaConcrete,
Units: &utils.Decimal{Big: decimal.New(1000, 0)},
CostIncrements: []*utils.CostIncrement{
{
FilterIDs: []string{"*string:~*req.ToR:*data"},
Increment: &utils.Decimal{Big: decimal.New(1, 0)},
FixedFee: &utils.Decimal{Big: decimal.New(0, 0)},
RecurrentFee: &utils.Decimal{Big: decimal.New(1, 0)},
},
},
},
},
}
cc := &ccMock{
calls: map[string]func(ctx *context.Context, args any, reply any) error{
utils.AccountSv1GetAccount: func(ctx *context.Context, args, reply any) error {
rplCast, canCast := reply.(*utils.Account)
if !canCast {
t.Errorf("Wrong argument type : %T", reply)
return nil
}
*rplCast = *customRply
return nil
},
},
}
rpcInternal := make(chan birpc.ClientConnector, 1)
rpcInternal <- cc
cM := NewConnManager(cfg)
cM.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal, utils.MetaAccounts), utils.AccountSv1, rpcInternal)
exp := &utils.Decimal{Big: decimal.New(1000, 0)}
if rcv, err := dDP.fieldAsInterface([]string{utils.MetaAccounts, "1001", "Balances", "ConcreteBalance1", "Units"}); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(rcv, exp) {
t.Errorf("expected: <%+v>,\nreceived: \n<%+v>", utils.ToJSON(exp), utils.ToJSON(rcv))
}
}
func TestDynamicDPfieldAsInterfaceMetaResources(t *testing.T) {
Cache.Clear(nil)
dDP := &DynamicDP{
resConns: []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaResources)},
tenant: "cgrates.org",
cache: utils.MapStorage{
"testField": "testValue",
},
ctx: context.Background(),
}
cfg := config.NewDefaultCGRConfig()
customRply := &utils.ResourceWithConfig{
Resource: &utils.Resource{
Tenant: "cgrates.org",
ID: "ResGroup2",
Usages: map[string]*utils.ResourceUsage{
"RU1": {
Tenant: "cgrates.org",
ID: "RU1",
Units: 9,
},
},
},
Config: &utils.ResourceProfile{
Tenant: "cgrates.org",
ID: "ResGroup2",
FilterIDs: []string{"*string:~*req.Account:1001"},
Limit: 10,
AllocationMessage: "Approved",
Weights: utils.DynamicWeights{
{
Weight: 10,
}},
ThresholdIDs: []string{utils.MetaNone},
},
}
cc := &ccMock{
calls: map[string]func(ctx *context.Context, args any, reply any) error{
utils.ResourceSv1GetResourceWithConfig: func(ctx *context.Context, args, reply any) error {
rplCast, canCast := reply.(*utils.ResourceWithConfig)
if !canCast {
t.Errorf("Wrong argument type : %T", reply)
return nil
}
*rplCast = *customRply
return nil
},
},
}
rpcInternal := make(chan birpc.ClientConnector, 1)
rpcInternal <- cc
cM := NewConnManager(cfg)
cM.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal, utils.MetaResources), utils.ResourceSv1, rpcInternal)
var exp float64 = 9
if rcv, err := dDP.fieldAsInterface([]string{utils.MetaResources, "ResGroup2", "TotalUsage"}); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(rcv, exp) {
t.Errorf("expected: <%+v>,\nreceived: \n<%+v>", utils.ToJSON(exp), utils.ToJSON(rcv))
}
}
func TestDynamicDPfieldAsInterfaceMetaStats(t *testing.T) {
Cache.Clear(nil)
dDP := &DynamicDP{
stsConns: []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaStats)},
tenant: "cgrates.org",
cache: utils.MapStorage{
"testField": "testValue",
},
ctx: context.Background(),
}
cfg := config.NewDefaultCGRConfig()
customRply := &map[string]*utils.Decimal{
utils.MetaACD: utils.NewDecimal(1, 0),
utils.MetaTCC: utils.NewDecimal(2, 0),
}
cc := &ccMock{
calls: map[string]func(ctx *context.Context, args any, reply any) error{
utils.StatSv1GetQueueDecimalMetrics: func(ctx *context.Context, args, reply any) error {
rplCast, canCast := reply.(*map[string]*utils.Decimal)
if !canCast {
t.Errorf("Wrong argument type : %T", reply)
return nil
}
*rplCast = *customRply
return nil
},
},
}
rpcInternal := make(chan birpc.ClientConnector, 1)
rpcInternal <- cc
cM := NewConnManager(cfg)
cM.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal, utils.MetaStats), utils.StatSv1, rpcInternal)
exp := utils.NewDecimal(1, 0)
if rcv, err := dDP.fieldAsInterface([]string{utils.MetaStats, "Stats2", utils.MetaACD}); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(utils.ToJSON(exp), utils.ToJSON(rcv)) {
t.Errorf("expected: <%+v><%T>,\nreceived: \n<%+v><%T>", utils.ToJSON(exp), exp, utils.ToJSON(rcv), rcv)
}
}