Files
cgrates/sessionmanager/sma_event_test.go

296 lines
11 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 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 sessionmanager
import (
"encoding/json"
"reflect"
"testing"
"github.com/cgrates/cgrates/utils"
)
var (
stasisStart = `{"application":"cgrates_auth","type":"StasisStart","timestamp":"2016-09-12T13:53:48.919+0200","args":["cgr_reqtype=*prepaid","cgr_destination=1003"],"channel":{"id":"1473681228.6","state":"Ring","name":"PJSIP/1001-00000004","caller":{"name":"1001","number":"1001"},"language":"en","connected":{"name":"","number":""},"accountcode":"","dialplan":{"context":"internal","exten":"1002","priority":2},"creationtime":"2016-09-12T13:53:48.918+0200"}}`
channelStateChange = `{"application":"cgrates_auth","type":"ChannelStateChange","timestamp":"2016-09-12T13:53:52.110+0200","channel":{"id":"1473681228.6","state":"Up","name":"PJSIP/1001-00000004","caller":{"name":"1001","number":"1001"},"language":"en","connected":{"name":"","number":"1002"},"accountcode":"","dialplan":{"context":"internal","exten":"1002","priority":3},"creationtime":"2016-09-12T13:53:48.918+0200"}}`
channelAnsweredDestroyed = `{"type":"ChannelDestroyed","timestamp":"2016-09-12T13:54:27.335+0200","application":"cgrates_auth","cause_txt":"Normal Clearing","channel":{"id":"1473681228.6","state":"Up","name":"PJSIP/1001-00000004","caller":{"name":"1001","number":"1001"},"language":"en","connected":{"name":"","number":"1002"},"accountcode":"","dialplan":{"context":"internal","exten":"1002","priority":3},"creationtime":"2016-09-12T13:53:48.918+0200"},"cause":16}`
channelUnansweredDestroyed = `{"type":"ChannelDestroyed","timestamp":"2016-09-12T18:00:18.121+0200","application":"cgrates_auth","cause_txt":"Normal Clearing","channel":{"id":"1473696018.2","state":"Ring","name":"PJSIP/1002-00000002","caller":{"name":"1002","number":"1002"},"language":"en","connected":{"name":"","number":""},"accountcode":"","dialplan":{"context":"internal","exten":"1002","priority":2},"creationtime":"2016-09-12T18:00:18.109+0200"},"cause":16}`
channelBusyDestroyed = `{"type":"ChannelDestroyed","timestamp":"2016-09-13T12:59:48.806+0200","application":"cgrates_auth","cause_txt":"User busy","channel":{"id":"1473764378.3","state":"Ring","name":"PJSIP/1001-00000002","caller":{"name":"1001","number":"1001"},"language":"en","connected":{"name":"","number":"1002"},"accountcode":"","dialplan":{"context":"internal","exten":"1002","priority":4},"creationtime":"2016-09-13T12:59:38.839+0200"},"cause":17}`
)
func TestSMAParseStasisArgs(t *testing.T) {
var ev map[string]interface{}
if err := json.Unmarshal([]byte(stasisStart), &ev); err != nil {
t.Error(err)
}
smaEv := NewSMAsteriskEvent(ev, "127.0.0.1")
expAppArgs := map[string]string{"cgr_reqtype": "*prepaid", "cgr_destination": "1003"}
if !reflect.DeepEqual(smaEv.cachedFields, expAppArgs) {
t.Errorf("Expecting: %+v, received: %+v", smaEv.cachedFields, expAppArgs)
}
ev = make(map[string]interface{}) // Clear previous data
if err := json.Unmarshal([]byte("{}"), &ev); err != nil {
t.Error(err)
}
smaEv = NewSMAsteriskEvent(ev, "127.0.0.1")
expAppArgs = map[string]string{}
if !reflect.DeepEqual(smaEv.cachedFields, expAppArgs) {
t.Errorf("Expecting: %+v, received: %+v", smaEv.cachedFields, expAppArgs)
}
}
func TestSMAEventType(t *testing.T) {
var ev map[string]interface{}
if err := json.Unmarshal([]byte(stasisStart), &ev); err != nil {
t.Error(err)
}
smaEv := NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.EventType() != "StasisStart" {
t.Error("Received:", smaEv.EventType())
}
ev = make(map[string]interface{}) // Clear previous data
if err := json.Unmarshal([]byte("{}"), &ev); err != nil {
t.Error(err)
}
smaEv = NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.EventType() != "" {
t.Error("Received:", smaEv.EventType())
}
}
func TestSMAEventChannelID(t *testing.T) {
var ev map[string]interface{}
if err := json.Unmarshal([]byte(stasisStart), &ev); err != nil {
t.Error(err)
}
smaEv := NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.ChannelID() != "1473681228.6" {
t.Error("Received:", smaEv.ChannelID())
}
ev = make(map[string]interface{}) // Clear previous data
if err := json.Unmarshal([]byte("{}"), &ev); err != nil {
t.Error(err)
}
smaEv = NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.ChannelID() != "" {
t.Error("Received:", smaEv.ChannelID())
}
}
func TestSMAEventOriginatorIP(t *testing.T) {
var ev map[string]interface{}
if err := json.Unmarshal([]byte(stasisStart), &ev); err != nil {
t.Error(err)
}
smaEv := NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.OriginatorIP() != "127.0.0.1" {
t.Error("Received:", smaEv.OriginatorIP())
}
}
func TestSMAEventAccount(t *testing.T) {
var ev map[string]interface{}
if err := json.Unmarshal([]byte(stasisStart), &ev); err != nil {
t.Error(err)
}
smaEv := NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.Account() != "1001" {
t.Error("Received:", smaEv.Account())
}
ev = make(map[string]interface{}) // Clear previous data
if err := json.Unmarshal([]byte("{}"), &ev); err != nil {
t.Error(err)
}
smaEv = NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.Account() != "" {
t.Error("Received:", smaEv.Account())
}
}
func TestSMAEventDestination(t *testing.T) {
var ev map[string]interface{}
if err := json.Unmarshal([]byte(stasisStart), &ev); err != nil {
t.Error(err)
}
smaEv := NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.Destination() != "1003" {
t.Error("Received:", smaEv.Destination())
}
ev = make(map[string]interface{}) // Clear previous data
if err := json.Unmarshal([]byte("{}"), &ev); err != nil {
t.Error(err)
}
smaEv = NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.Destination() != "" {
t.Error("Received:", smaEv.Destination())
}
}
func TestSMAEventTimestamp(t *testing.T) {
var ev map[string]interface{}
if err := json.Unmarshal([]byte(stasisStart), &ev); err != nil {
t.Error(err)
}
smaEv := NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.Timestamp() != "2016-09-12T13:53:48.919+0200" {
t.Error("Received:", smaEv.Timestamp())
}
ev = make(map[string]interface{}) // Clear previous data
if err := json.Unmarshal([]byte("{}"), &ev); err != nil {
t.Error(err)
}
smaEv = NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.Timestamp() != "" {
t.Error("Received:", smaEv.Timestamp())
}
}
func TestSMAEventRequestType(t *testing.T) {
var ev map[string]interface{}
if err := json.Unmarshal([]byte(stasisStart), &ev); err != nil {
t.Error(err)
}
smaEv := NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.RequestType() != "*prepaid" {
t.Error("Received:", smaEv.RequestType())
}
ev = make(map[string]interface{}) // Clear previous data
smaEv = NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.RequestType() != "" {
t.Error("Received:", smaEv.RequestType())
}
}
func TestSMAEventTenant(t *testing.T) {
var ev map[string]interface{}
if err := json.Unmarshal([]byte(stasisStart), &ev); err != nil {
t.Error(err)
}
smaEv := NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.Tenant() != "" {
t.Error("Received:", smaEv.Tenant())
}
ev = map[string]interface{}{"args": []interface{}{"cgr_tenant=cgrates.org"}} // Clear previous data
smaEv = NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.Tenant() != "cgrates.org" {
t.Error("Received:", smaEv.Tenant())
}
}
func TestSMAEventCategory(t *testing.T) {
var ev map[string]interface{}
if err := json.Unmarshal([]byte(stasisStart), &ev); err != nil {
t.Error(err)
}
smaEv := NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.Category() != "" {
t.Error("Received:", smaEv.Category())
}
ev = map[string]interface{}{"args": []interface{}{"cgr_category=premium_call"}} // Clear previous data
smaEv = NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.Category() != "premium_call" {
t.Error("Received:", smaEv.Category())
}
}
func TestSMAEventSubject(t *testing.T) {
var ev map[string]interface{}
if err := json.Unmarshal([]byte(stasisStart), &ev); err != nil {
t.Error(err)
}
smaEv := NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.Subject() != "" {
t.Error("Received:", smaEv.Subject())
}
ev = map[string]interface{}{"args": []interface{}{"cgr_subject=dan"}} // Clear previous data
smaEv = NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.Subject() != "dan" {
t.Error("Received:", smaEv.Subject())
}
}
func TestSMAEventPDD(t *testing.T) {
var ev map[string]interface{}
if err := json.Unmarshal([]byte(stasisStart), &ev); err != nil {
t.Error(err)
}
smaEv := NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.PDD() != "" {
t.Error("Received:", smaEv.PDD())
}
ev = map[string]interface{}{"args": []interface{}{"cgr_pdd=2.1"}} // Clear previous data
smaEv = NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.PDD() != "2.1" {
t.Error("Received:", smaEv.PDD())
}
}
func TestSMAEventSupplier(t *testing.T) {
var ev map[string]interface{}
if err := json.Unmarshal([]byte(stasisStart), &ev); err != nil {
t.Error(err)
}
smaEv := NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.Supplier() != "" {
t.Error("Received:", smaEv.Supplier())
}
ev = map[string]interface{}{"args": []interface{}{"cgr_supplier=supplier1"}} // Clear previous data
smaEv = NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.Supplier() != "supplier1" {
t.Error("Received:", smaEv.Supplier())
}
}
func TestSMAEventDisconnectCause(t *testing.T) {
var ev map[string]interface{}
if err := json.Unmarshal([]byte(stasisStart), &ev); err != nil {
t.Error(err)
}
smaEv := NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.DisconnectCause() != "" {
t.Error("Received:", smaEv.DisconnectCause())
}
ev = map[string]interface{}{"args": []interface{}{"cgr_disconnectcause=NORMAL_DISCONNECT"}} // Clear previous data
smaEv = NewSMAsteriskEvent(ev, "127.0.0.1")
if smaEv.DisconnectCause() != "NORMAL_DISCONNECT" {
t.Error("Received:", smaEv.DisconnectCause())
}
}
func TestSMAEventAsSMGenericSessionStart(t *testing.T) {
var ev map[string]interface{}
if err := json.Unmarshal([]byte(stasisStart), &ev); err != nil {
t.Error(err)
}
eSMGEv := SMGenericEvent{
utils.EVENT_NAME: utils.CGR_SESSION_START,
utils.ACCID: "1473681228.6",
utils.REQTYPE: "*prepaid",
utils.CDRHOST: "127.0.0.1",
utils.ACCOUNT: "1001",
utils.DESTINATION: "1003",
utils.SETUP_TIME: "2016-09-12T13:53:48.919+0200",
}
smaEv := NewSMAsteriskEvent(ev, "127.0.0.1")
if smgEv, err := smaEv.AsSMGenericSessionStart(); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(eSMGEv, smgEv) {
t.Errorf("Expecting: %+v, received: %+v", eSMGEv, smgEv)
}
}