mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
1757 lines
53 KiB
Go
1757 lines
53 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 apis
|
|
|
|
import (
|
|
"fmt"
|
|
"reflect"
|
|
"sort"
|
|
"testing"
|
|
|
|
"github.com/cgrates/birpc"
|
|
"github.com/cgrates/birpc/context"
|
|
"github.com/cgrates/cgrates/config"
|
|
"github.com/cgrates/cgrates/engine"
|
|
"github.com/cgrates/cgrates/utils"
|
|
)
|
|
|
|
func TestFiltersSetGetGetCountFilters(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
fltr := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "fltr_for_attr",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Account",
|
|
Values: []string{"1001"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
var reply string
|
|
err := admS.SetFilter(context.Background(), fltr, &reply)
|
|
if err != nil {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
|
|
}
|
|
if !reflect.DeepEqual(`"OK"`, utils.ToJSON(&reply)) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", `"OK"`, utils.ToJSON(&reply))
|
|
}
|
|
|
|
var replyGet engine.Filter
|
|
argsGet := &utils.TenantIDWithAPIOpts{
|
|
TenantID: &utils.TenantID{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "fltr_for_attr",
|
|
},
|
|
}
|
|
|
|
err = admS.GetFilter(context.Background(), argsGet, &replyGet)
|
|
if err != nil {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
|
|
}
|
|
if !reflect.DeepEqual(&replyGet, fltr.Filter) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", fltr.Filter, &replyGet)
|
|
}
|
|
|
|
var replyCnt int
|
|
argsCnt := &utils.ArgsItemIDs{
|
|
Tenant: utils.CGRateSorg,
|
|
}
|
|
err = admS.GetFiltersCount(context.Background(), argsCnt, &replyCnt)
|
|
if err != nil {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
|
|
}
|
|
if !reflect.DeepEqual(utils.ToJSON(&replyCnt), utils.ToJSON(1)) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", utils.ToJSON(1), utils.ToJSON(&replyCnt))
|
|
}
|
|
fltr2 := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "fltr_for_attr2",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Account",
|
|
Values: []string{"1002"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
var reply2 string
|
|
err = admS.SetFilter(context.Background(), fltr2, &reply2)
|
|
if err != nil {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
|
|
}
|
|
var replyCnt2 int
|
|
argsCnt2 := &utils.ArgsItemIDs{
|
|
Tenant: utils.CGRateSorg,
|
|
}
|
|
err = admS.GetFiltersCount(context.Background(), argsCnt2, &replyCnt2)
|
|
if err != nil {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
|
|
}
|
|
if !reflect.DeepEqual(utils.ToJSON(&replyCnt2), utils.ToJSON(2)) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", utils.ToJSON(2), utils.ToJSON(&replyCnt2))
|
|
}
|
|
|
|
args5 := &utils.ArgsItemIDs{}
|
|
var reply5 []string
|
|
err = admS.GetFilterIDs(context.Background(), args5, &reply5)
|
|
if err != nil {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
|
|
}
|
|
sort.Strings(reply5)
|
|
if !reflect.DeepEqual(reply5, []string{"fltr_for_attr", "fltr_for_attr2"}) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", []string{"fltr_for_attr", "fltr_for_attr2"}, reply5)
|
|
}
|
|
|
|
argRmv := &utils.TenantIDWithAPIOpts{
|
|
TenantID: &utils.TenantID{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "fltr_for_attr2",
|
|
},
|
|
APIOpts: nil,
|
|
}
|
|
var replyRmv string
|
|
err = admS.RemoveFilter(context.Background(), argRmv, &replyRmv)
|
|
if err != nil {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
|
|
}
|
|
if !reflect.DeepEqual(`"OK"`, utils.ToJSON(&replyRmv)) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", `"OK"`, utils.ToJSON(&replyRmv))
|
|
}
|
|
var replyCnt3 int
|
|
argsCnt3 := &utils.ArgsItemIDs{
|
|
Tenant: utils.CGRateSorg,
|
|
}
|
|
err = admS.GetFiltersCount(context.Background(), argsCnt3, &replyCnt3)
|
|
if err != nil {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
|
|
}
|
|
if !reflect.DeepEqual(utils.ToJSON(&replyCnt3), utils.ToJSON(1)) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", utils.ToJSON(1), utils.ToJSON(&replyCnt3))
|
|
}
|
|
|
|
args6 := &utils.ArgsItemIDs{}
|
|
var reply6 []string
|
|
err = admS.GetFilterIDs(context.Background(), args6, &reply6)
|
|
if err != nil {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
|
|
}
|
|
if !reflect.DeepEqual(reply6, []string{"fltr_for_attr"}) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", []string{"fltr_for_attr"}, reply6)
|
|
}
|
|
|
|
argRmv2 := &utils.TenantIDWithAPIOpts{
|
|
TenantID: &utils.TenantID{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "fltr_for_attr",
|
|
},
|
|
APIOpts: nil,
|
|
}
|
|
var replyRmv2 string
|
|
err = admS.RemoveFilter(context.Background(), argRmv2, &replyRmv2)
|
|
if err != nil {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
|
|
}
|
|
if !reflect.DeepEqual(`"OK"`, utils.ToJSON(&replyRmv2)) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", `"OK"`, utils.ToJSON(&replyRmv2))
|
|
}
|
|
|
|
var replyCnt4 int
|
|
argsCnt4 := &utils.ArgsItemIDs{}
|
|
err = admS.GetFiltersCount(context.Background(), argsCnt4, &replyCnt4)
|
|
if err == nil || err != utils.ErrNotFound {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", utils.ErrNotFound, err)
|
|
}
|
|
|
|
args7 := &utils.ArgsItemIDs{}
|
|
var reply7 []string
|
|
err = admS.GetFilterIDs(context.Background(), args7, &reply7)
|
|
if err == nil || err != utils.ErrNotFound {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", utils.ErrNotFound, err)
|
|
}
|
|
}
|
|
|
|
func TestFiltersSetFiltersMissingField(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
fltr := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
Tenant: utils.CGRateSorg,
|
|
},
|
|
}
|
|
var reply string
|
|
err := admS.SetFilter(context.Background(), fltr, &reply)
|
|
if err == nil || err.Error() != "MANDATORY_IE_MISSING: [ID]" {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "MANDATORY_IE_MISSING: [ID]", err)
|
|
}
|
|
if !reflect.DeepEqual(`""`, utils.ToJSON(&reply)) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", `""`, utils.ToJSON(&reply))
|
|
}
|
|
|
|
}
|
|
|
|
func TestFiltersSetFiltersTenantEmpty(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
fltr := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
ID: "fltr_for_attr",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Account",
|
|
Values: []string{"1002"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
var reply string
|
|
err := admS.SetFilter(context.Background(), fltr, &reply)
|
|
if err != nil {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
|
|
}
|
|
if !reflect.DeepEqual(`"OK"`, utils.ToJSON(&reply)) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", `"OK"`, utils.ToJSON(&reply))
|
|
}
|
|
var replyGet engine.Filter
|
|
argsGet := &utils.TenantIDWithAPIOpts{
|
|
TenantID: &utils.TenantID{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "fltr_for_attr",
|
|
},
|
|
}
|
|
|
|
err = admS.GetFilter(context.Background(), argsGet, &replyGet)
|
|
if err != nil {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
|
|
}
|
|
if !reflect.DeepEqual(&replyGet, fltr.Filter) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", fltr.Filter, &replyGet)
|
|
}
|
|
|
|
}
|
|
|
|
func TestFiltersSetFiltersGetFilterError(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB := &engine.DataDBMock{
|
|
GetFilterDrvF: func(ctx *context.Context, str1 string, str2 string) (*engine.Filter, error) {
|
|
return nil, utils.ErrNotImplemented
|
|
},
|
|
}
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
fltr := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "fltr_for_attr",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Account",
|
|
Values: []string{"1002"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
var reply string
|
|
err := admS.SetFilter(context.Background(), fltr, &reply)
|
|
if err == nil || err.Error() != "SERVER_ERROR: NOT_IMPLEMENTED" {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "SERVER_ERROR: NOT_IMPLEMENTED", err)
|
|
}
|
|
if !reflect.DeepEqual(`""`, utils.ToJSON(&reply)) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", `""`, utils.ToJSON(&reply))
|
|
}
|
|
|
|
}
|
|
|
|
func TestFiltersSetFiltersError(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB := &engine.DataDBMock{
|
|
GetFilterDrvF: func(ctx *context.Context, str1 string, str2 string) (*engine.Filter, error) {
|
|
return &engine.Filter{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "fltr_for_attr",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~testElement",
|
|
Values: []string{"testValue1", "testValue2"},
|
|
},
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~testElement2",
|
|
Values: []string{"testValue3", "testValue4"},
|
|
},
|
|
},
|
|
}, nil
|
|
},
|
|
GetIndexesDrvF: func(ctx *context.Context, idxItmType, tntCtx, idxKey, transactionID string) (indexes map[string]utils.StringSet, err error) {
|
|
return nil, utils.ErrNotImplemented
|
|
},
|
|
}
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
fltr := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "fltr_for_attr",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: "testType",
|
|
Element: "~testElement",
|
|
Values: []string{"testValue1", "testValue2"},
|
|
},
|
|
{
|
|
Type: "testType2",
|
|
Element: "~testElement2",
|
|
Values: []string{"testValue3", "testValue4"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
var reply string
|
|
err := admS.SetFilter(context.Background(), fltr, &reply)
|
|
if err == nil || err.Error() != "SERVER_ERROR: NOT_IMPLEMENTED" {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "SERVER_ERROR: NOT_IMPLEMENTED", err)
|
|
}
|
|
if !reflect.DeepEqual(`""`, utils.ToJSON(&reply)) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", `""`, utils.ToJSON(&reply))
|
|
}
|
|
|
|
}
|
|
|
|
func TestFiltersSetFiltersSetFilterError(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB := &engine.DataDBMock{
|
|
GetFilterDrvF: func(ctx *context.Context, str1 string, str2 string) (*engine.Filter, error) {
|
|
return nil, utils.ErrNotFound
|
|
},
|
|
GetIndexesDrvF: func(ctx *context.Context, idxItmType, tntCtx, idxKey, transactionID string) (indexes map[string]utils.StringSet, err error) {
|
|
return nil, utils.ErrNotImplemented
|
|
},
|
|
}
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
fltr := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "fltr_for_attr",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: "testType",
|
|
Element: "~testElement",
|
|
Values: []string{"testValue1", "testValue2"},
|
|
},
|
|
{
|
|
Type: "testType2",
|
|
Element: "~testElement2",
|
|
Values: []string{"testValue3", "testValue4"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
var reply string
|
|
err := admS.SetFilter(context.Background(), fltr, &reply)
|
|
if err == nil || err.Error() != "SERVER_ERROR: NOT_IMPLEMENTED" {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "SERVER_ERROR: NOT_IMPLEMENTED", err)
|
|
}
|
|
if !reflect.DeepEqual(`""`, utils.ToJSON(&reply)) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", `""`, utils.ToJSON(&reply))
|
|
}
|
|
|
|
}
|
|
|
|
func TestFiltersSetFiltersComposeCacheArgsForFilterError(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB := &engine.DataDBMock{
|
|
SetFilterDrvF: func(ctx *context.Context, fltr *engine.Filter) error {
|
|
|
|
return nil
|
|
},
|
|
GetFilterDrvF: func(ctx *context.Context, str1 string, str2 string) (*engine.Filter, error) {
|
|
return &engine.Filter{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "fltr_for_attr",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~testElement",
|
|
Values: []string{"testValue1", "testValue2"},
|
|
},
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~testElement2",
|
|
Values: []string{"testValue3", "testValue4"},
|
|
},
|
|
},
|
|
}, utils.ErrNotFound
|
|
},
|
|
GetIndexesDrvF: func(ctx *context.Context, idxItmType, tntCtx, idxKey, transactionID string) (indexes map[string]utils.StringSet, err error) {
|
|
return nil, utils.ErrNotImplemented
|
|
},
|
|
}
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
fltr := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "fltr_for_attr",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~testElement",
|
|
Values: []string{"testValue1", "testValue2"},
|
|
},
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~testElement2",
|
|
Values: []string{"testValue3", "testValue4"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
var reply string
|
|
err := admS.SetFilter(context.Background(), fltr, &reply)
|
|
if err == nil || err.Error() != "SERVER_ERROR: NOT_IMPLEMENTED" {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "SERVER_ERROR: NOT_IMPLEMENTED", err)
|
|
}
|
|
if !reflect.DeepEqual(`""`, utils.ToJSON(&reply)) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", `""`, utils.ToJSON(&reply))
|
|
}
|
|
|
|
}
|
|
|
|
func TestFiltersSetFiltersSetLoadIDsError(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB := &engine.DataDBMock{
|
|
SetFilterDrvF: func(ctx *context.Context, fltr *engine.Filter) error {
|
|
return nil
|
|
},
|
|
GetFilterDrvF: func(ctx *context.Context, str1 string, str2 string) (*engine.Filter, error) {
|
|
return &engine.Filter{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "fltr_for_attr",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~testElement",
|
|
Values: []string{"testValue1", "testValue2"},
|
|
},
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~testElement2",
|
|
Values: []string{"testValue3", "testValue4"},
|
|
},
|
|
},
|
|
}, nil
|
|
},
|
|
GetIndexesDrvF: func(ctx *context.Context, idxItmType, tntCtx, idxKey, transactionID string) (indexes map[string]utils.StringSet, err error) {
|
|
return nil, nil
|
|
},
|
|
SetLoadIDsDrvF: func(ctx *context.Context, loadIDs map[string]int64) error {
|
|
return utils.ErrNotImplemented
|
|
},
|
|
}
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
fltr := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "fltr_for_attr",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~testElement",
|
|
Values: []string{"testValue1", "testValue2"},
|
|
},
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~testElement2",
|
|
Values: []string{"testValue3", "testValue4"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
var reply string
|
|
err := admS.SetFilter(context.Background(), fltr, &reply)
|
|
if err == nil || err.Error() != "SERVER_ERROR: NOT_IMPLEMENTED" {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "SERVER_ERROR: NOT_IMPLEMENTED", err)
|
|
}
|
|
if !reflect.DeepEqual(`""`, utils.ToJSON(&reply)) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", `""`, utils.ToJSON(&reply))
|
|
}
|
|
|
|
}
|
|
func TestFiltersSetFiltersCacheForFilterError(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = "123"
|
|
cfg.AdminSCfg().CachesConns = []string{}
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB := &engine.DataDBMock{
|
|
SetFilterDrvF: func(ctx *context.Context, fltr *engine.Filter) error {
|
|
return nil
|
|
},
|
|
GetFilterDrvF: func(ctx *context.Context, str1 string, str2 string) (*engine.Filter, error) {
|
|
return &engine.Filter{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "fltr_for_attr",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~testElement",
|
|
Values: []string{"testValue1", "testValue2"},
|
|
},
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~testElement2",
|
|
Values: []string{"testValue3", "testValue4"},
|
|
},
|
|
},
|
|
}, nil
|
|
},
|
|
GetIndexesDrvF: func(ctx *context.Context, idxItmType, tntCtx, idxKey, transactionID string) (indexes map[string]utils.StringSet, err error) {
|
|
return nil, nil
|
|
},
|
|
SetLoadIDsDrvF: func(ctx *context.Context, loadIDs map[string]int64) error {
|
|
return nil
|
|
},
|
|
}
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
fltr := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "fltr_for_attr",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~testElement",
|
|
Values: []string{"testValue1", "testValue2"},
|
|
},
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~testElement2",
|
|
Values: []string{"testValue3", "testValue4"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
var reply string
|
|
err := admS.SetFilter(context.Background(), fltr, &reply)
|
|
if err == nil || err.Error() != "SERVER_ERROR: MANDATORY_IE_MISSING: [connIDs]" {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "SERVER_ERROR: MANDATORY_IE_MISSING: [connIDs]", err)
|
|
}
|
|
if !reflect.DeepEqual(`""`, utils.ToJSON(&reply)) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", `""`, utils.ToJSON(&reply))
|
|
}
|
|
|
|
}
|
|
|
|
func TestFiltersGetFilterNoTenant(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
fltr := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
ID: "fltr_for_attr",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Account",
|
|
Values: []string{"1002"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
var reply string
|
|
err := admS.SetFilter(context.Background(), fltr, &reply)
|
|
if err != nil {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
|
|
}
|
|
if !reflect.DeepEqual(`"OK"`, utils.ToJSON(&reply)) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", `"OK"`, utils.ToJSON(&reply))
|
|
}
|
|
var replyGet engine.Filter
|
|
argsGet := &utils.TenantIDWithAPIOpts{
|
|
TenantID: &utils.TenantID{
|
|
Tenant: utils.CGRateSorg,
|
|
ID: "fltr_for_attr",
|
|
},
|
|
}
|
|
|
|
err = admS.GetFilter(context.Background(), argsGet, &replyGet)
|
|
if err != nil {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
|
|
}
|
|
if !reflect.DeepEqual(&replyGet, fltr.Filter) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", fltr.Filter, &replyGet)
|
|
}
|
|
|
|
}
|
|
|
|
func TestFiltersGetFilterMissingField(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
fltr := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
ID: "fltr_for_attr",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Account",
|
|
Values: []string{"1002"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
var reply string
|
|
err := admS.SetFilter(context.Background(), fltr, &reply)
|
|
if err != nil {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
|
|
}
|
|
if !reflect.DeepEqual(`"OK"`, utils.ToJSON(&reply)) {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", `"OK"`, utils.ToJSON(&reply))
|
|
}
|
|
var replyGet engine.Filter
|
|
argsGet := &utils.TenantIDWithAPIOpts{
|
|
TenantID: &utils.TenantID{
|
|
Tenant: utils.CGRateSorg,
|
|
},
|
|
}
|
|
|
|
err = admS.GetFilter(context.Background(), argsGet, &replyGet)
|
|
if err == nil || err.Error() != "MANDATORY_IE_MISSING: [ID]" {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "MANDATORY_IE_MISSING: [ID]", err)
|
|
}
|
|
|
|
}
|
|
|
|
func TestFiltersGetFilterGetFilterError(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB := &engine.DataDBMock{
|
|
GetFilterDrvF: func(ctx *context.Context, str1 string, str2 string) (*engine.Filter, error) {
|
|
return nil, utils.ErrNotImplemented
|
|
},
|
|
}
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
var replyGet engine.Filter
|
|
argsGet := &utils.TenantIDWithAPIOpts{
|
|
TenantID: &utils.TenantID{
|
|
ID: "fltr_for_attr",
|
|
},
|
|
}
|
|
|
|
err := admS.GetFilter(context.Background(), argsGet, &replyGet)
|
|
if err == nil || err.Error() != "SERVER_ERROR: NOT_IMPLEMENTED" {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "SERVER_ERROR: NOT_IMPLEMENTED", err)
|
|
}
|
|
|
|
}
|
|
|
|
func TestFiltersGetFiltersCountError(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB := &engine.DataDBMock{
|
|
GetKeysForPrefixF: func(c *context.Context, s string) ([]string, error) {
|
|
return nil, utils.ErrNotImplemented
|
|
},
|
|
}
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
var reply int
|
|
args := &utils.ArgsItemIDs{}
|
|
|
|
err := admS.GetFiltersCount(context.Background(), args, &reply)
|
|
if err == nil || err.Error() != "NOT_IMPLEMENTED" {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "NOT_IMPLEMENTED", err)
|
|
}
|
|
|
|
}
|
|
|
|
func TestFiltersRemoveFilterMissingStructFieldError(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
var reply string
|
|
args := &utils.TenantIDWithAPIOpts{
|
|
TenantID: &utils.TenantID{},
|
|
APIOpts: nil,
|
|
}
|
|
|
|
err := admS.RemoveFilter(context.Background(), args, &reply)
|
|
if err == nil || err.Error() != "MANDATORY_IE_MISSING: [ID]" {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "MANDATORY_IE_MISSING: [ID]", err)
|
|
}
|
|
|
|
}
|
|
|
|
func TestFiltersRemoveFilterRemoveFilterError(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB := &engine.DataDBMock{
|
|
GetFilterDrvF: func(ctx *context.Context, str1 string, str2 string) (*engine.Filter, error) {
|
|
return nil, utils.ErrNotImplemented
|
|
},
|
|
}
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
var reply string
|
|
args := &utils.TenantIDWithAPIOpts{
|
|
TenantID: &utils.TenantID{
|
|
Tenant: "",
|
|
ID: "testID",
|
|
},
|
|
APIOpts: nil,
|
|
}
|
|
|
|
err := admS.RemoveFilter(context.Background(), args, &reply)
|
|
if err == nil || err.Error() != "SERVER_ERROR: NOT_IMPLEMENTED" {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "SERVER_ERROR: NOT_IMPLEMENTED", err)
|
|
}
|
|
|
|
}
|
|
|
|
type mockClientConn struct {
|
|
calls map[string]func(*context.Context, any, any) error
|
|
}
|
|
|
|
func (mCC *mockClientConn) Call(ctx *context.Context, serviceMethod string, args any, reply any) (err error) {
|
|
if call, has := mCC.calls[serviceMethod]; has {
|
|
return call(ctx, args, reply)
|
|
}
|
|
return utils.ErrUnsupporteServiceMethod
|
|
}
|
|
|
|
func TestFiltersSetFilterReloadCache(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
cfg.AdminSCfg().CachesConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaCaches)}
|
|
dataDB, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, nil)
|
|
expArgs := &utils.AttrReloadCacheWithAPIOpts{
|
|
APIOpts: map[string]any{
|
|
utils.MetaCache: utils.MetaReload,
|
|
},
|
|
Tenant: "cgrates.org",
|
|
FilterIDs: []string{"cgrates.org:FLTR_ID"},
|
|
}
|
|
ccM := &mockClientConn{
|
|
calls: map[string]func(ctx *context.Context, args any, reply any) error{
|
|
utils.CacheSv1ReloadCache: func(ctx *context.Context, args, reply any) error {
|
|
if !reflect.DeepEqual(args, expArgs) {
|
|
return fmt.Errorf("expected: <%+v>,\nreceived: <%+v>", utils.ToJSON(expArgs), utils.ToJSON(args))
|
|
}
|
|
return nil
|
|
},
|
|
},
|
|
}
|
|
rpcInternal := make(chan birpc.ClientConnector, 1)
|
|
rpcInternal <- ccM
|
|
cM := engine.NewConnManager(cfg)
|
|
cM.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal, utils.MetaCaches), utils.CacheSv1, rpcInternal)
|
|
adms := &AdminSv1{
|
|
cfg: cfg,
|
|
dm: dm,
|
|
connMgr: cM,
|
|
}
|
|
arg := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
ID: "FLTR_ID",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Account",
|
|
Values: []string{"1001"},
|
|
},
|
|
},
|
|
},
|
|
APIOpts: map[string]any{
|
|
utils.MetaCache: utils.MetaReload,
|
|
},
|
|
}
|
|
var reply string
|
|
|
|
if err := adms.SetFilter(context.Background(), arg, &reply); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
attrPrf := &utils.APIAttributeProfileWithAPIOpts{
|
|
APIAttributeProfile: &utils.APIAttributeProfile{
|
|
FilterIDs: []string{"FLTR_ID"},
|
|
ID: "ATTR_ID",
|
|
Weights: utils.DynamicWeights{
|
|
{
|
|
Weight: 10,
|
|
},
|
|
},
|
|
Attributes: []*utils.ExternalAttribute{
|
|
{
|
|
Path: "*req.Account",
|
|
Value: "1003",
|
|
Type: utils.MetaConstant,
|
|
},
|
|
},
|
|
},
|
|
APIOpts: map[string]any{
|
|
utils.MetaCache: utils.MetaNone,
|
|
},
|
|
}
|
|
|
|
if err := adms.SetAttributeProfile(context.Background(), attrPrf, &reply); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
thPrf := &engine.ThresholdProfileWithAPIOpts{
|
|
ThresholdProfile: &engine.ThresholdProfile{
|
|
ID: "THD_ID",
|
|
FilterIDs: []string{"FLTR_ID"},
|
|
MaxHits: 10,
|
|
Weights: utils.DynamicWeights{
|
|
{
|
|
Weight: 10,
|
|
},
|
|
},
|
|
},
|
|
APIOpts: map[string]any{
|
|
utils.MetaCache: utils.MetaNone,
|
|
},
|
|
}
|
|
|
|
if err := adms.SetThresholdProfile(context.Background(), thPrf, &reply); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
rsPrf := &utils.ResourceProfileWithAPIOpts{
|
|
ResourceProfile: &utils.ResourceProfile{
|
|
ID: "RES_ID",
|
|
FilterIDs: []string{"FLTR_ID"},
|
|
Weights: utils.DynamicWeights{
|
|
{
|
|
Weight: 10,
|
|
}},
|
|
},
|
|
APIOpts: map[string]any{
|
|
utils.MetaCache: utils.MetaNone,
|
|
},
|
|
}
|
|
|
|
if err := adms.SetResourceProfile(context.Background(), rsPrf, &reply); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
sqPrf := &engine.StatQueueProfileWithAPIOpts{
|
|
StatQueueProfile: &engine.StatQueueProfile{
|
|
ID: "SQ_ID",
|
|
FilterIDs: []string{"FLTR_ID"},
|
|
Weights: utils.DynamicWeights{
|
|
{
|
|
Weight: 10.0,
|
|
},
|
|
},
|
|
},
|
|
APIOpts: map[string]any{
|
|
utils.MetaCache: utils.MetaNone,
|
|
},
|
|
}
|
|
|
|
if err := adms.SetStatQueueProfile(context.Background(), sqPrf, &reply); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
arg = &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
ID: "FLTR_ID",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Account",
|
|
Values: []string{"1002"},
|
|
},
|
|
},
|
|
},
|
|
APIOpts: map[string]any{
|
|
utils.MetaCache: utils.MetaReload,
|
|
},
|
|
}
|
|
expArgs = &utils.AttrReloadCacheWithAPIOpts{
|
|
APIOpts: map[string]any{
|
|
utils.MetaCache: utils.MetaReload,
|
|
},
|
|
Tenant: "cgrates.org",
|
|
FilterIDs: []string{"cgrates.org:FLTR_ID"},
|
|
AttributeFilterIndexIDs: []string{"cgrates.org:*string:*req.Account:1001", "cgrates.org:*string:*req.Account:1002"},
|
|
ResourceFilterIndexIDs: []string{"cgrates.org:*string:*req.Account:1001", "cgrates.org:*string:*req.Account:1002"},
|
|
StatFilterIndexIDs: []string{"cgrates.org:*string:*req.Account:1001", "cgrates.org:*string:*req.Account:1002"},
|
|
ThresholdFilterIndexIDs: []string{"cgrates.org:*string:*req.Account:1001", "cgrates.org:*string:*req.Account:1002"},
|
|
}
|
|
|
|
if err := adms.SetFilter(context.Background(), arg, &reply); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
dm.DataDB()[utils.MetaDefault].Flush(utils.EmptyString)
|
|
}
|
|
|
|
func TestFiltersSetFilterClearCache(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
cfg.AdminSCfg().CachesConns = []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaCaches)}
|
|
dataDB, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, nil)
|
|
expArgs := &utils.AttrCacheIDsWithAPIOpts{
|
|
APIOpts: map[string]any{
|
|
utils.MetaCache: utils.MetaClear,
|
|
},
|
|
Tenant: "cgrates.org",
|
|
CacheIDs: []string{utils.CacheFilters},
|
|
}
|
|
ccM := &mockClientConn{
|
|
calls: map[string]func(ctx *context.Context, args any, reply any) error{
|
|
utils.CacheSv1Clear: func(ctx *context.Context, args, reply any) error {
|
|
sort.Strings(args.(*utils.AttrCacheIDsWithAPIOpts).CacheIDs)
|
|
if !reflect.DeepEqual(args, expArgs) {
|
|
return fmt.Errorf("expected: <%+v>,\nreceived: <%+v>", utils.ToJSON(expArgs), utils.ToJSON(args))
|
|
}
|
|
return nil
|
|
},
|
|
},
|
|
}
|
|
rpcInternal := make(chan birpc.ClientConnector, 1)
|
|
rpcInternal <- ccM
|
|
cM := engine.NewConnManager(cfg)
|
|
cM.AddInternalConn(utils.ConcatenatedKey(utils.MetaInternal, utils.MetaCaches), utils.CacheSv1, rpcInternal)
|
|
adms := &AdminSv1{
|
|
cfg: cfg,
|
|
dm: dm,
|
|
connMgr: cM,
|
|
}
|
|
arg := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
ID: "FLTR_ID",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Account",
|
|
Values: []string{"1001"},
|
|
},
|
|
},
|
|
},
|
|
APIOpts: map[string]any{
|
|
utils.MetaCache: utils.MetaClear,
|
|
},
|
|
}
|
|
var reply string
|
|
|
|
if err := adms.SetFilter(context.Background(), arg, &reply); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
attrPrf := &utils.APIAttributeProfileWithAPIOpts{
|
|
APIAttributeProfile: &utils.APIAttributeProfile{
|
|
FilterIDs: []string{"FLTR_ID"},
|
|
ID: "ATTR_ID",
|
|
Weights: utils.DynamicWeights{
|
|
{
|
|
Weight: 10,
|
|
},
|
|
},
|
|
Attributes: []*utils.ExternalAttribute{
|
|
{
|
|
Path: "*req.Account",
|
|
Value: "1003",
|
|
Type: utils.MetaConstant,
|
|
},
|
|
},
|
|
},
|
|
APIOpts: map[string]any{
|
|
utils.MetaCache: utils.MetaNone,
|
|
},
|
|
}
|
|
|
|
if err := adms.SetAttributeProfile(context.Background(), attrPrf, &reply); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
thPrf := &engine.ThresholdProfileWithAPIOpts{
|
|
ThresholdProfile: &engine.ThresholdProfile{
|
|
ID: "THD_ID",
|
|
FilterIDs: []string{"FLTR_ID"},
|
|
MaxHits: 10,
|
|
Weights: utils.DynamicWeights{
|
|
{
|
|
Weight: 10,
|
|
},
|
|
},
|
|
},
|
|
APIOpts: map[string]any{
|
|
utils.MetaCache: utils.MetaNone,
|
|
},
|
|
}
|
|
|
|
if err := adms.SetThresholdProfile(context.Background(), thPrf, &reply); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
rsPrf := &utils.ResourceProfileWithAPIOpts{
|
|
ResourceProfile: &utils.ResourceProfile{
|
|
ID: "RES_ID",
|
|
FilterIDs: []string{"FLTR_ID"},
|
|
Weights: utils.DynamicWeights{
|
|
{
|
|
Weight: 10,
|
|
}},
|
|
},
|
|
APIOpts: map[string]any{
|
|
utils.MetaCache: utils.MetaNone,
|
|
},
|
|
}
|
|
|
|
if err := adms.SetResourceProfile(context.Background(), rsPrf, &reply); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
sqPrf := &engine.StatQueueProfileWithAPIOpts{
|
|
StatQueueProfile: &engine.StatQueueProfile{
|
|
ID: "SQ_ID",
|
|
FilterIDs: []string{"FLTR_ID"},
|
|
Weights: utils.DynamicWeights{
|
|
{
|
|
Weight: 10,
|
|
},
|
|
},
|
|
},
|
|
APIOpts: map[string]any{
|
|
utils.MetaCache: utils.MetaNone,
|
|
},
|
|
}
|
|
|
|
if err := adms.SetStatQueueProfile(context.Background(), sqPrf, &reply); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
arg = &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
ID: "FLTR_ID",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Account",
|
|
Values: []string{"1002"},
|
|
},
|
|
},
|
|
},
|
|
APIOpts: map[string]any{
|
|
utils.MetaCache: utils.MetaClear,
|
|
},
|
|
}
|
|
expArgs = &utils.AttrCacheIDsWithAPIOpts{
|
|
APIOpts: map[string]any{
|
|
utils.MetaCache: utils.MetaClear,
|
|
},
|
|
Tenant: "cgrates.org",
|
|
CacheIDs: []string{utils.CacheAttributeFilterIndexes, utils.CacheThresholdFilterIndexes,
|
|
utils.CacheResourceFilterIndexes, utils.CacheStatFilterIndexes,
|
|
utils.CacheFilters},
|
|
}
|
|
sort.Strings(expArgs.CacheIDs)
|
|
|
|
if err := adms.SetFilter(context.Background(), arg, &reply); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
dm.DataDB()[utils.MetaDefault].Flush(utils.EmptyString)
|
|
}
|
|
func TestFiltersRemoveFilterSetLoadIDsError(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
cfg.CacheCfg().ReplicationConns = []string{"rep"}
|
|
cfg.CacheCfg().Partitions[utils.CacheReverseFilterIndexes].Replicate = false
|
|
cfg.RPCConns()["connID"] = &config.RPCConn{}
|
|
config.CgrConfig().DbCfg().Items[utils.CacheReverseFilterIndexes].Remote = true
|
|
config.CgrConfig().DbCfg().DBConns[utils.MetaDefault].RmtConns = []string{"connID"}
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB := &engine.DataDBMock{
|
|
GetFilterDrvF: func(ctx *context.Context, str1 string, str2 string) (*engine.Filter, error) {
|
|
return &engine.Filter{}, nil
|
|
},
|
|
GetIndexesDrvF: func(ctx *context.Context, idxItmType, tntCtx, idxKey, transactionID string) (indexes map[string]utils.StringSet, err error) {
|
|
return map[string]utils.StringSet{}, utils.ErrNotFound
|
|
},
|
|
SetIndexesDrvF: func(ctx *context.Context, idxItmType, tntCtx string, indexes map[string]utils.StringSet, commit bool, transactionID string) (err error) {
|
|
return utils.ErrNotFound
|
|
},
|
|
RemoveFilterDrvF: func(ctx *context.Context, str1 string, str2 string) error {
|
|
return nil
|
|
},
|
|
}
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
var reply string
|
|
args := &utils.TenantIDWithAPIOpts{
|
|
TenantID: &utils.TenantID{
|
|
Tenant: "",
|
|
ID: "testID",
|
|
},
|
|
APIOpts: nil,
|
|
}
|
|
|
|
err := admS.RemoveFilter(context.Background(), args, &reply)
|
|
if err == nil || err.Error() != "SERVER_ERROR: NOT_IMPLEMENTED" {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "SERVER_ERROR: NOT_IMPLEMENTED", err)
|
|
}
|
|
config.CgrConfig().DbCfg().DBConns[utils.MetaDefault].RmtConns = []string{}
|
|
config.CgrConfig().DbCfg().Items[utils.CacheReverseFilterIndexes].Remote = false
|
|
}
|
|
|
|
func TestFiltersRemoveFilterCallCacheForFilterError(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = "123"
|
|
cfg.CacheCfg().ReplicationConns = []string{"rep"}
|
|
cfg.CacheCfg().Partitions[utils.CacheReverseFilterIndexes].Replicate = false
|
|
cfg.RPCConns()["connID"] = &config.RPCConn{}
|
|
config.CgrConfig().DbCfg().Items[utils.CacheReverseFilterIndexes].Remote = true
|
|
config.CgrConfig().DbCfg().DBConns[utils.MetaDefault].RmtConns = []string{"connID"}
|
|
cfg.AdminSCfg().CachesConns = []string{}
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB := &engine.DataDBMock{
|
|
SetLoadIDsDrvF: func(ctx *context.Context, loadIDs map[string]int64) error {
|
|
return nil
|
|
},
|
|
GetFilterDrvF: func(ctx *context.Context, str1 string, str2 string) (*engine.Filter, error) {
|
|
return &engine.Filter{}, nil
|
|
},
|
|
GetIndexesDrvF: func(ctx *context.Context, idxItmType, tntCtx, idxKey, transactionID string) (indexes map[string]utils.StringSet, err error) {
|
|
return map[string]utils.StringSet{}, utils.ErrNotFound
|
|
},
|
|
SetIndexesDrvF: func(ctx *context.Context, idxItmType, tntCtx string, indexes map[string]utils.StringSet, commit bool, transactionID string) (err error) {
|
|
return utils.ErrNotFound
|
|
},
|
|
RemoveFilterDrvF: func(ctx *context.Context, str1 string, str2 string) error {
|
|
return nil
|
|
},
|
|
}
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
var reply string
|
|
args := &utils.TenantIDWithAPIOpts{
|
|
TenantID: &utils.TenantID{
|
|
Tenant: "",
|
|
ID: "testID",
|
|
},
|
|
APIOpts: nil,
|
|
}
|
|
|
|
err := admS.RemoveFilter(context.Background(), args, &reply)
|
|
if err == nil || err.Error() != "SERVER_ERROR: MANDATORY_IE_MISSING: [connIDs]" {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "SERVER_ERROR: MANDATORY_IE_MISSING: [connIDs]", err)
|
|
}
|
|
config.CgrConfig().DbCfg().DBConns[utils.MetaDefault].RmtConns = []string{}
|
|
config.CgrConfig().DbCfg().Items[utils.CacheReverseFilterIndexes].Remote = false
|
|
}
|
|
|
|
func TestFiltersGetFilterIDs(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB := &engine.DataDBMock{}
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
|
|
args6 := &utils.ArgsItemIDs{}
|
|
var reply6 []string
|
|
err := admS.GetFilterIDs(context.Background(), args6, &reply6)
|
|
if err == nil || err.Error() != "NOT_IMPLEMENTED" {
|
|
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "NOT_IMPLEMENTED", err)
|
|
}
|
|
}
|
|
|
|
func TestFiltersValidateFilterRuleOK(t *testing.T) {
|
|
fltrRules := []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Subject",
|
|
Values: []string{"1004", "6774", "22312"},
|
|
},
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*opts.Subsystems",
|
|
Values: []string{"*attributes"},
|
|
},
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~*req.Destinations",
|
|
Values: []string{"+0775", "+442"},
|
|
},
|
|
}
|
|
|
|
if err := validateFilterRules(fltrRules); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestFiltersValidateFilterRuleErr(t *testing.T) {
|
|
fltrRules := []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Subject",
|
|
Values: []string{"1004", "6774", "22312"},
|
|
},
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*opts.Subsystems",
|
|
Values: []string{"*attributes"},
|
|
},
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "",
|
|
Values: []string{},
|
|
},
|
|
}
|
|
|
|
experr := `there exists at least one filter rule that is not valid`
|
|
if err := validateFilterRules(fltrRules); err == nil ||
|
|
err.Error() != experr {
|
|
t.Errorf("expected: <%+v>, \nreceived: <%+v>", experr, err)
|
|
}
|
|
}
|
|
|
|
func TestFiltersFiltersMatchTrue(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
fltrS := engine.NewFilterS(cfg, connMgr, dm)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, fltrS)
|
|
args := &engine.ArgsFiltersMatch{
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: "cgrates.org",
|
|
ID: "EventTest",
|
|
Event: map[string]any{
|
|
utils.AccountField: "1001",
|
|
utils.Destination: "1002",
|
|
},
|
|
},
|
|
FilterIDs: []string{
|
|
"*string:~*req.Account:1001",
|
|
"*prefix:~*req.Destination:10",
|
|
},
|
|
}
|
|
var reply bool
|
|
if err := admS.FiltersMatch(context.Background(), args, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if reply != true {
|
|
t.Errorf("expected reply to be <%+v>", true)
|
|
}
|
|
}
|
|
|
|
func TestFiltersFiltersMatchFalse(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
fltrS := engine.NewFilterS(cfg, connMgr, dm)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, fltrS)
|
|
args := &engine.ArgsFiltersMatch{
|
|
CGREvent: &utils.CGREvent{
|
|
ID: "EventTest",
|
|
Event: map[string]any{
|
|
utils.AccountField: "1001",
|
|
utils.Destination: "2002",
|
|
},
|
|
},
|
|
FilterIDs: []string{
|
|
"*string:~*req.Account:1001",
|
|
"*prefix:~*req.Destination:10",
|
|
},
|
|
}
|
|
var reply bool
|
|
if err := admS.FiltersMatch(context.Background(), args, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if reply != false {
|
|
t.Errorf("expected reply to be <%+v>", false)
|
|
}
|
|
}
|
|
|
|
func TestFiltersFiltersMatchErr(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
fltrS := engine.NewFilterS(cfg, connMgr, dm)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, fltrS)
|
|
args := &engine.ArgsFiltersMatch{
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: "cgrates.org",
|
|
ID: "EventTest",
|
|
Event: map[string]any{
|
|
utils.AccountField: "1001",
|
|
utils.Destination: "1002",
|
|
},
|
|
},
|
|
FilterIDs: []string{
|
|
"*string.invalid:filter",
|
|
},
|
|
}
|
|
experr := `inline parse error for string: <*string.invalid:filter>`
|
|
var reply bool
|
|
if err := admS.FiltersMatch(context.Background(), args, &reply); err == nil || err.Error() != experr {
|
|
t.Errorf("expected: <%+v>, \nreceived: <%+v>", experr, err)
|
|
} else if reply != false {
|
|
t.Errorf("expected reply to be <%+v>", false)
|
|
}
|
|
}
|
|
|
|
func TestFiltersGetFiltersOK(t *testing.T) {
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
args1 := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
Tenant: "cgrates.org",
|
|
ID: "test_ID1",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Account",
|
|
Values: []string{"1002"},
|
|
},
|
|
},
|
|
},
|
|
APIOpts: nil,
|
|
}
|
|
|
|
var setReply string
|
|
if err := admS.SetFilter(context.Background(), args1, &setReply); err != nil {
|
|
t.Error(err)
|
|
} else if setReply != "OK" {
|
|
t.Error("Unexpected reply returned:", setReply)
|
|
}
|
|
|
|
args2 := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
Tenant: "cgrates.org",
|
|
ID: "test_ID2",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~*req.Destination",
|
|
Values: []string{"10"},
|
|
},
|
|
},
|
|
},
|
|
APIOpts: nil,
|
|
}
|
|
|
|
if err := admS.SetFilter(context.Background(), args2, &setReply); err != nil {
|
|
t.Error(err)
|
|
} else if setReply != "OK" {
|
|
t.Error("Unexpected reply returned:", setReply)
|
|
}
|
|
|
|
// this profile will not match
|
|
args3 := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
Tenant: "cgrates.org",
|
|
ID: "test2_ID1",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Account",
|
|
Values: []string{"1001"},
|
|
},
|
|
},
|
|
},
|
|
APIOpts: nil,
|
|
}
|
|
|
|
if err := admS.SetFilter(context.Background(), args3, &setReply); err != nil {
|
|
t.Error(err)
|
|
} else if setReply != "OK" {
|
|
t.Error("Unexpected reply returned:", setReply)
|
|
}
|
|
|
|
argsGet := &utils.ArgsItemIDs{
|
|
Tenant: "cgrates.org",
|
|
ItemsPrefix: "test_ID",
|
|
}
|
|
exp := []*engine.Filter{
|
|
{
|
|
Tenant: "cgrates.org",
|
|
ID: "test_ID1",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Account",
|
|
Values: []string{"1002"},
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Tenant: "cgrates.org",
|
|
ID: "test_ID2",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~*req.Destination",
|
|
Values: []string{"10"},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
var getReply []*engine.Filter
|
|
if err := admS.GetFilters(context.Background(), argsGet, &getReply); err != nil {
|
|
t.Error(err)
|
|
} else {
|
|
sort.Slice(getReply, func(i, j int) bool {
|
|
return getReply[i].ID < getReply[j].ID
|
|
})
|
|
for _, val := range exp {
|
|
val.Compile()
|
|
}
|
|
if !reflect.DeepEqual(getReply, exp) {
|
|
t.Errorf("expected: <%+v>, \nreceived: <%+v>",
|
|
utils.ToJSON(exp), utils.ToJSON(getReply))
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestFiltersGetFiltersGetIDsErr(t *testing.T) {
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
args := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
Tenant: "cgrates.org",
|
|
ID: "test_ID1",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Account",
|
|
Values: []string{"1002"},
|
|
},
|
|
},
|
|
},
|
|
APIOpts: nil,
|
|
}
|
|
|
|
var setReply string
|
|
if err := admS.SetFilter(context.Background(), args, &setReply); err != nil {
|
|
t.Error(err)
|
|
} else if setReply != "OK" {
|
|
t.Error("Unexpected reply returned:", setReply)
|
|
}
|
|
|
|
argsGet := &utils.ArgsItemIDs{
|
|
Tenant: "cgrates.org",
|
|
ItemsPrefix: "test_ID",
|
|
APIOpts: map[string]any{
|
|
utils.PageLimitOpt: 2,
|
|
utils.PageOffsetOpt: 4,
|
|
utils.PageMaxItemsOpt: 5,
|
|
},
|
|
}
|
|
|
|
experr := `SERVER_ERROR: maximum number of items exceeded`
|
|
var getReply []*engine.Filter
|
|
if err := admS.GetFilters(context.Background(), argsGet, &getReply); err == nil || err.Error() != experr {
|
|
t.Errorf("expected: <%+v>, \nreceived: <%+v>", experr, err)
|
|
}
|
|
}
|
|
|
|
func TestFiltersGetFiltersGetProfileErr(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
dbMock := &engine.DataDBMock{
|
|
SetFilterDrvF: func(*context.Context, *engine.Filter) error {
|
|
return nil
|
|
},
|
|
RemoveFilterDrvF: func(*context.Context, string, string) error {
|
|
return nil
|
|
},
|
|
GetKeysForPrefixF: func(c *context.Context, s string) ([]string, error) {
|
|
return []string{"ftr_cgrates.org:TEST"}, nil
|
|
},
|
|
}
|
|
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dbMock}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, nil)
|
|
adms := &AdminSv1{
|
|
cfg: cfg,
|
|
dm: dm,
|
|
}
|
|
|
|
var reply []*engine.Filter
|
|
experr := "SERVER_ERROR: NOT_IMPLEMENTED"
|
|
|
|
if err := adms.GetFilters(context.Background(),
|
|
&utils.ArgsItemIDs{
|
|
ItemsPrefix: "TEST",
|
|
}, &reply); err == nil || err.Error() != experr {
|
|
t.Errorf("expected: <%+v>, \nreceived: <%+v>", experr, err)
|
|
}
|
|
|
|
dm.DataDB()[utils.MetaDefault].Flush(utils.EmptyString)
|
|
}
|
|
|
|
func TestFiltersGetFilterIDsGetOptsErr(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
dbMock := &engine.DataDBMock{
|
|
GetFilterDrvF: func(*context.Context, string, string) (*engine.Filter, error) {
|
|
fltr := &engine.Filter{
|
|
Tenant: "cgrates.org",
|
|
ID: "TEST",
|
|
}
|
|
return fltr, nil
|
|
},
|
|
SetFilterDrvF: func(*context.Context, *engine.Filter) error {
|
|
return nil
|
|
},
|
|
RemoveFilterDrvF: func(*context.Context, string, string) error {
|
|
return nil
|
|
},
|
|
GetKeysForPrefixF: func(c *context.Context, s string) ([]string, error) {
|
|
return []string{"ftr_cgrates.org:key1", "ftr_cgrates.org:key2", "ftr_cgrates.org:key3"}, nil
|
|
},
|
|
}
|
|
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dbMock}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, nil)
|
|
adms := &AdminSv1{
|
|
cfg: cfg,
|
|
dm: dm,
|
|
}
|
|
|
|
var reply []string
|
|
experr := "cannot convert field<bool>: true to int"
|
|
|
|
if err := adms.GetFilterIDs(context.Background(),
|
|
&utils.ArgsItemIDs{
|
|
Tenant: "cgrates.org",
|
|
APIOpts: map[string]any{
|
|
utils.PageLimitOpt: true,
|
|
},
|
|
}, &reply); err == nil || err.Error() != experr {
|
|
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", experr, err)
|
|
}
|
|
|
|
dm.DataDB()[utils.MetaDefault].Flush(utils.EmptyString)
|
|
}
|
|
|
|
func TestFiltersGetFilterIDsPaginateErr(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
dbMock := &engine.DataDBMock{
|
|
GetFilterDrvF: func(*context.Context, string, string) (*engine.Filter, error) {
|
|
fltr := &engine.Filter{
|
|
Tenant: "cgrates.org",
|
|
ID: "TEST",
|
|
}
|
|
return fltr, nil
|
|
},
|
|
SetFilterDrvF: func(*context.Context, *engine.Filter) error {
|
|
return nil
|
|
},
|
|
RemoveFilterDrvF: func(*context.Context, string, string) error {
|
|
return nil
|
|
},
|
|
GetKeysForPrefixF: func(c *context.Context, s string) ([]string, error) {
|
|
return []string{"ftr_cgrates.org:key1", "ftr_cgrates.org:key2", "ftr_cgrates.org:key3"}, nil
|
|
},
|
|
}
|
|
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dbMock}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, nil)
|
|
adms := &AdminSv1{
|
|
cfg: cfg,
|
|
dm: dm,
|
|
}
|
|
|
|
var reply []string
|
|
experr := `SERVER_ERROR: maximum number of items exceeded`
|
|
|
|
if err := adms.GetFilterIDs(context.Background(),
|
|
&utils.ArgsItemIDs{
|
|
Tenant: "cgrates.org",
|
|
APIOpts: map[string]any{
|
|
utils.PageLimitOpt: 2,
|
|
utils.PageOffsetOpt: 4,
|
|
utils.PageMaxItemsOpt: 5,
|
|
},
|
|
}, &reply); err == nil || err.Error() != experr {
|
|
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", experr, err)
|
|
}
|
|
|
|
dm.DataDB()[utils.MetaDefault].Flush(utils.EmptyString)
|
|
}
|
|
|
|
func TestFiltersSetFilterNoRulesErr(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
fltr := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
ID: "fltr_for_attr",
|
|
Rules: []*engine.FilterRule{},
|
|
},
|
|
}
|
|
experr := `MANDATORY_IE_MISSING: [Filter Rules]`
|
|
var reply string
|
|
if err := admS.SetFilter(context.Background(), fltr, &reply); err == nil ||
|
|
err.Error() != experr {
|
|
t.Errorf("expected: <%+v>, \nreceived: <%+v>", experr, err)
|
|
}
|
|
}
|
|
|
|
func TestFiltersSetFilterInvalidRulesErr(t *testing.T) {
|
|
engine.Cache.Clear(nil)
|
|
cfg := config.NewDefaultCGRConfig()
|
|
cfg.GeneralCfg().DefaultCaching = utils.MetaNone
|
|
connMgr := engine.NewConnManager(cfg)
|
|
dataDB, _ := engine.NewInternalDB(nil, nil, nil, cfg.DbCfg().Items)
|
|
dbCM := engine.NewDBConnManager(map[string]engine.DataDB{utils.MetaDefault: dataDB}, cfg.DbCfg())
|
|
dm := engine.NewDataManager(dbCM, cfg, connMgr)
|
|
admS := NewAdminSv1(cfg, dm, connMgr, nil)
|
|
fltr := &engine.FilterWithAPIOpts{
|
|
Filter: &engine.Filter{
|
|
ID: "fltr_for_attr",
|
|
Rules: []*engine.FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Account",
|
|
},
|
|
},
|
|
},
|
|
}
|
|
experr := `SERVER_ERROR: there exists at least one filter rule that is not valid`
|
|
var reply string
|
|
if err := admS.SetFilter(context.Background(), fltr, &reply); err == nil ||
|
|
err.Error() != experr {
|
|
t.Errorf("expected: <%+v>, \nreceived: <%+v>", experr, err)
|
|
}
|
|
}
|