Added tests for NavigableMap.GetKeys()

This commit is contained in:
Trial97
2020-01-10 09:20:24 +02:00
parent 24b073199c
commit 9d6a0bb345
3 changed files with 128 additions and 2 deletions

View File

@@ -103,3 +103,75 @@ func TestFilterMatchingItemIDsForEvent(t *testing.T) {
t.Errorf("Expecting: %+v, received: %+v", prefixFilterID, aPrflIDs)
}
}
func TestFilterMatchingItemIDsForEvent2(t *testing.T) {
var stringFilter, prefixFilter, defaultFilter []*FilterRule
stringFilterID := "stringFilterID"
prefixFilterID := "prefixFilterID"
data := NewInternalDB(nil, nil)
dmMatch = NewDataManager(data, config.CgrConfig().CacheCfg(), nil)
context := utils.MetaRating
x, err := NewFilterRule(utils.MetaString, "~*req.CallCost.Account", []string{"1001"})
if err != nil {
t.Errorf("Error: %+v", err)
}
stringFilter = append(stringFilter, x)
attribStringF := &Filter{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "stringFilter",
Rules: stringFilter}
dmMatch.SetFilter(attribStringF)
x, err = NewFilterRule(utils.MetaPrefix, "~*req.CallCost.Field", []string{"profile"})
if err != nil {
t.Errorf("Error: %+v", err)
}
prefixFilter = append(prefixFilter, x)
attribPrefF := &Filter{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "prefFilter",
Rules: prefixFilter}
dmMatch.SetFilter(attribPrefF)
x, err = NewFilterRule(utils.MetaGreaterOrEqual, "~*req.Weight", []string{"200.00"})
if err != nil {
t.Errorf("Error: %+v", err)
}
defaultFilter = append(defaultFilter, x)
attribDefaultF := &Filter{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "defaultFilter",
Rules: defaultFilter}
dmMatch.SetFilter(attribDefaultF)
prefix := utils.ConcatenatedKey(config.CgrConfig().GeneralCfg().DefaultTenant, context)
atrRFI := NewFilterIndexer(dmMatch, utils.AttributeProfilePrefix, prefix)
atrRFI.IndexTPFilter(FilterToTPFilter(attribStringF), stringFilterID)
atrRFI.IndexTPFilter(FilterToTPFilter(attribPrefF), prefixFilterID)
err = atrRFI.StoreIndexes(true, utils.NonTransactional)
if err != nil {
t.Errorf("Error: %+v", err)
}
matchEV = map[string]interface{}{
utils.AnswerTime: time.Date(2014, 7, 14, 14, 30, 0, 0, time.UTC),
"CallCost": map[string]interface{}{"Account": 1001},
}
aPrflIDs, err := MatchingItemIDsForEvent(matchEV, nil, nil,
dmMatch, utils.CacheAttributeFilterIndexes, prefix, true, true)
if err != nil {
t.Errorf("Error: %+v", err)
}
_, has := aPrflIDs[stringFilterID]
if !has {
t.Errorf("Expecting: %+v, received: %+v", stringFilterID, aPrflIDs)
}
matchEV = map[string]interface{}{
"CallCost": map[string]interface{}{"Field": "profilePrefix"},
}
aPrflIDs, err = MatchingItemIDsForEvent(matchEV, nil, nil,
dmMatch, utils.CacheAttributeFilterIndexes, prefix, true, true)
if err != nil {
t.Errorf("Error: %+v", err)
}
_, has = aPrflIDs[prefixFilterID]
if !has {
t.Errorf("Expecting: %+v, received: %+v", prefixFilterID, aPrflIDs)
}
}