uppal/cgrates
1
0
Fork 1
mirror of https://github.com/cgrates/cgrates.git synced 2026-02-12 18:46:24 +05:00
Code Issues Packages Projects Releases Wiki Activity
Files
5f6faab51b2be7e8f86603a00a2c7dfa2ed8b108
cgrates/sessionmanager/smgeneric_test.go
DanB cdf8960a42 SMGeneric TestSetPassiveSession
2016-10-27 14:02:33 +02:00

345 lines
11 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 sessionmanager
import (
"reflect"
"testing"
"github.com/cgrates/cgrates/config"
"github.com/cgrates/cgrates/utils"
)
var smgCfg *config.CGRConfig
func init() {
smgCfg, _ = config.NewDefaultCGRConfig()
smgCfg.SmGenericConfig.SessionIndexes = []string{"Tenant", "Account", "Extra3", "Extra4"}
}
func TestSMGSessionIndexing(t *testing.T) {
smg := NewSMGeneric(smgCfg, nil, nil, "UTC")
smGev := SMGenericEvent{
utils.EVENT_NAME: "TEST_EVENT",
utils.TOR: "*voice",
utils.ACCID: "12345",
utils.DIRECTION: "*out",
utils.ACCOUNT: "account1",
utils.SUBJECT: "subject1",
utils.DESTINATION: "+4986517174963",
utils.CATEGORY: "call",
utils.TENANT: "cgrates.org",
utils.REQTYPE: "*prepaid",
utils.SETUP_TIME: "2015-11-09 14:21:24",
utils.ANSWER_TIME: "2015-11-09 14:22:02",
utils.USAGE: "1m23s",
utils.LastUsed: "21s",
utils.PDD: "300ms",
utils.SUPPLIER: "supplier1",
utils.DISCONNECT_CAUSE: "NORMAL_DISCONNECT",
utils.CDRHOST: "127.0.0.1",
"Extra1": "Value1",
"Extra2": 5,
"Extra3": "",
}
// Index first session
smgSession := &SMGSession{EventStart: smGev}
uuid := smGev.GetUUID()
smg.indexASession(uuid, smgSession)
eIndexes := map[string]map[string]utils.StringMap{
"Tenant": map[string]utils.StringMap{
"cgrates.org": utils.StringMap{
uuid: true,
},
},
"Account": map[string]utils.StringMap{
"account1": utils.StringMap{
uuid: true,
},
},
"Extra3": map[string]utils.StringMap{
utils.MetaEmpty: utils.StringMap{
uuid: true,
},
},
"Extra4": map[string]utils.StringMap{
utils.NOT_AVAILABLE: utils.StringMap{
uuid: true,
},
},
}
if !reflect.DeepEqual(eIndexes, smg.aSessionsIndex) {
t.Errorf("Expecting: %+v, received: %+v", eIndexes, smg.aSessionsIndex)
}
// Index seccond session
smGev2 := SMGenericEvent{
utils.EVENT_NAME: "TEST_EVENT2",
utils.ACCID: "12346",
utils.DIRECTION: "*out",
utils.ACCOUNT: "account2",
utils.DESTINATION: "+4986517174964",
utils.TENANT: "itsyscom.com",
"Extra3": "",
"Extra4": "info2",
}
uuid2 := smGev2.GetUUID()
smgSession2 := &SMGSession{EventStart: smGev2}
smg.indexASession(uuid2, smgSession2)
eIndexes = map[string]map[string]utils.StringMap{
"Tenant": map[string]utils.StringMap{
"cgrates.org": utils.StringMap{
uuid: true,
},
"itsyscom.com": utils.StringMap{
uuid2: true,
},
},
"Account": map[string]utils.StringMap{
"account1": utils.StringMap{
uuid: true,
},
"account2": utils.StringMap{
uuid2: true,
},
},
"Extra3": map[string]utils.StringMap{
utils.MetaEmpty: utils.StringMap{
uuid: true,
uuid2: true,
},
},
"Extra4": map[string]utils.StringMap{
utils.NOT_AVAILABLE: utils.StringMap{
uuid: true,
},
"info2": utils.StringMap{
uuid2: true,
},
},
}
if !reflect.DeepEqual(eIndexes, smg.aSessionsIndex) {
t.Errorf("Expecting: %+v, received: %+v", eIndexes, smg.aSessionsIndex)
}
// Unidex first session
smg.unindexASession(uuid)
eIndexes = map[string]map[string]utils.StringMap{
"Tenant": map[string]utils.StringMap{
"itsyscom.com": utils.StringMap{
uuid2: true,
},
},
"Account": map[string]utils.StringMap{
"account2": utils.StringMap{
uuid2: true,
},
},
"Extra3": map[string]utils.StringMap{
utils.MetaEmpty: utils.StringMap{
uuid2: true,
},
},
"Extra4": map[string]utils.StringMap{
"info2": utils.StringMap{
uuid2: true,
},
},
}
if !reflect.DeepEqual(eIndexes, smg.aSessionsIndex) {
t.Errorf("Expecting: %+v, received: %+v", eIndexes, smg.aSessionsIndex)
}
}
func TestSMGActiveSessions(t *testing.T) {
smg := NewSMGeneric(smgCfg, nil, nil, "UTC")
smGev1 := SMGenericEvent{
utils.EVENT_NAME: "TEST_EVENT",
utils.TOR: "*voice",
utils.ACCID: "111",
utils.DIRECTION: "*out",
utils.ACCOUNT: "account1",
utils.SUBJECT: "subject1",
utils.DESTINATION: "+4986517174963",
utils.CATEGORY: "call",
utils.TENANT: "cgrates.org",
utils.REQTYPE: "*prepaid",
utils.SETUP_TIME: "2015-11-09 14:21:24",
utils.ANSWER_TIME: "2015-11-09 14:22:02",
utils.USAGE: "1m23s",
utils.LastUsed: "21s",
utils.PDD: "300ms",
utils.SUPPLIER: "supplier1",
utils.DISCONNECT_CAUSE: "NORMAL_DISCONNECT",
utils.CDRHOST: "127.0.0.1",
"Extra1": "Value1",
"Extra2": 5,
"Extra3": "",
}
smg.recordASession(smGev1.GetUUID(), &SMGSession{EventStart: smGev1})
smGev2 := SMGenericEvent{
utils.EVENT_NAME: "TEST_EVENT",
utils.TOR: "*voice",
utils.ACCID: "222",
utils.DIRECTION: "*out",
utils.ACCOUNT: "account2",
utils.DESTINATION: "+4986517174963",
utils.CATEGORY: "call",
utils.TENANT: "itsyscom.com",
utils.REQTYPE: "*prepaid",
utils.ANSWER_TIME: "2015-11-09 14:22:02",
utils.USAGE: "1m23s",
utils.LastUsed: "21s",
utils.PDD: "300ms",
utils.SUPPLIER: "supplier2",
utils.DISCONNECT_CAUSE: "NORMAL_DISCONNECT",
utils.CDRHOST: "127.0.0.1",
"Extra1": "Value1",
"Extra3": "extra3",
}
smg.recordASession(smGev2.GetUUID(), &SMGSession{EventStart: smGev2})
if aSessions, _, err := smg.ActiveSessions(nil, false); err != nil {
t.Error(err)
} else if len(aSessions) != 2 {
t.Errorf("Received sessions: %%+v", aSessions)
}
if aSessions, _, err := smg.ActiveSessions(map[string]string{"Tenant": "itsyscom.com"}, false); err != nil {
t.Error(err)
} else if len(aSessions) != 1 {
t.Errorf("Received sessions: %%+v", aSessions)
}
if aSessions, _, err := smg.ActiveSessions(map[string]string{utils.TOR: "*voice"}, false); err != nil {
t.Error(err)
} else if len(aSessions) != 2 {
t.Errorf("Received sessions: %%+v", aSessions)
}
if aSessions, _, err := smg.ActiveSessions(map[string]string{"Extra3": utils.MetaEmpty}, false); err != nil {
t.Error(err)
} else if len(aSessions) != 1 {
t.Errorf("Received sessions: %+v", aSessions)
}
if aSessions, _, err := smg.ActiveSessions(map[string]string{utils.SUPPLIER: "supplier2"}, false); err != nil {
t.Error(err)
} else if len(aSessions) != 1 {
t.Errorf("Received sessions: %+v", aSessions)
}
}
func TestSetPassiveSession(t *testing.T) {
smg := NewSMGeneric(smgCfg, nil, nil, "UTC")
smGev := SMGenericEvent{
utils.EVENT_NAME: "TEST_EVENT",
utils.TOR: "*voice",
utils.ACCID: "12345",
utils.DIRECTION: "*out",
utils.ACCOUNT: "account1",
utils.SUBJECT: "subject1",
utils.DESTINATION: "+4986517174963",
utils.CATEGORY: "call",
utils.TENANT: "cgrates.org",
utils.REQTYPE: "*prepaid",
utils.SETUP_TIME: "2015-11-09 14:21:24",
utils.ANSWER_TIME: "2015-11-09 14:22:02",
utils.USAGE: "1m23s",
utils.LastUsed: "21s",
utils.PDD: "300ms",
utils.SUPPLIER: "supplier1",
utils.DISCONNECT_CAUSE: "NORMAL_DISCONNECT",
utils.CDRHOST: "127.0.0.1",
"Extra1": "Value1",
"Extra2": 5,
"Extra3": "",
}
// Index first session
smgSession := &SMGSession{EventStart: smGev, RunID: utils.META_DEFAULT}
if len(smg.passiveSessions) != 0 {
t.Errorf("PassiveSessions: %+v", smg.passiveSessions)
}
smg.setPassiveSession(smgSession)
if ss, hasIt := smg.passiveSessions[smGev.GetUUID()]; !hasIt || len(smg.passiveSessions) != 1 || len(ss) != 1 {
t.Errorf("PassiveSessions: %+v", smg.passiveSessions)
}
// Update session
smGev = SMGenericEvent{
utils.EVENT_NAME: "TEST_EVENT",
utils.TOR: "*voice",
utils.ACCID: "12345",
utils.DIRECTION: "*out",
utils.ACCOUNT: "account1",
utils.SUBJECT: "subject1",
utils.DESTINATION: "+4986517174963",
utils.CATEGORY: "call",
utils.TENANT: "cgrates.org",
utils.REQTYPE: "*prepaid",
utils.SETUP_TIME: "2015-11-09 14:21:24",
utils.ANSWER_TIME: "2015-11-09 14:22:02",
utils.USAGE: "2m33s",
utils.LastUsed: "21s",
utils.PDD: "300ms",
utils.SUPPLIER: "supplier1",
utils.DISCONNECT_CAUSE: "NORMAL_DISCONNECT",
utils.CDRHOST: "127.0.0.1",
"Extra1": "Value1",
"Extra2": 5,
"Extra3": "",
}
smgSession = &SMGSession{EventStart: smGev, RunID: utils.META_DEFAULT}
smg.setPassiveSession(smgSession) // Should only update in place
if ss, hasIt := smg.passiveSessions[smGev.GetUUID()]; !hasIt || len(smg.passiveSessions) != 1 || len(ss) != 1 {
t.Errorf("PassiveSessions: %+v", smg.passiveSessions)
} else if ss[0].EventStart[utils.USAGE] != "2m33s" {
t.Errorf("SMGSession.EventStart: %+v", ss[0].EventStart[utils.USAGE])
}
// Second run
smgSession = &SMGSession{EventStart: smGev, RunID: "second_test"}
smg.setPassiveSession(smgSession)
if ss, hasIt := smg.passiveSessions[smGev.GetUUID()]; !hasIt || len(smg.passiveSessions) != 1 || len(ss) != 2 {
t.Errorf("PassiveSessions: %+v", smg.passiveSessions)
}
// Update session
smGev = SMGenericEvent{
utils.EVENT_NAME: "TEST_EVENT",
utils.TOR: "*voice",
utils.ACCID: "22345",
utils.DIRECTION: "*out",
utils.ACCOUNT: "account1",
utils.SUBJECT: "subject1",
utils.DESTINATION: "+4986517174963",
utils.CATEGORY: "call",
utils.TENANT: "cgrates.org",
utils.REQTYPE: "*prepaid",
utils.SETUP_TIME: "2015-11-09 14:21:24",
utils.ANSWER_TIME: "2015-11-09 14:22:02",
utils.USAGE: "2m33s",
utils.LastUsed: "21s",
utils.PDD: "300ms",
utils.SUPPLIER: "supplier1",
utils.DISCONNECT_CAUSE: "NORMAL_DISCONNECT",
utils.CDRHOST: "127.0.0.1",
"Extra1": "Value1",
"Extra2": 5,
"Extra3": "",
}
smgSession = &SMGSession{EventStart: smGev, RunID: utils.META_DEFAULT}
smg.setPassiveSession(smgSession)
if ss, hasIt := smg.passiveSessions[smGev.GetUUID()]; !hasIt || len(smg.passiveSessions) != 2 || len(ss) != 1 {
t.Errorf("PassiveSessions: %+v", smg.passiveSessions)
} else if ss[0].EventStart[utils.USAGE] != "2m33s" {
t.Errorf("SMGSession.EventStart: %+v", ss[0].EventStart[utils.USAGE])
}
}
Reference in New Issue View Git Blame Copy Permalink