mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
New integration test in general tests for cdrs tenant change
This commit is contained in:
andronache
committed by
Dan Christian Bogos
Dan Christian Bogos
parent
36cf49811c
commit
92d879ea34
235
general_tests/sessions_tnt_change_cdr_it_test.go
Normal file
235
general_tests/sessions_tnt_change_cdr_it_test.go
Normal file
@@ -0,0 +1,235 @@
|
||||
// +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 (
|
||||
"net/rpc"
|
||||
"path"
|
||||
"reflect"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
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 (
|
||||
sesTntChngCdrsCfgDir string
|
||||
sesTntChngCdrsCfgPath string
|
||||
sesTntChngCdrsCfg *config.CGRConfig
|
||||
sesTntChngCdrsRPC *rpc.Client
|
||||
|
||||
sesTntChngCdrsTests = []func(t *testing.T){
|
||||
testSesTntChngCdrsLoadConfig,
|
||||
testSesTntChngCdrsResetDataDB,
|
||||
testSesTntChngCdrsResetStorDb,
|
||||
testSesTntChngCdrsStartEngine,
|
||||
testSesTntChngCdrsRPCConn,
|
||||
testSesTntChngCdrsSetChargerProfile1,
|
||||
testSesTntChngCdrsSetChargerProfile2,
|
||||
testChargerSCdrsAuthProcessEventAuth,
|
||||
testSesTntChngCdrsStopCgrEngine,
|
||||
}
|
||||
)
|
||||
|
||||
func TestSesCdrsTntChange(t *testing.T) {
|
||||
switch *dbType {
|
||||
case utils.MetaInternal:
|
||||
sesTntChngCfgDir = "tutinternal"
|
||||
case utils.MetaMySQL:
|
||||
sesTntChngCfgDir = "tutmysql"
|
||||
case utils.MetaMongo:
|
||||
sesTntChngCfgDir = "tutmongo"
|
||||
case utils.MetaPostgres:
|
||||
t.SkipNow()
|
||||
default:
|
||||
t.Fatal("Unknown Database type")
|
||||
}
|
||||
for _, sestest := range sesTntChngTests {
|
||||
t.Run(sesTntChngCfgDir, sestest)
|
||||
}
|
||||
}
|
||||
|
||||
func testSesTntChngCdrsLoadConfig(t *testing.T) {
|
||||
sesTntChngCfgPath = path.Join(*dataDir, "conf", "samples", sesTntChngCfgDir)
|
||||
if sesTntChngCfg, err = config.NewCGRConfigFromPath(sesTntChngCfgPath); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
func testSesTntChngCdrsResetDataDB(t *testing.T) {
|
||||
if err := engine.InitDataDb(sesTntChngCfg); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func testSesTntChngCdrsResetStorDb(t *testing.T) {
|
||||
if err := engine.InitStorDb(sesTntChngCfg); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func testSesTntChngCdrsStartEngine(t *testing.T) {
|
||||
if _, err := engine.StopStartEngine(sesTntChngCfgPath, *waitRater); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func testSesTntChngCdrsRPCConn(t *testing.T) {
|
||||
var err error
|
||||
sesTntChngRPC, err = newRPCClient(sesTntChngCfg.ListenCfg())
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func testSesTntChngCdrsSetChargerProfile1(t *testing.T) {
|
||||
var reply *engine.ChargerProfile
|
||||
if err := sesTntChngRPC.Call(utils.APIerSv1GetChargerProfile,
|
||||
&utils.TenantID{Tenant: "cgrates.org", ID: "Charger1"}, &reply); err == nil ||
|
||||
err.Error() != utils.ErrNotFound.Error() {
|
||||
t.Fatal(err)
|
||||
}
|
||||
chargerProfile := &v1.ChargerWithAPIOpts{
|
||||
ChargerProfile: &engine.ChargerProfile{
|
||||
Tenant: "cgrates.org",
|
||||
ID: "Charger1",
|
||||
RunID: utils.MetaDefault,
|
||||
AttributeIDs: []string{"*constant:*tenant:cgrates.ro;*constant:*req.Account:1234"},
|
||||
},
|
||||
}
|
||||
|
||||
var result string
|
||||
if err := sesTntChngRPC.Call(utils.APIerSv1SetChargerProfile, chargerProfile, &result); err != nil {
|
||||
t.Error(err)
|
||||
} else if result != utils.OK {
|
||||
t.Error("Unexpected reply returned", result)
|
||||
}
|
||||
|
||||
if err := sesTntChngRPC.Call(utils.APIerSv1GetChargerProfile,
|
||||
&utils.TenantID{Tenant: "cgrates.org", ID: "Charger1"}, &reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if !reflect.DeepEqual(chargerProfile.ChargerProfile, reply) {
|
||||
t.Errorf("Expecting : %+v, received: %+v", chargerProfile.ChargerProfile, reply)
|
||||
}
|
||||
}
|
||||
|
||||
func testSesTntChngCdrsSetChargerProfile2(t *testing.T) {
|
||||
var reply *engine.ChargerProfile
|
||||
if err := sesTntChngRPC.Call(utils.APIerSv1GetChargerProfile,
|
||||
&utils.TenantID{Tenant: "cgrates.org", ID: "Charger2"}, &reply); err == nil ||
|
||||
err.Error() != utils.ErrNotFound.Error() {
|
||||
t.Fatal(err)
|
||||
}
|
||||
chargerProfile := &v1.ChargerWithAPIOpts{
|
||||
ChargerProfile: &engine.ChargerProfile{
|
||||
Tenant: "cgrates.org",
|
||||
ID: "Charger2",
|
||||
RunID: utils.MetaRaw,
|
||||
AttributeIDs: []string{},
|
||||
},
|
||||
}
|
||||
|
||||
var result string
|
||||
if err := sesTntChngRPC.Call(utils.APIerSv1SetChargerProfile, chargerProfile, &result); err != nil {
|
||||
t.Error(err)
|
||||
} else if result != utils.OK {
|
||||
t.Error("Unexpected reply returned", result)
|
||||
}
|
||||
|
||||
if err := sesTntChngRPC.Call(utils.APIerSv1GetChargerProfile,
|
||||
&utils.TenantID{Tenant: "cgrates.org", ID: "Charger2"}, &reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if !reflect.DeepEqual(chargerProfile.ChargerProfile, reply) {
|
||||
t.Errorf("Expecting : %+v, received: %+v", chargerProfile.ChargerProfile, reply)
|
||||
}
|
||||
}
|
||||
|
||||
func testChargerSCdrsAuthProcessEventAuth(t *testing.T) {
|
||||
attrSetBalance := utils.AttrSetBalance{
|
||||
Tenant: "cgrates.org",
|
||||
Account: "1001",
|
||||
BalanceType: utils.MetaVoice,
|
||||
Value: float64(2 * time.Minute),
|
||||
Balance: map[string]interface{}{
|
||||
utils.ID: "testSes",
|
||||
utils.RatingSubject: "*zero1ms",
|
||||
},
|
||||
}
|
||||
var reply string
|
||||
if err := sesTntChngRPC.Call(utils.APIerSv2SetBalance, attrSetBalance, &reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.OK {
|
||||
t.Errorf("Received: %s", reply)
|
||||
}
|
||||
|
||||
attrSetBalance2 := utils.AttrSetBalance{
|
||||
Tenant: "cgrates.ro",
|
||||
Account: "1234",
|
||||
BalanceType: utils.MetaVoice,
|
||||
Value: float64(2 * time.Minute),
|
||||
Balance: map[string]interface{}{
|
||||
utils.ID: "testSes",
|
||||
utils.RatingSubject: "*zero1ms",
|
||||
},
|
||||
}
|
||||
var reply2 string
|
||||
if err := sesTntChngRPC.Call(utils.APIerSv2SetBalance, attrSetBalance2, &reply2); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply2 != utils.OK {
|
||||
t.Errorf("Received: %s", reply2)
|
||||
}
|
||||
|
||||
ev := &engine.ArgV1ProcessEvent{
|
||||
CGREvent: utils.CGREvent{
|
||||
Tenant: "cgrates.org",
|
||||
ID: "TestEv1",
|
||||
Event: map[string]interface{}{
|
||||
utils.ToR: utils.MetaVoice,
|
||||
utils.OriginID: "TestEv1",
|
||||
utils.RequestType: utils.MetaPrepaid,
|
||||
utils.AccountField: "1001",
|
||||
utils.Subject: "1001",
|
||||
utils.Destination: "1002",
|
||||
utils.Usage: time.Minute,
|
||||
},
|
||||
},
|
||||
}
|
||||
var rply sessions.V1AuthorizeReply
|
||||
if err := sesTntChngRPC.Call(utils.CDRsV1ProcessEvent, ev, &rply); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
expected := &sessions.V1AuthorizeReply{
|
||||
MaxUsage: (*time.Duration)(utils.Int64Pointer(60000000000)),
|
||||
}
|
||||
if !reflect.DeepEqual(utils.ToJSON(&expected), utils.ToJSON(&rply)) {
|
||||
t.Errorf("Expecting : %+v, received: %+v", utils.ToJSON(&expected), utils.ToJSON(&rply))
|
||||
}
|
||||
}
|
||||
|
||||
func testSesTntChngCdrsStopCgrEngine(t *testing.T) {
|
||||
if err := engine.KillEngine(100); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user