mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 10:06:24 +05:00
278 lines
8.1 KiB
Go
278 lines
8.1 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 migrator
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/cgrates/cgrates/config"
|
|
"github.com/cgrates/cgrates/engine"
|
|
"github.com/cgrates/cgrates/utils"
|
|
)
|
|
|
|
func TestFieldinfo2Attribute(t *testing.T) {
|
|
type testA struct {
|
|
Path string
|
|
FieldInfo string
|
|
Initial []*engine.Attribute
|
|
Expected []*engine.Attribute
|
|
}
|
|
tests := []testA{
|
|
{
|
|
Path: utils.AccountField,
|
|
FieldInfo: utils.MetaDefault,
|
|
Initial: make([]*engine.Attribute, 0),
|
|
Expected: make([]*engine.Attribute, 0),
|
|
},
|
|
{
|
|
Path: utils.AccountField,
|
|
FieldInfo: "",
|
|
Initial: make([]*engine.Attribute, 0),
|
|
Expected: make([]*engine.Attribute, 0),
|
|
},
|
|
{
|
|
Path: utils.AccountField,
|
|
FieldInfo: "^1003",
|
|
Initial: make([]*engine.Attribute, 0),
|
|
Expected: []*engine.Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + utils.AccountField,
|
|
Type: utils.MetaVariable,
|
|
Value: config.NewRSRParsersMustCompile("1003", utils.InfieldSep),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Path: utils.Subject,
|
|
FieldInfo: `~effective_caller_id_number:s/(\d+)/+$1/`,
|
|
Initial: make([]*engine.Attribute, 0),
|
|
Expected: []*engine.Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + utils.Subject,
|
|
Type: utils.MetaVariable,
|
|
Value: config.NewRSRParsersMustCompile(`~effective_caller_id_number:s/(\d+)/+$1/`, utils.InfieldSep),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Path: utils.Subject,
|
|
FieldInfo: "^call_1003",
|
|
Initial: []*engine.Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + utils.AccountField,
|
|
Type: utils.MetaVariable,
|
|
Value: config.NewRSRParsersMustCompile("1003", utils.InfieldSep),
|
|
},
|
|
},
|
|
Expected: []*engine.Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + utils.AccountField,
|
|
Type: utils.MetaVariable,
|
|
Value: config.NewRSRParsersMustCompile("1003", utils.InfieldSep),
|
|
},
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + utils.Subject,
|
|
Type: utils.MetaVariable,
|
|
Value: config.NewRSRParsersMustCompile("call_1003", utils.InfieldSep),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
for i, v := range tests {
|
|
if rply := fieldinfo2Attribute(v.Initial, v.Path, v.FieldInfo); !reflect.DeepEqual(v.Expected, rply) {
|
|
t.Errorf("For %v expected: %s ,recieved: %s", i, utils.ToJSON(v.Expected), utils.ToJSON(rply))
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDerivedChargers2AttributeProfile(t *testing.T) {
|
|
type testC struct {
|
|
DC *v1DerivedCharger
|
|
Tenant string
|
|
Key string
|
|
Filters []string
|
|
Expected *engine.AttributeProfile
|
|
}
|
|
tests := []testC{
|
|
{
|
|
DC: &v1DerivedCharger{
|
|
RequestTypeField: utils.MetaDefault,
|
|
CategoryField: "^*voice",
|
|
AccountField: "^1003",
|
|
},
|
|
Tenant: defaultTenant,
|
|
Key: "key1",
|
|
Filters: make([]string, 0),
|
|
Expected: &engine.AttributeProfile{
|
|
Tenant: defaultTenant,
|
|
ID: "key1",
|
|
Contexts: []string{utils.MetaChargers},
|
|
FilterIDs: make([]string, 0),
|
|
ActivationInterval: nil,
|
|
Attributes: []*engine.Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + utils.Category,
|
|
Type: utils.MetaVariable,
|
|
Value: config.NewRSRParsersMustCompile("*voice", utils.InfieldSep),
|
|
},
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + utils.AccountField,
|
|
Type: utils.MetaVariable,
|
|
Value: config.NewRSRParsersMustCompile("1003", utils.InfieldSep),
|
|
},
|
|
},
|
|
Blocker: false,
|
|
Weight: 10,
|
|
},
|
|
},
|
|
{
|
|
DC: &v1DerivedCharger{
|
|
RequestTypeField: utils.MetaDefault,
|
|
CategoryField: "^*voice",
|
|
AccountField: "^1003",
|
|
SubjectField: "call_1003_to_1004",
|
|
DestinationField: "^1004",
|
|
},
|
|
Tenant: defaultTenant,
|
|
Key: "key1",
|
|
Filters: []string{"*string:~*req.Subject:1005"},
|
|
Expected: &engine.AttributeProfile{
|
|
Tenant: defaultTenant,
|
|
ID: "key1",
|
|
Contexts: []string{utils.MetaChargers},
|
|
FilterIDs: []string{"*string:~*req.Subject:1005"},
|
|
ActivationInterval: nil,
|
|
Attributes: []*engine.Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + utils.Category,
|
|
Type: utils.MetaVariable,
|
|
Value: config.NewRSRParsersMustCompile("*voice", utils.InfieldSep),
|
|
},
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + utils.AccountField,
|
|
Type: utils.MetaVariable,
|
|
Value: config.NewRSRParsersMustCompile("1003", utils.InfieldSep),
|
|
},
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + utils.Subject,
|
|
Type: utils.MetaVariable,
|
|
Value: config.NewRSRParsersMustCompile("call_1003_to_1004", utils.InfieldSep),
|
|
},
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + utils.Destination,
|
|
Type: utils.MetaVariable,
|
|
Value: config.NewRSRParsersMustCompile("1004", utils.InfieldSep),
|
|
},
|
|
},
|
|
Blocker: false,
|
|
Weight: 10,
|
|
},
|
|
},
|
|
}
|
|
for i, v := range tests {
|
|
if rply := derivedChargers2AttributeProfile(v.DC, v.Tenant, v.Key, v.Filters); !reflect.DeepEqual(v.Expected, rply) {
|
|
t.Errorf("For %v expected: %s ,recieved: %s", i, utils.ToJSON(v.Expected), utils.ToJSON(rply))
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDerivedChargers2Charger(t *testing.T) {
|
|
type testB struct {
|
|
DC *v1DerivedCharger
|
|
Tenant string
|
|
Key string
|
|
Filters []string
|
|
Expected *engine.ChargerProfile
|
|
}
|
|
tests := []testB{
|
|
{
|
|
DC: &v1DerivedCharger{
|
|
RunID: "runID",
|
|
RunFilters: "~Header4:s/a/${1}b/{*duration_seconds&*round:2}(b&c)",
|
|
RequestTypeField: utils.MetaDefault,
|
|
CategoryField: "^*voice",
|
|
AccountField: "^1003",
|
|
},
|
|
Tenant: defaultTenant,
|
|
Key: "key2",
|
|
Filters: []string{
|
|
"*string:~*req.Category:*voice1",
|
|
"*string:~*req.Account:1001",
|
|
},
|
|
Expected: &engine.ChargerProfile{
|
|
Tenant: defaultTenant,
|
|
ID: "key2",
|
|
FilterIDs: []string{
|
|
"*string:~*req.Category:*voice1",
|
|
"*string:~*req.Account:1001",
|
|
"*rsr:~*req.Header4:b|c",
|
|
},
|
|
ActivationInterval: nil,
|
|
RunID: "runID",
|
|
AttributeIDs: make([]string, 0),
|
|
Weight: 10,
|
|
},
|
|
},
|
|
{
|
|
DC: &v1DerivedCharger{
|
|
RunID: "runID2",
|
|
RunFilters: "~Account(^1003)",
|
|
AccountField: "^1003",
|
|
},
|
|
Tenant: defaultTenant,
|
|
Key: "key2",
|
|
Filters: []string{},
|
|
Expected: &engine.ChargerProfile{
|
|
Tenant: defaultTenant,
|
|
ID: "key2",
|
|
FilterIDs: []string{"*rsr:~*req.Account:^1003"},
|
|
ActivationInterval: nil,
|
|
RunID: "runID2",
|
|
AttributeIDs: make([]string, 0),
|
|
Weight: 10,
|
|
},
|
|
},
|
|
}
|
|
for i, v := range tests {
|
|
if rply := derivedChargers2Charger(v.DC, v.Tenant, v.Key, v.Filters); !reflect.DeepEqual(v.Expected, rply) {
|
|
t.Errorf("For %v expected: %s ,\n recieved: %s", i, utils.ToJSON(v.Expected), utils.ToJSON(rply))
|
|
}
|
|
}
|
|
}
|
|
func TestMigrateFilterV3(t *testing.T) {
|
|
v1flt := &v1Filter{
|
|
Tenant: "cgrates.org",
|
|
ID: "test-id",
|
|
Rules: []*v1FilterRule{{Type: "type1", FieldName: "field1", Values: []string{"value1"}}},
|
|
ActivationInterval: nil,
|
|
}
|
|
result := migrateFilterV3(v1flt)
|
|
|
|
if result.Tenant != v1flt.Tenant {
|
|
t.Errorf("expected Tenant %v, got %v", v1flt.Tenant, result.Tenant)
|
|
}
|
|
if result.ID != v1flt.ID {
|
|
t.Errorf("expected ID %v, got %v", v1flt.ID, result.ID)
|
|
}
|
|
if len(result.Rules) != len(v1flt.Rules) {
|
|
t.Errorf("expected %d rules, got %d", len(v1flt.Rules), len(result.Rules))
|
|
}
|
|
}
|