cgrates/general_tests/sessions_benchmark_it_test.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 general_tests

import (
	"fmt"
	"os/exec"
	"path"
	"testing"
	"time"

	"github.com/cgrates/birpc"
	"github.com/cgrates/birpc/context"
	v1 "github.com/cgrates/cgrates/apier/v1"
	"github.com/cgrates/cgrates/config"
	"github.com/cgrates/cgrates/engine"
	"github.com/cgrates/cgrates/sessions"
	"github.com/cgrates/cgrates/utils"
)

var (
	sesPCfgDir  string
	sesPCfgPath string
	sesPCfg     *config.CGRConfig
	sesPRPC     *birpc.Client

	sesPTests = []func(t *testing.T){
		testSesPItLoadConfig,
		testSesPItResetDataDB,
		testSesPItResetStorDB,
		testSesPItStartEngine,
		// testSesPItStartEngineWithProfiling,
		testSesPItRPCConn,
		testSesPItBenchmark,
		// testSesPItCheckAccounts,
		// testSesPItStopCgrEngine,
	}
)

func TestSesPIt(t *testing.T) {
	switch *utils.DBType {
	case utils.MetaInternal:
		sesPCfgDir = "tutinternal"
	case utils.MetaMySQL:
		sesPCfgDir = "tutmysql"
	case utils.MetaMongo:
		sesPCfgDir = "tutmongo"
	case utils.MetaPostgres:
		t.SkipNow()
	default:
		t.Fatal("Unknown Database type")
	}
	for _, stest := range sesPTests {
		t.Run(sesPCfgDir, stest)
	}
}

func testSesPItLoadConfig(t *testing.T) {
	var err error
	sesPCfgPath = path.Join(*utils.DataDir, "conf", "samples", sesPCfgDir)
	if sesPCfg, err = config.NewCGRConfigFromPath(sesPCfgPath); err != nil {
		t.Error(err)
	}
}

func testSesPItResetDataDB(t *testing.T) {
	if err := engine.InitDataDb(sesPCfg); err != nil {
		t.Fatal(err)
	}
}

func testSesPItResetStorDB(t *testing.T) {
	if err := engine.InitStorDb(sesPCfg); err != nil {
		t.Fatal(err)
	}
}

func testSesPItStartEngine(t *testing.T) {
	if _, err := engine.StopStartEngine(sesPCfgPath, *utils.WaitRater); err != nil {
		t.Fatal(err)
	}
}

func testSesPItStartEngineWithProfiling(t *testing.T) {
	engine := exec.Command("cgr-engine", "-config_path", fmt.Sprintf("/usr/share/cgrates/conf/samples/%s", sesPCfgDir), "-cpuprof_dir=~/prof/")
	if err := engine.Start(); err != nil {
		t.Error(err)
	}
}

func testSesPItRPCConn(t *testing.T) {
	sesPRPC = engine.NewRPCClient(t, sesPCfg.ListenCfg())
}

func setAccounts(t *testing.T, n int) (err error) {
	for i := 0; i < n; i++ {
		// fmt.Println(fmt.Sprintf("acc%d", i))
		if err := setAccBalance(fmt.Sprintf("acc%d", i)); err != nil {
			t.Error(err)
		}
	}
	return nil
}

func getAccounts(ids []string) (accounts *[]any, err error) {
	var reply *[]any

	attr := &utils.AttrGetAccounts{
		Tenant:     "cgrates.org",
		AccountIDs: ids,
	}
	err = sesPRPC.Call(context.Background(), "APIerSv1.GetAccounts", attr, &reply)
	if err != nil {
		return
	}
	accounts = reply
	return
}

func setAccBalance(acc string) (err error) {
	args := utils.AttrSetBalance{
		Tenant:      "cgrates.org",
		Account:     acc,
		BalanceType: utils.MetaVoice,
		Value:       float64(24 * time.Hour),
		Balance: map[string]any{
			utils.ID:            "TestBalance",
			utils.RatingSubject: "*zero1ms",
		},
	}
	var reply string

	err = sesPRPC.Call(context.Background(), utils.APIerSv1SetBalance, args, &reply)
	return err
}

func initSes(n int) (err error) {
	var accIDs []string
	for i := 0; i < n; i++ {
		accIDs = append(accIDs, fmt.Sprintf("acc%d", i))
	}
	_, err = getAccounts(accIDs)
	if err != nil {
		return
	}

	initArgs := &sessions.V1InitSessionArgs{
		InitSession: true,
		CGREvent: &utils.CGREvent{
			Tenant: "cgrates.org",
			ID:     utils.UUIDSha1Prefix(),
			Event: map[string]any{
				utils.EventName:    "TEST_EVENT",
				utils.OriginID:     utils.UUIDSha1Prefix(),
				utils.ToR:          utils.MetaVoice,
				utils.Category:     "call",
				utils.Tenant:       "cgrates.org",
				utils.AccountField: "1001",
				utils.Subject:      "1001",
				utils.Destination:  "1002",
				utils.RequestType:  utils.MetaPrepaid,
				utils.AnswerTime:   time.Date(2016, time.January, 5, 18, 31, 05, 0, time.UTC),
			},
			APIOpts: map[string]any{
				utils.OptsDebitInterval: 2 * time.Second,
			},
		},
	}

	var initRpl *sessions.V1InitSessionReply

	for i := 0; i < n; i++ {
		initArgs.CGREvent.Event[utils.AccountField] = accIDs[i]
		initArgs.CGREvent.Event[utils.OriginID] = utils.UUIDSha1Prefix()
		if err = sesPRPC.Call(context.Background(), utils.SessionSv1InitiateSession,
			initArgs, &initRpl); err != nil {
			return
		}
	}
	return
}

func testSesPItBenchmark(t *testing.T) {
	// add 2 charger profiles
	var reply string
	n := 100
	args := &v1.ChargerWithAPIOpts{
		ChargerProfile: &engine.ChargerProfile{
			Tenant:       "cgrates.org",
			ID:           "Default",
			RunID:        "runID1",
			AttributeIDs: []string{"*none"},
		},
		APIOpts: map[string]any{},
	}
	if err := sesPRPC.Call(context.Background(), utils.APIerSv1SetChargerProfile, args, &reply); err != nil {
		t.Error(err)
	} else if reply != "OK" {
		t.Error("Expected OK")
	}

	args2 := &v1.ChargerWithAPIOpts{
		ChargerProfile: &engine.ChargerProfile{
			Tenant:       "cgrates.org",
			ID:           "Default",
			RunID:        "runID2",
			AttributeIDs: []string{"*none"},
		},
		APIOpts: map[string]any{},
	}

	if err := sesPRPC.Call(context.Background(), utils.APIerSv1SetChargerProfile, args2, &reply); err != nil {
		t.Error(err)
	} else if reply != "OK" {
		t.Error("Expected OK")
	}

	////////// charger profiles set

	if err := setAccounts(t, n); err != nil {
		t.Error(err)
	}

	if err := initSes(n); err != nil {
		t.Error(err)
	}
	var statusRpl map[string]any

	if err := sesPRPC.Call(context.Background(), utils.CoreSv1Status, nil, &statusRpl); err != nil {
		t.Error(err)
	}
}

func testSesPItStopCgrEngine(t *testing.T) {
	if err := engine.KillEngine(100); err != nil {
		t.Error(err)
	}
}