mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-15 05:09:54 +05:00
426 lines
13 KiB
Go
426 lines
13 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"
|
|
|
|
"github.com/cgrates/cgrates/config"
|
|
"github.com/cgrates/cgrates/utils"
|
|
)
|
|
|
|
func TestActionProfileSort(t *testing.T) {
|
|
testStruct := &ActionProfiles{
|
|
{
|
|
Tenant: "test_tenantA",
|
|
ID: "test_idA",
|
|
Weight: 1,
|
|
},
|
|
{
|
|
Tenant: "test_tenantB",
|
|
ID: "test_idB",
|
|
Weight: 2,
|
|
},
|
|
{
|
|
Tenant: "test_tenantC",
|
|
ID: "test_idC",
|
|
Weight: 3,
|
|
},
|
|
}
|
|
expStruct := &ActionProfiles{
|
|
{
|
|
Tenant: "test_tenantC",
|
|
ID: "test_idC",
|
|
Weight: 3,
|
|
},
|
|
|
|
{
|
|
Tenant: "test_tenantB",
|
|
ID: "test_idB",
|
|
Weight: 2,
|
|
},
|
|
{
|
|
Tenant: "test_tenantA",
|
|
ID: "test_idA",
|
|
Weight: 1,
|
|
},
|
|
}
|
|
testStruct.Sort()
|
|
if !reflect.DeepEqual(expStruct, testStruct) {
|
|
t.Errorf("Expected: %s ,received: %s", utils.ToJSON(expStruct), utils.ToJSON(testStruct))
|
|
}
|
|
}
|
|
|
|
func TestActionAPDiktatRSRValues(t *testing.T) {
|
|
apdDiktat := APDiktat{
|
|
valRSR: config.RSRParsers{
|
|
&config.RSRParser{
|
|
Rules: ">;q=0.7;expires=3600",
|
|
},
|
|
&config.RSRParser{
|
|
Rules: ">;q=0.7;expires=3600",
|
|
},
|
|
},
|
|
}
|
|
rsrPars, err := apdDiktat.RSRValues(";")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
expected := config.RSRParsers{
|
|
&config.RSRParser{
|
|
Rules: ">;q=0.7;expires=3600",
|
|
},
|
|
&config.RSRParser{
|
|
Rules: ">;q=0.7;expires=3600",
|
|
},
|
|
}
|
|
if !reflect.DeepEqual(rsrPars, expected) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", utils.ToJSON(expected), utils.ToJSON(rsrPars))
|
|
|
|
}
|
|
}
|
|
|
|
func TestActionAPDiktatRSRValuesNil(t *testing.T) {
|
|
apdDiktat := APDiktat{}
|
|
rsrPars, err := apdDiktat.RSRValues(";")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
var expected config.RSRParsers
|
|
if !reflect.DeepEqual(rsrPars, expected) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", utils.ToJSON(expected), utils.ToJSON(rsrPars))
|
|
|
|
}
|
|
}
|
|
|
|
func TestActionAPDiktatRSRValuesError(t *testing.T) {
|
|
apdDiktat := APDiktat{
|
|
Value: "val`val2val3",
|
|
}
|
|
expErr := "Closed unspilit syntax"
|
|
_, err := apdDiktat.RSRValues(";")
|
|
if err == nil || err.Error() != expErr {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expErr, err)
|
|
}
|
|
}
|
|
func TestAPDiktatRSRValues(t *testing.T) {
|
|
dk := &APDiktat{Value: "1001"}
|
|
if rply, err := dk.RSRValues(utils.InfieldSep); err != nil {
|
|
return
|
|
} else if exp := config.NewRSRParsersMustCompile("1001", utils.InfieldSep); !reflect.DeepEqual(exp, rply) {
|
|
t.Errorf("Expected: %+v ,received: %+v", exp, rply)
|
|
}
|
|
}
|
|
|
|
func TestActionProfileSet(t *testing.T) {
|
|
ap := ActionProfile{Targets: make(map[string]utils.StringSet)}
|
|
exp := ActionProfile{
|
|
Tenant: "cgrates.org",
|
|
ID: "ID",
|
|
FilterIDs: []string{"fltr1", "*string:~*req.Account:1001"},
|
|
Schedule: utils.MetaNow,
|
|
Weight: 10,
|
|
Targets: map[string]utils.StringSet{
|
|
utils.MetaAccounts: utils.NewStringSet([]string{"1001", "1002"}),
|
|
utils.MetaThresholds: utils.NewStringSet([]string{"TH1", "TH2"}),
|
|
},
|
|
Actions: []*APAction{{
|
|
ID: "acc1",
|
|
Type: "val1",
|
|
FilterIDs: []string{"fltr1"},
|
|
Blocker: true,
|
|
TTL: 10,
|
|
Opts: map[string]interface{}{
|
|
"opt0": "val1",
|
|
"opt1": "val1",
|
|
"opt2": "val1",
|
|
"opt3": utils.MapStorage{"opt4": "val1"},
|
|
},
|
|
Diktats: []*APDiktat{{
|
|
Path: "path",
|
|
Value: "val1",
|
|
}},
|
|
}},
|
|
}
|
|
if err := ap.Set([]string{}, "", false, utils.EmptyString); err != utils.ErrWrongPath {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{"NotAField"}, "", false, utils.EmptyString); err != utils.ErrWrongPath {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{"NotAField", "1"}, "", false, utils.EmptyString); err != utils.ErrWrongPath {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := ap.Set([]string{utils.Tenant}, "cgrates.org", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.ID}, "ID", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.FilterIDs}, "fltr1;*string:~*req.Account:1001", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Schedule}, utils.MetaNow, false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Weight}, 10, false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Targets + "[" + utils.MetaAccounts + "]"}, "1001;1002", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Targets, utils.MetaThresholds}, "TH1;TH2", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := ap.Set([]string{utils.Actions + "[acc1]", utils.ID}, "acc1", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", "Wrong", "path", "2"}, "acc1", false, utils.EmptyString); err != utils.ErrWrongPath {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", "Wrong", "path"}, "acc1", false, utils.EmptyString); err != utils.ErrWrongPath {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", "Wrong"}, "acc1", false, utils.EmptyString); err != utils.ErrWrongPath {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.Opts}, "opt0:val1", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.Opts + "[opt1]"}, "val1", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.Opts, "opt2"}, "val1", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.Opts, "opt3", "opt4"}, "val1", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.Type}, "val1", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.FilterIDs}, "fltr1", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.Blocker}, "true", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.TTL}, "10", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.Diktats, utils.Path}, "path", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.Diktats, utils.Value}, "val1", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := ap.Actions[0].Set(nil, "", false); err != utils.ErrWrongPath {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(exp, ap) {
|
|
t.Errorf("Expected %v \n but received \n %v", utils.ToJSON(exp), utils.ToJSON(ap))
|
|
}
|
|
}
|
|
func TestActionAPDiktatRSRValues(t *testing.T) {
|
|
apdDiktat := APDiktat{
|
|
valRSR: config.RSRParsers{
|
|
&config.RSRParser{
|
|
Rules: ">;q=0.7;expires=3600",
|
|
},
|
|
&config.RSRParser{
|
|
Rules: ">;q=0.7;expires=3600",
|
|
},
|
|
},
|
|
}
|
|
rsrPars, err := apdDiktat.RSRValues(";")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
expected := config.RSRParsers{
|
|
&config.RSRParser{
|
|
Rules: ">;q=0.7;expires=3600",
|
|
},
|
|
&config.RSRParser{
|
|
Rules: ">;q=0.7;expires=3600",
|
|
},
|
|
}
|
|
if !reflect.DeepEqual(rsrPars, expected) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", utils.ToJSON(expected), utils.ToJSON(rsrPars))
|
|
|
|
}
|
|
}
|
|
|
|
func TestActionAPDiktatRSRValuesNil(t *testing.T) {
|
|
apdDiktat := APDiktat{}
|
|
rsrPars, err := apdDiktat.RSRValues(";")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
var expected config.RSRParsers
|
|
if !reflect.DeepEqual(rsrPars, expected) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", utils.ToJSON(expected), utils.ToJSON(rsrPars))
|
|
|
|
}
|
|
}
|
|
|
|
func TestActionAPDiktatRSRValuesError(t *testing.T) {
|
|
apdDiktat := APDiktat{
|
|
Value: "val`val2val3",
|
|
}
|
|
expErr := "Closed unspilit syntax"
|
|
_, err := apdDiktat.RSRValues(";")
|
|
if err == nil || err.Error() != expErr {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expErr, err)
|
|
}
|
|
|
|
}
|
|
func TestAPDiktatRSRValues(t *testing.T) {
|
|
dk := &APDiktat{Value: "1001"}
|
|
if rply, err := dk.RSRValues(utils.InfieldSep); err != nil {
|
|
return
|
|
} else if exp := config.NewRSRParsersMustCompile("1001", utils.InfieldSep); !reflect.DeepEqual(exp, rply) {
|
|
t.Errorf("Expected: %+v ,received: %+v", exp, rply)
|
|
}
|
|
}
|
|
|
|
func TestActionProfileSet(t *testing.T) {
|
|
ap := ActionProfile{Targets: make(map[string]utils.StringSet)}
|
|
exp := ActionProfile{
|
|
Tenant: "cgrates.org",
|
|
ID: "ID",
|
|
FilterIDs: []string{"fltr1", "*string:~*req.Account:1001"},
|
|
Schedule: utils.MetaNow,
|
|
Weight: 10,
|
|
Targets: map[string]utils.StringSet{
|
|
utils.MetaAccounts: utils.NewStringSet([]string{"1001", "1002"}),
|
|
utils.MetaThresholds: utils.NewStringSet([]string{"TH1", "TH2"}),
|
|
},
|
|
Actions: []*APAction{{
|
|
ID: "acc1",
|
|
Type: "val1",
|
|
FilterIDs: []string{"fltr1"},
|
|
Blocker: true,
|
|
TTL: 10,
|
|
Opts: map[string]interface{}{
|
|
"opt0": "val1",
|
|
"opt1": "val1",
|
|
"opt2": "val1",
|
|
"opt3": utils.MapStorage{"opt4": "val1"},
|
|
},
|
|
Diktats: []*APDiktat{{
|
|
Path: "path",
|
|
Value: "val1",
|
|
}},
|
|
}},
|
|
}
|
|
if err := ap.Set([]string{}, "", false, utils.EmptyString); err != utils.ErrWrongPath {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{"NotAField"}, "", false, utils.EmptyString); err != utils.ErrWrongPath {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{"NotAField", "1"}, "", false, utils.EmptyString); err != utils.ErrWrongPath {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := ap.Set([]string{utils.Tenant}, "cgrates.org", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.ID}, "ID", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.FilterIDs}, "fltr1;*string:~*req.Account:1001", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Schedule}, utils.MetaNow, false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Weight}, 10, false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Targets + "[" + utils.MetaAccounts + "]"}, "1001;1002", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Targets, utils.MetaThresholds}, "TH1;TH2", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := ap.Set([]string{utils.Actions + "[acc1]", utils.ID}, "acc1", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", "Wrong", "path", "2"}, "acc1", false, utils.EmptyString); err != utils.ErrWrongPath {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", "Wrong", "path"}, "acc1", false, utils.EmptyString); err != utils.ErrWrongPath {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", "Wrong"}, "acc1", false, utils.EmptyString); err != utils.ErrWrongPath {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.Opts}, "opt0:val1", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.Opts + "[opt1]"}, "val1", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.Opts, "opt2"}, "val1", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.Opts, "opt3", "opt4"}, "val1", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.Type}, "val1", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.FilterIDs}, "fltr1", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.Blocker}, "true", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.TTL}, "10", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.Diktats, utils.Path}, "path", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := ap.Set([]string{utils.Actions, "acc1", utils.Diktats, utils.Value}, "val1", false, utils.EmptyString); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if err := ap.Actions[0].Set(nil, "", false); err != utils.ErrWrongPath {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(exp, ap) {
|
|
t.Errorf("Expected %v \n but received \n %v", utils.ToJSON(exp), utils.ToJSON(ap))
|
|
}
|
|
}
|