mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
3062113a61
The rpcclient constructor could not see the centralized Encoding flag because it didn't have the necessary build constraints. Added the constraints in lib_test.go files where it wasn't alone. In all the other cases, it was moved to the first file where it was needed.
410 lines
14 KiB
Go
410 lines
14 KiB
Go
//go:build integration
|
|
// +build integration
|
|
|
|
/*
|
|
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 sessions
|
|
|
|
import (
|
|
"errors"
|
|
"path"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/cgrates/birpc"
|
|
"github.com/cgrates/birpc/context"
|
|
"github.com/cgrates/birpc/jsonrpc"
|
|
"github.com/cgrates/cgrates/config"
|
|
"github.com/cgrates/cgrates/engine"
|
|
"github.com/cgrates/cgrates/utils"
|
|
)
|
|
|
|
var (
|
|
sessionsBiRPCCfgPath string
|
|
sessionsBiRPCCfgDIR string
|
|
sessionsBiRPCCfg *config.CGRConfig
|
|
sessionsBiRPC *birpc.BirpcClient
|
|
disconnectEvChan = make(chan utils.CGREvent, 1)
|
|
err error
|
|
sessionsTests = []func(t *testing.T){
|
|
testSessionsBiRPCInitCfg,
|
|
testSessionsBiRPCResetDataDb,
|
|
testSessionsBiRPCResetStorDb,
|
|
testSessionsBiRPCStartEngine,
|
|
testSessionsBiRPCApierRpcConn,
|
|
testSessionsBiRPCTPFromFolder,
|
|
testSessionsBiRPCSessionAutomaticDisconnects,
|
|
testSessionsBiRPCSessionOriginatorTerminate,
|
|
testSessionsBiRPCStopCgrEngine,
|
|
}
|
|
)
|
|
|
|
// Tests starts here
|
|
func TestSessionsBiRPC(t *testing.T) {
|
|
switch *utils.DBType {
|
|
case utils.MetaInternal:
|
|
sessionsBiRPCCfgDIR = "smg_automatic_debits_internal"
|
|
case utils.MetaMySQL:
|
|
sessionsBiRPCCfgDIR = "smg_automatic_debits_mysql"
|
|
case utils.MetaMongo:
|
|
sessionsBiRPCCfgDIR = "smg_automatic_debits_mongo"
|
|
case utils.MetaPostgres:
|
|
t.SkipNow()
|
|
default:
|
|
t.Fatal("Unknown Database type")
|
|
}
|
|
for _, stest := range sessionsTests {
|
|
t.Run(sessionsBiRPCCfgDIR, stest)
|
|
}
|
|
}
|
|
|
|
type smock struct{}
|
|
|
|
func (*smock) DisconnectSession(ctx *context.Context,
|
|
args utils.CGREvent, reply *string) error {
|
|
disconnectEvChan <- args
|
|
*reply = utils.OK
|
|
return nil
|
|
}
|
|
|
|
func testSessionsBiRPCInitCfg(t *testing.T) {
|
|
sessionsBiRPCCfgPath = path.Join(*utils.DataDir, "conf", "samples", sessionsBiRPCCfgDIR)
|
|
// Init config first
|
|
sessionsBiRPCCfg, err = config.NewCGRConfigFromPath(sessionsBiRPCCfgPath)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
// Remove data in both rating and accounting db
|
|
func testSessionsBiRPCResetDataDb(t *testing.T) {
|
|
if err := engine.InitDataDb(sessionsBiRPCCfg); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
// Wipe out the cdr database
|
|
func testSessionsBiRPCResetStorDb(t *testing.T) {
|
|
if err := engine.InitStorDb(sessionsBiRPCCfg); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
// Start CGR Engine
|
|
func testSessionsBiRPCStartEngine(t *testing.T) {
|
|
if _, err := engine.StopStartEngine(sessionsBiRPCCfgPath, *utils.WaitRater); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
// Connect rpc client to rater
|
|
func testSessionsBiRPCApierRpcConn(t *testing.T) {
|
|
srv, err := birpc.NewService(new(smock), utils.AgentV1, true)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
dummyClnt, err := utils.NewBiJSONrpcClient(sessionsBiRPCCfg.SessionSCfg().ListenBijson,
|
|
srv)
|
|
if err != nil { // First attempt is to make sure multiple clients are supported
|
|
t.Fatal(err)
|
|
}
|
|
if sessionsBiRPC, err = utils.NewBiJSONrpcClient(sessionsBiRPCCfg.SessionSCfg().ListenBijson,
|
|
srv); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if sessionsRPC, err = newRPCClient(sessionsBiRPCCfg.ListenCfg()); err != nil { // Connect also simple RPC so we can check accounts and such
|
|
t.Fatal(err)
|
|
}
|
|
dummyClnt.Close() // close so we don't get EOF error when disconnecting server
|
|
}
|
|
|
|
// Load the tariff plan, creating accounts and their balances
|
|
func testSessionsBiRPCTPFromFolder(t *testing.T) {
|
|
attrs := &utils.AttrLoadTpFromFolder{FolderPath: path.Join(*utils.DataDir, "tariffplans", "oldtutorial")}
|
|
var loadInst utils.LoadInstance
|
|
if err := sessionsRPC.Call(context.Background(), utils.APIerSv2LoadTariffPlanFromFolder, attrs, &loadInst); err != nil {
|
|
t.Error(err)
|
|
}
|
|
time.Sleep(time.Duration(*utils.WaitRater) * time.Millisecond) // Give time for scheduler to execute topups
|
|
}
|
|
|
|
func testSessionsBiRPCSessionAutomaticDisconnects(t *testing.T) {
|
|
// Create a balance with 1 second inside and rating increments of 1ms (to be compatible with debit interval)
|
|
attrSetBalance := utils.AttrSetBalance{Tenant: "cgrates.org",
|
|
Account: "TestSessionsBiRPCSessionAutomaticDisconnects",
|
|
BalanceType: utils.MetaVoice,
|
|
Value: 0.01 * float64(time.Second),
|
|
Balance: map[string]any{
|
|
utils.ID: "TestSessionsBiRPCSessionAutomaticDisconnects",
|
|
utils.RatingSubject: "*zero1ms",
|
|
},
|
|
}
|
|
var reply string
|
|
if err := sessionsRPC.Call(context.Background(), utils.APIerSv2SetBalance, attrSetBalance, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if reply != utils.OK {
|
|
t.Errorf("Received: %s", reply)
|
|
}
|
|
var acnt *engine.Account
|
|
attrGetAcnt := &utils.AttrGetAccount{
|
|
Tenant: attrSetBalance.Tenant,
|
|
Account: attrSetBalance.Account,
|
|
}
|
|
eAcntVal := 0.01 * float64(time.Second)
|
|
if err := sessionsRPC.Call(context.Background(), utils.APIerSv2GetAccount, attrGetAcnt, &acnt); err != nil {
|
|
t.Error(err)
|
|
} else if acnt.BalanceMap[utils.MetaVoice].GetTotalValue() != eAcntVal {
|
|
t.Errorf("Expecting: %f, received: %f", eAcntVal,
|
|
acnt.BalanceMap[utils.MetaVoice].GetTotalValue())
|
|
}
|
|
|
|
initArgs := &V1InitSessionArgs{
|
|
InitSession: true,
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: "cgrates.org",
|
|
ID: "TestSessionsBiRPCSessionAutomaticDisconnects",
|
|
Event: map[string]any{
|
|
utils.EventName: "TEST_EVENT",
|
|
utils.ToR: utils.MetaVoice,
|
|
utils.OriginID: "123451",
|
|
utils.AccountField: attrSetBalance.Account,
|
|
utils.Subject: attrSetBalance.Account,
|
|
utils.Destination: "1004",
|
|
utils.Category: "call",
|
|
utils.Tenant: attrSetBalance.Tenant,
|
|
utils.RequestType: utils.MetaPrepaid,
|
|
utils.SetupTime: time.Date(2016, time.January, 5, 18, 30, 59, 0, time.UTC),
|
|
utils.AnswerTime: time.Date(2016, time.January, 5, 18, 31, 05, 0, time.UTC),
|
|
utils.Usage: 200 * time.Millisecond,
|
|
},
|
|
},
|
|
}
|
|
|
|
var initRpl *V1InitSessionReply
|
|
if err := sessionsBiRPC.Call(context.Background(), utils.SessionSv1InitiateSession,
|
|
initArgs, &initRpl); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
time.Sleep(10 * time.Millisecond) // give some time to allow the session to be created
|
|
expMaxUsage := 3 * time.Hour // MaxCallDuration from config
|
|
if initRpl.MaxUsage == nil || *initRpl.MaxUsage != expMaxUsage {
|
|
t.Errorf("Expecting : %+v, received: %+v", expMaxUsage, initRpl.MaxUsage)
|
|
}
|
|
// Make sure we are receiving a disconnect event
|
|
select {
|
|
case <-time.After(100 * time.Millisecond):
|
|
t.Error("Did not receive disconnect event")
|
|
case disconnectEv := <-disconnectEvChan:
|
|
if engine.NewMapEvent(disconnectEv.Event).GetStringIgnoreErrors(utils.OriginID) != initArgs.CGREvent.Event[utils.OriginID] {
|
|
t.Errorf("Unexpected event received: %+v", disconnectEv)
|
|
}
|
|
initArgs.CGREvent.Event[utils.Usage] = disconnectEv.Event[utils.Usage]
|
|
}
|
|
termArgs := &V1TerminateSessionArgs{
|
|
TerminateSession: true,
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: "cgrates.org",
|
|
ID: "TestSessionsDataLastUsedData",
|
|
Event: map[string]any{
|
|
utils.EventName: "TEST_EVENT",
|
|
utils.ToR: utils.MetaVoice,
|
|
utils.OriginID: "123451",
|
|
utils.AccountField: attrSetBalance.Account,
|
|
utils.Subject: attrSetBalance.Account,
|
|
utils.Destination: "1004",
|
|
utils.Category: "call",
|
|
utils.Tenant: attrSetBalance.Tenant,
|
|
utils.RequestType: utils.MetaPrepaid,
|
|
utils.SetupTime: time.Date(2016, time.January, 5, 18, 30, 59, 0, time.UTC),
|
|
utils.AnswerTime: time.Date(2016, time.January, 5, 18, 31, 05, 0, time.UTC),
|
|
utils.Usage: initArgs.CGREvent.Event[utils.Usage],
|
|
},
|
|
},
|
|
}
|
|
|
|
var rpl string
|
|
if err := sessionsBiRPC.Call(context.Background(), utils.SessionSv1TerminateSession, termArgs, &rpl); err != nil || rpl != utils.OK {
|
|
t.Error(err)
|
|
}
|
|
time.Sleep(100 * time.Millisecond) // Give time for debits to occur
|
|
if err := sessionsRPC.Call(context.Background(), utils.APIerSv2GetAccount, attrGetAcnt, &acnt); err != nil {
|
|
t.Error(err)
|
|
} else if acnt.BalanceMap[utils.MetaVoice].GetTotalValue() != 0 {
|
|
t.Errorf("Balance should be empty, have: %f", acnt.BalanceMap[utils.MetaVoice].GetTotalValue())
|
|
}
|
|
if err := sessionsBiRPC.Call(context.Background(), utils.SessionSv1ProcessCDR, termArgs.CGREvent, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if reply != utils.OK {
|
|
t.Errorf("Received reply: %s", reply)
|
|
}
|
|
time.Sleep(100 * time.Millisecond)
|
|
var cdrs []*engine.ExternalCDR
|
|
req := utils.RPCCDRsFilter{RunIDs: []string{utils.MetaDefault},
|
|
DestinationPrefixes: []string{"1004"}}
|
|
if err := sessionsRPC.Call(context.Background(), utils.APIerSv2GetCDRs, &req, &cdrs); err != nil {
|
|
t.Error("Unexpected error: ", err.Error())
|
|
} else if len(cdrs) != 1 {
|
|
t.Error("Unexpected number of CDRs returned: ", len(cdrs))
|
|
} else {
|
|
if cdrs[0].Usage != "10ms" {
|
|
t.Errorf("Unexpected CDR Usage received, cdr: %v %+v ", cdrs[0].Usage, cdrs[0])
|
|
} else if cdrs[0].CostSource != utils.MetaSessionS {
|
|
t.Errorf("Unexpected CDR CostSource received, cdr: %v %+v ", cdrs[0].CostSource, cdrs[0])
|
|
}
|
|
}
|
|
}
|
|
|
|
func testSessionsBiRPCSessionOriginatorTerminate(t *testing.T) {
|
|
attrSetBalance := utils.AttrSetBalance{
|
|
Tenant: "cgrates.org",
|
|
Account: "TestSessionsBiRPCSessionOriginatorTerminate",
|
|
BalanceType: utils.MetaVoice,
|
|
Value: float64(time.Second),
|
|
Balance: map[string]any{
|
|
utils.ID: "TestSessionsBiRPCSessionOriginatorTerminate",
|
|
utils.RatingSubject: "*zero1ms",
|
|
},
|
|
}
|
|
var reply string
|
|
if err := sessionsRPC.Call(context.Background(), utils.APIerSv2SetBalance, attrSetBalance, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if reply != utils.OK {
|
|
t.Errorf("Received: %s", reply)
|
|
}
|
|
var acnt *engine.Account
|
|
attrGetAcnt := &utils.AttrGetAccount{Tenant: attrSetBalance.Tenant, Account: attrSetBalance.Account}
|
|
eAcntVal := 1.0 * float64(time.Second)
|
|
if err := sessionsRPC.Call(context.Background(), utils.APIerSv2GetAccount, attrGetAcnt, &acnt); err != nil {
|
|
t.Error(err)
|
|
} else if acnt.BalanceMap[utils.MetaVoice].GetTotalValue() != eAcntVal {
|
|
t.Errorf("Expecting: %f, received: %f", eAcntVal, acnt.BalanceMap[utils.MetaVoice].GetTotalValue())
|
|
}
|
|
|
|
initArgs := &V1InitSessionArgs{
|
|
InitSession: true,
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: "cgrates.org",
|
|
ID: "TestSessionsBiRPCSessionOriginatorTerminate",
|
|
Event: map[string]any{
|
|
utils.EventName: "TEST_EVENT",
|
|
utils.ToR: utils.MetaVoice,
|
|
utils.OriginID: "123452",
|
|
utils.AccountField: attrSetBalance.Account,
|
|
utils.Subject: attrSetBalance.Account,
|
|
utils.Destination: "1005",
|
|
utils.Category: "call",
|
|
utils.Tenant: attrSetBalance.Tenant,
|
|
utils.RequestType: utils.MetaPrepaid,
|
|
utils.SetupTime: time.Date(2016, time.January, 5, 18, 30, 59, 0, time.UTC),
|
|
utils.AnswerTime: time.Date(2016, time.January, 5, 18, 31, 05, 0, time.UTC),
|
|
utils.Usage: 200 * time.Millisecond,
|
|
},
|
|
},
|
|
}
|
|
|
|
var initRpl *V1InitSessionReply
|
|
if err := sessionsBiRPC.Call(context.Background(), utils.SessionSv1InitiateSession,
|
|
initArgs, &initRpl); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
expMaxUsage := 3 * time.Hour // MaxCallDuration from config
|
|
if initRpl.MaxUsage == nil || *initRpl.MaxUsage != expMaxUsage {
|
|
t.Errorf("Expecting : %+v, received: %+v", expMaxUsage, initRpl.MaxUsage)
|
|
}
|
|
|
|
time.Sleep(10 * time.Millisecond) // Give time for debits to occur
|
|
|
|
termArgs := &V1TerminateSessionArgs{
|
|
TerminateSession: true,
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: "cgrates.org",
|
|
ID: "TestSessionsBiRPCSessionOriginatorTerminate",
|
|
Event: map[string]any{
|
|
utils.EventName: "TEST_EVENT",
|
|
utils.ToR: utils.MetaVoice,
|
|
utils.OriginID: "123452",
|
|
utils.AccountField: attrSetBalance.Account,
|
|
utils.Subject: attrSetBalance.Account,
|
|
utils.Destination: "1005",
|
|
utils.Category: "call",
|
|
utils.Tenant: attrSetBalance.Tenant,
|
|
utils.RequestType: utils.MetaPrepaid,
|
|
utils.SetupTime: time.Date(2016, time.January, 5, 18, 30, 59, 0, time.UTC),
|
|
utils.AnswerTime: time.Date(2016, time.January, 5, 18, 31, 05, 0, time.UTC),
|
|
utils.Usage: 7 * time.Millisecond,
|
|
},
|
|
},
|
|
}
|
|
|
|
var rpl string
|
|
if err := sessionsBiRPC.Call(context.Background(), utils.SessionSv1TerminateSession, termArgs, &rpl); err != nil || rpl != utils.OK {
|
|
t.Error(err)
|
|
}
|
|
|
|
time.Sleep(50 * time.Millisecond) // Give time for debits to occur
|
|
if err := sessionsRPC.Call(context.Background(), utils.APIerSv2GetAccount, attrGetAcnt, &acnt); err != nil {
|
|
t.Error(err)
|
|
} else if acnt.BalanceMap[utils.MetaVoice].GetTotalValue() > 0.995*float64(time.Second) { // FixMe: should be not 0.93?
|
|
t.Errorf("Balance value: %f", acnt.BalanceMap[utils.MetaVoice].GetTotalValue())
|
|
}
|
|
|
|
if err := sessionsRPC.Call(context.Background(), utils.SessionSv1ProcessCDR, termArgs.CGREvent, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if reply != utils.OK {
|
|
t.Errorf("Received reply: %s", reply)
|
|
}
|
|
time.Sleep(10 * time.Millisecond)
|
|
|
|
var cdrs []*engine.ExternalCDR
|
|
req := utils.RPCCDRsFilter{RunIDs: []string{utils.MetaDefault},
|
|
DestinationPrefixes: []string{"1005"}}
|
|
if err := sessionsRPC.Call(context.Background(), utils.APIerSv2GetCDRs, &req, &cdrs); err != nil {
|
|
t.Error("Unexpected error: ", err.Error())
|
|
} else if len(cdrs) != 1 {
|
|
t.Error("Unexpected number of CDRs returned: ", len(cdrs))
|
|
} else {
|
|
if cdrs[0].Usage != "7ms" {
|
|
t.Errorf("Unexpected CDR Usage received, cdr: %v %+v ", cdrs[0].Usage, cdrs[0])
|
|
} else if cdrs[0].CostSource != utils.MetaSessionS {
|
|
t.Errorf("Unexpected CDR CostSource received, cdr: %v %+v ", cdrs[0].CostSource, cdrs[0])
|
|
}
|
|
}
|
|
}
|
|
|
|
func testSessionsBiRPCStopCgrEngine(t *testing.T) {
|
|
if err := sessionsBiRPC.Close(); err != nil { // Close the connection so we don't get EOF warnings from client
|
|
t.Error(err)
|
|
}
|
|
if err := engine.KillEngine(*utils.WaitRater); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func newRPCClient(cfg *config.ListenCfg) (c *birpc.Client, err error) {
|
|
switch *utils.Encoding {
|
|
case utils.MetaJSON:
|
|
return jsonrpc.Dial(utils.TCP, cfg.RPCJSONListen)
|
|
case utils.MetaGOB:
|
|
return birpc.Dial(utils.TCP, cfg.RPCGOBListen)
|
|
default:
|
|
return nil, errors.New("UNSUPPORTED_RPC")
|
|
}
|
|
}
|