mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-13 02:56:24 +05:00
Add test for TPLcrRules
This commit is contained in:
TeoV
@@ -39,6 +39,15 @@ type AttrGetTPLcrRules struct {
|
||||
LcrRuleId string // Lcr id
|
||||
}
|
||||
|
||||
type AttrRemTPLcrRules struct {
|
||||
TPid string // Tariff plan id
|
||||
Direction string
|
||||
Tenant string
|
||||
Category string
|
||||
Account string
|
||||
Subject string
|
||||
}
|
||||
|
||||
// Queries specific LcrRules profile on tariff plan
|
||||
func (self *ApierV1) GetTPLcrRule(attr AttrGetTPLcrRules, reply *utils.TPLcrRules) error {
|
||||
if missing := utils.MissingStructFields(&attr, []string{"TPid", "LcrRuleId"}); len(missing) != 0 { //Params missing
|
||||
@@ -47,9 +56,10 @@ func (self *ApierV1) GetTPLcrRule(attr AttrGetTPLcrRules, reply *utils.TPLcrRule
|
||||
filter := &utils.TPLcrRules{TPid: attr.TPid}
|
||||
filter.SetId(attr.LcrRuleId)
|
||||
if lcrs, err := self.StorDb.GetTPLCRs(filter); err != nil {
|
||||
return utils.NewErrServerError(err)
|
||||
} else if len(lcrs) == 0 {
|
||||
return utils.ErrNotFound
|
||||
if err.Error() != utils.ErrNotFound.Error() {
|
||||
err = utils.NewErrServerError(err)
|
||||
}
|
||||
return err
|
||||
} else {
|
||||
*reply = *lcrs[0]
|
||||
}
|
||||
@@ -77,11 +87,11 @@ func (self *ApierV1) GetTPLcrRuleIds(attrs AttrGetTPLcrIds, reply *[]string) err
|
||||
}
|
||||
|
||||
// Removes specific LcrRules on Tariff plan
|
||||
func (self *ApierV1) RemTPLcrRule(attrs AttrGetTPLcrRules, reply *string) error {
|
||||
if missing := utils.MissingStructFields(&attrs, []string{"TPid", "LcrRulesId"}); len(missing) != 0 { //Params missing
|
||||
func (self *ApierV1) RemTPLcrRule(attrs AttrRemTPLcrRules, reply *string) error {
|
||||
if missing := utils.MissingStructFields(&attrs, []string{"TPid", "Direction", "Tenant", "Category", "Account", "Subject"}); len(missing) != 0 { //Params missing
|
||||
return utils.NewErrMandatoryIeMissing(missing...)
|
||||
}
|
||||
if err := self.StorDb.RemTpData(utils.TBLTPLcrs, attrs.TPid, map[string]string{"tag": attrs.LcrRuleId}); err != nil {
|
||||
if err := self.StorDb.RemTpData(utils.TBLTPLcrs, attrs.TPid, map[string]string{"direction": attrs.Direction, "tenant": attrs.Tenant, "category": attrs.Category, "account": attrs.Account, "subject": attrs.Subject}); err != nil {
|
||||
return utils.NewErrServerError(err)
|
||||
} else {
|
||||
*reply = utils.OK
|
||||
|
||||
259
apier/v1/tplcrrules_it_test.go
Normal file
259
apier/v1/tplcrrules_it_test.go
Normal file
@@ -0,0 +1,259 @@
|
||||
// +build offline_tp
|
||||
|
||||
/*
|
||||
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 v1
|
||||
|
||||
import (
|
||||
"github.com/cgrates/cgrates/config"
|
||||
"github.com/cgrates/cgrates/engine"
|
||||
"github.com/cgrates/cgrates/utils"
|
||||
"net/rpc"
|
||||
"net/rpc/jsonrpc"
|
||||
"path"
|
||||
"reflect"
|
||||
"testing"
|
||||
)
|
||||
|
||||
var (
|
||||
tpLcrRuleCfgPath string
|
||||
tpLcrRuleCfg *config.CGRConfig
|
||||
tpLcrRuleRPC *rpc.Client
|
||||
tpLcrRuleDataDir = "/usr/share/cgrates"
|
||||
tpLcrRules *utils.TPLcrRules
|
||||
tpLcrRuleDelay int
|
||||
tpLcrRuleConfigDIR string //run tests for specific configuration
|
||||
tpLcrRuleID = "*out:cgrates.org:call:1001:*any"
|
||||
)
|
||||
|
||||
var sTestsTPLcrRules = []func(t *testing.T){
|
||||
testTPLcrRulesInitCfg,
|
||||
testTPLcrRulesResetStorDb,
|
||||
testTPLcrRulesStartEngine,
|
||||
testTPLcrRulesRpcConn,
|
||||
testTPLcrRulesGetTPLcrRulesBeforeSet,
|
||||
testTPLcrRulesSetTPLcrRules,
|
||||
testTPLcrRulesGetTPLcrRulesAfterSet,
|
||||
testTPLcrRulesGetTPDestinationIds,
|
||||
testTPLcrRulesUpdateTPLcrRules,
|
||||
testTPLcrRulesGetTPLcrRulesAfterUpdate,
|
||||
testTPLcrRulesRemTPLcrRules,
|
||||
testTPLcrRulesGetTPLcrRulesAfterRemove,
|
||||
testTPLcrRulesKillEngine,
|
||||
}
|
||||
|
||||
//Test start here
|
||||
func TestTPLcrRulesITMySql(t *testing.T) {
|
||||
tpLcrRuleConfigDIR = "tutmysql"
|
||||
for _, stest := range sTestsTPLcrRules {
|
||||
t.Run(tpLcrRuleConfigDIR, stest)
|
||||
}
|
||||
}
|
||||
|
||||
func TestTPLcrRulesITMongo(t *testing.T) {
|
||||
tpLcrRuleConfigDIR = "tutmongo"
|
||||
for _, stest := range sTestsTPLcrRules {
|
||||
t.Run(tpLcrRuleConfigDIR, stest)
|
||||
}
|
||||
}
|
||||
|
||||
func TestTPLcrRulesITPG(t *testing.T) {
|
||||
tpLcrRuleConfigDIR = "tutpostgres"
|
||||
for _, stest := range sTestsTPLcrRules {
|
||||
t.Run(tpLcrRuleConfigDIR, stest)
|
||||
}
|
||||
}
|
||||
|
||||
func testTPLcrRulesInitCfg(t *testing.T) {
|
||||
var err error
|
||||
tpLcrRuleCfgPath = path.Join(tpLcrRuleDataDir, "conf", "samples", tpLcrRuleConfigDIR)
|
||||
tpLcrRuleCfg, err = config.NewCGRConfigFromFolder(tpLcrRuleCfgPath)
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
tpLcrRuleCfg.DataFolderPath = tpLcrRuleDataDir // Share DataFolderPath through config towards StoreDb for Flush()
|
||||
config.SetCgrConfig(tpLcrRuleCfg)
|
||||
switch tpLcrRuleConfigDIR {
|
||||
case "tutmongo": // Mongo needs more time to reset db, need to investigate
|
||||
tpLcrRuleDelay = 2000
|
||||
default:
|
||||
tpLcrRuleDelay = 1000
|
||||
}
|
||||
}
|
||||
|
||||
// Wipe out the cdr database
|
||||
func testTPLcrRulesResetStorDb(t *testing.T) {
|
||||
if err := engine.InitStorDb(tpLcrRuleCfg); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
// Start CGR Engine
|
||||
func testTPLcrRulesStartEngine(t *testing.T) {
|
||||
if _, err := engine.StopStartEngine(tpLcrRuleCfgPath, tpLcrRuleDelay); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
// Connect rpc client to rater
|
||||
func testTPLcrRulesRpcConn(t *testing.T) {
|
||||
var err error
|
||||
tpLcrRuleRPC, err = jsonrpc.Dial("tcp", tpLcrRuleCfg.RPCJSONListen) // We connect over JSON so we can also troubleshoot if needed
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func testTPLcrRulesGetTPLcrRulesBeforeSet(t *testing.T) {
|
||||
var reply *utils.TPRatingPlan
|
||||
if err := tpLcrRuleRPC.Call("ApierV1.GetTPLcrRule", &AttrGetTPLcrRules{TPid: "TPLRC1", LcrRuleId: tpLcrRuleID}, &reply); err == nil || err.Error() != utils.ErrNotFound.Error() {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
func testTPLcrRulesSetTPLcrRules(t *testing.T) {
|
||||
tpLcrRules = &utils.TPLcrRules{
|
||||
TPid: "TPLRC1",
|
||||
Direction: "*out",
|
||||
Tenant: "cgrates.org",
|
||||
Category: "call",
|
||||
Account: "1001",
|
||||
Subject: "*any",
|
||||
Rules: []*utils.TPLcrRule{
|
||||
&utils.TPLcrRule{
|
||||
DestinationId: "DST_1002",
|
||||
RpCategory: "lcr_profile1",
|
||||
Strategy: "*static",
|
||||
StrategyParams: "suppl2;suppl1",
|
||||
ActivationTime: "05:00:00",
|
||||
Weight: 10,
|
||||
},
|
||||
&utils.TPLcrRule{
|
||||
DestinationId: "*any",
|
||||
RpCategory: "lcr_profile1",
|
||||
Strategy: "*highest_cost",
|
||||
StrategyParams: "",
|
||||
ActivationTime: "05:00:00",
|
||||
Weight: 10,
|
||||
},
|
||||
},
|
||||
}
|
||||
var result string
|
||||
if err := tpLcrRuleRPC.Call("ApierV1.SetTPLcrRule", tpLcrRules, &result); err != nil {
|
||||
t.Error(err)
|
||||
} else if result != utils.OK {
|
||||
t.Error("Unexpected reply returned", result)
|
||||
}
|
||||
}
|
||||
|
||||
func testTPLcrRulesGetTPLcrRulesAfterSet(t *testing.T) {
|
||||
var reply *utils.TPLcrRules
|
||||
if err := tpLcrRuleRPC.Call("ApierV1.GetTPLcrRule", &AttrGetTPLcrRules{TPid: "TPLRC1", LcrRuleId: tpLcrRuleID}, &reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if !reflect.DeepEqual(tpLcrRules.TPid, reply.TPid) {
|
||||
t.Errorf("Expecting : %+v, received: %+v", tpLcrRules.TPid, reply.TPid)
|
||||
} else if !reflect.DeepEqual(tpLcrRules.GetLcrRuleId(), reply.GetLcrRuleId()) {
|
||||
t.Errorf("Expecting : %+v, received: %+v", tpLcrRules.GetLcrRuleId(), reply.GetLcrRuleId())
|
||||
} else if !reflect.DeepEqual(len(tpLcrRules.Rules), len(reply.Rules)) {
|
||||
t.Errorf("Expecting : %+v, received: %+v", len(tpLcrRules.Rules), len(reply.Rules))
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func testTPLcrRulesGetTPDestinationIds(t *testing.T) {
|
||||
var result []string
|
||||
expectedTPID := []string{"*out:cgrates.org:call:1001:*any"}
|
||||
if err := tpLcrRuleRPC.Call("ApierV1.GetTPLcrRuleIds", &AttrGetTPLcrIds{TPid: "TPLRC1"}, &result); err != nil {
|
||||
t.Error(err)
|
||||
} else if !reflect.DeepEqual(expectedTPID, result) {
|
||||
t.Errorf("Expecting: %+v, received: %+v", expectedTPID, result)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func testTPLcrRulesUpdateTPLcrRules(t *testing.T) {
|
||||
tpLcrRules.Rules = []*utils.TPLcrRule{
|
||||
&utils.TPLcrRule{
|
||||
DestinationId: "DST_1002",
|
||||
RpCategory: "lcr_profile1",
|
||||
Strategy: "*static",
|
||||
StrategyParams: "suppl2;suppl1",
|
||||
ActivationTime: "03:00:00",
|
||||
Weight: 10,
|
||||
},
|
||||
&utils.TPLcrRule{
|
||||
DestinationId: "*any",
|
||||
RpCategory: "lcr_profile1",
|
||||
Strategy: "*highest_cost",
|
||||
StrategyParams: "",
|
||||
ActivationTime: "05:00:00",
|
||||
Weight: 10,
|
||||
},
|
||||
&utils.TPLcrRule{
|
||||
DestinationId: "*any",
|
||||
RpCategory: "lcr_profile2",
|
||||
Strategy: "*load_distribution",
|
||||
StrategyParams: "supplier1:5",
|
||||
ActivationTime: "01:00:00",
|
||||
Weight: 10,
|
||||
},
|
||||
}
|
||||
var result string
|
||||
if err := tpLcrRuleRPC.Call("ApierV1.SetTPLcrRule", tpLcrRules, &result); err != nil {
|
||||
t.Error(err)
|
||||
} else if result != utils.OK {
|
||||
t.Error("Unexpected reply returned", result)
|
||||
}
|
||||
}
|
||||
|
||||
func testTPLcrRulesGetTPLcrRulesAfterUpdate(t *testing.T) {
|
||||
var reply *utils.TPLcrRules
|
||||
if err := tpLcrRuleRPC.Call("ApierV1.GetTPLcrRule", &AttrGetTPLcrRules{TPid: "TPLRC1", LcrRuleId: tpLcrRuleID}, &reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if !reflect.DeepEqual(tpLcrRules.TPid, reply.TPid) {
|
||||
t.Errorf("Expecting : %+v, received: %+v", tpLcrRules.TPid, reply.TPid)
|
||||
} else if !reflect.DeepEqual(tpLcrRules.GetLcrRuleId(), reply.GetLcrRuleId()) {
|
||||
t.Errorf("Expecting : %+v, received: %+v", tpLcrRules.GetLcrRuleId(), reply.GetLcrRuleId())
|
||||
} else if !reflect.DeepEqual(len(tpLcrRules.Rules), len(reply.Rules)) {
|
||||
t.Errorf("Expecting : %+v, received: %+v", len(tpLcrRules.Rules), len(reply.Rules))
|
||||
}
|
||||
}
|
||||
|
||||
func testTPLcrRulesRemTPLcrRules(t *testing.T) {
|
||||
var resp string
|
||||
if err := tpLcrRuleRPC.Call("ApierV1.RemTPLcrRule", &AttrRemTPLcrRules{TPid: tpLcrRules.TPid, Direction: tpLcrRules.Direction, Tenant: tpLcrRules.Tenant, Category: tpLcrRules.Category, Account: tpLcrRules.Account, Subject: tpLcrRules.Subject}, &resp); err != nil {
|
||||
t.Error(err)
|
||||
} else if resp != utils.OK {
|
||||
t.Error("Unexpected reply returned", resp)
|
||||
}
|
||||
}
|
||||
|
||||
func testTPLcrRulesGetTPLcrRulesAfterRemove(t *testing.T) {
|
||||
var reply *utils.TPRatingPlan
|
||||
if err := tpLcrRuleRPC.Call("ApierV1.GetTPLcrRule", &AttrGetTPLcrRules{TPid: "TPLRC1", LcrRuleId: tpLcrRuleID}, &reply); err == nil || err.Error() != utils.ErrNotFound.Error() {
|
||||
t.Error(err)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func testTPLcrRulesKillEngine(t *testing.T) {
|
||||
if err := engine.KillEngine(tpLcrRuleDelay); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user