Files
cgrates/apis/filters_test.go
2025-10-19 13:15:57 +02:00

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)
}
}