mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
358 lines
12 KiB
Go
358 lines
12 KiB
Go
//go:build performance
|
|
// +build performance
|
|
|
|
/*
|
|
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 Affero 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 Affero General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Affero General Public License
|
|
along with this program. If not, see <https://www.gnu.org/licenses/>
|
|
*/
|
|
package general_tests
|
|
|
|
// import (
|
|
// "flag"
|
|
// "fmt"
|
|
// "path"
|
|
// "sync"
|
|
// "testing"
|
|
// "time"
|
|
|
|
// "github.com/cgrates/birpc"
|
|
// "github.com/cgrates/birpc/context"
|
|
// v1 "github.com/cgrates/cgrates/apier/v1"
|
|
// v2 "github.com/cgrates/cgrates/apier/v2"
|
|
// "github.com/cgrates/cgrates/config"
|
|
// "github.com/cgrates/cgrates/engine"
|
|
// "github.com/cgrates/cgrates/sessions"
|
|
// "github.com/cgrates/cgrates/utils"
|
|
// )
|
|
|
|
// var (
|
|
// sCncrCfgDIR, sCncrCfgPath string
|
|
// sCncrCfg *config.CGRConfig
|
|
// sCncrRPC *birpc.Client
|
|
|
|
// sCncrSessions = flag.Int("sessions", 100000, "maximum concurrent sessions created")
|
|
// sCncrCps = flag.Int("cps", 50000, "maximum requests per second sent out")
|
|
|
|
// cpsPool = make(chan struct{}, *sCncrCps)
|
|
// acntIDs = make(chan string, 1)
|
|
// wg sync.WaitGroup
|
|
// )
|
|
|
|
// // Tests starting here
|
|
// func TestSCncrInternal(t *testing.T) {
|
|
// sCncrCfgDIR = "sessinternal"
|
|
// for _, tst := range sTestsSCncrIT {
|
|
// t.Run("InternalConn", tst)
|
|
// }
|
|
// }
|
|
|
|
// // Tests starting here
|
|
// func TestSCncrJSON(t *testing.T) {
|
|
// sCncrCfgDIR = "sessintjson"
|
|
// for _, tst := range sTestsSCncrIT {
|
|
// t.Run("JSONConn", tst)
|
|
// }
|
|
// }
|
|
|
|
// // subtests to be executed
|
|
// var sTestsSCncrIT = []func(t *testing.T){
|
|
// testSCncrInitConfig,
|
|
// testSCncrInitDataDB,
|
|
// testSCncrInitStorDB,
|
|
// testSCncrStartEngine,
|
|
// testSCncrRPCConn,
|
|
// testSCncrLoadTP,
|
|
// testSCncrRunSessions,
|
|
// testSCncrKillEngine,
|
|
// }
|
|
|
|
// func testSCncrInitConfig(t *testing.T) {
|
|
// sCncrCfgPath = path.Join(*utils.DataDir, "conf", "samples", sCncrCfgDIR)
|
|
// if sCncrCfg, err = config.NewCGRConfigFromPath(sCncrCfgPath); err != nil {
|
|
// t.Fatal(err)
|
|
// }
|
|
// }
|
|
|
|
// func testSCncrInitDataDB(t *testing.T) {
|
|
// if err := engine.InitDataDb(sCncrCfg); err != nil {
|
|
// t.Fatal(err)
|
|
// }
|
|
// }
|
|
|
|
// // InitDb so we can rely on count
|
|
// func testSCncrInitStorDB(t *testing.T) {
|
|
// if err := engine.InitStorDb(sCncrCfg); err != nil {
|
|
// t.Fatal(err)
|
|
// }
|
|
// }
|
|
|
|
// func testSCncrStartEngine(t *testing.T) {
|
|
// if _, err := engine.StopStartEngine(sCncrCfgPath, *utils.WaitRater); err != nil {
|
|
// t.Fatal(err)
|
|
// }
|
|
// }
|
|
|
|
// // Connect rpc client to rater
|
|
// func testSCncrRPCConn(t *testing.T) {
|
|
// var err error
|
|
// sCncrRPC, err = newRPCClient(sCncrCfg.ListenCfg()) // We connect over JSON so we can also troubleshoot if needed
|
|
// if err != nil {
|
|
// t.Fatal("Could not connect to rater: ", err.Error())
|
|
// }
|
|
// }
|
|
|
|
// func testSCncrKillEngine(t *testing.T) {
|
|
// if err := engine.KillEngine(*utils.WaitRater); err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
// }
|
|
|
|
// func testSCncrLoadTP(t *testing.T) {
|
|
// var loadInst string
|
|
// if err := sCncrRPC.Call(context.Background(), utils.APIerSv1LoadTariffPlanFromFolder,
|
|
// &utils.AttrLoadTpFromFolder{FolderPath: path.Join(
|
|
// *utils.DataDir, "tariffplans", "tp1cnt")}, &loadInst); err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
// attrPrfl := &v2.AttributeWithAPIOpts{
|
|
// ExternalAttributeProfile: &engine.ExternalAttributeProfile{
|
|
// Tenant: "cgrates.org",
|
|
// ID: "AttrConcurrentSessions",
|
|
// Contexts: []string{utils.MetaAny},
|
|
// Attributes: []*engine.ExternalAttribute{
|
|
// {
|
|
// Path: utils.MetaReq + utils.NestingSep + "TestType",
|
|
// Value: "ConcurrentSessions",
|
|
// },
|
|
// },
|
|
// Weight: 20,
|
|
// },
|
|
// }
|
|
// var resAttrSet string
|
|
// if err := sCncrRPC.Call(context.Background(), utils.APIerSv2SetAttributeProfile, attrPrfl, &resAttrSet); err != nil {
|
|
// t.Error(err)
|
|
// } else if resAttrSet != utils.OK {
|
|
// t.Errorf("unexpected reply returned: <%s>", resAttrSet)
|
|
// }
|
|
// }
|
|
|
|
// func testSCncrRunSessions(t *testing.T) {
|
|
// acntIDsSet := utils.NewStringSet(nil)
|
|
// bufferTopup := 8760 * time.Hour
|
|
// for i := 0; i < *sCncrSessions; i++ {
|
|
// acntID := fmt.Sprintf("100%d", utils.RandomInteger(100, 200))
|
|
// if !acntIDsSet.Has(acntID) {
|
|
// // Special balance BUFFER to cover concurrency on MAIN one
|
|
// argsAddBalance := &v1.AttrAddBalance{
|
|
// Tenant: "cgrates.org",
|
|
// Account: acntID,
|
|
// BalanceType: utils.MetaVoice,
|
|
// Value: float64(bufferTopup.Nanoseconds()),
|
|
// Balance: map[string]any{
|
|
// utils.ID: "BUFFER",
|
|
// },
|
|
// }
|
|
// var addBlcRply string
|
|
// if err = sCncrRPC.Call(context.Background(), utils.APIerSv1AddBalance, argsAddBalance, &addBlcRply); err != nil {
|
|
// t.Error(err)
|
|
// } else if addBlcRply != utils.OK {
|
|
// t.Errorf("received: <%s>", addBlcRply)
|
|
// }
|
|
// acntIDsSet.Add(acntID)
|
|
// }
|
|
// acntIDs <- acntID
|
|
// wg.Add(1)
|
|
// go t.Run(fmt.Sprintf("RunSession#%d", i), testRunSession)
|
|
// }
|
|
// wg.Wait()
|
|
// for acntID := range acntIDsSet.Data() {
|
|
// // make sure the account was properly refunded
|
|
// var acnt *engine.Account
|
|
// acntAttrs := &utils.AttrGetAccount{
|
|
// Tenant: "cgrates.org",
|
|
// Account: acntID}
|
|
// if err = sCncrRPC.Call(context.Background(), utils.APIerSv2GetAccount, acntAttrs, &acnt); err != nil {
|
|
// return
|
|
// } else if vcBlnc := acnt.BalanceMap[utils.MetaVoice].GetTotalValue(); float64(bufferTopup.Nanoseconds())-vcBlnc > 1000000.0 { // eliminate rounding errors
|
|
// t.Errorf("unexpected voice balance received: %+v", utils.ToIJSON(acnt))
|
|
// } else if mnBlnc := acnt.BalanceMap[utils.MetaMonetary].GetTotalValue(); mnBlnc != 0 {
|
|
// t.Errorf("unexpected monetary balance received: %+v", utils.ToIJSON(acnt))
|
|
// }
|
|
// }
|
|
// }
|
|
|
|
// // runSession runs one session
|
|
// func testRunSession(t *testing.T) {
|
|
// defer wg.Done() // decrease group counter one out from test
|
|
// cpsPool <- struct{}{} // push here up to cps
|
|
// go func() { // allow more requests after a second
|
|
// time.Sleep(time.Second)
|
|
// <-cpsPool
|
|
// }()
|
|
// acntID := <-acntIDs
|
|
// originID := utils.GenUUID() // each test with it's own OriginID
|
|
// // topup as much as we know we need for one session
|
|
// mainTopup := 90 * time.Second
|
|
// var addBlcRply string
|
|
// argsAddBalance := &v1.AttrAddBalance{
|
|
// Tenant: "cgrates.org",
|
|
// Account: acntID,
|
|
// BalanceType: utils.MetaVoice,
|
|
// Value: float64(mainTopup.Nanoseconds()),
|
|
// Balance: map[string]any{
|
|
// utils.ID: "MAIN",
|
|
// utils.Weight: 10,
|
|
// },
|
|
// }
|
|
// if err = sCncrRPC.Call(context.Background(), utils.APIerSv1AddBalance, argsAddBalance, &addBlcRply); err != nil {
|
|
// t.Error(err)
|
|
// } else if addBlcRply != utils.OK {
|
|
// t.Errorf("received: <%s> to APIerSv1.AddBalance", addBlcRply)
|
|
// }
|
|
// time.Sleep(time.Duration(
|
|
// utils.RandomInteger(0, 100)) * time.Millisecond) // randomize between tests
|
|
|
|
// // Auth the session
|
|
// authDur := 5 * time.Minute
|
|
// authArgs := &sessions.V1AuthorizeArgs{
|
|
// GetMaxUsage: true,
|
|
// CGREvent: &utils.CGREvent{
|
|
// Tenant: "cgrates.org",
|
|
// ID: fmt.Sprintf("TestSCncrAuth%s", originID),
|
|
// Event: map[string]any{
|
|
// utils.Tenant: "cgrates.org",
|
|
// utils.OriginID: originID,
|
|
// utils.RequestType: utils.MetaPrepaid,
|
|
// utils.AccountField: acntID,
|
|
// utils.Destination: fmt.Sprintf("%s%s", acntID, acntID),
|
|
// utils.SetupTime: time.Now(),
|
|
// utils.Usage: authDur,
|
|
// },
|
|
// },
|
|
// }
|
|
// var rplyAuth sessions.V1AuthorizeReply
|
|
// if err := sCncrRPC.Call(context.Background(), utils.SessionSv1AuthorizeEvent, authArgs, &rplyAuth); err != nil {
|
|
// t.Error(err)
|
|
// }
|
|
// time.Sleep(time.Duration(
|
|
// utils.RandomInteger(0, 100)) * time.Millisecond)
|
|
|
|
// // Init the session
|
|
// initUsage := time.Minute
|
|
// initArgs := &sessions.V1InitSessionArgs{
|
|
// InitSession: true,
|
|
// GetAttributes: true,
|
|
// CGREvent: &utils.CGREvent{
|
|
// Tenant: "cgrates.org",
|
|
// ID: fmt.Sprintf("TestSCncrInit%s", originID),
|
|
// Event: map[string]any{
|
|
// utils.OriginID: originID,
|
|
// utils.RequestType: utils.MetaPrepaid,
|
|
// utils.AccountField: acntID,
|
|
// utils.Destination: fmt.Sprintf("%s%s", acntID, acntID),
|
|
// utils.AnswerTime: time.Now(),
|
|
// utils.Usage: initUsage,
|
|
// },
|
|
// },
|
|
// }
|
|
// var rplyInit sessions.V1InitSessionReply
|
|
// if err := sCncrRPC.Call(context.Background(), utils.SessionSv1InitiateSession,
|
|
// initArgs, &rplyInit); err != nil {
|
|
// t.Error(err)
|
|
// } else if rplyInit.MaxUsage == 0 {
|
|
// t.Errorf("unexpected MaxUsage at init: %v", rplyInit.MaxUsage)
|
|
// }
|
|
// time.Sleep(time.Duration(
|
|
// utils.RandomInteger(0, 100)) * time.Millisecond)
|
|
|
|
// // Update the session with relocate
|
|
// initOriginID := originID
|
|
// originID = utils.GenUUID()
|
|
// updtUsage := time.Minute
|
|
// updtArgs := &sessions.V1UpdateSessionArgs{
|
|
// GetAttributes: true,
|
|
// UpdateSession: true,
|
|
// CGREvent: &utils.CGREvent{
|
|
// Tenant: "cgrates.org",
|
|
// ID: fmt.Sprintf("TestSCncrUpdate%s", originID),
|
|
// Event: map[string]any{
|
|
// utils.OriginID: originID,
|
|
// utils.InitialOriginID: initOriginID,
|
|
// utils.Usage: updtUsage,
|
|
// },
|
|
// },
|
|
// }
|
|
// var rplyUpdt sessions.V1UpdateSessionReply
|
|
// if err := sCncrRPC.Call(context.Background(), utils.SessionSv1UpdateSession,
|
|
// updtArgs, &rplyUpdt); err != nil {
|
|
// t.Error(err)
|
|
// } else if rplyUpdt.MaxUsage == 0 {
|
|
// t.Errorf("unexpected MaxUsage at update: %v", rplyUpdt.MaxUsage)
|
|
// }
|
|
// time.Sleep(time.Duration(
|
|
// utils.RandomInteger(0, 100)) * time.Millisecond)
|
|
|
|
// // Terminate the session
|
|
// trmntArgs := &sessions.V1TerminateSessionArgs{
|
|
// TerminateSession: true,
|
|
// CGREvent: &utils.CGREvent{
|
|
// Tenant: "cgrates.org",
|
|
// ID: fmt.Sprintf("TestSCncrTerminate%s", originID),
|
|
// Event: map[string]any{
|
|
// utils.OriginID: originID,
|
|
// utils.Usage: 90 * time.Second,
|
|
// },
|
|
// },
|
|
// }
|
|
// var rplyTrmnt string
|
|
// if err := sCncrRPC.Call(context.Background(), utils.SessionSv1TerminateSession,
|
|
// trmntArgs, &rplyTrmnt); err != nil {
|
|
// t.Error(err)
|
|
// } else if rplyTrmnt != utils.OK {
|
|
// t.Errorf("received: <%s> to SessionSv1.Terminate", rplyTrmnt)
|
|
// }
|
|
// time.Sleep(time.Duration(
|
|
// utils.RandomInteger(0, 100)) * time.Millisecond)
|
|
|
|
// // processCDR
|
|
// argsCDR := &utils.CGREventWithOpts{
|
|
// CGREvent: &utils.CGREvent{
|
|
// Tenant: "cgrates.org",
|
|
// ID: fmt.Sprintf("TestSCncrCDR%s", originID),
|
|
// Event: map[string]any{
|
|
// utils.OriginID: originID,
|
|
// },
|
|
// },
|
|
// }
|
|
// var rplyCDR string
|
|
// if err := sCncrRPC.Call(context.Background(), utils.SessionSv1ProcessCDR,
|
|
// argsCDR, &rplyCDR); err != nil {
|
|
// t.Error(err)
|
|
// } else if rplyCDR != utils.OK {
|
|
// t.Errorf("received: <%s> to ProcessCDR", rplyCDR)
|
|
// }
|
|
// time.Sleep(20 * time.Millisecond)
|
|
// var cdrs []*engine.ExternalCDR
|
|
// argCDRs := utils.RPCCDRsFilter{OriginIDs: []string{originID}}
|
|
// if err := sCncrRPC.Call(context.Background(), utils.APIerSv2GetCDRs, &argCDRs, &cdrs); err != nil {
|
|
// t.Error(err)
|
|
// } else if len(cdrs) != 1 {
|
|
// t.Errorf("unexpected number of CDRs returned: %d", len(cdrs))
|
|
// } else if cdrs[0].Usage != "1m30s" {
|
|
// t.Errorf("unexpected usage of CDR: %+v", cdrs[0])
|
|
// }
|
|
// }
|