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

Add unit tests for UpdateCGREvent method

Browse Source
This commit is contained in:
ionutboangiu
2024-12-11 10:49:51 +02:00
committed by Dan Christian Bogos
parent 9db1fac75f
commit 4746a23c24
1 changed files with 134 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

134
agents/asterisk_event_test.go
View File

@@ -25,6 +25,7 @@ import (
"github.com/cgrates/cgrates/config"
"github.com/cgrates/cgrates/sessions"
"github.com/cgrates/cgrates/utils"
"github.com/google/go-cmp/cmp"
)
var (
@@ -448,3 +449,136 @@ func TestSMAEventV1TerminateSessionArgs(t *testing.T) {
t.Errorf("Expecting: %+v, received: %+v", utils.ToJSON(exp2), utils.ToJSON(rcv))
}
}
func TestAsteriskEventUpdateCGREventChannelStateChange(t *testing.T) {
var ariEv map[string]any
if err := json.Unmarshal([]byte(channelStateChange), &ariEv); err != nil {
t.Fatal(err)
}
smaEv := NewSMAsteriskEvent(ariEv, "127.0.0.1", "")
cgrEv := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "99b61ee",
Event: map[string]any{
utils.Account: "1001",
utils.Destination: "1002",
utils.EVENT_NAME: "SMA_AUTHORIZATION",
utils.OriginHost: "127.0.0.1",
utils.OriginID: "1733904242.0",
utils.RequestType: "*prepaid",
utils.SetupTime: "2024-12-11T10:04:02.180+0200",
utils.Source: "AsteriskAgent",
utils.CGRFlags: "*accounts,*attributes,*resources,*stats,*suppliers,*thresholds",
},
}
alterableFields := []string{
"channel.name:channel_name",
"channel.non_existing_field",
"channel.caller.number:channel_caller_number",
"channel.connected.number:channel_connected_number",
"channel.dialplan.exten:extension",
"channel.dialplan.priority",
}
want := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "99b61ee",
Event: map[string]any{
utils.Account: "1001",
utils.Destination: "1002",
utils.EVENT_NAME: "SMA_SESSION_START", // updated
utils.OriginHost: "127.0.0.1",
utils.OriginID: "1733904242.0",
utils.RequestType: "*prepaid",
utils.SetupTime: "2024-12-11T10:04:02.180+0200",
utils.Source: "AsteriskAgent",
utils.CGRFlags: "*accounts,*attributes,*resources,*stats,*suppliers,*thresholds",
// new fields
utils.AnswerTime: "2016-09-12T13:53:52.110+0200",
"channel_name": "PJSIP/1001-00000004",
"channel_caller_number": "1001",
"channel_connected_number": "1002",
"extension": "1002",
"priority": "3",
},
}
if err := smaEv.UpdateCGREvent(cgrEv, alterableFields); err != nil {
t.Fatalf("UpdateCGREvent unexpected err: %v", err)
}
if diff := cmp.Diff(cgrEv, want); diff != "" {
t.Errorf("UpdateCGREvent: event was not updated as expected (-want +got): \n%s", diff)
}
}
func TestAsteriskEventUpdateCGREventChannelDestroyed(t *testing.T) {
var ariEv map[string]any
if err := json.Unmarshal([]byte(channelAnsweredDestroyed), &ariEv); err != nil {
t.Fatal(err)
}
smaEv := NewSMAsteriskEvent(ariEv, "127.0.0.1", "")
cgrEv := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "99b61ee",
Event: map[string]any{
utils.Account: "1001",
utils.Destination: "1002",
utils.EVENT_NAME: "SMA_SESSION_START",
utils.OriginHost: "127.0.0.1",
utils.OriginID: "1733904242.0",
utils.RequestType: "*prepaid",
utils.SetupTime: "2024-12-11T10:04:02.180+0200",
utils.Source: "AsteriskAgent",
utils.CGRFlags: "*accounts,*attributes,*resources,*stats,*suppliers,*thresholds",
// added by ChannelStateChange handler
utils.AnswerTime: "2016-09-12T13:53:52.110+0200",
"channel_name": "PJSIP/1001-00000004",
"channel_caller_number": "1001",
"channel_connected_number": "1002",
"extension": "1002",
"priority": "3",
},
}
alterableFields := []string{
"channel.name:channel_name",
"channel.non_existing_field",
"channel.caller.number:channel_caller_number",
"channel.connected.number:channel_connected_number",
"channel.dialplan.exten:extension",
"channel.dialplan.priority",
}
want := &utils.CGREvent{
Tenant: "cgrates.org",
ID: "99b61ee",
Event: map[string]any{
utils.Account: "1001",
utils.Destination: "1002",
utils.EVENT_NAME: "SMA_SESSION_TERMINATE", // updated
utils.OriginHost: "127.0.0.1",
utils.OriginID: "1733904242.0",
utils.RequestType: "*prepaid",
utils.SetupTime: "2024-12-11T10:04:02.180+0200",
utils.Source: "AsteriskAgent",
utils.CGRFlags: "*accounts,*attributes,*resources,*stats,*suppliers,*thresholds",
utils.AnswerTime: "2016-09-12T13:53:52.110+0200",
"channel_name": "PJSIP/1001-00000004",
"channel_caller_number": "1001",
"channel_connected_number": "1002",
"extension": "1002",
"priority": "3",
// new fields
utils.Usage: "35.225s",
utils.DISCONNECT_CAUSE: "Normal Clearing",
},
}
if err := smaEv.UpdateCGREvent(cgrEv, alterableFields); err != nil {
t.Fatalf("UpdateCGREvent unexpected err: %v", err)
}
if diff := cmp.Diff(cgrEv, want); diff != "" {
t.Errorf("UpdateCGREvent: event was not updated as expected (-want +got): \n%s", diff)
}
}