Files
cgrates/config/statscfg_test.go
2025-10-13 09:57:41 +02:00

370 lines
12 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 config
import (
"reflect"
"testing"
"time"
"github.com/cgrates/cgrates/utils"
)
func TestStatSCfgloadFromJsonCfgCase1(t *testing.T) {
cfgJSON := &StatServJsonCfg{
Enabled: utils.BoolPointer(true),
Indexed_selects: utils.BoolPointer(true),
Store_interval: utils.StringPointer("2"),
Store_uncompressed_limit: utils.IntPointer(10),
Thresholds_conns: &[]string{utils.MetaInternal, "*conn1"},
String_indexed_fields: &[]string{"*req.string"},
Prefix_indexed_fields: &[]string{"*req.index1", "*req.index2"},
Suffix_indexed_fields: &[]string{"*req.index1", "*req.index2"},
Exists_indexed_fields: &[]string{"*req.index1", "*req.index2"},
Notexists_indexed_fields: &[]string{"*req.index1", "*req.index2"},
Nested_fields: utils.BoolPointer(true),
}
expected := &StatSCfg{
Enabled: true,
IndexedSelects: true,
StoreInterval: 2,
StoreUncompressedLimit: 10,
ThresholdSConns: []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaThresholds), "*conn1"},
StringIndexedFields: &[]string{"*req.string"},
PrefixIndexedFields: &[]string{"*req.index1", "*req.index2"},
SuffixIndexedFields: &[]string{"*req.index1", "*req.index2"},
ExistsIndexedFields: &[]string{"*req.index1", "*req.index2"},
NotExistsIndexedFields: &[]string{"*req.index1", "*req.index2"},
NestedFields: true,
Opts: &StatsOpts{
ProfileIDs: []*DynamicStringSliceOpt{},
ProfileIgnoreFilters: []*DynamicBoolOpt{{}},
RoundingDecimals: []*DynamicIntOpt{},
},
EEsConns: []string{},
}
jsonCfg := NewDefaultCGRConfig()
if err := jsonCfg.statsCfg.loadFromJSONCfg(cfgJSON); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expected, jsonCfg.statsCfg) {
t.Errorf("Expected %+v \n, received %+v", utils.ToJSON(expected), utils.ToJSON(jsonCfg.statsCfg))
}
cfgJSON = nil
if err := jsonCfg.statsCfg.loadFromJSONCfg(cfgJSON); err != nil {
t.Error(err)
}
}
func TestStatSCfgloadFromJsonCfgOptsNil(t *testing.T) {
statsOpt := &StatsOpts{
ProfileIDs: []*DynamicStringSliceOpt{
{
Values: []string{},
},
},
ProfileIgnoreFilters: []*DynamicBoolOpt{
{
value: false,
},
},
}
exp := &StatsOpts{
ProfileIDs: []*DynamicStringSliceOpt{
{
Values: []string{},
},
},
ProfileIgnoreFilters: []*DynamicBoolOpt{
{
value: false,
},
},
}
statsOpt.loadFromJSONCfg(nil)
if !reflect.DeepEqual(statsOpt, exp) {
t.Errorf("Expected %v \n but received \n %v", exp, statsOpt)
}
}
func TestStatSCfgloadFromJsonCfgCase2(t *testing.T) {
statscfgJSON := &StatServJsonCfg{
Store_interval: utils.StringPointer("1ss"),
}
expected := "time: unknown unit \"ss\" in duration \"1ss\""
jsonCfg := NewDefaultCGRConfig()
if err := jsonCfg.statsCfg.loadFromJSONCfg(statscfgJSON); err == nil || err.Error() != expected {
t.Errorf("Expected %+v, received %+v", expected, err)
}
}
func TestStatSCfgAsMapInterface(t *testing.T) {
cfgJSONStr := `{
"stats": {},
}`
eMap := map[string]any{
utils.EnabledCfg: false,
utils.StoreIntervalCfg: utils.EmptyString,
utils.StoreUncompressedLimitCfg: 0,
utils.ThresholdSConnsCfg: []string{},
utils.IndexedSelectsCfg: true,
utils.PrefixIndexedFieldsCfg: []string{},
utils.SuffixIndexedFieldsCfg: []string{},
utils.ExistsIndexedFieldsCfg: []string{},
utils.NotExistsIndexedFieldsCfg: []string{},
utils.NestedFieldsCfg: false,
utils.OptsCfg: map[string]any{
utils.MetaProfileIDs: []*DynamicStringSliceOpt{},
utils.MetaProfileIgnoreFilters: []*DynamicBoolOpt{{}},
utils.OptsRoundingDecimals: []*DynamicIntOpt{},
},
utils.EEsConnsCfg: []string{},
utils.EEsExporterIDsCfg: []string(nil),
}
if cgrCfg, err := NewCGRConfigFromJSONStringWithDefaults(cfgJSONStr); err != nil {
t.Error(err)
} else if rcv := cgrCfg.statsCfg.AsMapInterface(); !reflect.DeepEqual(rcv, eMap) {
t.Errorf("Expected %+v \n, received %+v", eMap, rcv)
}
}
func TestStatSCfgAsMapInterface1(t *testing.T) {
cfgJSONStr := `{
"stats": {
"enabled": true,
"store_interval": "72h",
"store_uncompressed_limit": 1,
"thresholds_conns": ["*internal:*thresholds", "*conn1"],
"indexed_selects":false,
"string_indexed_fields": ["*req.string"],
"prefix_indexed_fields": ["*req.prefix_indexed_fields1","*req.prefix_indexed_fields2"],
"suffix_indexed_fields":["*req.suffix_indexed_fields"],
"exists_indexed_fields":["*req.exists_indexed_fields"],
"notexists_indexed_fields":["*req.notexists_indexed_fields"],
"nested_fields": true,
},
}`
eMap := map[string]any{
utils.EnabledCfg: true,
utils.StoreIntervalCfg: "72h0m0s",
utils.StoreUncompressedLimitCfg: 1,
utils.ThresholdSConnsCfg: []string{utils.MetaInternal, "*conn1"},
utils.IndexedSelectsCfg: false,
utils.StringIndexedFieldsCfg: []string{"*req.string"},
utils.PrefixIndexedFieldsCfg: []string{"*req.prefix_indexed_fields1", "*req.prefix_indexed_fields2"},
utils.SuffixIndexedFieldsCfg: []string{"*req.suffix_indexed_fields"},
utils.ExistsIndexedFieldsCfg: []string{"*req.exists_indexed_fields"},
utils.NotExistsIndexedFieldsCfg: []string{"*req.notexists_indexed_fields"},
utils.NestedFieldsCfg: true,
utils.OptsCfg: map[string]any{
utils.MetaProfileIDs: []*DynamicStringSliceOpt{},
utils.MetaProfileIgnoreFilters: []*DynamicBoolOpt{{}},
utils.OptsRoundingDecimals: []*DynamicIntOpt{},
},
utils.EEsConnsCfg: []string{},
utils.EEsExporterIDsCfg: []string(nil),
}
if cgrCfg, err := NewCGRConfigFromJSONStringWithDefaults(cfgJSONStr); err != nil {
t.Error(err)
} else if rcv := cgrCfg.statsCfg.AsMapInterface(); !reflect.DeepEqual(rcv, eMap) {
t.Errorf("Expected %+v \n, received %+v", eMap, rcv)
}
}
func TestStatSCfgClone(t *testing.T) {
ban := &StatSCfg{
Enabled: true,
IndexedSelects: true,
StoreInterval: 2,
StoreUncompressedLimit: 10,
ThresholdSConns: []string{utils.ConcatenatedKey(utils.MetaInternal, utils.MetaThresholds), "*conn1"},
StringIndexedFields: &[]string{"*req.index1"},
PrefixIndexedFields: &[]string{"*req.index1", "*req.index2"},
SuffixIndexedFields: &[]string{"*req.index1", "*req.index2"},
NestedFields: true,
Opts: &StatsOpts{},
}
rcv := ban.Clone()
if !reflect.DeepEqual(ban, rcv) {
t.Errorf("Expected: %+v\nReceived: %+v", utils.ToJSON(ban), utils.ToJSON(rcv))
}
if rcv.ThresholdSConns[1] = ""; ban.ThresholdSConns[1] != "*conn1" {
t.Errorf("Expected clone to not modify the cloned")
}
if (*rcv.StringIndexedFields)[0] = ""; (*ban.StringIndexedFields)[0] != "*req.index1" {
t.Errorf("Expected clone to not modify the cloned")
}
if (*rcv.PrefixIndexedFields)[0] = ""; (*ban.PrefixIndexedFields)[0] != "*req.index1" {
t.Errorf("Expected clone to not modify the cloned")
}
if (*rcv.SuffixIndexedFields)[0] = ""; (*ban.SuffixIndexedFields)[0] != "*req.index1" {
t.Errorf("Expected clone to not modify the cloned")
}
}
func TestDiffStatServJsonCfg(t *testing.T) {
var d *StatServJsonCfg
v1 := &StatSCfg{
Enabled: false,
IndexedSelects: false,
StoreInterval: 1 * time.Second,
StoreUncompressedLimit: 2,
ThresholdSConns: []string{"*localhost"},
StringIndexedFields: &[]string{"*req.index1"},
PrefixIndexedFields: &[]string{"*req.index2"},
SuffixIndexedFields: &[]string{"*req.index3"},
NestedFields: false,
Opts: &StatsOpts{
ProfileIDs: []*DynamicStringSliceOpt{
{
Tenant: "cgrates.org",
Values: []string{"statsid1"},
},
},
ProfileIgnoreFilters: []*DynamicBoolOpt{
{
Tenant: "cgrates.org",
value: false,
},
},
RoundingDecimals: []*DynamicIntOpt{
{
Tenant: "cgrates.org",
value: 1,
},
},
},
}
v2 := &StatSCfg{
Enabled: true,
IndexedSelects: true,
StoreInterval: 2 * time.Second,
StoreUncompressedLimit: 3,
ThresholdSConns: []string{"*birpc"},
StringIndexedFields: &[]string{"*req.index11"},
PrefixIndexedFields: &[]string{"*req.index22"},
SuffixIndexedFields: &[]string{"*req.index33"},
NestedFields: true,
Opts: &StatsOpts{
ProfileIDs: []*DynamicStringSliceOpt{
{
Tenant: "cgrates.net",
Values: []string{"statsid2"},
},
},
ProfileIgnoreFilters: []*DynamicBoolOpt{
{
Tenant: "cgrates.net",
value: true,
},
},
RoundingDecimals: []*DynamicIntOpt{
{
Tenant: "cgrates.net",
value: 2,
},
},
},
}
expected := &StatServJsonCfg{
Enabled: utils.BoolPointer(true),
Indexed_selects: utils.BoolPointer(true),
Store_interval: utils.StringPointer("2s"),
Store_uncompressed_limit: utils.IntPointer(3),
Thresholds_conns: &[]string{"*birpc"},
String_indexed_fields: &[]string{"*req.index11"},
Prefix_indexed_fields: &[]string{"*req.index22"},
Suffix_indexed_fields: &[]string{"*req.index33"},
Nested_fields: utils.BoolPointer(true),
Opts: &StatsOptsJson{
ProfileIDs: []*DynamicStringSliceOpt{
{
Tenant: "cgrates.net",
Values: []string{"statsid2"},
},
},
ProfileIgnoreFilters: []*DynamicInterfaceOpt{
{
Tenant: "cgrates.net",
Value: true,
},
},
RoundingDecimals: []*DynamicInterfaceOpt{
{
Tenant: "cgrates.net",
Value: 2,
},
},
},
}
rcv := diffStatServJsonCfg(d, v1, v2)
if !reflect.DeepEqual(rcv, expected) {
t.Errorf("Expected %v \n but received \n %v", utils.ToJSON(expected), utils.ToJSON(rcv))
}
v1 = v2
expected = &StatServJsonCfg{
Opts: &StatsOptsJson{},
}
rcv = diffStatServJsonCfg(d, v1, v2)
if !reflect.DeepEqual(rcv, expected) {
t.Errorf("Expected %v \n but received \n %v", utils.ToJSON(expected), utils.ToJSON(rcv))
}
}
func TestStatSCloneSection(t *testing.T) {
statsCfg := &StatSCfg{
Enabled: false,
IndexedSelects: false,
StoreInterval: 1 * time.Second,
StoreUncompressedLimit: 2,
ThresholdSConns: []string{"*localhost"},
StringIndexedFields: &[]string{"*req.index1"},
PrefixIndexedFields: &[]string{"*req.index2"},
SuffixIndexedFields: &[]string{"*req.index3"},
NestedFields: false,
Opts: &StatsOpts{
ProfileIDs: []*DynamicStringSliceOpt{},
},
}
exp := &StatSCfg{
Enabled: false,
IndexedSelects: false,
StoreInterval: 1 * time.Second,
StoreUncompressedLimit: 2,
ThresholdSConns: []string{"*localhost"},
StringIndexedFields: &[]string{"*req.index1"},
PrefixIndexedFields: &[]string{"*req.index2"},
SuffixIndexedFields: &[]string{"*req.index3"},
NestedFields: false,
Opts: &StatsOpts{
ProfileIDs: []*DynamicStringSliceOpt{},
},
}
rcv := statsCfg.CloneSection()
if !reflect.DeepEqual(rcv, exp) {
t.Errorf("Expected %v \n but received \n %v", utils.ToJSON(exp), utils.ToJSON(rcv))
}
}