mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 10:06:24 +05:00
test newFilterIndex
This commit is contained in:
ionutboangiu
committed by
Dan Christian Bogos
Dan Christian Bogos
parent
10493ad9c0
commit
7b0f4b1aed
@@ -20,6 +20,8 @@ package engine
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"reflect"
|
||||
"strconv"
|
||||
"testing"
|
||||
|
||||
"github.com/cgrates/cgrates/config"
|
||||
@@ -207,3 +209,558 @@ func TestLibIndexNewFilterIndexGetFilterErrNotFound(t *testing.T) {
|
||||
t.Fatalf("Expected error %v, got %v", expectedErr, err)
|
||||
}
|
||||
}
|
||||
func TestLibIndex_newFilterIndex(t *testing.T) {
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
dataDB, err := NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
dm := NewDataManager(dataDB, cfg.CacheCfg(), nil)
|
||||
|
||||
for i := range 2 {
|
||||
idx := strconv.Itoa(i + 1)
|
||||
flt := &Filter{
|
||||
Tenant: "cgrates.org",
|
||||
ID: "FLTR_" + idx,
|
||||
Rules: []*FilterRule{
|
||||
{
|
||||
Type: utils.MetaString,
|
||||
Element: "~*req.Field" + idx,
|
||||
Values: []string{"val" + idx},
|
||||
},
|
||||
{
|
||||
Type: utils.MetaPrefix,
|
||||
Element: "~*req.Field" + idx,
|
||||
Values: []string{"val"},
|
||||
},
|
||||
{
|
||||
Type: utils.MetaSuffix,
|
||||
Element: "~*req.Field" + idx,
|
||||
Values: []string{idx},
|
||||
},
|
||||
{
|
||||
Type: utils.MetaExists,
|
||||
Element: "~*req.Field" + idx,
|
||||
},
|
||||
},
|
||||
}
|
||||
if err := dm.SetFilter(flt, true); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
filterIDsList := [][]string{
|
||||
{"*prefix:~*req.Field2:val", "*empty:~*req.Field3:", "FLTR_1"},
|
||||
{"*suffix:~*req.Field1:1", "*notstring:~*req.Field2:val1", "FLTR_2"},
|
||||
{"*exists:~*req.Field2:", "*suffix:~*req.Field1:1", "FLTR_1"},
|
||||
}
|
||||
|
||||
for i, filterIDs := range filterIDsList {
|
||||
idx := strconv.Itoa(i + 1)
|
||||
if err := dm.SetAttributeProfile(&AttributeProfile{
|
||||
Tenant: "cgrates.org",
|
||||
ID: "ATTR_" + idx,
|
||||
FilterIDs: filterIDs,
|
||||
Contexts: []string{utils.MetaCDRs},
|
||||
}, true); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
wantIndexes := map[string]utils.StringSet{
|
||||
"*prefix:*req.Field1:val": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_3": {},
|
||||
},
|
||||
"*prefix:*req.Field2:val": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_2": {},
|
||||
},
|
||||
"*string:*req.Field1:val1": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_3": {},
|
||||
},
|
||||
"*string:*req.Field2:val2": {
|
||||
"ATTR_2": {},
|
||||
},
|
||||
"*suffix:*req.Field1:1": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_2": {},
|
||||
"ATTR_3": {},
|
||||
},
|
||||
"*suffix:*req.Field2:2": {
|
||||
"ATTR_2": {},
|
||||
},
|
||||
}
|
||||
tntCtx := utils.ConcatenatedKey("cgrates.org", utils.MetaCDRs)
|
||||
gotIndexes, err := dm.GetIndexes(utils.CacheAttributeFilterIndexes, tntCtx, "", false, false)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if !reflect.DeepEqual(gotIndexes, wantIndexes) {
|
||||
t.Errorf("dm.GetIndexes() = %s, want %s", utils.ToJSON(gotIndexes), utils.ToJSON(wantIndexes))
|
||||
}
|
||||
|
||||
tests := []struct {
|
||||
name string
|
||||
idxItmType string
|
||||
tnt string
|
||||
ctx string
|
||||
itemID string
|
||||
filterIDs []string
|
||||
newFlt *Filter
|
||||
want map[string]utils.StringSet
|
||||
wantErr bool
|
||||
}{
|
||||
{
|
||||
name: "no filterIDs",
|
||||
idxItmType: utils.CacheAttributeFilterIndexes,
|
||||
want: map[string]utils.StringSet{
|
||||
"*none:*any:*any": {},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "newFlt with no filterIDs",
|
||||
idxItmType: utils.CacheAttributeFilterIndexes,
|
||||
newFlt: &Filter{
|
||||
Tenant: "cgrates.org",
|
||||
ID: "FLTR_TEST",
|
||||
Rules: []*FilterRule{
|
||||
{
|
||||
Type: utils.MetaString,
|
||||
Element: "~*req.Field",
|
||||
Values: []string{"val"},
|
||||
},
|
||||
},
|
||||
},
|
||||
want: map[string]utils.StringSet{
|
||||
"*none:*any:*any": {},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "broken reference",
|
||||
idxItmType: utils.CacheAttributeFilterIndexes,
|
||||
itemID: "ATTR_1",
|
||||
filterIDs: []string{"FLTR_1"},
|
||||
wantErr: true,
|
||||
},
|
||||
{
|
||||
name: "unindexable filter",
|
||||
idxItmType: utils.CacheAttributeFilterIndexes,
|
||||
filterIDs: []string{"*exists:~*req.Field1:"},
|
||||
want: make(map[string]utils.StringSet),
|
||||
},
|
||||
{
|
||||
name: "dynamic element, constant value",
|
||||
idxItmType: utils.CacheAttributeFilterIndexes,
|
||||
filterIDs: []string{"*string:~*req.Field1:val1"},
|
||||
want: map[string]utils.StringSet{
|
||||
"*string:*req.Field1:val1": {},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "constant element, dynamic value",
|
||||
idxItmType: utils.CacheAttributeFilterIndexes,
|
||||
filterIDs: []string{"*string:val1:~*req.Field1"},
|
||||
want: map[string]utils.StringSet{
|
||||
"*string:*req.Field1:val1": {},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "dynamic element, dynamic value",
|
||||
idxItmType: utils.CacheAttributeFilterIndexes,
|
||||
filterIDs: []string{"*string:~*req.Field1:~*req.Field1"},
|
||||
want: make(map[string]utils.StringSet),
|
||||
},
|
||||
{
|
||||
name: "constant element, constant value",
|
||||
idxItmType: utils.CacheAttributeFilterIndexes,
|
||||
filterIDs: []string{"*string:val1:val1"},
|
||||
want: make(map[string]utils.StringSet),
|
||||
},
|
||||
{
|
||||
name: "filter and tenant without context",
|
||||
idxItmType: utils.CacheAttributeFilterIndexes,
|
||||
tnt: "cgrates.org",
|
||||
// ctx: utils.MetaCDRs,
|
||||
filterIDs: []string{"*string:~*req.Field1:val1"},
|
||||
want: map[string]utils.StringSet{
|
||||
"*string:*req.Field1:val1": {},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "filter and tenant with context (without references)",
|
||||
idxItmType: utils.CacheAttributeFilterIndexes,
|
||||
tnt: "cgrates.org",
|
||||
ctx: utils.MetaCDRs,
|
||||
filterIDs: []string{"*string:~*req.Random:val"},
|
||||
want: map[string]utils.StringSet{
|
||||
"*string:*req.Random:val": {},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "filter and tenant with context (with references)",
|
||||
idxItmType: utils.CacheAttributeFilterIndexes,
|
||||
tnt: "cgrates.org",
|
||||
ctx: utils.MetaCDRs,
|
||||
filterIDs: []string{"*string:~*req.Field1:val1"},
|
||||
want: map[string]utils.StringSet{
|
||||
"*string:*req.Field1:val1": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_3": {},
|
||||
},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "filterID of newFlt",
|
||||
idxItmType: utils.CacheAttributeFilterIndexes,
|
||||
tnt: "cgrates.org",
|
||||
ctx: utils.MetaCDRs,
|
||||
filterIDs: []string{"FLTR_TEST"},
|
||||
newFlt: &Filter{
|
||||
Tenant: "cgrates.org",
|
||||
ID: "FLTR_TEST",
|
||||
Rules: []*FilterRule{
|
||||
{
|
||||
Type: utils.MetaString,
|
||||
Element: "~*req.Random",
|
||||
Values: []string{"val"},
|
||||
},
|
||||
{
|
||||
Type: utils.MetaString,
|
||||
Element: "~*req.Field1",
|
||||
Values: []string{"val1"},
|
||||
},
|
||||
},
|
||||
},
|
||||
want: map[string]utils.StringSet{
|
||||
"*string:*req.Field1:val1": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_3": {},
|
||||
},
|
||||
"*string:*req.Random:val": {},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "all indexable filters with references",
|
||||
idxItmType: utils.CacheAttributeFilterIndexes,
|
||||
tnt: "cgrates.org",
|
||||
ctx: "*cdrs",
|
||||
filterIDs: []string{
|
||||
"FLTR_1",
|
||||
"FLTR_2",
|
||||
"*prefix:~*req.Field2:val",
|
||||
"*suffix:~*req.Field1:1",
|
||||
"*suffix:~*req.Field1:1",
|
||||
},
|
||||
want: map[string]utils.StringSet{
|
||||
"*prefix:*req.Field1:val": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_3": {},
|
||||
},
|
||||
"*prefix:*req.Field2:val": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_2": {},
|
||||
},
|
||||
"*string:*req.Field1:val1": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_3": {},
|
||||
},
|
||||
"*string:*req.Field2:val2": {
|
||||
"ATTR_2": {},
|
||||
},
|
||||
"*suffix:*req.Field1:1": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_2": {},
|
||||
"ATTR_3": {},
|
||||
},
|
||||
"*suffix:*req.Field2:2": {
|
||||
"ATTR_2": {},
|
||||
},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "all filters",
|
||||
idxItmType: utils.CacheAttributeFilterIndexes,
|
||||
tnt: "cgrates.org",
|
||||
ctx: "*cdrs",
|
||||
filterIDs: []string{
|
||||
"FLTR_1",
|
||||
"FLTR_2",
|
||||
"*prefix:~*req.Field2:val",
|
||||
"*suffix:~*req.Field1:1",
|
||||
"*exists:~*req.Field2:",
|
||||
"*empty:~*req.Field3:",
|
||||
"*notstring:~*req.Field2:val1",
|
||||
},
|
||||
want: map[string]utils.StringSet{
|
||||
"*prefix:*req.Field1:val": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_3": {},
|
||||
},
|
||||
"*prefix:*req.Field2:val": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_2": {},
|
||||
},
|
||||
"*string:*req.Field1:val1": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_3": {},
|
||||
},
|
||||
"*string:*req.Field2:val2": {
|
||||
"ATTR_2": {},
|
||||
},
|
||||
"*suffix:*req.Field1:1": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_2": {},
|
||||
"ATTR_3": {},
|
||||
},
|
||||
"*suffix:*req.Field2:2": {
|
||||
"ATTR_2": {},
|
||||
},
|
||||
},
|
||||
},
|
||||
{
|
||||
name: "all filters (with extra unreferenced filters)",
|
||||
idxItmType: utils.CacheAttributeFilterIndexes,
|
||||
tnt: "cgrates.org",
|
||||
ctx: "*cdrs",
|
||||
filterIDs: []string{
|
||||
"FLTR_1",
|
||||
"FLTR_2",
|
||||
"*prefix:~*req.Field2:val",
|
||||
"*suffix:~*req.Field1:1",
|
||||
"*exists:~*req.Field2:",
|
||||
"*empty:~*req.Field3:",
|
||||
"*notstring:~*req.Field2:val1",
|
||||
"*prefix:~*req.Field4:val",
|
||||
"*string:~*req.Field5:val5",
|
||||
},
|
||||
want: map[string]utils.StringSet{
|
||||
"*prefix:*req.Field1:val": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_3": {},
|
||||
},
|
||||
"*prefix:*req.Field2:val": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_2": {},
|
||||
},
|
||||
"*string:*req.Field1:val1": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_3": {},
|
||||
},
|
||||
"*string:*req.Field2:val2": {
|
||||
"ATTR_2": {},
|
||||
},
|
||||
"*suffix:*req.Field1:1": {
|
||||
"ATTR_1": {},
|
||||
"ATTR_2": {},
|
||||
"ATTR_3": {},
|
||||
},
|
||||
"*suffix:*req.Field2:2": {
|
||||
"ATTR_2": {},
|
||||
},
|
||||
"*prefix:*req.Field4:val": {},
|
||||
"*string:*req.Field5:val5": {},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
got, gotErr := newFilterIndex(dm, tt.idxItmType, tt.tnt, tt.ctx, tt.itemID, tt.filterIDs, tt.newFlt)
|
||||
if gotErr != nil {
|
||||
if !tt.wantErr {
|
||||
t.Errorf("prepareFilterIndexMap() failed: %v", gotErr)
|
||||
}
|
||||
return
|
||||
}
|
||||
if tt.wantErr {
|
||||
t.Fatal("prepareFilterIndexMap() succeeded unexpectedly")
|
||||
}
|
||||
if !reflect.DeepEqual(got, tt.want) {
|
||||
t.Errorf("prepareFilterIndexMap() = %s, want %s", utils.ToJSON(got), utils.ToJSON(tt.want))
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// func TestLibIndex_prepareFilterIndexMap(t *testing.T) {
|
||||
// cfg := config.NewDefaultCGRConfig()
|
||||
// dataDB := NewInternalDB(nil, nil, true, true, cfg.DataDbCfg().Items)
|
||||
// dm := NewDataManager(dataDB, cfg.CacheCfg(), nil)
|
||||
//
|
||||
// var flt *Filter // to be used as newFlt
|
||||
// for i := range 2 {
|
||||
// idx := strconv.Itoa(i + 1)
|
||||
// flt = &Filter{
|
||||
// Tenant: "cgrates.org",
|
||||
// ID: "FLTR_" + idx,
|
||||
// Rules: []*FilterRule{
|
||||
// {
|
||||
// Type: utils.MetaString,
|
||||
// Element: "~*req.Field" + idx,
|
||||
// Values: []string{"val" + idx},
|
||||
// },
|
||||
// {
|
||||
// Type: utils.MetaPrefix,
|
||||
// Element: "~*req.Field" + idx,
|
||||
// Values: []string{"val"},
|
||||
// },
|
||||
// {
|
||||
// Type: utils.MetaSuffix,
|
||||
// Element: "~*req.Field" + idx,
|
||||
// Values: []string{idx},
|
||||
// },
|
||||
// {
|
||||
// Type: utils.MetaExists,
|
||||
// Element: "~*req.Field" + idx,
|
||||
// },
|
||||
// },
|
||||
// }
|
||||
// if err := dm.SetFilter(flt, true); err != nil {
|
||||
// t.Fatal(err)
|
||||
// }
|
||||
// }
|
||||
//
|
||||
// filterIDsList := [][]string{
|
||||
// {"*prefix:~*req.Field2:val", "FLTR_1"},
|
||||
// {"*suffix:~*req.Field1:1", "FLTR_2"},
|
||||
// {"*exists:~*req.Field2:", "*suffix:~*req.Field1:1", "FLTR_1"},
|
||||
// }
|
||||
//
|
||||
// for i, filterIDs := range filterIDsList {
|
||||
// idx := strconv.Itoa(i + 1)
|
||||
// if err := dm.SetChargerProfile(&ChargerProfile{
|
||||
// Tenant: "cgrates.org",
|
||||
// ID: "CP_" + idx,
|
||||
// FilterIDs: filterIDs,
|
||||
// RunID: "DEFAULT" + idx,
|
||||
// AttributeIDs: []string{"*none"},
|
||||
// }, true); err != nil {
|
||||
// t.Fatal(err)
|
||||
// }
|
||||
// if err := dm.SetAttributeProfile(&AttributeProfile{
|
||||
// Tenant: "cgrates.org",
|
||||
// ID: "ATTR_CDRS_" + idx,
|
||||
// FilterIDs: filterIDs,
|
||||
// Contexts: []string{utils.MetaCDRs},
|
||||
// }, true); err != nil {
|
||||
// t.Fatal(err)
|
||||
// }
|
||||
// if err := dm.SetAttributeProfile(&AttributeProfile{
|
||||
// Tenant: "cgrates.org",
|
||||
// ID: "ATTR_SESSIONS_" + idx,
|
||||
// FilterIDs: filterIDs,
|
||||
// Contexts: []string{utils.MetaSessionS},
|
||||
// }, true); err != nil {
|
||||
// t.Fatal(err)
|
||||
// }
|
||||
// }
|
||||
//
|
||||
// wantTemplate := func(prefix string) map[string]utils.StringSet {
|
||||
// return map[string]utils.StringSet{
|
||||
// "*prefix:*req.Field1:val": {
|
||||
// prefix + "_1": {},
|
||||
// prefix + "_3": {},
|
||||
// },
|
||||
// "*prefix:*req.Field2:val": {
|
||||
// prefix + "_1": {},
|
||||
// prefix + "_2": {},
|
||||
// },
|
||||
// "*string:*req.Field1:val1": {
|
||||
// prefix + "_1": {},
|
||||
// prefix + "_3": {},
|
||||
// },
|
||||
// "*string:*req.Field2:val2": {
|
||||
// prefix + "_2": {},
|
||||
// },
|
||||
// "*suffix:*req.Field1:1": {
|
||||
// prefix + "_1": {},
|
||||
// prefix + "_2": {},
|
||||
// prefix + "_3": {},
|
||||
// },
|
||||
// "*suffix:*req.Field2:2": {
|
||||
// prefix + "_2": {},
|
||||
// },
|
||||
// }
|
||||
// }
|
||||
//
|
||||
// argsList := []struct {
|
||||
// itemType string
|
||||
// ctx string
|
||||
// prefix string // of the profile (CP, ATTR_CDRS, ATTR_SESSIONS)
|
||||
// }{
|
||||
// {
|
||||
// itemType: utils.CacheChargerFilterIndexes,
|
||||
// ctx: "",
|
||||
// prefix: "CP",
|
||||
// },
|
||||
// {
|
||||
// itemType: utils.CacheAttributeFilterIndexes,
|
||||
// ctx: utils.MetaCDRs,
|
||||
// prefix: "ATTR_CDRS",
|
||||
// },
|
||||
// {
|
||||
// itemType: utils.CacheAttributeFilterIndexes,
|
||||
// ctx: utils.MetaSessionS,
|
||||
// prefix: "ATTR_SESSIONS",
|
||||
// },
|
||||
// }
|
||||
//
|
||||
// for _, args := range argsList {
|
||||
// wantIndexes := wantTemplate(args.prefix)
|
||||
// tntCtx := "cgrates.org"
|
||||
// if args.ctx != "" {
|
||||
// tntCtx = utils.ConcatenatedKey(tntCtx, args.ctx)
|
||||
// }
|
||||
// gotIndexes, err := dm.GetIndexes(args.itemType, tntCtx, "", false, false)
|
||||
// if err != nil {
|
||||
// t.Fatal(err)
|
||||
// }
|
||||
// if !reflect.DeepEqual(gotIndexes, wantIndexes) {
|
||||
// t.Errorf("dm.GetIndexes() = %s, want %s", utils.ToJSON(gotIndexes), utils.ToJSON(wantIndexes))
|
||||
// }
|
||||
//
|
||||
// }
|
||||
//
|
||||
// tests := []struct {
|
||||
// name string
|
||||
// idxItmType string
|
||||
// tnt string
|
||||
// ctx string
|
||||
// itemID string
|
||||
// filterIDs []string
|
||||
// newFlt *Filter
|
||||
// want map[string]utils.StringSet
|
||||
// wantErr bool
|
||||
// }{
|
||||
// {
|
||||
// name: "test1",
|
||||
// idxItmType: utils.CacheAttributeFilterIndexes,
|
||||
// tnt: "cgrates.org",
|
||||
// ctx: "*cdrs",
|
||||
// itemID: "ATTR_CDRS",
|
||||
// filterIDs: []string{"FLTR_1"},
|
||||
// newFlt: flt,
|
||||
// want: map[string]utils.StringSet{},
|
||||
// },
|
||||
// }
|
||||
//
|
||||
// for _, tt := range tests {
|
||||
// t.Run(tt.name, func(t *testing.T) {
|
||||
// got, gotErr := prepareFilterIndexMap(dm, tt.idxItmType, tt.tnt, tt.ctx, tt.itemID, tt.filterIDs, tt.newFlt)
|
||||
// if gotErr != nil {
|
||||
// if !tt.wantErr {
|
||||
// t.Errorf("prepareFilterIndexMap() failed: %v", gotErr)
|
||||
// }
|
||||
// return
|
||||
// }
|
||||
// if tt.wantErr {
|
||||
// t.Fatal("prepareFilterIndexMap() succeeded unexpectedly")
|
||||
// }
|
||||
// if !reflect.DeepEqual(got, tt.want) {
|
||||
// t.Errorf("prepareFilterIndexMap() = %s, want %s", utils.ToJSON(got), utils.ToJSON(tt.want))
|
||||
// }
|
||||
// })
|
||||
// }
|
||||
// }
|
||||
|
||||
Reference in New Issue
Block a user