/*
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 utils

import (
	"reflect"
	"slices"
	"sort"
	"strings"
	"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 TestStringMapAsPrefixedString(t *testing.T) {
	tests := []struct {
		name   string
		input  StringMap
		expStr string
	}{
		{
			name: "NormalDest",
			input: StringMap{
				"DEST_1001": true,
				"DEST_1002": true,
			},
			expStr: "DEST_1001;DEST_1002",
		},
		{
			name: "ExcludedDest",
			input: StringMap{
				"DEST_1003": false,
				"DEST_1004": false,
			},
			expStr: "!DEST_1003;!DEST_1004",
		},
		{
			name: "MixedDest",
			input: StringMap{
				"DEST_1001": true,
				"DEST_1004": false,
			},
			expStr: "DEST_1001;!DEST_1004",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := tt.input.AsPrefixedString()

			gotDest := strings.Split(got, InfieldSep)
			expDest := strings.Split(tt.expStr, InfieldSep)

			slices.Sort(gotDest)
			slices.Sort(expDest)

			for i := range gotDest {
				if gotDest[i] != expDest[i] {
					t.Errorf("Mismatch at index %d! Expected: %s, Received: %s", i, expDest[i], gotDest[i])
				}
			}

		})
	}
}

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

}