Repaired test for Account actions

This commit is contained in:
andronache
2021-11-16 16:32:11 +02:00
committed by Dan Christian Bogos
parent f9356016fd
commit 3a07459d92

View File

@@ -21,337 +21,338 @@ along with this program. If not, see <http://www.gnu.org/licenses/>
package general_tests
// import (
// "net/rpc"
// "path"
// "reflect"
// "strconv"
// "testing"
// "time"
import (
"path"
"reflect"
"strconv"
"testing"
"time"
// "github.com/cgrates/cgrates/config"
// "github.com/cgrates/cgrates/engine"
// "github.com/cgrates/cgrates/utils"
// )
"github.com/cgrates/birpc"
"github.com/cgrates/birpc/context"
"github.com/cgrates/cgrates/config"
"github.com/cgrates/cgrates/engine"
"github.com/cgrates/cgrates/utils"
)
// var (
// accPrfCfgPath string
// accPrfCfg *config.CGRConfig
// accSRPC *rpc.Client
// accPrfConfigDIR string //run tests for specific configuration
var (
accPrfCfgPath string
accPrfCfg *config.CGRConfig
accSRPC *birpc.Client
accPrfConfigDIR string //run tests for specific configuration
// sTestsAccPrf = []func(t *testing.T){
// testAccActionsInitCfg,
// testAccActionsInitDataDb,
// testAccActionsResetStorDb,
// testAccActionsStartEngine,
// testAccActionsRPCConn,
// testAccActionsSetActionProfile,
// testAccActionsExecuteAction,
// testAccActionsSetActionProfile,
// testAccActionsExecuteAction2,
// testAccActionsGetAccountAfterActions,
// testAccActionsSetActionProfile2,
// testAccActionsExecuteAction3,
// testAccActionsGetAccountAfterRemActions,
// testAccActionsKillEngine,
// }
// )
sTestsAccPrf = []func(t *testing.T){
testAccActionsInitCfg,
testAccActionsInitDataDb,
testAccActionsResetStorDb,
testAccActionsStartEngine,
testAccActionsRPCConn,
testAccActionsSetActionProfile,
testAccActionsExecuteAction,
testAccActionsSetActionProfile,
testAccActionsExecuteAction2,
testAccActionsGetAccountAfterActions,
testAccActionsSetActionProfile2,
testAccActionsExecuteAction3,
testAccActionsGetAccountAfterRemActions,
testAccActionsKillEngine,
}
)
// //Test start here
// func TestAccActionsIT(t *testing.T) {
// switch *dbType {
// case utils.MetaInternal:
// accPrfConfigDIR = "tutinternal"
// case utils.MetaMySQL:
// accPrfConfigDIR = "tutmysql"
// case utils.MetaMongo:
// accPrfConfigDIR = "tutmongo"
// case utils.MetaPostgres:
// t.SkipNow()
// default:
// t.Fatal("Unknown Database type")
// }
// for _, stest := range sTestsAccPrf {
// t.Run(accPrfConfigDIR, stest)
// }
// }
//Test start here
func TestAccActionsIT(t *testing.T) {
switch *dbType {
case utils.MetaInternal:
accPrfConfigDIR = "tutinternal"
case utils.MetaMySQL:
accPrfConfigDIR = "tutmysql"
case utils.MetaMongo:
accPrfConfigDIR = "tutmongo"
case utils.MetaPostgres:
t.SkipNow()
default:
t.Fatal("Unknown Database type")
}
for _, stest := range sTestsAccPrf {
t.Run(accPrfConfigDIR, stest)
}
}
// func testAccActionsInitCfg(t *testing.T) {
// var err error
// accPrfCfgPath = path.Join(*dataDir, "conf", "samples", accPrfConfigDIR)
// if accPrfCfg, err = config.NewCGRConfigFromPath(accPrfCfgPath); err != nil {
// t.Error(err)
// }
// }
func testAccActionsInitCfg(t *testing.T) {
var err error
accPrfCfgPath = path.Join(*dataDir, "conf", "samples", accPrfConfigDIR)
if accPrfCfg, err = config.NewCGRConfigFromPath(context.Background(), accPrfCfgPath); err != nil {
t.Error(err)
}
}
// func testAccActionsInitDataDb(t *testing.T) {
// if err := engine.InitDataDB(accPrfCfg); err != nil {
// t.Fatal(err)
// }
// }
func testAccActionsInitDataDb(t *testing.T) {
if err := engine.InitDataDB(accPrfCfg); err != nil {
t.Fatal(err)
}
}
// // Wipe out the cdr database
// func testAccActionsResetStorDb(t *testing.T) {
// if err := engine.InitStorDB(accPrfCfg); err != nil {
// t.Fatal(err)
// }
// }
// Wipe out the cdr database
func testAccActionsResetStorDb(t *testing.T) {
if err := engine.InitStorDB(accPrfCfg); err != nil {
t.Fatal(err)
}
}
// // Start CGR Engine
// func testAccActionsStartEngine(t *testing.T) {
// if _, err := engine.StopStartEngine(accPrfCfgPath, *waitRater); err != nil {
// t.Fatal(err)
// }
// }
// Start CGR Engine
func testAccActionsStartEngine(t *testing.T) {
if _, err := engine.StopStartEngine(accPrfCfgPath, *waitRater); err != nil {
t.Fatal(err)
}
}
// // Connect rpc client to rater
// func testAccActionsRPCConn(t *testing.T) {
// var err error
// accSRPC, err = newRPCClient(accPrfCfg.ListenCfg()) // We connect over JSON so we can also troubleshoot if needed
// if err != nil {
// t.Fatal(err)
// }
// }
// Connect rpc client to rater
func testAccActionsRPCConn(t *testing.T) {
var err error
accSRPC, err = newRPCClient(accPrfCfg.ListenCfg()) // We connect over JSON so we can also troubleshoot if needed
if err != nil {
t.Fatal(err)
}
}
// func testAccActionsSetActionProfile(t *testing.T) {
// actPrf := &engine.ActionProfileWithAPIOpts{
// ActionProfile: &engine.ActionProfile{
// Tenant: "cgrates.org",
// ID: "CREATE_ACC",
// FilterIDs: []string{"*string:~*req.Account:1001"},
// Weight: 0,
// Targets: map[string]utils.StringSet{utils.MetaAccounts: {"1001": {}}},
// Schedule: utils.MetaASAP,
// Actions: []*engine.APAction{
// {
// ID: "SET_NEW_BAL",
// FilterIDs: []string{"*exists:*opts.BAL_NEW:"},
// Type: utils.MetaSetBalance,
// Diktats: []*engine.APDiktat{
// {
// Path: "*account.ThresholdIDs",
// Value: utils.MetaNone,
// },
// {
// Path: "*balance.MONETARY.Type",
// Value: utils.MetaConcrete,
// },
// {
// Path: "*balance.MONETARY.Units",
// Value: "1048576",
// },
// {
// Path: "*balance.MONETARY.Weights",
// Value: "`;0`",
// },
// {
// Path: "*balance.MONETARY.CostIncrements",
// Value: "`*string:~*req.ToR:*data;1024;0;0.01`",
// },
// },
// },
// {
// ID: "SET_ADD_BAL",
// FilterIDs: []string{"*exists:*opts.BAL_ADD:"},
// Type: utils.MetaAddBalance,
// Diktats: []*engine.APDiktat{
// {
// Path: "*balance.VOICE.Type",
// Value: utils.MetaAbstract,
// },
// {
// Path: "*balance.VOICE.Units",
// Value: strconv.FormatInt((3 * time.Hour).Nanoseconds(), 10),
// },
// {
// Path: "*balance.VOICE.FilterIDs",
// Value: "`*string:~*req.ToR:*voice`",
// },
// {
// Path: "*balance.VOICE.Weights",
// Value: "`;2`",
// },
// {
// Path: "*balance.VOICE.CostIncrements",
// Value: "`*string:~*req.ToR:*voice;1000000000;0;0.01`",
// },
// },
// },
// },
// },
// APIOpts: map[string]interface{}{},
// }
// var reply string
// if err := accSRPC.Call(utils.APIerSv1SetActionProfile, actPrf, &reply); err != nil {
// t.Error(err)
// } else if reply != utils.OK {
// t.Error("Unexpected reply returned", reply)
// }
// var result *engine.ActionProfile
// if err := accSRPC.Call(utils.APIerSv1GetActionProfile, &utils.TenantIDWithAPIOpts{
// TenantID: &utils.TenantID{Tenant: actPrf.Tenant, ID: actPrf.ID}}, &result); err != nil {
// t.Error(err)
// } else if !reflect.DeepEqual(actPrf.ActionProfile, result) {
// t.Errorf("Expecting : %+v, received: %+v", actPrf.ActionProfile, result)
// }
// }
func testAccActionsSetActionProfile(t *testing.T) {
actPrf := &engine.ActionProfileWithAPIOpts{
ActionProfile: &engine.ActionProfile{
Tenant: "cgrates.org",
ID: "CREATE_ACC",
FilterIDs: []string{"*string:~*req.Account:1001"},
Weight: 0,
Targets: map[string]utils.StringSet{utils.MetaAccounts: {"1001": {}}},
Schedule: utils.MetaASAP,
Actions: []*engine.APAction{
{
ID: "SET_NEW_BAL",
FilterIDs: []string{"*exists:*opts.BAL_NEW:"},
Type: utils.MetaSetBalance,
Diktats: []*engine.APDiktat{
{
Path: "*account.ThresholdIDs",
Value: utils.MetaNone,
},
{
Path: "*balance.MONETARY.Type",
Value: utils.MetaConcrete,
},
{
Path: "*balance.MONETARY.Units",
Value: "1048576",
},
{
Path: "*balance.MONETARY.Weights",
Value: "`;0`",
},
{
Path: "*balance.MONETARY.CostIncrements",
Value: "`*string:~*req.ToR:*data;1024;0;0.01`",
},
},
},
{
ID: "SET_ADD_BAL",
FilterIDs: []string{"*exists:*opts.BAL_ADD:"},
Type: utils.MetaAddBalance,
Diktats: []*engine.APDiktat{
{
Path: "*balance.VOICE.Type",
Value: utils.MetaAbstract,
},
{
Path: "*balance.VOICE.Units",
Value: strconv.FormatInt((3 * time.Hour).Nanoseconds(), 10),
},
{
Path: "*balance.VOICE.FilterIDs",
Value: "`*string:~*req.ToR:*voice`",
},
{
Path: "*balance.VOICE.Weights",
Value: "`;2`",
},
{
Path: "*balance.VOICE.CostIncrements",
Value: "`*string:~*req.ToR:*voice;1000000000;0;0.01`",
},
},
},
},
},
APIOpts: map[string]interface{}{},
}
var reply string
if err := accSRPC.Call(context.Background(), utils.AdminSv1SetActionProfile, actPrf, &reply); err != nil {
t.Error(err)
} else if reply != utils.OK {
t.Error("Unexpected reply returned", reply)
}
var result *engine.ActionProfile
if err := accSRPC.Call(context.Background(), utils.AdminSv1GetActionProfile, &utils.TenantIDWithAPIOpts{
TenantID: &utils.TenantID{Tenant: actPrf.Tenant, ID: actPrf.ID}}, &result); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(actPrf.ActionProfile, result) {
t.Errorf("Expecting : %+v, received: %+v", actPrf.ActionProfile, result)
}
}
// func testAccActionsExecuteAction(t *testing.T) {
// var reply string
// if err := accSRPC.Call(utils.ActionSv1ExecuteActions, &utils.CGREvent{
// Tenant: "cgrates.org",
// Event: map[string]interface{}{
// "Account": 1001,
// },
// APIOpts: map[string]interface{}{
// "BAL_NEW": true,
// "BAL_ADD": true,
// },
// }, &reply); err != nil {
// t.Error(err)
// } else if reply != utils.OK {
// t.Error("Unexpected reply returned", reply)
// }
// }
func testAccActionsExecuteAction(t *testing.T) {
var reply string
if err := accSRPC.Call(context.Background(), utils.ActionSv1ExecuteActions, &utils.CGREvent{
Tenant: "cgrates.org",
Event: map[string]interface{}{
"Account": 1001,
},
APIOpts: map[string]interface{}{
"BAL_NEW": true,
"BAL_ADD": true,
},
}, &reply); err != nil {
t.Error(err)
} else if reply != utils.OK {
t.Error("Unexpected reply returned", reply)
}
}
// func testAccActionsExecuteAction2(t *testing.T) {
// var reply string
// if err := accSRPC.Call(utils.ActionSv1ExecuteActions, &utils.CGREvent{
// Tenant: "cgrates.org",
// Event: map[string]interface{}{
// "Account": 1001,
// },
// APIOpts: map[string]interface{}{
// "BAL_NEW": true,
// "BAL_ADD": true,
// utils.OptsActionsActionProfileIDs: []string{"CREATE_ACC"},
// },
// }, &reply); err != nil {
// t.Error(err)
// } else if reply != utils.OK {
// t.Error("Unexpected reply returned", reply)
// }
// }
func testAccActionsExecuteAction2(t *testing.T) {
var reply string
if err := accSRPC.Call(context.Background(), utils.ActionSv1ExecuteActions, &utils.CGREvent{
Tenant: "cgrates.org",
Event: map[string]interface{}{
"Account": 1001,
},
APIOpts: map[string]interface{}{
"BAL_NEW": true,
"BAL_ADD": true,
utils.OptsActionsProfileIDs: []string{"CREATE_ACC"},
},
}, &reply); err != nil {
t.Error(err)
} else if reply != utils.OK {
t.Error("Unexpected reply returned", reply)
}
}
// func testAccActionsGetAccountAfterActions(t *testing.T) {
// accPrf := &utils.Account{
// Tenant: "cgrates.org",
// ID: "1001",
// Balances: map[string]*utils.Balance{
// "MONETARY": {
// ID: "MONETARY",
// Weights: utils.DynamicWeights{{}},
// Type: utils.MetaConcrete,
// Units: utils.NewDecimalFromFloat64(1048576),
// CostIncrements: []*utils.CostIncrement{{
// FilterIDs: []string{"*string:~*req.ToR:*data"},
// Increment: utils.NewDecimalFromFloat64(1024),
// FixedFee: utils.NewDecimalFromFloat64(0),
// RecurrentFee: utils.NewDecimalFromFloat64(0.01),
// }},
// },
// "VOICE": {
// ID: "VOICE",
// FilterIDs: []string{"*string:~*req.ToR:*voice"},
// Weights: utils.DynamicWeights{{Weight: 2}},
// Type: utils.MetaAbstract,
// Units: utils.NewDecimalFromFloat64(2 * 10800000000000),
// CostIncrements: []*utils.CostIncrement{{
// FilterIDs: []string{"*string:~*req.ToR:*voice"},
// Increment: utils.NewDecimalFromFloat64(1000000000),
// FixedFee: utils.NewDecimalFromFloat64(0),
// RecurrentFee: utils.NewDecimalFromFloat64(0.01),
// }},
// },
// },
// ThresholdIDs: []string{utils.MetaNone},
// }
// var result *utils.Account
// if err := accSRPC.Call(utils.APIerSv1GetAccount, &utils.TenantIDWithAPIOpts{
// TenantID: &utils.TenantID{Tenant: "cgrates.org", ID: "1001"}}, &result); err != nil {
// t.Error(err)
// } else if !reflect.DeepEqual(accPrf, result) {
// t.Errorf("Expecting : %s, received: %s", utils.ToJSON(accPrf), utils.ToJSON(result))
// }
// }
func testAccActionsGetAccountAfterActions(t *testing.T) {
accPrf := &utils.Account{
Tenant: "cgrates.org",
ID: "1001",
Balances: map[string]*utils.Balance{
"MONETARY": {
ID: "MONETARY",
Weights: utils.DynamicWeights{{}},
Type: utils.MetaConcrete,
Units: utils.NewDecimalFromFloat64(1048576),
CostIncrements: []*utils.CostIncrement{{
FilterIDs: []string{"*string:~*req.ToR:*data"},
Increment: utils.NewDecimalFromFloat64(1024),
FixedFee: utils.NewDecimalFromFloat64(0),
RecurrentFee: utils.NewDecimalFromFloat64(0.01),
}},
},
"VOICE": {
ID: "VOICE",
FilterIDs: []string{"*string:~*req.ToR:*voice"},
Weights: utils.DynamicWeights{{Weight: 2}},
Type: utils.MetaAbstract,
Units: utils.NewDecimalFromFloat64(2 * 10800000000000),
CostIncrements: []*utils.CostIncrement{{
FilterIDs: []string{"*string:~*req.ToR:*voice"},
Increment: utils.NewDecimalFromFloat64(1000000000),
FixedFee: utils.NewDecimalFromFloat64(0),
RecurrentFee: utils.NewDecimalFromFloat64(0.01),
}},
},
},
ThresholdIDs: []string{utils.MetaNone},
}
var result *utils.Account
if err := accSRPC.Call(context.Background(), utils.AdminSv1GetAccount, &utils.TenantIDWithAPIOpts{
TenantID: &utils.TenantID{Tenant: "cgrates.org", ID: "1001"}}, &result); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(accPrf, result) {
t.Errorf("Expecting : %s, received: %s", utils.ToJSON(accPrf), utils.ToJSON(result))
}
}
// func testAccActionsSetActionProfile2(t *testing.T) {
// actPrf := &engine.ActionProfileWithAPIOpts{
// ActionProfile: &engine.ActionProfile{
// Tenant: "cgrates.org",
// ID: "REM_ACC",
// FilterIDs: []string{"*string:~*req.Account:1001"},
// Weight: 0,
// Targets: map[string]utils.StringSet{utils.MetaAccounts: {"1001": {}}},
// Schedule: utils.MetaASAP,
// Actions: []*engine.APAction{
// {
// ID: "REM_BAL",
// Type: utils.MetaRemBalance,
// Diktats: []*engine.APDiktat{
// {
// Path: "MONETARY",
// },
// {
// Path: "VOICE",
// },
// },
// },
// },
// },
// APIOpts: map[string]interface{}{},
// }
// var reply string
// if err := accSRPC.Call(utils.APIerSv1SetActionProfile, actPrf, &reply); err != nil {
// t.Error(err)
// } else if reply != utils.OK {
// t.Error("Unexpected reply returned", reply)
// }
// var result *engine.ActionProfile
// if err := accSRPC.Call(utils.APIerSv1GetActionProfile, &utils.TenantIDWithAPIOpts{
// TenantID: &utils.TenantID{Tenant: actPrf.Tenant, ID: actPrf.ID}}, &result); err != nil {
// t.Error(err)
// } else if !reflect.DeepEqual(actPrf.ActionProfile, result) {
// t.Errorf("Expecting : %+v, received: %+v", actPrf.ActionProfile, result)
// }
// }
func testAccActionsSetActionProfile2(t *testing.T) {
actPrf := &engine.ActionProfileWithAPIOpts{
ActionProfile: &engine.ActionProfile{
Tenant: "cgrates.org",
ID: "REM_ACC",
FilterIDs: []string{"*string:~*req.Account:1001"},
Weight: 0,
Targets: map[string]utils.StringSet{utils.MetaAccounts: {"1001": {}}},
Schedule: utils.MetaASAP,
Actions: []*engine.APAction{
{
ID: "REM_BAL",
Type: utils.MetaRemBalance,
Diktats: []*engine.APDiktat{
{
Path: "MONETARY",
},
{
Path: "VOICE",
},
},
},
},
},
APIOpts: map[string]interface{}{},
}
var reply string
if err := accSRPC.Call(context.Background(), utils.AdminSv1SetActionProfile, actPrf, &reply); err != nil {
t.Error(err)
} else if reply != utils.OK {
t.Error("Unexpected reply returned", reply)
}
var result *engine.ActionProfile
if err := accSRPC.Call(context.Background(), utils.AdminSv1GetActionProfile, &utils.TenantIDWithAPIOpts{
TenantID: &utils.TenantID{Tenant: actPrf.Tenant, ID: actPrf.ID}}, &result); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(actPrf.ActionProfile, result) {
t.Errorf("Expecting : %+v, received: %+v", actPrf.ActionProfile, result)
}
}
// func testAccActionsExecuteAction3(t *testing.T) {
// var reply string
// if err := accSRPC.Call(utils.ActionSv1ExecuteActions, &utils.CGREvent{
// Tenant: "cgrates.org",
// Event: map[string]interface{}{
// "Account": 1001,
// },
// APIOpts: map[string]interface{}{
// utils.OptsActionsActionProfileIDs: []string{"REM_ACC"},
// },
// }, &reply); err != nil {
// t.Error(err)
// } else if reply != utils.OK {
// t.Error("Unexpected reply returned", reply)
// }
// }
func testAccActionsExecuteAction3(t *testing.T) {
var reply string
if err := accSRPC.Call(context.Background(), utils.ActionSv1ExecuteActions, &utils.CGREvent{
Tenant: "cgrates.org",
Event: map[string]interface{}{
"Account": 1001,
},
APIOpts: map[string]interface{}{
utils.OptsActionsProfileIDs: []string{"REM_ACC"},
},
}, &reply); err != nil {
t.Error(err)
} else if reply != utils.OK {
t.Error("Unexpected reply returned", reply)
}
}
// func testAccActionsGetAccountAfterRemActions(t *testing.T) {
// accPrf := &utils.Account{
// Tenant: "cgrates.org",
// ID: "1001",
// Balances: map[string]*utils.Balance{},
// ThresholdIDs: []string{utils.MetaNone},
// }
// var result *utils.Account
// if err := accSRPC.Call(utils.APIerSv1GetAccount, &utils.TenantIDWithAPIOpts{
// TenantID: &utils.TenantID{Tenant: "cgrates.org", ID: "1001"}}, &result); err != nil {
// t.Error(err)
// } else if !reflect.DeepEqual(accPrf, result) {
// t.Errorf("Expecting : %s, received: %s", utils.ToJSON(accPrf), utils.ToJSON(result))
// }
// }
func testAccActionsGetAccountAfterRemActions(t *testing.T) {
accPrf := &utils.Account{
Tenant: "cgrates.org",
ID: "1001",
Balances: map[string]*utils.Balance{},
ThresholdIDs: []string{utils.MetaNone},
}
var result *utils.Account
if err := accSRPC.Call(context.Background(), utils.AdminSv1GetAccount, &utils.TenantIDWithAPIOpts{
TenantID: &utils.TenantID{Tenant: "cgrates.org", ID: "1001"}}, &result); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(accPrf, result) {
t.Errorf("Expecting : %s, received: %s", utils.ToJSON(accPrf), utils.ToJSON(result))
}
}
// func testAccActionsKillEngine(t *testing.T) {
// if err := engine.KillEngine(100); err != nil {
// t.Error(err)
// }
// }
func testAccActionsKillEngine(t *testing.T) {
if err := engine.KillEngine(100); err != nil {
t.Error(err)
}
}