Files
cgrates/utils/map_test.go
2022-10-19 09:43:33 +02:00

556 lines
16 KiB
Go

/*
Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
Copyright (C) ITsysCOM GmbH
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>
*/
package utils
import (
"reflect"
"sort"
"testing"
)
func TestMapKeysStringMapParse(t *testing.T) {
if sm := ParseStringMap(EmptyString); len(sm) != 0 {
t.Errorf("Expecting %+v, received %+v", 0, len(sm))
}
if sm := ParseStringMap(MetaZero); len(sm) != 0 {
t.Errorf("Expecting %+v, received %+v", 0, len(sm))
}
if sm := ParseStringMap("1;2;3;4"); len(sm) != 4 {
t.Error("Error pasring map: ", sm)
}
if sm := ParseStringMap("1;2;!3;4"); len(sm) != 4 {
t.Error("Error pasring map: ", sm)
} else if sm["3"] != false {
t.Error("Error parsing negative: ", sm)
}
sm := ParseStringMap("1;2;!3;4")
if include, has := sm["2"]; include != true && has != true {
t.Error("Error detecting positive: ", sm)
}
if include, has := sm["3"]; include != false && has != true {
t.Error("Error detecting negative: ", sm)
}
if include, has := sm["5"]; include != false && has != false {
t.Error("Error detecting missing: ", sm)
}
eOut := make(StringMap)
if rcv := ParseStringMap(MetaZero); !reflect.DeepEqual(eOut, rcv) {
t.Errorf("Expecting: %+v, received: %+v", eOut, rcv)
}
}
func TestEqual(t *testing.T) {
t1 := NewStringMap("val1")
t2 := NewStringMap("val2")
result := t1.Equal(t2)
expected := false
if result != expected {
t.Error("Expecting:", expected, ", received:", result)
}
}
func TestIsEmpty(t *testing.T) {
t1 := NewStringMap("val1")
result := t1.IsEmpty()
expected := false
if result != expected {
t.Error("Expecting:", expected, ", received:", result)
}
}
func TestMapStringToInt64(t *testing.T) {
t1 := map[string]int64{"test": int64(21)}
t2 := map[string]string{"test": "21"}
t3, err := MapStringToInt64(t2)
if err != nil {
t.Error("Got Error: ", err)
}
if !reflect.DeepEqual(t1, t3) {
t.Errorf("Expecting: %+v, received: %+v", t1, t3)
}
}
func TestMapHasKey(t *testing.T) {
mp := ParseStringMap("Item1;Item2;Item3")
if mp.HasKey("Item1") != true {
t.Errorf("Expecting: true, received: %+v", mp.HasKey("Item1"))
}
if mp.HasKey("Item4") != false {
t.Errorf("Expecting: true, received: %+v", mp.HasKey("Item4"))
}
}
func TestMapSubsystemIDsFromSlice(t *testing.T) {
sls := []string{"*event", "*thresholds:*ids:ID1&ID2&ID3", "*thresholds:*derived_reply", "*attributes:*disabled", "*stats:*ids:ID"}
eMp := FlagsWithParams{
"*event": map[string][]string{},
"*thresholds": map[string][]string{MetaIDs: {"ID1", "ID2", "ID3"}, MetaDerivedReply: {}},
"*attributes": map[string][]string{"*disabled": {}},
"*stats": map[string][]string{MetaIDs: {"ID"}},
}
if mp := FlagsWithParamsFromSlice(sls); !reflect.DeepEqual(mp, eMp) {
t.Errorf("Expecting: %+v, received: %+v", eMp, mp)
}
}
func TestMapSubsystemIDsHasKey(t *testing.T) {
sls := []string{"*event", "*thresholds:*ids:ID1&ID2&ID3", "*attributes", "*stats:*ids:ID"}
eMp := FlagsWithParams{
"*event": map[string][]string{},
"*thresholds": map[string][]string{MetaIDs: {"ID1", "ID2", "ID3"}},
"*attributes": map[string][]string{},
"*stats": map[string][]string{MetaIDs: {"ID"}},
}
mp := FlagsWithParamsFromSlice(sls)
if !reflect.DeepEqual(mp, eMp) {
t.Errorf("Expecting: %+v, received: %+v", eMp, mp)
}
if has := mp.Has("*event"); !has {
t.Errorf("Expecting: true, received: %+v", has)
}
if has := mp.Has("*thresholds"); !has {
t.Errorf("Expecting: true, received: %+v", has)
}
if has := mp.Has("*resources"); has {
t.Errorf("Expecting: false, received: %+v", has)
}
}
func TestMapSubsystemIDsGetIDs(t *testing.T) {
sls := []string{"*event", "*thresholds:*ids:ID1&ID2&ID3", "*attributes", "*stats:*ids:ID"}
eMp := FlagsWithParams{
"*event": map[string][]string{},
"*thresholds": map[string][]string{MetaIDs: {"ID1", "ID2", "ID3"}},
"*attributes": map[string][]string{},
"*stats": map[string][]string{MetaIDs: {"ID"}},
}
mp := FlagsWithParamsFromSlice(sls)
if !reflect.DeepEqual(mp, eMp) {
t.Errorf("Expecting: %+v, received: %+v", eMp, mp)
}
eIDs := []string{"ID1", "ID2", "ID3"}
if ids := mp.ParamsSlice("*thresholds", MetaIDs); !reflect.DeepEqual(ids, eIDs) {
t.Errorf("Expecting: %+v, received: %+v", eIDs, ids)
}
eIDs = nil
if ids := mp.ParamsSlice("*event", MetaIDs); !reflect.DeepEqual(ids, eIDs) {
t.Errorf("Expecting: %+v, received: %+v", eIDs, ids)
}
if ids := mp.ParamsSlice("*test", MetaIDs); !reflect.DeepEqual(ids, eIDs) {
t.Errorf("Expecting: %+v, received: %+v", eIDs, ids)
}
}
func TestFlagsToSlice(t *testing.T) {
sls := []string{"*event", "*thresholds:*ids:ID1&ID2&ID3", "*attributes", "*stats:*ids:ID", "*routes:*derived_reply"}
eMp := FlagsWithParams{
"*event": map[string][]string{},
"*thresholds": map[string][]string{MetaIDs: {"ID1", "ID2", "ID3"}},
"*attributes": map[string][]string{},
"*stats": map[string][]string{MetaIDs: {"ID"}},
"*routes": map[string][]string{MetaDerivedReply: {}},
}
mp := FlagsWithParamsFromSlice(sls)
if !reflect.DeepEqual(mp, eMp) {
t.Errorf("Expecting: %+v, received: %+v", eMp, mp)
}
sort.Strings(sls)
flgSls := mp.SliceFlags()
sort.Strings(flgSls)
if !reflect.DeepEqual(flgSls, sls) {
t.Errorf("Expecting: %+v, received: %+v", sls, flgSls)
}
}
func TestFlagsWithParamsGetBool(t *testing.T) {
flagsWithParams := &FlagsWithParams{
"test": map[string][]string{"false": {}, "string2": {}},
"test2": map[string][]string{"string2": {}},
"test3": map[string][]string{"true": {}},
"empty": map[string][]string{},
}
key := "notpresent"
if rcv := flagsWithParams.GetBool(key); rcv != false {
t.Errorf("Expecting: false, received: %+v", ToJSON(rcv))
}
key = "empty"
if rcv := flagsWithParams.GetBool(key); rcv != true {
t.Errorf("Expecting: false, received: %+v", ToJSON(rcv))
}
key = "test"
if rcv := flagsWithParams.GetBool(key); rcv != false {
t.Errorf("Expecting: false, received: %+v", ToJSON(rcv))
}
key = "test2"
if rcv := flagsWithParams.GetBool(key); rcv != true {
t.Errorf("Expecting: true, received: %+v", ToJSON(rcv))
}
key = "test3"
if rcv := flagsWithParams.GetBool(key); rcv != true {
t.Errorf("Expecting: true, received: %+v", ToJSON(rcv))
}
}
func TestFlagsWithParamsValue(t *testing.T) {
flagsWithParams := &FlagsWithParams{
"test": map[string][]string{"string2": {}},
"empty": map[string][]string{},
}
key := "notpresent"
if rcv := flagsWithParams.ParamValue(key); rcv != EmptyString {
t.Errorf("Expecting: %q, received: %+v", EmptyString, rcv)
}
key = "empty"
if rcv := flagsWithParams.ParamValue(key); rcv != EmptyString {
t.Errorf("Expecting: %q, received: %+v", EmptyString, rcv)
}
key = "test"
if rcv := flagsWithParams.ParamValue(key); rcv != "string2" {
t.Errorf("Expecting: string2, received: %+v", rcv)
}
}
func TestFlagParamsValue(t *testing.T) {
flagsWithParams := &FlagParams{
"test": []string{"string2"},
"empty": []string{},
}
key := "notpresent"
if rcv := flagsWithParams.ParamValue(key); rcv != EmptyString {
t.Errorf("Expecting: %q, received: %+v", EmptyString, rcv)
}
key = "empty"
if rcv := flagsWithParams.ParamValue(key); rcv != EmptyString {
t.Errorf("Expecting: %q, received: %+v", EmptyString, rcv)
}
key = "test"
if rcv := flagsWithParams.ParamValue(key); rcv != "string2" {
t.Errorf("Expecting: string2, received: %+v", rcv)
}
}
func TestFlagParamsAdd(t *testing.T) {
flgs := make(FlagParams)
exp := FlagParams{
MetaIDs: []string{"id1", "id2"},
}
flgs.Add([]string{MetaIDs, "id1&id2", "ignored"})
if !reflect.DeepEqual(flgs, exp) {
t.Errorf("Expecting: %+v, received: %+v", exp, flgs)
}
}
func TestFlagsToSlice2(t *testing.T) {
sls := []string{"*event", "*thresholds:*ids:ID1&ID2&ID3", "*attributes", "*stats:*ids:ID", "*routes:*derived_reply", "*cdrs:*attributes", "*cdrs:*stats:ID"}
eMp := FlagsWithParams{
"*event": map[string][]string{},
"*thresholds": map[string][]string{MetaIDs: {"ID1", "ID2", "ID3"}},
"*attributes": map[string][]string{},
"*stats": map[string][]string{MetaIDs: {"ID"}},
"*routes": map[string][]string{MetaDerivedReply: {}},
"*cdrs": map[string][]string{MetaAttributes: {}, MetaStats: {"ID"}},
}
mp := FlagsWithParamsFromSlice(sls)
if !reflect.DeepEqual(mp, eMp) {
t.Errorf("Expecting: %+v, received: %+v", ToJSON(eMp), ToJSON(mp))
}
sort.Strings(sls)
flgSls := mp.SliceFlags()
sort.Strings(flgSls)
if !reflect.DeepEqual(flgSls, sls) {
t.Errorf("Expecting: %+v, received: %+v", sls, flgSls)
}
sls = []string{"*attributes", "*stats:ID"}
flgSls = mp["*cdrs"].SliceFlags()
sort.Strings(flgSls)
if !reflect.DeepEqual(flgSls, sls) {
t.Errorf("Expecting: %+v, received: %+v", sls, flgSls)
}
}
func TestNewStringMap(t *testing.T) {
expected := StringMap{"item1": true, "item2": true, "negitem1": false}
received := NewStringMap("item1", "item2", "!negitem1")
if !reflect.DeepEqual(expected, received) {
t.Errorf("Expecting: %+v, received: %+v", expected, received)
}
}
func TestNewStringMap2(t *testing.T) {
expected := StringMap{"test": true, "test!": true, "t!est": true}
received := NewStringMap("test", "test!", "t!est")
if !reflect.DeepEqual(expected, received) {
t.Errorf("Expecting: %+v, received: %+v", expected, received)
}
}
func TestStringMapEqual1(t *testing.T) {
expected := false
received := new(StringMap).Equal(StringMap{"test": true})
if expected != received {
t.Errorf("Expecting: %+v, received: %+v", expected, received)
}
}
func TestStringMapEqual2(t *testing.T) {
expected := false
received := StringMap{"test": true, "test2": true}.Equal(StringMap{"test": true})
if expected != received {
t.Errorf("Expecting: %+v, received: %+v", expected, received)
}
}
func TestStringMapEqual3(t *testing.T) {
expected := true
received := StringMap{"test": true, "test2": true}.Equal(StringMap{"test": true, "test2": true})
if expected != received {
t.Errorf("Expecting: %+v, received: %+v", expected, received)
}
}
func TestStringMapIncludes1(t *testing.T) {
expected := false
received := StringMap{"test": true}.Includes(StringMap{"test": true, "test2": true})
if expected != received {
t.Errorf("Expecting: %+v, received: %+v", expected, received)
}
}
func TestStringMapIncludes2(t *testing.T) {
expected := false
received := StringMap{"test": true, "test2": true}.Includes(StringMap{"test3": true})
if expected != received {
t.Errorf("Expecting: %+v, received: %+v", expected, received)
}
}
func TestStringMapIncludes3(t *testing.T) {
expected := true
received := StringMap{"test": true, "test2": true}.Includes(StringMap{"test": true, "test2": true})
if expected != received {
t.Errorf("Expecting: %+v, received: %+v", expected, received)
}
}
func TestStringMapSlice(t *testing.T) {
expected := []string{"test", "test2"}
received := StringMap{"test": true, "test2": true}.Slice()
sort.Strings(received)
sort.Strings(expected)
if !reflect.DeepEqual(expected, received) {
t.Errorf("Expecting: %+v, received: %+v", expected, received)
}
}
func TestStringMapClone(t *testing.T) {
expected := StringMap{"test": true, "test2": true}
received := StringMap{"test": true, "test2": true}.Clone()
if !reflect.DeepEqual(expected, received) {
t.Errorf("Expecting: %+v, received: %+v", expected, received)
}
}
func TestStringMapString(t *testing.T) {
received := StringMap{"test": true, "test2": true}.String()
if received != "test;test2" && received != "test2;test" {
t.Errorf("Expecting: test or test2, received: %+v", received)
}
}
func TestStringMapGetOneEmpty(t *testing.T) {
expected := EmptyString
received := StringMap{}.GetOne()
if !reflect.DeepEqual(expected, received) {
t.Errorf("Expecting: %+v, received: %+v", expected, received)
}
}
func TestStringMapGetOneNotEmpty(t *testing.T) {
received := StringMap{"test": true, "test2": true}.GetOne()
if received != "test" && received != "test2" {
t.Errorf("Expecting: test or test2, received: %+v", received)
}
}
func TestMapStringToInt64Err(t *testing.T) {
t2 := map[string]string{"test": "a"}
_, err := MapStringToInt64(t2)
if err == nil {
t.Error("Got Error: ", err)
}
}
func TestFlagsWithParamsClone(t *testing.T) {
fWp := FlagsWithParams{
MetaEvent: {},
MetaRoutes: nil,
MetaThresholds: {MetaIDs: {"ID1", "ID2", "ID3"}, MetaDerivedReply: {}},
MetaAttributes: {"*disabled": {}},
MetaStats: {MetaIDs: {"ID"}},
}
cln := fWp.Clone()
if !reflect.DeepEqual(cln, fWp) {
t.Errorf("Expecting: %+v, received: %+v", ToJSON(fWp), ToJSON(cln))
}
cln[MetaDispatchers] = FlagParams{}
if _, has := fWp[MetaDispatchers]; has {
t.Errorf("Expected clone to not modify the cloned")
}
cln[MetaThresholds][MetaIDs][0] = ""
if fWp[MetaThresholds][MetaIDs][0] != "ID1" {
t.Errorf("Expected clone to not modify the cloned")
}
fWp = nil
cln = fWp.Clone()
if !reflect.DeepEqual(cln, fWp) {
t.Errorf("Expecting: %+v, received: %+v", ToJSON(fWp), ToJSON(cln))
}
}
func TestStringMapFieldAsInterfaceNotNil(t *testing.T) {
sm := StringMap{}
fldPath := []string{"first", "second"}
if val, _ := sm.FieldAsInterface(fldPath); val != nil {
t.Error("expected nil")
}
sm["first"] = true
fldPath = []string{"second"}
if val, _ := sm.FieldAsInterface(fldPath); val != nil {
t.Error("Should be nil")
}
fldPath[0] = "first"
if _, val := sm.FieldAsInterface(fldPath); val != nil {
t.Error("Should not be nil")
}
}
func TestStringMapFieldAsString(t *testing.T) {
sm := StringMap{}
fldPath := []string{"Hello"}
if _, val := sm.FieldAsString(fldPath); val == nil {
t.Error("Should not be nil")
}
sm["first"] = true
fldPath[0] = "first"
if _, val := sm.FieldAsString(fldPath); val != nil {
t.Error("Should be nil")
}
}
func TestStringMapHasKey(t *testing.T) {
sm := StringMap{"first": true}
key := "first"
exp := true
if v := sm.HasKey(key); v != exp {
t.Error("Expected true")
}
exp = false
if v := sm.HasKey(key); v == exp {
t.Error("Expected false")
}
}
func TestStringMapStringToInt64(t *testing.T) {
mymap := map[string]string{
"first": "1",
"second": "two",
}
exp := map[string]int64{
"first": 1,
"second": 2,
}
if val, _ := MapStringToInt64(mymap); reflect.DeepEqual(val, exp) == true {
t.Errorf("expected %v got %v", ToJSON(exp), ToJSON(val))
}
mymap["second"] = "2"
if val, _ := MapStringToInt64(mymap); reflect.DeepEqual(val, exp) == false {
t.Errorf("expected %v got %v", ToJSON(exp), ToJSON(val))
}
}
func TestStringMapIsEmpty(t *testing.T) {
sm := StringMap{"foo": true}
exp := false
if val := sm.IsEmpty(); val != exp {
t.Error("Should be false")
}
}
func TestStringMapFromSlice(t *testing.T) {
s := []string{"foo", "", "!baz", "bar"}
exp := map[string]bool{
"foo": true,
"baz": false,
"bar": true,
}
if val := StringMapFromSlice(s); reflect.DeepEqual(val, exp) {
t.Errorf("expected %v got %v", ToJSON(exp), ToJSON(val))
}
}
func TestStringMapNewStringMap(t *testing.T) {
s := []string{"foo", "", "!baz", "bar"}
exp := map[string]bool{
"foo": true,
"baz": false,
"bar": true,
}
if val := NewStringMap(s...); reflect.DeepEqual(val, exp) {
t.Errorf("expected %v got %v", ToJSON(exp), ToJSON(val))
}
}
func TestStringMapMapParse(t *testing.T) {
s := MetaZero
exp := make(StringMap)
if val := ParseStringMap(s); !reflect.DeepEqual(exp, val) {
t.Errorf("expected %v go t %v", ToJSON(exp), ToJSON(val))
}
s = "foo;!bar;baz"
exp = StringMap{
"foo": true,
"bar": false,
"baz": true,
}
if val := ParseStringMap(s); !reflect.DeepEqual(exp, val) {
t.Errorf("expected %v got %v", ToJSON(exp), ToJSON(val))
}
}