uppal/cgrates
1
0
Fork 1
mirror of https://github.com/cgrates/cgrates.git synced 2026-02-11 18:16:24 +05:00
Code Issues Packages Projects Releases Wiki Activity
Files
f46ea874ba088fcaeee42f511c71d3032ae4b430
cgrates/agents/kamevent_test.go
armirveliaj 1854afd74d Add coverage tests on agents and config
2024-09-10 21:10:24 +02:00

1100 lines
37 KiB
Go
Raw Blame History

/*
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 agents
import (
"encoding/json"
"errors"
"reflect"
"strings"
"testing"
"time"
"github.com/cgrates/cgrates/config"
"github.com/cgrates/cgrates/engine"
"github.com/cgrates/cgrates/sessions"
"github.com/cgrates/cgrates/utils"
"github.com/google/go-cmp/cmp"
)
var kamEv = KamEvent{KamTRIndex: "29223", KamTRLabel: "698469260",
"callid": "ODVkMDI2Mzc2MDY5N2EzODhjNTAzNTdlODhiZjRlYWQ", "from_tag": "eb082607", "to_tag": "4ea9687f", "cgr_account": "dan",
"cgr_reqtype": utils.MetaPrepaid, "cgr_subject": "dan", "cgr_destination": "+4986517174963", "cgr_tenant": "itsyscom.com",
"cgr_duration": "20", utils.CGRRoute: "suppl2", utils.CGRDisconnectCause: "200", "extra1": "val1", "extra2": "val2"}
func TestNewKamEvent(t *testing.T) {
evStr := `{"event":"CGR_CALL_END",
"callid":"46c01a5c249b469e76333fc6bfa87f6a@0:0:0:0:0:0:0:0",
"from_tag":"bf71ad59",
"to_tag":"7351fecf",
"cgr_reqtype":"*postpaid",
"cgr_account":"1001",
"cgr_destination":"1002",
"cgr_answertime":"1419839310",
"cgr_duration":"3",
"cgr_route":"supplier2",
"cgr_disconnectcause": "200",
"cgr_pdd": "4"}`
eKamEv := KamEvent{
"event": "CGR_CALL_END",
"callid": "46c01a5c249b469e76333fc6bfa87f6a@0:0:0:0:0:0:0:0",
"from_tag": "bf71ad59",
"to_tag": "7351fecf",
"cgr_reqtype": utils.MetaPostpaid,
"cgr_account": "1001",
"cgr_destination": "1002",
"cgr_answertime": "1419839310",
"cgr_duration": "3",
"cgr_pdd": "4",
utils.CGRRoute: "supplier2",
utils.CGRDisconnectCause: "200",
utils.OriginHost: utils.KamailioAgent,
}
if kamEv, err := NewKamEvent([]byte(evStr), utils.KamailioAgent, ""); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(eKamEv, kamEv) {
t.Error("Received: ", kamEv)
}
}
func TestKamEvMissingParameter(t *testing.T) {
kamEv = KamEvent{EVENT: CGR_CALL_END,
"callid": "46c01a5c249b469e76333fc6bfa87f6a@0:0:0:0:0:0:0:0",
"from_tag": "bf71ad59", "to_tag": "7351fecf",
"cgr_reqtype": utils.MetaPostpaid, "cgr_account": "1001",
"cgr_answertime": "1419839310", "cgr_duration": "3", "cgr_pdd": "4",
utils.CGRRoute: "supplier2",
utils.CGRDisconnectCause: "200"}
if missingParam := kamEv.MissingParameter(); missingParam != true {
t.Errorf("Expecting: true, received:%+v ", missingParam)
}
}
func TestKamEvAsMapStringInterface(t *testing.T) {
kamEv := KamEvent{"event": "CGR_CALL_END",
"callid": "46c01a5c249b469e76333fc6bfa87f6a@0:0:0:0:0:0:0:0",
"from_tag": "bf71ad59", "to_tag": "7351fecf",
"cgr_reqtype": utils.MetaPostpaid, "cgr_account": "1001",
"cgr_destination": "1002", "cgr_answertime": "1419839310",
"cgr_duration": "3", "cgr_pdd": "4",
utils.CGRRoute: "supplier2",
utils.CGRDisconnectCause: "200"}
expMp := make(map[string]any)
expMp["cgr_account"] = "1001"
expMp["cgr_duration"] = "3"
expMp["cgr_pdd"] = "4"
expMp["cgr_destination"] = "1002"
expMp[utils.CGRRoute] = "supplier2"
expMp["cgr_answertime"] = "1419839310"
expMp[utils.CGRDisconnectCause] = "200"
expMp["callid"] = "46c01a5c249b469e76333fc6bfa87f6a@0:0:0:0:0:0:0:0"
expMp["from_tag"] = "bf71ad59"
expMp["to_tag"] = "7351fecf"
expMp["cgr_reqtype"] = utils.MetaPostpaid
expMp[utils.RequestType] = utils.MetaRated
expMp[utils.Source] = utils.KamailioAgent
rcv := kamEv.AsMapStringInterface()
if !reflect.DeepEqual(expMp, rcv) {
t.Errorf("Expecting: %+v, received: %+v", expMp, rcv)
}
}
func TestKamEvAsCGREvent(t *testing.T) {
timezone := config.CgrConfig().GeneralCfg().DefaultTimezone
kamEv := KamEvent{"event": "CGR_CALL_END",
"callid": "46c01a5c249b469e76333fc6bfa87f6a@0:0:0:0:0:0:0:0",
"from_tag": "bf71ad59", "to_tag": "7351fecf",
"cgr_reqtype": utils.MetaPostpaid, "cgr_account": "1001",
"cgr_destination": "1002", "cgr_answertime": "1419839310",
"cgr_duration": "3", "cgr_pdd": "4",
utils.CGRRoute: "supplier2",
utils.CGRDisconnectCause: "200"}
sTime, err := utils.ParseTimeDetectLayout(kamEv[utils.AnswerTime], timezone)
if err != nil {
return
}
expected := &utils.CGREvent{
Tenant: utils.FirstNonEmpty(kamEv[utils.Tenant],
config.CgrConfig().GeneralCfg().DefaultTenant),
ID: utils.UUIDSha1Prefix(),
Time: &sTime,
Event: kamEv.AsMapStringInterface(),
}
if rcv, err := kamEv.AsCGREvent(timezone); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expected.Tenant, rcv.Tenant) {
t.Errorf("Expecting: %+v, received: %+v", expected.Tenant, rcv.Tenant)
} else if !reflect.DeepEqual(expected.Time, rcv.Time) {
t.Errorf("Expecting: %+v, received: %+v", expected.Time, rcv.Time)
} else if !reflect.DeepEqual(expected.Event, rcv.Event) {
t.Errorf("Expecting: %+v, received: %+v", expected.Event, rcv.Event)
}
}
func TestKamEvV1AuthorizeArgs(t *testing.T) {
timezone := config.CgrConfig().GeneralCfg().DefaultTimezone
kamEv := KamEvent{"event": "CGR_CALL_END",
"callid": "46c01a5c249b469e76333fc6bfa87f6a@0:0:0:0:0:0:0:0",
"from_tag": "bf71ad59", "to_tag": "7351fecf",
"cgr_reqtype": utils.MetaPostpaid, "cgr_account": "1001",
"cgr_destination": "1002", "cgr_answertime": "1419839310",
"cgr_duration": "3", "cgr_pdd": "4",
utils.CGRRoute: "supplier2",
utils.CGRDisconnectCause: "200",
utils.CGRFlags: "*accounts;*routes;*routes_event_cost;*routes_ignore_errors"}
sTime, err := utils.ParseTimeDetectLayout(kamEv[utils.AnswerTime], timezone)
if err != nil {
return
}
expected := &sessions.V1AuthorizeArgs{
GetMaxUsage: true,
CGREvent: &utils.CGREvent{
Tenant: utils.FirstNonEmpty(kamEv[utils.Tenant],
config.CgrConfig().GeneralCfg().DefaultTenant),
ID: utils.UUIDSha1Prefix(),
Time: &sTime,
Event: kamEv.AsMapStringInterface(),
},
GetRoutes: true,
RoutesIgnoreErrors: true,
RoutesMaxCost: utils.MetaEventCost,
}
rcv := kamEv.V1AuthorizeArgs()
if !reflect.DeepEqual(expected.CGREvent.Tenant, rcv.CGREvent.Tenant) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Tenant, rcv.CGREvent.Tenant)
} else if !reflect.DeepEqual(expected.CGREvent.Time, rcv.CGREvent.Time) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Time, rcv.CGREvent.Time)
} else if !reflect.DeepEqual(expected.CGREvent.Event, rcv.CGREvent.Event) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Event, rcv.CGREvent.Event)
} else if !reflect.DeepEqual(expected.CGREvent.Event, rcv.CGREvent.Event) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Event, rcv.CGREvent.Event)
} else if !reflect.DeepEqual(expected.GetMaxUsage, rcv.GetMaxUsage) {
t.Errorf("Expecting: %+v, received: %+v", expected.GetMaxUsage, rcv.GetMaxUsage)
} else if !reflect.DeepEqual(expected.GetRoutes, rcv.GetRoutes) {
t.Errorf("Expecting: %+v, received: %+v", expected.GetRoutes, rcv.GetRoutes)
} else if !reflect.DeepEqual(expected.GetAttributes, rcv.GetAttributes) {
t.Errorf("Expecting: %+v, received: %+v", expected.GetAttributes, rcv.GetAttributes)
} else if !reflect.DeepEqual(expected.RoutesMaxCost, rcv.RoutesMaxCost) {
t.Errorf("Expecting: %+v, received: %+v", expected.RoutesMaxCost, rcv.RoutesMaxCost)
} else if !reflect.DeepEqual(expected.RoutesIgnoreErrors, rcv.RoutesIgnoreErrors) {
t.Errorf("Expecting: %+v, received: %+v", expected.RoutesIgnoreErrors, rcv.RoutesIgnoreErrors)
}
}
func TestKamEvV1AuthorizeArgs2(t *testing.T) {
timezone := config.CgrConfig().GeneralCfg().DefaultTimezone
kamEv := KamEvent{"event": "CGR_CALL_END",
"callid": "46c01a5c249b469e76333fc6bfa87f6a@0:0:0:0:0:0:0:0",
"from_tag": "bf71ad59", "to_tag": "7351fecf",
"cgr_reqtype": utils.MetaPostpaid, "cgr_account": "1001",
"cgr_destination": "1002", "cgr_answertime": "1419839310",
"cgr_duration": "3", "cgr_pdd": "4",
utils.CGRRoute: "supplier2",
utils.CGRDisconnectCause: "200",
utils.CGRFlags: "*accounts;*routes;*routes_maxcost:100;*routes_ignore_errors"}
sTime, err := utils.ParseTimeDetectLayout(kamEv[utils.AnswerTime], timezone)
if err != nil {
return
}
expected := &sessions.V1AuthorizeArgs{
GetMaxUsage: true,
CGREvent: &utils.CGREvent{
Tenant: utils.FirstNonEmpty(kamEv[utils.Tenant],
config.CgrConfig().GeneralCfg().DefaultTenant),
ID: utils.UUIDSha1Prefix(),
Time: &sTime,
Event: kamEv.AsMapStringInterface(),
},
GetRoutes: true,
RoutesIgnoreErrors: true,
RoutesMaxCost: "100",
}
rcv := kamEv.V1AuthorizeArgs()
if !reflect.DeepEqual(expected.CGREvent.Tenant, rcv.CGREvent.Tenant) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Tenant, rcv.CGREvent.Tenant)
} else if !reflect.DeepEqual(expected.CGREvent.Time, rcv.CGREvent.Time) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Time, rcv.CGREvent.Time)
} else if !reflect.DeepEqual(expected.CGREvent.Event, rcv.CGREvent.Event) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Event, rcv.CGREvent.Event)
} else if !reflect.DeepEqual(expected.CGREvent.Event, rcv.CGREvent.Event) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Event, rcv.CGREvent.Event)
} else if !reflect.DeepEqual(expected.GetMaxUsage, rcv.GetMaxUsage) {
t.Errorf("Expecting: %+v, received: %+v", expected.GetMaxUsage, rcv.GetMaxUsage)
} else if !reflect.DeepEqual(expected.GetRoutes, rcv.GetRoutes) {
t.Errorf("Expecting: %+v, received: %+v", expected.GetRoutes, rcv.GetRoutes)
} else if !reflect.DeepEqual(expected.GetAttributes, rcv.GetAttributes) {
t.Errorf("Expecting: %+v, received: %+v", expected.GetAttributes, rcv.GetAttributes)
} else if !reflect.DeepEqual(expected.RoutesMaxCost, rcv.RoutesMaxCost) {
t.Errorf("Expecting: %+v, received: %+v", expected.RoutesMaxCost, rcv.RoutesMaxCost)
} else if !reflect.DeepEqual(expected.RoutesIgnoreErrors, rcv.RoutesIgnoreErrors) {
t.Errorf("Expecting: %+v, received: %+v", expected.RoutesIgnoreErrors, rcv.RoutesIgnoreErrors)
}
}
func TestKamEvAsKamAuthReply(t *testing.T) {
timezone := config.CgrConfig().GeneralCfg().DefaultTimezone
kamEv := KamEvent{"event": "CGR_CALL_END",
"callid": "46c01a5c249b469e76333fc6bfa87f6a@0:0:0:0:0:0:0:0",
"from_tag": "bf71ad59", "to_tag": "7351fecf",
"cgr_reqtype": utils.MetaPostpaid, "cgr_account": "1001",
"cgr_destination": "1002", "cgr_answertime": "1419839310",
"cgr_duration": "3", "cgr_pdd": "4",
utils.CGRRoute: "supplier2",
utils.CGRDisconnectCause: "200"}
sTime, err := utils.ParseTimeDetectLayout(kamEv[utils.AnswerTime], timezone)
if err != nil {
return
}
authArgs := &sessions.V1AuthorizeArgs{
GetMaxUsage: true,
CGREvent: &utils.CGREvent{
Tenant: utils.FirstNonEmpty(kamEv[utils.Tenant],
config.CgrConfig().GeneralCfg().DefaultTenant),
ID: utils.UUIDSha1Prefix(),
Time: &sTime,
Event: kamEv.AsMapStringInterface(),
},
}
authRply := &sessions.V1AuthorizeReply{
MaxUsage: utils.DurationPointer(5 * time.Second),
}
expected := &KamReply{
Event: CGR_AUTH_REPLY,
MaxUsage: 5,
}
if rcv, err := kamEv.AsKamAuthReply(authArgs, authRply, nil); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expected, rcv) {
t.Errorf("Expecting: %+v, received: %+v", expected, rcv)
}
kamEv = KamEvent{"event": "CGR_PROFILE_REQUEST",
"Tenant": "cgrates.org", "Account": "1001",
KamReplyRoute: "CGR_PROFILE_REPLY"}
authArgs = &sessions.V1AuthorizeArgs{
GetAttributes: true,
CGREvent: &utils.CGREvent{
Tenant: utils.FirstNonEmpty(kamEv[utils.Tenant],
config.CgrConfig().GeneralCfg().DefaultTenant),
ID: utils.UUIDSha1Prefix(),
Time: &sTime,
Event: kamEv.AsMapStringInterface(),
},
}
authRply = &sessions.V1AuthorizeReply{
Attributes: &engine.AttrSProcessEventReply{
MatchedProfiles: []string{"ATTR_1001_ACCOUNT_PROFILE"},
AlteredFields: []string{"*req.Password", utils.MetaReq + utils.NestingSep + utils.RequestType},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "TestKamEvAsKamAuthReply",
Event: map[string]any{
utils.Tenant: "cgrates.org",
utils.AccountField: "1001",
"Password": "check123",
utils.RequestType: utils.MetaPrepaid,
},
},
},
}
expected = &KamReply{
Event: "CGR_PROFILE_REPLY",
Attributes: "Password:check123,RequestType:*prepaid",
}
if rcv, err := kamEv.AsKamAuthReply(authArgs, authRply, nil); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expected, rcv) {
t.Errorf("Expecting: %+v, received: %+v", expected, rcv)
}
}
func TestKamEvV1InitSessionArgs(t *testing.T) {
timezone := config.CgrConfig().GeneralCfg().DefaultTimezone
kamEv := KamEvent{"event": "CGR_CALL_END",
"callid": "46c01a5c249b469e76333fc6bfa87f6a@0:0:0:0:0:0:0:0",
"from_tag": "bf71ad59", "to_tag": "7351fecf",
"cgr_reqtype": utils.MetaPostpaid, "cgr_account": "1001",
"cgr_destination": "1002", "cgr_answertime": "1419839310",
"cgr_duration": "3", "cgr_pdd": "4",
utils.CGRRoute: "supplier2",
utils.CGRDisconnectCause: "200"}
sTime, err := utils.ParseTimeDetectLayout(kamEv[utils.AnswerTime], timezone)
if err != nil {
return
}
expected := &sessions.V1InitSessionArgs{
InitSession: true,
CGREvent: &utils.CGREvent{
Tenant: utils.FirstNonEmpty(kamEv[utils.Tenant],
config.CgrConfig().GeneralCfg().DefaultTenant),
ID: utils.UUIDSha1Prefix(),
Time: &sTime,
Event: kamEv.AsMapStringInterface(),
},
}
rcv := kamEv.V1InitSessionArgs()
if !reflect.DeepEqual(expected.CGREvent.Tenant, rcv.CGREvent.Tenant) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Tenant, rcv.CGREvent.Tenant)
} else if !reflect.DeepEqual(expected.CGREvent.Time, rcv.CGREvent.Time) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Time, rcv.CGREvent.Time)
} else if !reflect.DeepEqual(expected.CGREvent.Event, rcv.CGREvent.Event) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Event, rcv.CGREvent.Event)
} else if !reflect.DeepEqual(expected.CGREvent.Event, rcv.CGREvent.Event) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Event, rcv.CGREvent.Event)
} else if !reflect.DeepEqual(expected.InitSession, rcv.InitSession) {
t.Errorf("Expecting: %+v, received: %+v", expected.InitSession, rcv.InitSession)
}
}
func TestKamEvV1TerminateSessionArgs(t *testing.T) {
timezone := config.CgrConfig().GeneralCfg().DefaultTimezone
kamEv := KamEvent{"event": "CGR_CALL_END",
"callid": "46c01a5c249b469e76333fc6bfa87f6a@0:0:0:0:0:0:0:0",
"from_tag": "bf71ad59", "to_tag": "7351fecf",
"cgr_reqtype": utils.MetaPostpaid, "cgr_account": "1001",
"cgr_destination": "1002", "cgr_answertime": "1419839310",
"cgr_duration": "3", "cgr_pdd": "4",
utils.CGRRoute: "supplier2",
utils.CGRDisconnectCause: "200"}
sTime, err := utils.ParseTimeDetectLayout(kamEv[utils.AnswerTime], timezone)
if err != nil {
return
}
expected := &sessions.V1TerminateSessionArgs{
TerminateSession: true,
CGREvent: &utils.CGREvent{
Tenant: utils.FirstNonEmpty(kamEv[utils.Tenant],
config.CgrConfig().GeneralCfg().DefaultTenant),
ID: utils.UUIDSha1Prefix(),
Time: &sTime,
Event: kamEv.AsMapStringInterface(),
},
}
rcv := kamEv.V1TerminateSessionArgs()
if !reflect.DeepEqual(expected.CGREvent.Tenant, rcv.CGREvent.Tenant) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Tenant, rcv.CGREvent.Tenant)
} else if !reflect.DeepEqual(expected.CGREvent.Time, rcv.CGREvent.Time) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Time, rcv.CGREvent.Time)
} else if !reflect.DeepEqual(expected.CGREvent.Event, rcv.CGREvent.Event) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Event, rcv.CGREvent.Event)
} else if !reflect.DeepEqual(expected.CGREvent.Event, rcv.CGREvent.Event) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Event, rcv.CGREvent.Event)
} else if !reflect.DeepEqual(expected.TerminateSession, rcv.TerminateSession) {
t.Errorf("Expecting: %+v, received: %+v", expected.TerminateSession, rcv.TerminateSession)
}
}
func TestKamEvV1ProcessMessageArgs(t *testing.T) {
timezone := config.CgrConfig().GeneralCfg().DefaultTimezone
kamEv := KamEvent{"event": "CGR_PROCESS_MESSAGE",
"callid": "46c01a5c249b469e76333fc6bfa87f6a@0:0:0:0:0:0:0:0",
"from_tag": "bf71ad59", "to_tag": "7351fecf",
"cgr_reqtype": utils.MetaPostpaid, "cgr_account": "1001",
"cgr_destination": "1002", "cgr_answertime": "1419839310",
"cgr_duration": "3", "cgr_pdd": "4",
utils.CGRRoute: "supplier2",
utils.CGRDisconnectCause: "200"}
sTime, err := utils.ParseTimeDetectLayout(kamEv[utils.AnswerTime], timezone)
if err != nil {
return
}
expected := &sessions.V1ProcessMessageArgs{
CGREvent: &utils.CGREvent{
Tenant: utils.FirstNonEmpty(kamEv[utils.Tenant],
config.CgrConfig().GeneralCfg().DefaultTenant),
ID: utils.UUIDSha1Prefix(),
Time: &sTime,
Event: kamEv.AsMapStringInterface(),
},
}
rcv := kamEv.V1ProcessMessageArgs()
if !reflect.DeepEqual(expected.CGREvent.Tenant, rcv.CGREvent.Tenant) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Tenant, rcv.CGREvent.Tenant)
} else if !reflect.DeepEqual(expected.CGREvent.Time, rcv.CGREvent.Time) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Time, rcv.CGREvent.Time)
} else if !reflect.DeepEqual(expected.CGREvent.Event, rcv.CGREvent.Event) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Event, rcv.CGREvent.Event)
} else if !reflect.DeepEqual(expected.CGREvent.Event, rcv.CGREvent.Event) {
t.Errorf("Expecting: %+v, received: %+v", expected.CGREvent.Event, rcv.CGREvent.Event)
}
}
func TestKamEvAsKamProcessEventReply(t *testing.T) {
timezone := config.CgrConfig().GeneralCfg().DefaultTimezone
kamEv := KamEvent{"event": "CGR_PROCESS_MESSAGE",
"callid": "46c01a5c249b469e76333fc6bfa87f6a@0:0:0:0:0:0:0:0",
"from_tag": "bf71ad59", "to_tag": "7351fecf",
"cgr_reqtype": utils.MetaPostpaid, "cgr_account": "1001",
"cgr_destination": "1002", "cgr_answertime": "1419839310",
"cgr_duration": "3", "cgr_pdd": "4",
utils.CGRRoute: "supplier2",
utils.CGRDisconnectCause: "200"}
sTime, err := utils.ParseTimeDetectLayout(kamEv[utils.AnswerTime], timezone)
if err != nil {
return
}
procEvArgs := &sessions.V1ProcessMessageArgs{
Debit: true,
CGREvent: &utils.CGREvent{
Tenant: utils.FirstNonEmpty(kamEv[utils.Tenant],
config.CgrConfig().GeneralCfg().DefaultTenant),
ID: utils.UUIDSha1Prefix(),
Time: &sTime,
Event: kamEv.AsMapStringInterface(),
},
}
procEvhRply := &sessions.V1ProcessMessageReply{
MaxUsage: utils.DurationPointer(5 * time.Second),
}
expected := &KamReply{
Event: CGR_PROCESS_MESSAGE,
MaxUsage: 5,
}
if rcv, err := kamEv.AsKamProcessMessageReply(procEvArgs, procEvhRply, nil); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expected, rcv) {
t.Errorf("Expecting: %+v, received: %+v", expected, rcv)
}
kamEv = KamEvent{"event": "CGR_PROFILE_REQUEST",
"Tenant": "cgrates.org", "Account": "1001",
KamReplyRoute: "CGR_PROFILE_REPLY"}
procEvArgs = &sessions.V1ProcessMessageArgs{
GetAttributes: true,
CGREvent: &utils.CGREvent{
Tenant: utils.FirstNonEmpty(kamEv[utils.Tenant],
config.CgrConfig().GeneralCfg().DefaultTenant),
ID: utils.UUIDSha1Prefix(),
Time: &sTime,
Event: kamEv.AsMapStringInterface(),
},
}
procEvhRply = &sessions.V1ProcessMessageReply{
Attributes: &engine.AttrSProcessEventReply{
MatchedProfiles: []string{"ATTR_1001_ACCOUNT_PROFILE"},
AlteredFields: []string{"*req.Password", utils.MetaReq + utils.NestingSep + utils.RequestType},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "TestKamEvAsKamAuthReply",
Event: map[string]any{
utils.Tenant: "cgrates.org",
utils.AccountField: "1001",
"Password": "check123",
utils.RequestType: utils.MetaPrepaid,
},
},
},
}
expected = &KamReply{
Event: "CGR_PROFILE_REPLY",
Attributes: "Password:check123,RequestType:*prepaid",
}
if rcv, err := kamEv.AsKamProcessMessageReply(procEvArgs, procEvhRply, nil); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expected, rcv) {
t.Errorf("Expecting: %+v, received: %+v", expected, rcv)
}
}
func TestAgentsNewKamSessionDisconnect(t *testing.T) {
hEntry := "entry123"
hID := "id123"
reason := "test reason"
got := NewKamSessionDisconnect(hEntry, hID, reason)
want := &KamSessionDisconnect{
Event: CGR_SESSION_DISCONNECT,
HashEntry: hEntry,
HashId: hID,
Reason: reason,
}
if diff := cmp.Diff(got, want); diff != "" {
t.Errorf("NewKamSessionDisconnect() mismatch (-got +want):\n%s", diff)
}
}
func TestAgentsKamEvent_String(t *testing.T) {
ke := KamEvent{
"EventName": "TestEvent",
"Data": "TestData",
}
got := ke.String()
want := `{"Data":"TestData","EventName":"TestEvent"}`
if got != want {
t.Errorf("KamEvent.String() = %v, want %v", got, want)
}
}
func TestKameventKamSessionDisconnect(t *testing.T) {
ksd := KamSessionDisconnect{
Event: "disconnect",
HashEntry: "123",
HashId: "2012",
Reason: "timeout",
}
want := `{"Event":"disconnect","HashEntry":"123","HashId":"2012","Reason":"timeout"}`
got := ksd.String()
if got != want {
t.Errorf("String() Want: %s, Got: %s", want, got)
}
}
func TestKameventNewKamDlgReply(t *testing.T) {
kamEvData := []byte(`{"Event":"dialog","Jsonrpl_body":{"Id":1,"Jsonrpc":"2.0","Result":[{"call-id":"123456","Caller":{"Tag":"tag123"},"variables":[{"cgrOriginID":"id123","cgrOriginHost":"host123"}]}]}}`)
expected := KamDlgReply{
Event: "dialog",
Jsonrpl_body: &kamJsonDlgBody{
Id: 1,
Jsonrpc: "2.0",
Result: []*kamDlgInfo{
{
CallId: "123456",
Caller: &kamCallerDlg{
Tag: "tag123",
},
Variables: []struct {
CgrOriginID string `json:"cgrOriginID,omitempty"`
CgrOriginHost string `json:"cgrOriginHost,omitempty"`
}{
{
CgrOriginID: "id123",
CgrOriginHost: "host123",
},
},
},
},
},
}
result, err := NewKamDlgReply(kamEvData)
if err != nil {
t.Errorf("Unexpected error: %v", err)
return
}
expectedJSON, _ := json.Marshal(expected)
resultJSON, _ := json.Marshal(result)
if string(expectedJSON) != string(resultJSON) {
t.Errorf("NewKamDlgReply() returned unexpected result. Expected: %s, Got: %s", expectedJSON, resultJSON)
}
invalidData := []byte("{invalid json")
_, err = NewKamDlgReply(invalidData)
if err == nil {
t.Errorf("Expected error when unmarshaling invalid data. Got nil error")
}
}
func TestKameventStringKdr(t *testing.T) {
event := "dialog"
jsonBody := &kamJsonDlgBody{
Id: 1,
Jsonrpc: "2.0",
Result: []*kamDlgInfo{
{
CallId: "123456",
Caller: &kamCallerDlg{
Tag: "tag123",
},
Variables: []struct {
CgrOriginID string `json:"cgrOriginID,omitempty"`
CgrOriginHost string `json:"cgrOriginHost,omitempty"`
}{
{
CgrOriginID: "id123",
CgrOriginHost: "host123",
},
},
},
},
}
reply := KamDlgReply{
Event: event,
Jsonrpl_body: jsonBody,
}
expectedJSON, err := json.Marshal(reply)
if err != nil {
t.Errorf("Unexpected error marshalling KamDlgReply: %v", err)
return
}
result := reply.String()
if result != string(expectedJSON) {
t.Errorf("KamDlgReply.String() returned unexpected result. Expected: %s, Got: %s", string(expectedJSON), result)
}
}
func TestKamEventKamReplyString(t *testing.T) {
krply := &KamReply{
Event: "testEvent",
TransactionIndex: "123",
TransactionLabel: "testLabel",
Attributes: "testAttributes",
ResourceAllocation: "testaAllocation",
MaxUsage: 10,
Routes: "testRoute",
Thresholds: "testThreshold",
StatQueues: "testQueue",
Error: "testError",
}
result := krply.String()
var unmarshalledResult map[string]interface{}
err := json.Unmarshal([]byte(result), &unmarshalledResult)
if err != nil {
t.Errorf("String() method does not return a valid JSON string: %v", err)
}
if unmarshalledResult["Event"] != krply.Event {
t.Errorf("Event mismatch. Expected: %s, Got: %v", krply.Event, unmarshalledResult["Event"])
}
}
func TestKamEventProcessMessageEmptyReply(t *testing.T) {
kev := KamEvent{
"KamReplyRoute": "CGR_PROCESS_MESSAGE",
}
kar := kev.AsKamProcessMessageEmptyReply()
if kar.Event != "CGR_PROCESS_MESSAGE" {
t.Errorf("expected event name 'CGR_PROCESS_MESSAGE', got %s", kar.Event)
}
}
func TestKamEventProcessCDRReply(t *testing.T) {
kev := KamEvent{
"KamReplyRoute": "CGR_PROCESS_CDR",
"KamTRIndex": "123",
"KamTRLabel": "456",
}
cgrEvWithArgDisp := &utils.CGREvent{}
rply := "reply"
var rplyErr error
kar, err := kev.AsKamProcessCDRReply(cgrEvWithArgDisp, &rply, rplyErr)
if kar.Event != "CGR_PROCESS_CDR" {
t.Errorf("expected event name 'CGR_PROCESS_CDR', got %s", kar.Event)
}
if kar.Error != "" {
t.Errorf("unexpected error: %s", kar.Error)
}
if err != nil {
t.Errorf("unexpected error: %v", err)
}
}
func TestKamEventV1ProcessCDRArgs(t *testing.T) {
kev := KamEvent{
"KamReplyRoute": "CGR_PROCESS_CDR",
"KamTRIndex": "123",
"KamTRLabel": "456",
}
args := kev.V1ProcessCDRArgs()
if args != nil {
t.Errorf("Expected non-nil CGREvent, got nil")
}
kev = KamEvent{}
args = kev.V1ProcessCDRArgs()
if args != nil {
t.Errorf("Expected nil CGREvent for error case, got %+v", args)
}
}
func TestKamEventMissingParameter(t *testing.T) {
tests := []struct {
name string
kev KamEvent
expected bool
}{
{
name: "CGR_AUTH_REQUEST with missing parameters",
kev: KamEvent{
EVENT: CGR_AUTH_REQUEST,
KamTRIndex: "",
KamTRLabel: "",
},
expected: true,
},
{
name: "CGR_CALL_START with missing parameters",
kev: KamEvent{
EVENT: CGR_CALL_START,
KamHashEntry: "",
KamHashID: "",
},
expected: true,
},
{
name: "CGR_PROCESS_MESSAGE with required parameters present",
kev: KamEvent{
EVENT: CGR_PROCESS_MESSAGE,
KamTRIndex: "index",
KamTRLabel: "label",
},
expected: false,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
result := tt.kev.MissingParameter()
if result != tt.expected {
t.Errorf("Expected MissingParameter() to be %v, but got %v", tt.expected, result)
}
})
}
}
func TestKamEventUnmarshalError(t *testing.T) {
validJSON := []byte(`{"key1": "value1", "key2": "value2"}`)
iJSON := []byte(`{"key1": "value1", "key2": "value2"`)
t.Run("Valid JSON", func(t *testing.T) {
_, err := NewKamEvent(validJSON, "alias", "address")
if err != nil {
t.Errorf("Expected no error for valid JSON, but got: %v", err)
}
})
t.Run("Invalid JSON", func(t *testing.T) {
_, err := NewKamEvent(iJSON, "alias", "address")
if err == nil {
t.Error("Expected error for invalid JSON, but got none")
}
})
}
func TestKamEventProcessMessageEmptyReplyY(t *testing.T) {
kevWithReplyRoute := KamEvent{
KamReplyRoute: "Route",
KamTRIndex: "index1",
KamTRLabel: "label1",
}
replyWithRoute := kevWithReplyRoute.AsKamProcessMessageEmptyReply()
if replyWithRoute.Event == "EventName" {
t.Errorf("Route, but got %s", replyWithRoute.Event)
}
kevWithoutReplyRoute := KamEvent{
KamTRIndex: "index2",
KamTRLabel: "label2",
}
replyWithoutRoute := kevWithoutReplyRoute.AsKamProcessMessageEmptyReply()
if replyWithoutRoute.Event != CGR_PROCESS_MESSAGE {
t.Errorf("Expected Event to be '%s', but got %s", CGR_PROCESS_MESSAGE, replyWithoutRoute.Event)
}
}
func TestAsKamProcessCDRReply(t *testing.T) {
cgrEv := &utils.CGREvent{}
kevWithReplyRoute := KamEvent{
KamReplyRoute: "EventName",
KamTRIndex: "index1",
KamTRLabel: "label1",
}
replyWithRoute, _ := kevWithReplyRoute.AsKamProcessCDRReply(cgrEv, nil, nil)
if replyWithRoute.Event != "EventName" {
t.Errorf("Expected Event to be 'EventName', but got %s", replyWithRoute.Event)
}
kevWithoutReplyRoute := KamEvent{
KamTRIndex: "index2",
KamTRLabel: "label2",
}
replyWithoutRoute, _ := kevWithoutReplyRoute.AsKamProcessCDRReply(cgrEv, nil, nil)
if replyWithoutRoute.Event != CGR_PROCESS_CDR {
t.Errorf("Expected Event to be '%s', but got %s", CGR_PROCESS_CDR, replyWithoutRoute.Event)
}
}
func TestAsKamProcessMessageReplyProcessStats(t *testing.T) {
kev := KamEvent{
KamTRIndex: "index1",
KamTRLabel: "label1",
}
procEvArgs := &sessions.V1ProcessMessageArgs{
ProcessStats: true,
}
tStatQueueIDs := []string{"queue1", "queue2", "queue3"}
procEvReply := &sessions.V1ProcessMessageReply{
StatQueueIDs: &tStatQueueIDs,
}
reply, _ := kev.AsKamProcessMessageReply(procEvArgs, procEvReply, nil)
if reply.Event != CGR_PROCESS_MESSAGE {
t.Errorf("Expected Event to be '%s', but got '%s'", CGR_PROCESS_MESSAGE, reply.Event)
}
if reply.TransactionIndex != "index1" {
t.Errorf("Expected TransactionIndex to be 'index1', but got '%s'", reply.TransactionIndex)
}
if reply.TransactionLabel != "label1" {
t.Errorf("Expected TransactionLabel to be 'label1', but got '%s'", reply.TransactionLabel)
}
}
func TestKamEventProcessMessageReplyProcessThresholds(t *testing.T) {
kev := KamEvent{
KamTRIndex: "index1",
KamTRLabel: "label1",
}
procEvArgs := &sessions.V1ProcessMessageArgs{
ProcessThresholds: true,
}
tThresholdIDs := []string{"threshold1", "threshold2", "threshold3"}
procEvReply := &sessions.V1ProcessMessageReply{
ThresholdIDs: &tThresholdIDs,
}
reply, _ := kev.AsKamProcessMessageReply(procEvArgs, procEvReply, nil)
if reply.Event != CGR_PROCESS_MESSAGE {
t.Errorf("Expected Event to be '%s', but got '%s'", CGR_PROCESS_MESSAGE, reply.Event)
}
if reply.TransactionIndex != "index1" {
t.Errorf("Expected TransactionIndex to be 'index1', but got '%s'", reply.TransactionIndex)
}
if reply.TransactionLabel != "label1" {
t.Errorf("Expected TransactionLabel to be 'label1', but got '%s'", reply.TransactionLabel)
}
}
func TestKamEventProcessCDRReplyError(t *testing.T) {
kev := KamEvent{
KamTRIndex: "index",
KamTRLabel: "label",
}
cgrEvWithArgDisp := &utils.CGREvent{}
rply := ""
rplyErr := errors.New("test error")
kar, err := kev.AsKamProcessCDRReply(cgrEvWithArgDisp, &rply, rplyErr)
if err != nil {
t.Errorf("expected no error, got %v", err)
}
if kar == nil {
t.Fatalf("expected kar to be non-nil")
}
if kar.TransactionIndex != "index" {
t.Errorf("expected TransactionIndex to be 'index', got %v", kar.TransactionIndex)
}
if kar.TransactionLabel != "label" {
t.Errorf("expected TransactionLabel to be 'label', got %v", kar.TransactionLabel)
}
if kar.Event != CGR_PROCESS_CDR {
t.Errorf("expected Event to be %v, got %v", CGR_PROCESS_CDR, kar.Event)
}
if kar.Error != "test error" {
t.Errorf("expected Error to be 'test error', got %v", kar.Error)
}
}
func TestAsKamProcessMessageReplyError(t *testing.T) {
kev := KamEvent{
KamTRIndex: "index",
KamTRLabel: "label",
}
procEvArgs := &sessions.V1ProcessMessageArgs{}
procEvReply := &sessions.V1ProcessMessageReply{}
rplyErr := errors.New("error")
kar, err := kev.AsKamProcessMessageReply(procEvArgs, procEvReply, rplyErr)
if err != nil {
t.Errorf("expected no error, got %v", err)
}
if kar == nil {
t.Fatalf("expected kar to be non-nil")
}
if kar.Error != "error" {
t.Errorf("expected Error to be 'error', got %v", kar.Error)
}
}
func TestV1ProcessCDRArgsError(t *testing.T) {
kev := KamEvent{}
args := kev.V1ProcessCDRArgs()
if args != nil {
t.Errorf("expected args to be nil when AsCGREvent returns an error, got %v", args)
}
}
func TestKamEventAuthReplyProcessStats(t *testing.T) {
kev := KamEvent{
KamTRIndex: "index",
KamTRLabel: "label",
}
authArgs := &sessions.V1AuthorizeArgs{
ProcessStats: true,
}
statQueueIDs := []string{"stat1", "stat2"}
authReply := &sessions.V1AuthorizeReply{
StatQueueIDs: &statQueueIDs,
}
rplyErr := error(nil)
kar, err := kev.AsKamAuthReply(authArgs, authReply, rplyErr)
if err != nil {
t.Errorf("expected no error, got %v", err)
}
if kar == nil {
t.Fatalf("expected kar to be non-nil")
}
expectedStatQueues := strings.Join(statQueueIDs, utils.FieldsSep)
if kar.StatQueues != expectedStatQueues {
t.Errorf("expected StatQueues to be '%v', got '%v'", expectedStatQueues, kar.StatQueues)
}
}
func TestKamEventAsKamAuthReplyProcessThresholds(t *testing.T) {
kev := KamEvent{
KamTRIndex: "index",
KamTRLabel: "label",
}
authArgs := &sessions.V1AuthorizeArgs{
ProcessThresholds: true,
}
thresholdIDs := []string{"threshold1", "threshold2"}
authReply := &sessions.V1AuthorizeReply{
ThresholdIDs: &thresholdIDs,
}
rplyErr := error(nil)
kar, err := kev.AsKamAuthReply(authArgs, authReply, rplyErr)
if err != nil {
t.Errorf("expected no error, got %v", err)
}
if kar == nil {
t.Fatalf("expected kar to be non-nil")
}
expectedThresholds := strings.Join(thresholdIDs, utils.FieldsSep)
if kar.Thresholds != expectedThresholds {
t.Errorf("expected Thresholds to be '%v', got '%v'", expectedThresholds, kar.Thresholds)
}
}
func TestKamEventAsKamAuthReplyAuthorizeResources(t *testing.T) {
kev := KamEvent{
KamTRIndex: "index",
KamTRLabel: "label",
}
authArgs := &sessions.V1AuthorizeArgs{
AuthorizeResources: true,
}
resourceAllocation := "resource_allocation_value"
authReply := &sessions.V1AuthorizeReply{
ResourceAllocation: &resourceAllocation,
}
rplyErr := error(nil)
kar, err := kev.AsKamAuthReply(authArgs, authReply, rplyErr)
if err != nil {
t.Errorf("expected no error, got %v", err)
}
if kar == nil {
t.Fatalf("expected kar to be non-nil")
}
if kar.ResourceAllocation != resourceAllocation {
t.Errorf("expected ResourceAllocation to be '%v', got '%v'", resourceAllocation, kar.ResourceAllocation)
}
}
func TestKamEventV1InitSessionArgsNoCGRFlags(t *testing.T) {
kev := KamEvent{}
args := kev.V1InitSessionArgs()
if args == nil {
t.Fatalf("expected args to be non-nil")
}
if !args.InitSession {
t.Error("expected InitSession to be true")
}
}
func TestKamEventAsMapStringInterfaceUsageKey(t *testing.T) {
kev := KamEvent{
"Usage": "123",
"Key": "value",
}
expected := map[string]any{
"Usage": "123s",
"Key": "value",
}
result := kev.AsMapStringInterface()
if reflect.DeepEqual(result, expected) {
t.Errorf("expected %v, got %v", expected, result)
}
}
func TestKamEventMissingParameterUnsupportedEvent(t *testing.T) {
kev := KamEvent{
EVENT: "unsupported_event",
}
if result := kev.MissingParameter(); !result {
t.Errorf("Expected true for unsupported event, got false")
}
}
func TestKamReplyString(t *testing.T) {
tests := []struct {
name string
reply KamReply
expected string
}{
{
name: "Complete data",
reply: KamReply{
Event: "event1",
TransactionIndex: "index1",
TransactionLabel: "label1",
Attributes: "attr1",
ResourceAllocation: "resource1",
MaxUsage: 100,
Routes: "route1,route2",
Thresholds: "threshold1",
StatQueues: "queue1",
Error: "none",
},
expected: `{"Event":"event1","TransactionIndex":"index1","TransactionLabel":"label1","Attributes":"attr1","ResourceAllocation":"resource1","MaxUsage":100,"Routes":"route1,route2","Thresholds":"threshold1","StatQueues":"queue1","Error":"none"}`,
},
{
name: "Empty fields",
reply: KamReply{
Event: "",
TransactionIndex: "",
TransactionLabel: "",
Attributes: "",
ResourceAllocation: "",
MaxUsage: 0,
Routes: "",
Thresholds: "",
StatQueues: "",
Error: "",
},
expected: `{"Event":"","TransactionIndex":"","TransactionLabel":"","Attributes":"","ResourceAllocation":"","MaxUsage":0,"Routes":"","Thresholds":"","StatQueues":"","Error":""}`,
},
{
name: "Zero MaxUsage",
reply: KamReply{
Event: "event2",
TransactionIndex: "index2",
TransactionLabel: "label2",
Attributes: "attr2",
ResourceAllocation: "resource2",
MaxUsage: 0,
Routes: "route3",
Thresholds: "threshold2",
StatQueues: "queue2",
Error: "error2",
},
expected: `{"Event":"event2","TransactionIndex":"index2","TransactionLabel":"label2","Attributes":"attr2","ResourceAllocation":"resource2","MaxUsage":0,"Routes":"route3","Thresholds":"threshold2","StatQueues":"queue2","Error":"error2"}`,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
actual := tt.reply.String()
if actual != tt.expected {
t.Errorf("Test %s failed: expected %s, got %s", tt.name, tt.expected, actual)
}
})
}
}
Reference in New Issue View Git Blame Copy Permalink