Files
cgrates/apier/v1/apier_it_test.go
2025-10-29 19:44:04 +01:00

2639 lines
114 KiB
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 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 v1
import (
"encoding/json"
"fmt"
"net/http"
"net/url"
"os"
"path"
"reflect"
"sort"
"strings"
"testing"
"time"
"github.com/cgrates/birpc"
"github.com/cgrates/birpc/context"
"github.com/cgrates/cgrates/config"
"github.com/cgrates/cgrates/dispatchers"
"github.com/cgrates/cgrates/ees"
"github.com/cgrates/cgrates/engine"
"github.com/cgrates/cgrates/scheduler"
"github.com/cgrates/cgrates/servmanager"
"github.com/cgrates/cgrates/utils"
"github.com/cgrates/ltcache"
amqp "github.com/rabbitmq/amqp091-go"
)
// ToDo: Replace rpc.Client with internal rpc server and Apier using internal map as both data and stor so we can run the tests non-local
/*
README:
Enable local tests by passing '-local' to the go test command
It is expected that the data folder of CGRateS exists at path /usr/share/cgrates/data or passed via command arguments.
Prior running the tests, create database and users by running:
mysql -pyourrootpwd < /usr/share/cgrates/data/storage/mysql/create_db_with_users.sql
What these tests do:
* Flush tables in storDb to start clean.
* Start engine with default configuration and give it some time to listen (here caching can slow down, hence the command argument parameter).
* Connect rpc client depending on encoding defined in configuration.
* Execute remote Apis and test their replies(follow testtp scenario so we can test load in dataDb also).
*/
var (
cfgPath string
cfg *config.CGRConfig
rater *birpc.Client
APIerSv1ConfigDIR string
apierTests = []func(t *testing.T){
testApierLoadConfig,
testApierCreateDirs,
testApierInitDataDb,
testApierInitStorDb,
testApierStartEngine,
testApierRpcConn,
testApierTPTiming,
testApierTPDestination,
testApierTPRate,
testApierTPDestinationRate,
testApierTPRatingPlan,
testApierTPRatingProfile,
testApierTPActions,
testApierTPActionPlan,
testApierTPActionTriggers,
testApierTPAccountActions,
testApierLoadRatingPlan,
testApierLoadRatingProfile,
testApierLoadRatingProfileWithoutTenant,
testApierLoadAccountActions,
testApierReloadScheduler,
testApierSetRatingProfile,
testApierSetRatingProfileWithoutTenant,
testApierRemoveRatingProfilesWithoutTenant,
testAPIerSv1GetRatingProfile,
testAPIerSv1GetRatingProfileWithoutTenant,
testAPIerSv1GetRatingProfileIDsWithoutTenant,
testApierReloadCache,
testApierGetActionTrigger,
testApierGetDestination,
testApierGetRatingPlan,
testApierRemoveRatingPlan,
testApierAddBalance,
testApierExecuteAction,
testApierExecuteActionWithoutTenant,
testApierSetActions,
testApierGetActions,
testApierSetActionPlan,
testApierAddTriggeredAction,
testApierGetAccountActionTriggers,
testApierAddTriggeredAction2,
testApierGetAccountActionTriggers2,
testApierSetAccountActionTriggers,
testApierRemAccountActionTriggers,
testApierSetAccount,
testApierGetAccountActionPlan,
testApierGetAccountActionPlanWithoutTenant,
testApierITGetScheduledActionsForAccount,
testApierRemUniqueIDActionTiming,
testApierRemUniqueIDActionTimingWithoutTenant,
testApierGetAccount,
testApierTriggersExecute,
testApierResetDataBeforeLoadFromFolder,
testApierLoadTariffPlanFromFolder,
testApierComputeReverse,
testApierResetDataAfterLoadFromFolder,
testApierSetChargerS,
testApierGetAccountAfterLoad,
testApierResponderGetCost,
testApierMaxDebitInexistentAcnt,
testApierCdrServer,
testApierITGetCdrs,
testApierITProcessCdr,
testApierGetCallCostLog,
testApierITSetDestination,
testApierITGetScheduledActions,
testApierITGetDataCost,
testApierITGetCost,
testApierInitDataDb2,
testApierInitStorDb2,
testApierReloadCache2,
testApierReloadScheduler2,
testApierImportTPFromFolderPath,
testApierLoadTariffPlanFromStorDbDryRun,
testApierGetCacheStats2,
testApierLoadTariffPlanFromStorDb,
testApierStartStopServiceStatus,
testApierReplayFldPosts,
testApierGetDataDBVesions,
testApierGetStorDBVesions,
testApierBackwardsCompatible,
testApierStopEngine,
//start test for cache options
testApierLoadConfig,
testApierCreateDirs,
testApierInitDataDb,
testApierInitStorDb,
testApierStartEngine,
testApierRpcConn,
testApierTPTiming,
testApierTPDestination,
testApierTPRate,
testApierTPDestinationRate,
testApierTPRatingPlan,
testApierLoadRatingPlan,
testRatingProfileCachingMetaNone,
testRatingProfileCachingMetaLoad,
testRatingProfileCachingMetaReload1,
testRatingProfileCachingMetaReload2,
testRatingProfileCachingMetaRemove,
testApierStopEngine,
}
)
func TestApierIT(t *testing.T) {
switch *utils.DBType {
case utils.MetaInternal:
t.SkipNow() // need tests redesign
case utils.MetaMySQL:
APIerSv1ConfigDIR = "apier_mysql"
case utils.MetaMongo:
APIerSv1ConfigDIR = "apier_mongo"
case utils.MetaPostgres:
t.SkipNow()
default:
t.Fatal("Unknown Database type")
}
for _, stest := range apierTests {
t.Run(APIerSv1ConfigDIR, stest)
}
}
func testApierLoadConfig(t *testing.T) {
var err error
cfgPath = path.Join(*utils.DataDir, "conf", "samples", APIerSv1ConfigDIR) // no need for a new config with *gob transport in this case
if cfg, err = config.NewCGRConfigFromPath(cfgPath); err != nil {
t.Error(err)
}
}
func testApierCreateDirs(t *testing.T) {
for _, pathDir := range []string{"/var/log/cgrates/ers/in", "/var/log/cgrates/ers/out"} {
if err := os.RemoveAll(pathDir); err != nil {
t.Fatal("Error removing folder: ", pathDir, err)
}
if err := os.MkdirAll(pathDir, 0755); err != nil {
t.Fatal("Error creating folder: ", pathDir, err)
}
}
}
func testApierInitDataDb(t *testing.T) {
if err := engine.InitDataDB(cfg); err != nil {
t.Fatal(err)
}
}
// Empty tables before using them
func testApierInitStorDb(t *testing.T) {
if err := engine.InitStorDb(cfg); err != nil {
t.Fatal(err)
}
}
// Start engine
func testApierStartEngine(t *testing.T) {
if _, err := engine.StopStartEngine(cfgPath, *utils.WaitRater); err != nil {
t.Fatal(err)
}
}
// Connect rpc client to rater
func testApierRpcConn(t *testing.T) {
var err error
rater, err = newRPCClient(cfg.ListenCfg()) // We connect over JSON so we can also troubleshoot if needed
if err != nil {
t.Fatal("Could not connect to rater: ", err.Error())
}
}
// Test here TPTiming APIs
func testApierTPTiming(t *testing.T) {
// ALWAYS,*any,*any,*any,*any,00:00:00
tmAlways := &utils.ApierTPTiming{TPid: utils.TestSQL,
ID: "ALWAYS",
Years: utils.MetaAny,
Months: utils.MetaAny,
MonthDays: utils.MetaAny,
WeekDays: utils.MetaAny,
Time: "00:00:00",
}
tmAlways2 := new(utils.ApierTPTiming)
*tmAlways2 = *tmAlways
tmAlways2.ID = "ALWAYS2"
tmAsap := &utils.ApierTPTiming{
TPid: utils.TestSQL,
ID: "ASAP",
Years: utils.MetaAny,
Months: utils.MetaAny,
MonthDays: utils.MetaAny,
WeekDays: utils.MetaAny,
Time: "*asap",
}
var reply string
for _, tm := range []*utils.ApierTPTiming{tmAlways, tmAsap, tmAlways2} {
if err := rater.Call(context.Background(), utils.APIerSv1SetTPTiming, &tm, &reply); err != nil {
t.Error("Got error on APIerSv1.SetTPTiming: ", err.Error())
} else if reply != utils.OK {
t.Error("Unexpected reply received when calling APIerSv1.SetTPTiming: ", reply)
}
}
// Check second set
if err := rater.Call(context.Background(), utils.APIerSv1SetTPTiming, &tmAlways, &reply); err != nil {
t.Error("Got error on second APIerSv1.SetTPTiming: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.SetTPTiming got reply: ", reply)
}
// Check missing params
if err := rater.Call(context.Background(), utils.APIerSv1SetTPTiming, new(utils.ApierTPTiming), &reply); err == nil {
t.Error("Calling APIerSv1.SetTPTiming, expected error, received: ", reply)
} else if err.Error() != "MANDATORY_IE_MISSING: [TPid ID Years Months MonthDays WeekDays Time]" {
t.Error("Calling APIerSv1.SetTPTiming got unexpected error: ", err.Error())
}
// Test get
var rplyTmAlways2 *utils.ApierTPTiming
if err := rater.Call(context.Background(), utils.APIerSv1GetTPTiming, &AttrGetTPTiming{tmAlways2.TPid, tmAlways2.ID}, &rplyTmAlways2); err != nil {
t.Error("Calling APIerSv1.GetTPTiming, got error: ", err.Error())
} else if !reflect.DeepEqual(tmAlways2, rplyTmAlways2) {
t.Errorf("Calling APIerSv1.GetTPTiming expected: %v, received: %v", tmAlways, rplyTmAlways2)
}
// Test remove
if err := rater.Call(context.Background(), utils.APIerSv1RemoveTPTiming, AttrGetTPTiming{tmAlways2.TPid, tmAlways2.ID}, &reply); err != nil {
t.Error("Calling APIerSv1.RemoveTPTiming, got error: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.RemoveTPTiming received: ", reply)
}
// Test getIds
var rplyTmIds []string
expectedTmIds := []string{"ALWAYS", "ASAP"}
if err := rater.Call(context.Background(), utils.APIerSv1GetTPTimingIds, &AttrGetTPTimingIds{tmAlways.TPid, utils.PaginatorWithSearch{}}, &rplyTmIds); err != nil {
t.Error("Calling APIerSv1.GetTPTimingIds, got error: ", err.Error())
}
sort.Strings(expectedTmIds)
sort.Strings(rplyTmIds)
if !reflect.DeepEqual(expectedTmIds, rplyTmIds) {
t.Errorf("Calling APIerSv1.GetTPTimingIds expected: %v, received: %v", expectedTmIds, rplyTmIds)
}
}
// Test here TPTiming APIs
func testApierTPDestination(t *testing.T) {
var reply string
dstDe := &utils.TPDestination{TPid: utils.TestSQL, ID: "GERMANY", Prefixes: []string{"+49"}}
dstDeMobile := &utils.TPDestination{TPid: utils.TestSQL, ID: "GERMANY_MOBILE", Prefixes: []string{"+4915", "+4916", "+4917"}}
dstFs := &utils.TPDestination{TPid: utils.TestSQL, ID: "FS_USERS", Prefixes: []string{"10"}}
dstDe2 := new(utils.TPDestination)
*dstDe2 = *dstDe // Data which we use for remove, still keeping the sample data to check proper loading
dstDe2.ID = "GERMANY2"
for _, dst := range []*utils.TPDestination{dstDe, dstDeMobile, dstFs, dstDe2} {
if err := rater.Call(context.Background(), utils.APIerSv1SetTPDestination, dst, &reply); err != nil {
t.Error("Got error on APIerSv1.SetTPDestination: ", err.Error())
} else if reply != utils.OK {
t.Error("Unexpected reply received when calling APIerSv1.SetTPDestination: ", reply)
}
}
// Check second set
if err := rater.Call(context.Background(), utils.APIerSv1SetTPDestination, dstDe2, &reply); err != nil {
t.Error("Got error on second APIerSv1.SetTPDestination: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.SetTPDestination got reply: ", reply)
}
// Check missing params
if err := rater.Call(context.Background(), utils.APIerSv1SetTPDestination, new(utils.TPDestination), &reply); err == nil {
t.Error("Calling APIerSv1.SetTPDestination, expected error, received: ", reply)
} else if err.Error() != "MANDATORY_IE_MISSING: [TPid ID Prefixes]" {
t.Error("Calling APIerSv1.SetTPDestination got unexpected error: ", err.Error())
}
// Test get
var rplyDstDe2 *utils.TPDestination
if err := rater.Call(context.Background(), utils.APIerSv1GetTPDestination, &AttrGetTPDestination{dstDe2.TPid, dstDe2.ID}, &rplyDstDe2); err != nil {
t.Error("Calling APIerSv1.GetTPDestination, got error: ", err.Error())
} else if !reflect.DeepEqual(dstDe2, rplyDstDe2) {
t.Errorf("Calling APIerSv1.GetTPDestination expected: %v, received: %v", dstDe2, rplyDstDe2)
}
// Test remove
if err := rater.Call(context.Background(), utils.APIerSv1RemoveTPDestination, &AttrGetTPDestination{dstDe2.TPid, dstDe2.ID}, &reply); err != nil {
t.Error("Calling APIerSv1.RemoveTPTiming, got error: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.RemoveTPTiming received: ", reply)
}
// Test getIds
var rplyDstIds []string
expectedDstIds := []string{"FS_USERS", "GERMANY", "GERMANY_MOBILE"}
if err := rater.Call(context.Background(), utils.APIerSv1GetTPDestinationIDs, &AttrGetTPDestinationIds{TPid: dstDe.TPid}, &rplyDstIds); err != nil {
t.Error("Calling APIerSv1.GetTPDestinationIDs, got error: ", err.Error())
}
sort.Strings(expectedDstIds)
sort.Strings(rplyDstIds)
if !reflect.DeepEqual(expectedDstIds, rplyDstIds) {
t.Errorf("Calling APIerSv1.GetTPDestinationIDs expected: %v, received: %v", expectedDstIds, rplyDstIds)
}
}
// Test here TPRateRALs APIs
func testApierTPRate(t *testing.T) {
var reply string
rt := &utils.TPRateRALs{TPid: utils.TestSQL, ID: "RT_FS_USERS", RateSlots: []*utils.RateSlot{
{ConnectFee: 0, Rate: 0, RateUnit: "60s", RateIncrement: "60s", GroupIntervalStart: "0s"},
}}
rt2 := new(utils.TPRateRALs)
*rt2 = *rt
rt2.ID = "RT_FS_USERS2"
for _, r := range []*utils.TPRateRALs{rt, rt2} {
if err := rater.Call(context.Background(), utils.APIerSv1SetTPRate, r, &reply); err != nil {
t.Error("Got error on APIerSv1.SetTPRate: ", err.Error())
} else if reply != utils.OK {
t.Error("Unexpected reply received when calling APIerSv1.SetTPRate: ", reply)
}
}
// Check second set
if err := rater.Call(context.Background(), utils.APIerSv1SetTPRate, rt2, &reply); err != nil {
t.Error("Got error on second APIerSv1.SetTPRate: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.SetTPRate got reply: ", reply)
}
// Check missing params
if err := rater.Call(context.Background(), utils.APIerSv1SetTPRate, new(utils.TPRateRALs), &reply); err == nil {
t.Error("Calling APIerSv1.SetTPDestination, expected error, received: ", reply)
} else if err.Error() != "MANDATORY_IE_MISSING: [TPid ID RateSlots]" {
t.Error("Calling APIerSv1.SetTPRate got unexpected error: ", err.Error())
}
// Test get
var rplyRt2 *utils.TPRateRALs
if err := rater.Call(context.Background(), utils.APIerSv1GetTPRate, &AttrGetTPRate{rt2.TPid, rt2.ID}, &rplyRt2); err != nil {
t.Error("Calling APIerSv1.GetTPRate, got error: ", err.Error())
} else if !reflect.DeepEqual(rt2, rplyRt2) {
t.Errorf("Calling APIerSv1.GetTPRate expected: %+v, received: %+v", rt2, rplyRt2)
}
// Test remove
if err := rater.Call(context.Background(), utils.APIerSv1RemoveTPRate, &AttrGetTPRate{rt2.TPid, rt2.ID}, &reply); err != nil {
t.Error("Calling APIerSv1.RemoveTPRate, got error: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.RemoveTPRate received: ", reply)
}
// Test getIds
var rplyRtIds []string
expectedRtIds := []string{"RT_FS_USERS"}
if err := rater.Call(context.Background(), utils.APIerSv1GetTPRateIds, &AttrGetTPRateIds{rt.TPid, utils.PaginatorWithSearch{}}, &rplyRtIds); err != nil {
t.Error("Calling APIerSv1.GetTPRateIds, got error: ", err.Error())
} else if !reflect.DeepEqual(expectedRtIds, rplyRtIds) {
t.Errorf("Calling APIerSv1.GetTPDestinationIDs expected: %v, received: %v", expectedRtIds, rplyRtIds)
}
}
// Test here TPDestinationRate APIs
func testApierTPDestinationRate(t *testing.T) {
var reply string
dr := &utils.TPDestinationRate{TPid: utils.TestSQL, ID: "DR_FREESWITCH_USERS", DestinationRates: []*utils.DestinationRate{
{DestinationId: "FS_USERS", RateId: "RT_FS_USERS", RoundingMethod: "*up", RoundingDecimals: 2},
}}
drDe := &utils.TPDestinationRate{TPid: utils.TestSQL, ID: "DR_FREESWITCH_USERS", DestinationRates: []*utils.DestinationRate{
{DestinationId: "GERMANY_MOBILE", RateId: "RT_FS_USERS", RoundingMethod: "*up", RoundingDecimals: 2},
}}
dr2 := new(utils.TPDestinationRate)
*dr2 = *dr
dr2.ID = utils.TestSQL
for _, d := range []*utils.TPDestinationRate{dr, dr2, drDe} {
if err := rater.Call(context.Background(), utils.APIerSv1SetTPDestinationRate, d, &reply); err != nil {
t.Error("Got error on APIerSv1.SetTPDestinationRate: ", err.Error())
} else if reply != utils.OK {
t.Error("Unexpected reply received when calling APIerSv1.SetTPDestinationRate: ", reply)
}
}
// Check second set
if err := rater.Call(context.Background(), utils.APIerSv1SetTPDestinationRate, dr2, &reply); err != nil {
t.Error("Got error on second APIerSv1.SetTPDestinationRate: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.SetTPDestinationRate got reply: ", reply)
}
// Check missing params
if err := rater.Call(context.Background(), utils.APIerSv1SetTPDestinationRate, new(utils.TPDestinationRate), &reply); err == nil {
t.Error("Calling APIerSv1.SetTPDestination, expected error, received: ", reply)
} else if err.Error() != "MANDATORY_IE_MISSING: [TPid ID DestinationRates]" {
t.Error("Calling APIerSv1.SetTPDestinationRate got unexpected error: ", err.Error())
}
// Test get
var rplyDr2 *utils.TPDestinationRate
if err := rater.Call(context.Background(), utils.APIerSv1GetTPDestinationRate, &AttrGetTPDestinationRate{dr2.TPid, dr2.ID, utils.Paginator{}}, &rplyDr2); err != nil {
t.Error("Calling APIerSv1.GetTPDestinationRate, got error: ", err.Error())
} else if !reflect.DeepEqual(dr2, rplyDr2) {
t.Errorf("Calling APIerSv1.GetTPDestinationRate expected: %v, received: %v", dr2, rplyDr2)
}
// Test remove
if err := rater.Call(context.Background(), utils.APIerSv1RemoveTPDestinationRate, &AttrGetTPDestinationRate{dr2.TPid, dr2.ID, utils.Paginator{}}, &reply); err != nil {
t.Error("Calling APIerSv1.RemoveTPRate, got error: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.RemoveTPRate received: ", reply)
}
// Test getIds
var rplyDrIds []string
expectedDrIds := []string{"DR_FREESWITCH_USERS"}
if err := rater.Call(context.Background(), utils.APIerSv1GetTPDestinationRateIds, &AttrTPDestinationRateIds{dr.TPid, utils.PaginatorWithSearch{}}, &rplyDrIds); err != nil {
t.Error("Calling APIerSv1.GetTPDestinationRateIds, got error: ", err.Error())
} else if !reflect.DeepEqual(expectedDrIds, rplyDrIds) {
t.Errorf("Calling APIerSv1.GetTPDestinationRateIds expected: %v, received: %v", expectedDrIds, rplyDrIds)
}
}
// Test here TPRatingPlan APIs
func testApierTPRatingPlan(t *testing.T) {
var reply string
rp := &utils.TPRatingPlan{TPid: utils.TestSQL, ID: "RETAIL1", RatingPlanBindings: []*utils.TPRatingPlanBinding{
{DestinationRatesId: "DR_FREESWITCH_USERS", TimingId: "ALWAYS", Weight: 10},
}}
rpTst := new(utils.TPRatingPlan)
*rpTst = *rp
rpTst.ID = utils.TestSQL
for _, rpl := range []*utils.TPRatingPlan{rp, rpTst} {
if err := rater.Call(context.Background(), utils.APIerSv1SetTPRatingPlan, rpl, &reply); err != nil {
t.Error("Got error on APIerSv1.SetTPRatingPlan: ", err.Error())
} else if reply != utils.OK {
t.Error("Unexpected reply received when calling APIerSv1.SetTPRatingPlan: ", reply)
}
}
// Check second set
if err := rater.Call(context.Background(), utils.APIerSv1SetTPRatingPlan, rpTst, &reply); err != nil {
t.Error("Got error on second APIerSv1.SetTPRatingPlan: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.SetTPRatingPlan got reply: ", reply)
}
// Check missing params
if err := rater.Call(context.Background(), utils.APIerSv1SetTPRatingPlan, new(utils.TPRatingPlan), &reply); err == nil {
t.Error("Calling APIerSv1.SetTPRatingPlan, expected error, received: ", reply)
} else if err.Error() != "MANDATORY_IE_MISSING: [TPid ID RatingPlanBindings]" {
t.Error("Calling APIerSv1.SetTPRatingPlan got unexpected error: ", err.Error())
}
// Test get
var rplyRpTst *utils.TPRatingPlan
if err := rater.Call(context.Background(), utils.APIerSv1GetTPRatingPlan, &AttrGetTPRatingPlan{TPid: rpTst.TPid, ID: rpTst.ID}, &rplyRpTst); err != nil {
t.Error("Calling APIerSv1.GetTPRatingPlan, got error: ", err.Error())
} else if !reflect.DeepEqual(rpTst, rplyRpTst) {
t.Errorf("Calling APIerSv1.GetTPRatingPlan expected: %v, received: %v", rpTst, rplyRpTst)
}
// Test remove
if err := rater.Call(context.Background(), utils.APIerSv1RemoveTPRatingPlan, &AttrGetTPRatingPlan{TPid: rpTst.TPid, ID: rpTst.ID}, &reply); err != nil {
t.Error("Calling APIerSv1.RemoveTPRatingPlan, got error: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.RemoveTPRatingPlan received: ", reply)
}
// Test getIds
var rplyRpIds []string
expectedRpIds := []string{"RETAIL1"}
if err := rater.Call(context.Background(), utils.APIerSv1GetTPRatingPlanIds, &AttrGetTPRatingPlanIds{rp.TPid, utils.PaginatorWithSearch{}}, &rplyRpIds); err != nil {
t.Error("Calling APIerSv1.GetTPRatingPlanIds, got error: ", err.Error())
} else if !reflect.DeepEqual(expectedRpIds, rplyRpIds) {
t.Errorf("Calling APIerSv1.GetTPRatingPlanIds expected: %v, received: %v", expectedRpIds, rplyRpIds)
}
}
// Test here TPRatingPlan APIs
func testApierTPRatingProfile(t *testing.T) {
var reply string
rpf := &utils.TPRatingProfile{
TPid: utils.TestSQL,
LoadId: utils.TestSQL,
Tenant: "cgrates.org",
Category: "call",
Subject: utils.MetaAny,
RatingPlanActivations: []*utils.TPRatingActivation{{
ActivationTime: "2012-01-01T00:00:00Z",
RatingPlanId: "RETAIL1",
FallbackSubjects: utils.EmptyString,
}},
}
rpfTst := new(utils.TPRatingProfile)
*rpfTst = *rpf
rpfTst.Subject = utils.TestSQL
for _, rp := range []*utils.TPRatingProfile{rpf, rpfTst} {
if err := rater.Call(context.Background(), utils.APIerSv1SetTPRatingProfile, rp, &reply); err != nil {
t.Error("Got error on APIerSv1.SetTPRatingProfile: ", err.Error())
} else if reply != utils.OK {
t.Error("Unexpected reply received when calling APIerSv1.SetTPRatingProfile: ", reply)
}
}
// Check second set
if err := rater.Call(context.Background(), utils.APIerSv1SetTPRatingProfile, rpfTst, &reply); err != nil {
t.Error("Got error on second APIerSv1.SetTPRatingProfile: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.SetTPRatingProfile got reply: ", reply)
}
// Check missing params
if err := rater.Call(context.Background(), utils.APIerSv1SetTPRatingProfile, new(utils.TPRatingProfile), &reply); err == nil {
t.Error("Calling APIerSv1.SetTPRatingProfile, expected error, received: ", reply)
} else if err.Error() != "MANDATORY_IE_MISSING: [TPid LoadId Category Subject RatingPlanActivations]" {
t.Error("Calling APIerSv1.SetTPRatingProfile got unexpected error: ", err.Error())
}
// Test get
var rplyRpf *utils.TPRatingProfile
if err := rater.Call(context.Background(), utils.APIerSv1GetTPRatingProfile, &AttrGetTPRatingProfile{TPid: rpfTst.TPid, RatingProfileID: utils.ConcatenatedKey(rpfTst.LoadId, rpfTst.Tenant, rpfTst.Category, rpfTst.Subject)}, &rplyRpf); err != nil {
t.Error("Calling APIerSv1.GetTPRatingProfiles, got error: ", err.Error())
} else if !reflect.DeepEqual(rpfTst, rplyRpf) {
t.Errorf("Calling APIerSv1.GetTPRatingProfiles expected: %v, received: %v", rpfTst, rplyRpf)
}
// Test remove
if err := rater.Call(context.Background(), utils.APIerSv1RemoveTPRatingProfile, &AttrGetTPRatingProfile{TPid: rpfTst.TPid, RatingProfileID: utils.ConcatenatedKey(rpfTst.LoadId, rpfTst.Tenant, rpfTst.Category, rpfTst.Subject)}, &reply); err != nil {
t.Error("Calling APIerSv1.RemoveTPRatingProfile, got error: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.RemoveTPRatingProfile received: ", reply)
}
// Test getLoadIds
var rplyRpIds []string
expectedRpIds := []string{utils.TestSQL}
if err := rater.Call(context.Background(), utils.APIerSv1GetTPRatingProfileLoadIds, &utils.AttrTPRatingProfileIds{TPid: rpf.TPid}, &rplyRpIds); err != nil {
t.Error("Calling APIerSv1.GetTPRatingProfileLoadIds, got error: ", err.Error())
} else if !reflect.DeepEqual(expectedRpIds, rplyRpIds) {
t.Errorf("Calling APIerSv1.GetTPRatingProfileLoadIds expected: %v, received: %v", expectedRpIds, rplyRpIds)
}
}
func testApierTPActions(t *testing.T) {
var reply string
act := &utils.TPActions{TPid: utils.TestSQL,
ID: "PREPAID_10", Actions: []*utils.TPAction{
{Identifier: "*topup_reset", BalanceType: utils.MetaMonetary,
Units: "10", ExpiryTime: "*unlimited",
DestinationIds: utils.MetaAny, BalanceWeight: "10", Weight: 10},
}}
actWarn := &utils.TPActions{TPid: utils.TestSQL, ID: "WARN_VIA_HTTP", Actions: []*utils.TPAction{
{Identifier: "*http_post", ExtraParameters: "http://localhost:8000", Weight: 10},
}}
actLog := &utils.TPActions{TPid: utils.TestSQL, ID: "LOG_BALANCE", Actions: []*utils.TPAction{
{Identifier: "*log", Weight: 10},
}}
actTst := new(utils.TPActions)
*actTst = *act
actTst.ID = utils.TestSQL
for _, ac := range []*utils.TPActions{act, actWarn, actTst, actLog} {
if err := rater.Call(context.Background(), utils.APIerSv1SetTPActions, ac, &reply); err != nil {
t.Error("Got error on APIerSv1.SetTPActions: ", err.Error())
} else if reply != utils.OK {
t.Error("Unexpected reply received when calling APIerSv1.SetTPActions: ", reply)
}
}
// Check second set
if err := rater.Call(context.Background(), utils.APIerSv1SetTPActions, actTst, &reply); err != nil {
t.Error("Got error on second APIerSv1.SetTPActions: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.SetTPActions got reply: ", reply)
}
// Check missing params
if err := rater.Call(context.Background(), utils.APIerSv1SetTPActions, new(utils.TPActions), &reply); err == nil {
t.Error("Calling APIerSv1.SetTPActions, expected error, received: ", reply)
} else if err.Error() != "MANDATORY_IE_MISSING: [TPid ID Actions]" {
t.Error("Calling APIerSv1.SetTPActions got unexpected error: ", err.Error())
}
// Test get
var rplyActs *utils.TPActions
if err := rater.Call(context.Background(), utils.APIerSv1GetTPActions, &AttrGetTPActions{TPid: actTst.TPid, ID: actTst.ID}, &rplyActs); err != nil {
t.Error("Calling APIerSv1.GetTPActions, got error: ", err.Error())
} else if !reflect.DeepEqual(actTst, rplyActs) {
t.Errorf("Calling APIerSv1.GetTPActions expected: %v, received: %v", actTst, rplyActs)
}
// Test remove
if err := rater.Call(context.Background(), utils.APIerSv1RemoveTPActions, &AttrGetTPActions{TPid: actTst.TPid, ID: actTst.ID}, &reply); err != nil {
t.Error("Calling APIerSv1.RemoveTPActions, got error: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.RemoveTPActions received: ", reply)
}
// Test getIds
var rplyIds []string
expectedIds := []string{"LOG_BALANCE", "PREPAID_10", "WARN_VIA_HTTP"}
if err := rater.Call(context.Background(), utils.APIerSv1GetTPActionIds, &AttrGetTPActionIds{TPid: actTst.TPid}, &rplyIds); err != nil {
t.Error("Calling APIerSv1.GetTPActionIds, got error: ", err.Error())
}
sort.Strings(expectedIds)
sort.Strings(rplyIds)
if !reflect.DeepEqual(expectedIds, rplyIds) {
t.Errorf("Calling APIerSv1.GetTPActionIds expected: %v, received: %v", expectedIds, rplyIds)
}
}
func testApierTPActionPlan(t *testing.T) {
var reply string
at := &utils.TPActionPlan{TPid: utils.TestSQL, ID: "PREPAID_10", ActionPlan: []*utils.TPActionTiming{
{ActionsId: "PREPAID_10", TimingId: "ASAP", Weight: 10},
}}
atTst := new(utils.TPActionPlan)
*atTst = *at
atTst.ID = utils.TestSQL
for _, act := range []*utils.TPActionPlan{at, atTst} {
if err := rater.Call(context.Background(), utils.APIerSv1SetTPActionPlan, act, &reply); err != nil {
t.Error("Got error on APIerSv1.SetTPActionPlan: ", err.Error())
} else if reply != utils.OK {
t.Error("Unexpected reply received when calling APIerSv1.SetTPActionPlan: ", reply)
}
}
// Check second set
if err := rater.Call(context.Background(), utils.APIerSv1SetTPActionPlan, atTst, &reply); err != nil {
t.Error("Got error on second APIerSv1.SetTPActionPlan: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.SetTPActionPlan got reply: ", reply)
}
// Check missing params
if err := rater.Call(context.Background(), utils.APIerSv1SetTPActionPlan, new(utils.TPActionPlan), &reply); err == nil {
t.Error("Calling APIerSv1.SetTPActionPlan, expected error, received: ", reply)
} else if err.Error() != "MANDATORY_IE_MISSING: [TPid ID ActionPlan]" {
t.Error("Calling APIerSv1.SetTPActionPlan got unexpected error: ", err.Error())
}
// Test get
var rplyActs *utils.TPActionPlan
if err := rater.Call(context.Background(), utils.APIerSv1GetTPActionPlan, &AttrGetTPActionPlan{TPid: atTst.TPid, ID: atTst.ID}, &rplyActs); err != nil {
t.Error("Calling APIerSv1.GetTPActionPlan, got error: ", err.Error())
} else if !reflect.DeepEqual(atTst, rplyActs) {
t.Errorf("Calling APIerSv1.GetTPActionPlan expected: %v, received: %v", atTst, rplyActs)
}
// Test remove
if err := rater.Call(context.Background(), utils.APIerSv1RemoveTPActionPlan, &AttrGetTPActionPlan{TPid: atTst.TPid, ID: atTst.ID}, &reply); err != nil {
t.Error("Calling APIerSv1.RemoveTPActionPlan, got error: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.RemoveTPActionPlan received: ", reply)
}
// Test getIds
var rplyIds []string
expectedIds := []string{"PREPAID_10"}
if err := rater.Call(context.Background(), utils.APIerSv1GetTPActionPlanIds, &AttrGetTPActionPlanIds{TPid: atTst.TPid}, &rplyIds); err != nil {
t.Error("Calling APIerSv1.GetTPActionPlanIds, got error: ", err.Error())
} else if !reflect.DeepEqual(expectedIds, rplyIds) {
t.Errorf("Calling APIerSv1.GetTPActionPlanIds expected: %v, received: %v", expectedIds, rplyIds)
}
}
func testApierTPActionTriggers(t *testing.T) {
var reply string
at := &utils.TPActionTriggers{
TPid: utils.TestSQL,
ID: "STANDARD_TRIGGERS",
ActionTriggers: []*utils.TPActionTrigger{{
Id: "STANDARD_TRIGGERS",
UniqueID: "MYFIRSTTRIGGER",
BalanceType: utils.MetaMonetary,
ThresholdType: "*min_balance",
ThresholdValue: 2,
ActionsId: "LOG_BALANCE",
Weight: 10,
}},
}
atTst := new(utils.TPActionTriggers)
*atTst = *at
atTst.ID = utils.TestSQL
atTst.ActionTriggers[0].Id = utils.TestSQL
for _, act := range []*utils.TPActionTriggers{at, atTst} {
if err := rater.Call(context.Background(), utils.APIerSv1SetTPActionTriggers, act, &reply); err != nil {
t.Error("Got error on APIerSv1.SetTPActionTriggers: ", err.Error())
} else if reply != utils.OK {
t.Error("Unexpected reply received when calling APIerSv1.SetTPActionTriggers: ", reply)
}
}
// Check second set
if err := rater.Call(context.Background(), utils.APIerSv1SetTPActionTriggers, atTst, &reply); err != nil {
t.Error("Got error on second APIerSv1.SetTPActionTriggers: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.SetTPActionTriggers got reply: ", reply)
}
// Check missing params
if err := rater.Call(context.Background(), utils.APIerSv1SetTPActionTriggers, new(utils.TPActionTriggers), &reply); err == nil {
t.Error("Calling APIerSv1.SetTPActionTriggers, expected error, received: ", reply)
} else if err.Error() != "MANDATORY_IE_MISSING: [TPid ID]" {
t.Error("Calling APIerSv1.SetTPActionTriggers got unexpected error: ", err.Error())
}
// Test get
var rplyActs *utils.TPActionTriggers
if err := rater.Call(context.Background(), utils.APIerSv1GetTPActionTriggers, &AttrGetTPActionTriggers{TPid: atTst.TPid, ID: atTst.ID}, &rplyActs); err != nil {
t.Errorf("Calling APIerSv1.GetTPActionTriggers %s, got error: %s", atTst.ID, err.Error())
} else if !reflect.DeepEqual(atTst, rplyActs) {
t.Errorf("Calling APIerSv1.GetTPActionTriggers expected: %+v, received: %+v", utils.ToJSON(atTst), utils.ToJSON(rplyActs))
}
// Test remove
if err := rater.Call(context.Background(), utils.APIerSv1RemoveTPActionTriggers, &AttrGetTPActionTriggers{TPid: atTst.TPid, ID: atTst.ID}, &reply); err != nil {
t.Error("Calling APIerSv1.RemoveTPActionTriggers, got error: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.RemoveTPActionTriggers received: ", reply)
}
// Test getIds
var rplyIds []string
expectedIds := []string{"STANDARD_TRIGGERS"}
if err := rater.Call(context.Background(), utils.APIerSv1GetTPActionTriggerIds, &AttrGetTPActionTriggerIds{TPid: atTst.TPid}, &rplyIds); err != nil {
t.Error("Calling APIerSv1.GetTPActionTriggerIds, got error: ", err.Error())
} else if !reflect.DeepEqual(expectedIds, rplyIds) {
t.Errorf("Calling APIerSv1.GetTPActionTriggerIds expected: %v, received: %v", expectedIds, rplyIds)
}
}
// Test here TPAccountActions APIs
func testApierTPAccountActions(t *testing.T) {
var reply string
aa1 := &utils.TPAccountActions{TPid: utils.TestSQL, LoadId: utils.TestSQL, Tenant: "cgrates.org",
Account: "1001", ActionPlanId: "PREPAID_10", ActionTriggersId: "STANDARD_TRIGGERS"}
aa2 := &utils.TPAccountActions{TPid: utils.TestSQL, LoadId: utils.TestSQL, Tenant: "cgrates.org",
Account: "1002", ActionPlanId: "PREPAID_10", ActionTriggersId: "STANDARD_TRIGGERS"}
aa3 := &utils.TPAccountActions{TPid: utils.TestSQL, LoadId: utils.TestSQL, Tenant: "cgrates.org",
Account: "1003", ActionPlanId: "PREPAID_10", ActionTriggersId: "STANDARD_TRIGGERS"}
aa4 := &utils.TPAccountActions{TPid: utils.TestSQL, LoadId: utils.TestSQL, Tenant: "cgrates.org",
Account: "1004", ActionPlanId: "PREPAID_10", ActionTriggersId: "STANDARD_TRIGGERS"}
aa5 := &utils.TPAccountActions{TPid: utils.TestSQL, LoadId: utils.TestSQL, Tenant: "cgrates.org",
Account: "1005", ActionPlanId: "PREPAID_10", ActionTriggersId: "STANDARD_TRIGGERS"}
aaTst := new(utils.TPAccountActions)
*aaTst = *aa1
aaTst.Account = utils.TestSQL
for _, aact := range []*utils.TPAccountActions{aa1, aa2, aa3, aa4, aa5, aaTst} {
if err := rater.Call(context.Background(), utils.APIerSv1SetTPAccountActions, aact, &reply); err != nil {
t.Error("Got error on APIerSv1.SetTPAccountActions: ", err.Error())
} else if reply != utils.OK {
t.Error("Unexpected reply received when calling APIerSv1.SetTPAccountActions: ", reply)
}
}
// Check second set
if err := rater.Call(context.Background(), utils.APIerSv1SetTPAccountActions, aaTst, &reply); err != nil {
t.Error("Got error on second APIerSv1.SetTPAccountActions: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.SetTPAccountActions got reply: ", reply)
}
// Check missing params
if err := rater.Call(context.Background(), utils.APIerSv1SetTPAccountActions, new(utils.TPAccountActions), &reply); err == nil {
t.Error("Calling APIerSv1.SetTPAccountActions, expected error, received: ", reply)
} else if err.Error() != "MANDATORY_IE_MISSING: [TPid LoadId Account ActionPlanId]" {
t.Error("Calling APIerSv1.SetTPAccountActions got unexpected error: ", err.Error())
}
// Test get
var rplyaa *utils.TPAccountActions
if err := rater.Call(context.Background(), utils.APIerSv1GetTPAccountActions, &AttrGetTPAccountActions{TPid: aaTst.TPid, AccountActionsId: aaTst.GetId()}, &rplyaa); err != nil {
t.Error("Calling APIerSv1.GetTPAccountActions, got error: ", err.Error())
} else if !reflect.DeepEqual(aaTst, rplyaa) {
t.Errorf("Calling APIerSv1.GetTPAccountActions expected: %v, received: %v", aaTst, rplyaa)
}
// Test remove
if err := rater.Call(context.Background(), utils.APIerSv1RemoveTPAccountActions, &AttrGetTPAccountActions{TPid: aaTst.TPid, AccountActionsId: aaTst.GetId()}, &reply); err != nil {
t.Error("Calling APIerSv1.RemoveTPAccountActions, got error: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.RemoveTPAccountActions received: ", reply)
}
// Test getLoadIds
var rplyRpIds []string
expectedRpIds := []string{utils.TestSQL}
if err := rater.Call(context.Background(), utils.APIerSv1GetTPAccountActionLoadIds, &AttrGetTPAccountActionIds{TPid: aaTst.TPid}, &rplyRpIds); err != nil {
t.Error("Calling APIerSv1.GetTPAccountActionLoadIds, got error: ", err.Error())
} else if !reflect.DeepEqual(expectedRpIds, rplyRpIds) {
t.Errorf("Calling APIerSv1.GetTPAccountActionLoadIds expected: %v, received: %v", expectedRpIds, rplyRpIds)
}
}
// Test here LoadRatingPlan
func testApierLoadRatingPlan(t *testing.T) {
var reply string
if err := rater.Call(context.Background(), utils.APIerSv1LoadRatingPlan, &AttrLoadRatingPlan{TPid: utils.TestSQL, RatingPlanId: "RETAIL1"}, &reply); err != nil {
t.Error("Got error on APIerSv1.LoadRatingPlan: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.LoadRatingPlan got reply: ", reply)
}
}
// Test here LoadRatingProfile
func testApierLoadRatingProfile(t *testing.T) {
var reply string
rpf := &utils.TPRatingProfile{
TPid: utils.TestSQL, LoadId: utils.TestSQL,
Tenant: "cgrates.org", Category: "call", Subject: "*any"}
startTime := time.Now()
if err := rater.Call(context.Background(), utils.APIerSv1LoadRatingProfile, &rpf, &reply); err != nil {
t.Error("Got error on APIerSv1.LoadRatingProfile: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.LoadRatingProfile got reply: ", reply)
}
elapsedTime := time.Since(startTime)
expectedDuration := 1 * time.Second
if elapsedTime < expectedDuration || elapsedTime >= 2*time.Second {
t.Errorf("Expected elapsed time of at least %v, but got %v", expectedDuration, elapsedTime)
}
}
func testApierLoadRatingProfileWithoutTenant(t *testing.T) {
var reply string
rpf := &utils.TPRatingProfile{
TPid: utils.TestSQL, LoadId: utils.TestSQL,
Category: "call", Subject: "*any"}
if err := rater.Call(context.Background(), utils.APIerSv1LoadRatingProfile, &rpf, &reply); err != nil {
t.Error("Got error on APIerSv1.LoadRatingProfile: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.LoadRatingProfile got reply: ", reply)
}
}
// Test here LoadAccountActions
func testApierLoadAccountActions(t *testing.T) {
var rcvStats map[string]*ltcache.CacheStats
expectedStats := engine.GetDefaultEmptyCacheStats() // Make sure nothing in cache so far
expectedStats[utils.CacheLoadIDs].Items = 2 // we loaded the ratingprofiles
expectedStats[utils.CacheRPCConnections].Items = 1
if err := rater.Call(context.Background(), utils.CacheSv1GetCacheStats, new(utils.AttrCacheIDsWithAPIOpts), &rcvStats); err != nil {
t.Error("Got error on CacheSv1.GetCacheStats: ", err.Error())
} else if !reflect.DeepEqual(expectedStats, rcvStats) {
t.Errorf("Calling CacheSv1.GetCacheStats expected: %+v,\n received: %+v", utils.ToJSON(expectedStats), utils.ToJSON(rcvStats))
}
var reply string
aa1 := &utils.TPAccountActions{TPid: utils.TestSQL, LoadId: utils.TestSQL, Tenant: "cgrates.org", Account: "1001"}
if err := rater.Call(context.Background(), utils.APIerSv1LoadAccountActions, aa1, &reply); err != nil {
t.Error("Got error on APIerSv1.LoadAccountActions: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.LoadAccountActions got reply: ", reply)
}
time.Sleep(10 * time.Millisecond)
// expectedStats[utils.CacheAccountActionPlans].Items = 1 // was removed because the accountActionPlans are only set in DB not in Cache
expectedStats[utils.CacheActionPlans].Items = 1
expectedStats[utils.CacheActions].Items = 1
expectedStats[utils.CacheLoadIDs].Items = 6
expectedStats[utils.CacheRPCConnections].Items = 1
if err := rater.Call(context.Background(), utils.CacheSv1GetCacheStats, new(utils.AttrCacheIDsWithAPIOpts), &rcvStats); err != nil {
t.Error("Got error on CacheSv1.GetCacheStats: ", err.Error())
} else if !reflect.DeepEqual(expectedStats, rcvStats) {
t.Errorf("Calling CacheSv1.GetCacheStats expected: %+v, \n received: %+v", utils.ToJSON(expectedStats), utils.ToJSON(rcvStats))
}
}
// Test here ReloadScheduler
func testApierReloadScheduler(t *testing.T) {
var reply string
// Simple test that command is executed without errors
if err := rater.Call(context.Background(), utils.SchedulerSv1Reload, utils.StringWithAPIOpts{}, &reply); err != nil {
t.Error("Got error on SchedulerSv1.Reload: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling SchedulerSv1.Reload got reply: ", reply)
}
}
// Test here SetRatingProfile
func testApierSetRatingProfile(t *testing.T) {
var reply string
rpa := &utils.TPRatingActivation{ActivationTime: "2012-01-01T00:00:00Z", RatingPlanId: "RETAIL1", FallbackSubjects: "dan2"}
rpf := &utils.AttrSetRatingProfile{Tenant: "cgrates.org", Category: "call",
Subject: "dan", RatingPlanActivations: []*utils.TPRatingActivation{rpa}}
startTime := time.Now()
if err := rater.Call(context.Background(), utils.APIerSv1SetRatingProfile, &rpf, &reply); err != nil {
t.Error("Got error on APIerSv1.SetRatingProfile: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.SetRatingProfile got reply: ", reply)
}
elapsedTime := time.Since(startTime)
expectedDuration := 1 * time.Second
if elapsedTime < expectedDuration || elapsedTime >= 2*time.Second {
t.Errorf("Expected elapsed time of at least %v, but got %v", expectedDuration, elapsedTime)
}
var rcvStats map[string]*ltcache.CacheStats
expectedStats := engine.GetDefaultEmptyCacheStats()
// expectedStats[utils.CacheAccountActionPlans].Items = 1 // was removed because the accountActionPlans are only set in DB not in Cache
expectedStats[utils.CacheActionPlans].Items = 1
expectedStats[utils.CacheActions].Items = 1
expectedStats[utils.CacheRatingProfiles].Items = 1
expectedStats[utils.CacheRPCConnections].Items = 1
expectedStats[utils.CacheLoadIDs].Items = 6
if err := rater.Call(context.Background(), utils.CacheSv1GetCacheStats, new(utils.AttrCacheIDsWithAPIOpts), &rcvStats); err != nil {
t.Error("Got error on CacheSv1.GetCacheStats: ", err.Error())
} else if !reflect.DeepEqual(expectedStats, rcvStats) {
t.Errorf("Calling CacheSv1.GetCacheStats expected: %+v, received: %+v", utils.ToJSON(expectedStats), utils.ToJSON(rcvStats))
}
// Calling the second time should not raise EXISTS
if err := rater.Call(context.Background(), utils.APIerSv1SetRatingProfile, &rpf, &reply); err != nil {
t.Error("Unexpected result on duplication: ", err.Error())
}
// Make sure rates were loaded for account dan
// Test here ResponderGetCost
tStart, _ := utils.ParseTimeDetectLayout("2013-08-07T17:30:00Z", utils.EmptyString)
tEnd, _ := utils.ParseTimeDetectLayout("2013-08-07T17:31:30Z", utils.EmptyString)
cd := &engine.CallDescriptorWithAPIOpts{
CallDescriptor: &engine.CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "dan",
Account: "dan",
Destination: "+4917621621391",
DurationIndex: 90,
TimeStart: tStart,
TimeEnd: tEnd,
},
}
var cc engine.CallCost
// Simple test that command is executed without errors
if err := rater.Call(context.Background(), utils.ResponderGetCost, cd, &cc); err != nil {
t.Error("Got error on Responder.GetCost: ", err.Error())
} else if cc.Cost != 0 {
t.Errorf("Calling Responder.GetCost got callcost: %v", cc.Cost)
}
expectedStats[utils.CacheRatingPlans].Items = 1
expectedStats[utils.CacheReverseDestinations].Items = 10
if err := rater.Call(context.Background(), utils.CacheSv1GetCacheStats, new(utils.AttrCacheIDsWithAPIOpts), &rcvStats); err != nil {
t.Error("Got error on CacheSv1.GetCacheStats: ", err.Error())
} else if !reflect.DeepEqual(expectedStats, rcvStats) {
t.Errorf("Calling CacheSv1.GetCacheStats expected: %+v, received: %+v", utils.ToJSON(expectedStats), utils.ToJSON(rcvStats))
}
}
func testAPIerSv1GetRatingProfile(t *testing.T) {
var rpl engine.RatingProfile
attrGetRatingPlan := &utils.AttrGetRatingProfile{
Tenant: "cgrates.org", Category: "call", Subject: "dan"}
actTime, err := utils.ParseTimeDetectLayout("2012-01-01T00:00:00Z", utils.EmptyString)
if err != nil {
t.Error(err)
}
expected := engine.RatingProfile{
Id: "*out:cgrates.org:call:dan",
RatingPlanActivations: engine.RatingPlanActivations{
{
ActivationTime: actTime,
RatingPlanId: "RETAIL1",
FallbackKeys: []string{"*out:cgrates.org:call:dan2"},
},
{
ActivationTime: actTime,
RatingPlanId: "RETAIL1",
FallbackKeys: []string{"*out:cgrates.org:call:dan2"},
},
},
}
if err := rater.Call(context.Background(), utils.APIerSv1GetRatingProfile, attrGetRatingPlan, &rpl); err != nil {
t.Errorf("Got error on APIerSv1.GetRatingProfile: %+v", err)
} else if !reflect.DeepEqual(expected, rpl) {
t.Errorf("Calling APIerSv1.GetRatingProfile expected: %+v, received: %+v", utils.ToJSON(expected), utils.ToJSON(rpl))
}
attrGetRatingPlan.Subject = utils.EmptyString
if err := rater.Call(context.Background(), utils.APIerSv1GetRatingProfile, attrGetRatingPlan, &rpl); err == nil {
t.Errorf("Expected error on APIerSv1.GetRatingProfile, received : %+v", rpl)
}
attrGetRatingPlan.Subject = "dan"
attrGetRatingPlan.Tenant = "other_tenant"
if err := rater.Call(context.Background(), utils.APIerSv1GetRatingProfile, attrGetRatingPlan, &rpl); err.Error() != utils.ErrNotFound.Error() {
t.Errorf("Expected error on APIerSv1.GetRatingProfile, received : %+v", err)
}
expectedIds := []string{"call:dan", "call:*any"}
var result []string
if err := rater.Call(context.Background(), utils.APIerSv1GetRatingProfileIDs, &utils.PaginatorWithTenant{Tenant: "cgrates.org"}, &result); err != nil {
t.Error(err)
} else if len(expectedIds) != len(result) {
t.Errorf("Expecting : %+v, received: %+v", expected, result)
}
}
func testAPIerSv1GetRatingProfileIDsWithoutTenant(t *testing.T) {
expectedIds := []string{"call:dan", "call:*any"}
var result []string
if err := rater.Call(context.Background(), utils.APIerSv1GetRatingProfileIDs, &utils.PaginatorWithTenant{}, &result); err != nil {
t.Error(err)
} else {
sort.Strings(expectedIds)
sort.Strings(result)
if !reflect.DeepEqual(expectedIds, result) {
t.Errorf("Expected %+v, received %+v", expectedIds, result)
}
}
}
func testAPIerSv1GetRatingProfileWithoutTenant(t *testing.T) {
var rpl engine.RatingProfile
attrGetRatingPlan := &utils.AttrGetRatingProfile{
Category: "call", Subject: "dan"}
actTime, err := utils.ParseTimeDetectLayout("2012-01-01T00:00:00Z", utils.EmptyString)
if err != nil {
t.Error(err)
}
expected := engine.RatingProfile{
Id: "*out:cgrates.org:call:dan",
RatingPlanActivations: engine.RatingPlanActivations{
{
ActivationTime: actTime,
RatingPlanId: "RETAIL1",
FallbackKeys: []string{"*out:cgrates.org:call:dan2"},
},
{
ActivationTime: actTime,
RatingPlanId: "RETAIL1",
FallbackKeys: []string{"*out:cgrates.org:call:dan2"},
},
},
}
if err := rater.Call(context.Background(), utils.APIerSv1GetRatingProfile, &attrGetRatingPlan, &rpl); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expected, rpl) {
t.Errorf("Calling APIerSv1.GetRatingProfile expected: %+v \n, received: %+v", utils.ToJSON(expected), utils.ToJSON(rpl))
}
}
func testApierGetActionTrigger(t *testing.T) {
//nothing to get from database
var atrs *engine.ActionTriggers
if err := rater.Call(context.Background(), utils.APIerSv1GetActionTriggers,
&AttrGetActionTriggers{GroupIDs: []string{"TEST_ID1"}}, &atrs); err.Error() != utils.ErrNotFound.Error() {
t.Error(err)
}
//set an ActionTrigger in database
var reply string
if err := rater.Call(context.Background(), utils.APIerSv1SetActionTrigger,
engine.AttrSetActionTrigger{
GroupID: "TEST_ID1",
UniqueID: "TEST_ID2",
}, &reply); err != nil {
t.Error(err)
}
//get from database and compare
newActTrg := &engine.ActionTriggers{
{
ID: "TEST_ID1",
UniqueID: "TEST_ID2",
Balance: &engine.BalanceFilter{},
},
}
if err := rater.Call(context.Background(), utils.APIerSv1GetActionTriggers,
&AttrGetActionTriggers{GroupIDs: []string{"TEST_ID1"}}, &atrs); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(newActTrg, atrs) {
t.Errorf("Expected %+v \n, received %+v", utils.ToJSON(newActTrg), utils.ToJSON(atrs))
}
//remove the ActionTrigger from dataBase
if err := rater.Call(context.Background(), utils.APIerSv1RemoveActionTrigger,
&AttrRemoveActionTrigger{GroupID: "TEST_ID1"}, &reply); err != nil {
t.Error(err)
} else if reply != utils.OK {
t.Errorf("Unexpected result")
}
//nothing to get from dataBase
if err := rater.Call(context.Background(), utils.APIerSv1GetActionTriggers,
&AttrGetActionTriggers{GroupIDs: []string{"TEST_ID1"}}, &atrs); err.Error() != utils.ErrNotFound.Error() {
t.Error(err)
}
}
// Test here ReloadCache
func testApierReloadCache(t *testing.T) {
var reply string
arc := new(utils.AttrReloadCacheWithAPIOpts)
// Simple test that command is executed without errors
if err := rater.Call(context.Background(), utils.CacheSv1ReloadCache, arc, &reply); err != nil {
t.Error("Got error on CacheSv1.ReloadCache: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling CacheSv1.ReloadCache got reply: ", reply)
}
var rcvStats map[string]*ltcache.CacheStats
expectedStats := engine.GetDefaultEmptyCacheStats()
// expectedStats[utils.CacheAccountActionPlans].Items = 1 // was removed because the accountActionPlans are only set in DB not in Cache
expectedStats[utils.CacheActionPlans].Items = 1
expectedStats[utils.CacheActions].Items = 1
expectedStats[utils.CacheRatingProfiles].Items = 3
expectedStats[utils.CacheRatingPlans].Items = 1
expectedStats[utils.CacheReverseDestinations].Items = 10
expectedStats[utils.CacheLoadIDs].Items = 6
expectedStats[utils.CacheRPCConnections].Items = 1
if err := rater.Call(context.Background(), utils.CacheSv1GetCacheStats, new(utils.AttrCacheIDsWithAPIOpts), &rcvStats); err != nil {
t.Error("Got error on CacheSv1.GetCacheStats: ", err.Error())
} else if !reflect.DeepEqual(expectedStats, rcvStats) {
t.Errorf("Calling CacheSv1.GetCacheStats expected: %+v,\n received: %+v", utils.ToJSON(expectedStats), utils.ToJSON(rcvStats))
}
}
// Test here GetDestination
func testApierGetDestination(t *testing.T) {
reply := new(engine.Destination)
dstId := "GERMANY_MOBILE"
expectedReply := &engine.Destination{Id: dstId, Prefixes: []string{"+4915", "+4916", "+4917"}}
if err := rater.Call(context.Background(), utils.APIerSv1GetDestination, &dstId, reply); err != nil {
t.Error("Got error on APIerSv1.GetDestination: ", err.Error())
} else if !reflect.DeepEqual(expectedReply, reply) {
t.Errorf("Calling APIerSv1.GetDestination expected: %v, received: %v", expectedReply, reply)
}
}
// Test here GetRatingPlan
func testApierGetRatingPlan(t *testing.T) {
reply := new(engine.RatingPlan)
rplnId := "RETAIL1"
if err := rater.Call(context.Background(), utils.APIerSv1GetRatingPlan, &rplnId, reply); err != nil {
t.Error("Got error on APIerSv1.GetRatingPlan: ", err.Error())
}
// Check parts of info received since a full one is not possible due to unique map keys inside reply
if reply.Id != rplnId {
t.Error("Unexpected id received", reply.Id)
}
if len(reply.Timings) != 1 || len(reply.Ratings) != 1 {
t.Error("Unexpected number of items received")
}
riRate := &engine.RIRate{ConnectFee: 0, RoundingMethod: "*up", RoundingDecimals: 2, Rates: []*engine.RGRate{
{GroupIntervalStart: 0, Value: 0, RateIncrement: 60 * time.Second, RateUnit: 60 * time.Second},
}}
for _, rating := range reply.Ratings {
riRateJson, _ := json.Marshal(rating)
if !reflect.DeepEqual(rating, riRate) {
t.Errorf("Unexpected riRate received: %s", riRateJson)
// {"Id":"RT_FS_USERS","ConnectFee":0,"Rates":[{"GroupIntervalStart":0,"Value":0,"RateIncrement":60000000000,"RateUnit":60000000000}],"RoundingMethod":"*up","RoundingDecimals":0}
}
}
}
func testApierRemoveRatingPlan(t *testing.T) {
rplnId := "RETAIL1"
var reply string
err := rater.Call(context.Background(), utils.APIerSv1RemoveRatingPlan, &rplnId, &reply)
if err != nil {
t.Error(err)
}
if reply != utils.OK {
t.Errorf("Expected %s, received %s", utils.OK, reply)
}
//get rating plan (the one that was removed. should return 'err not found')
var ratingPlan *engine.RatingPlan
err = rater.Call(context.Background(), utils.APIerSv1GetRatingPlan, &rplnId, ratingPlan)
if err == nil || err.Error() != utils.ErrNotFound.Error() {
t.Errorf("Expecting error: %s, received: %v", utils.ErrNotFound, err)
}
}
// Test here AddBalance
func testApierAddBalance(t *testing.T) {
var reply string
attrs := &AttrAddBalance{Tenant: "cgrates.org", Account: "1001", BalanceType: utils.MetaMonetary, Value: 1.5}
if err := rater.Call(context.Background(), utils.APIerSv1AddBalance, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.AddBalance: ", err.Error())
} else if reply != utils.OK {
t.Errorf("Calling APIerSv1.AddBalance received: %s", reply)
}
attrs = &AttrAddBalance{Tenant: "cgrates.org", Account: "dan", BalanceType: utils.MetaMonetary, Value: 1.5}
if err := rater.Call(context.Background(), utils.APIerSv1AddBalance, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.AddBalance: ", err.Error())
} else if reply != utils.OK {
t.Errorf("Calling APIerSv1.AddBalance received: %s", reply)
}
attrs = &AttrAddBalance{Tenant: "cgrates.org", Account: "dan2", BalanceType: utils.MetaMonetary, Value: 1.5}
if err := rater.Call(context.Background(), utils.APIerSv1AddBalance, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.AddBalance: ", err.Error())
} else if reply != utils.OK {
t.Errorf("Calling APIerSv1.AddBalance received: %s", reply)
}
attrs = &AttrAddBalance{Tenant: "cgrates.org", Account: "dan3", BalanceType: utils.MetaMonetary, Value: 1.5}
if err := rater.Call(context.Background(), utils.APIerSv1AddBalance, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.AddBalance: ", err.Error())
} else if reply != utils.OK {
t.Errorf("Calling APIerSv1.AddBalance received: %s", reply)
}
attrs = &AttrAddBalance{Tenant: "cgrates.org", Account: "dan3", BalanceType: utils.MetaMonetary, Value: 2.1}
if err := rater.Call(context.Background(), utils.APIerSv1AddBalance, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.AddBalance: ", err.Error())
} else if reply != utils.OK {
t.Errorf("Calling APIerSv1.AddBalance received: %s", reply)
}
attrs = &AttrAddBalance{Tenant: "cgrates.org", Account: "dan6", BalanceType: utils.MetaMonetary, Value: 2.1}
if err := rater.Call(context.Background(), utils.APIerSv1AddBalance, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.AddBalance: ", err.Error())
} else if reply != utils.OK {
t.Errorf("Calling APIerSv1.AddBalance received: %s", reply)
}
attrs = &AttrAddBalance{Tenant: "cgrates.org", Account: "dan6", BalanceType: utils.MetaMonetary, Value: 1, Overwrite: true}
if err := rater.Call(context.Background(), utils.APIerSv1AddBalance, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.AddBalance: ", err.Error())
} else if reply != utils.OK {
t.Errorf("Calling APIerSv1.AddBalance received: %s", reply)
}
}
// Test here ExecuteAction
func testApierExecuteAction(t *testing.T) {
var reply string
// Add balance to a previously known account
attrs := utils.AttrExecuteAction{Tenant: "cgrates.org", Account: "dan2", ActionsId: "PREPAID_10"}
if err := rater.Call(context.Background(), utils.APIerSv1ExecuteAction, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.ExecuteAction: ", err.Error())
} else if reply != utils.OK {
t.Errorf("Calling APIerSv1.ExecuteAction received: %s", reply)
}
reply2 := utils.EmptyString
// Add balance to an account which does n exist
attrs = utils.AttrExecuteAction{Tenant: "cgrates.org", Account: "dan2", ActionsId: "DUMMY_ACTION"}
if err := rater.Call(context.Background(), utils.APIerSv1ExecuteAction, attrs, &reply2); err == nil || reply2 == utils.OK {
t.Error("Expecting error on APIerSv1.ExecuteAction.", err, reply2)
}
}
func testApierExecuteActionWithoutTenant(t *testing.T) {
var reply string
// Add balance to a previously known account
attrs := utils.AttrExecuteAction{Account: "dan2", ActionsId: "PREPAID_10"}
if err := rater.Call(context.Background(), utils.APIerSv1ExecuteAction, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.ExecuteAction: ", err.Error())
} else if reply != utils.OK {
t.Errorf("Calling APIerSv1.ExecuteAction received: %s", reply)
}
reply2 := utils.EmptyString
// Add balance to an account which does n exist
attrs = utils.AttrExecuteAction{Account: "dan2", ActionsId: "DUMMY_ACTION"}
if err := rater.Call(context.Background(), utils.APIerSv1ExecuteAction, attrs, &reply2); err == nil || reply2 == utils.OK {
t.Error("Expecting error on APIerSv1.ExecuteAction.", err, reply2)
}
}
func testApierSetActions(t *testing.T) {
act1 := &V1TPAction{Identifier: utils.MetaTopUpReset, BalanceType: utils.MetaMonetary, Units: 75.0, ExpiryTime: utils.MetaUnlimited, Weight: 20.0}
attrs1 := &V1AttrSetActions{ActionsId: "ACTS_1", Actions: []*V1TPAction{act1}}
reply1 := utils.EmptyString
if err := rater.Call(context.Background(), utils.APIerSv1SetActions, &attrs1, &reply1); err != nil {
t.Error("Got error on APIerSv1.SetActions: ", err.Error())
} else if reply1 != utils.OK {
t.Errorf("Calling APIerSv1.SetActions received: %s", reply1)
}
// Calling the second time should raise EXISTS
if err := rater.Call(context.Background(), utils.APIerSv1SetActions, &attrs1, &reply1); err == nil || err.Error() != "EXISTS" {
t.Error("Unexpected result on duplication: ", err.Error())
}
}
func testApierGetActions(t *testing.T) {
expectActs := []*utils.TPAction{
{Identifier: utils.MetaTopUpReset, BalanceType: utils.MetaMonetary,
Units: "75", BalanceWeight: "0", BalanceBlocker: "false",
BalanceDisabled: "false", ExpiryTime: utils.MetaUnlimited, Weight: 20.0}}
var reply []*utils.TPAction
if err := rater.Call(context.Background(), utils.APIerSv1GetActions, utils.StringPointer("ACTS_1"), &reply); err != nil {
t.Error("Got error on APIerSv1.GetActions: ", err.Error())
} else if !reflect.DeepEqual(expectActs, reply) {
t.Errorf("Expected: %v, received: %v", utils.ToJSON(expectActs), utils.ToJSON(reply))
}
}
func testApierSetActionPlan(t *testing.T) {
atm1 := &engine.AttrActionPlan{ActionsId: "ACTS_1", MonthDays: "1", Time: "00:00:00", Weight: 20.0}
atms1 := &engine.AttrSetActionPlan{Id: "ATMS_1", ActionPlan: []*engine.AttrActionPlan{atm1}}
reply1 := utils.EmptyString
if err := rater.Call(context.Background(), utils.APIerSv1SetActionPlan, &atms1, &reply1); err != nil {
t.Error("Got error on APIerSv1.SetActionPlan: ", err.Error())
} else if reply1 != utils.OK {
t.Errorf("Calling APIerSv1.SetActionPlan received: %s", reply1)
}
// Calling the second time should raise EXISTS
if err := rater.Call(context.Background(), utils.APIerSv1SetActionPlan, &atms1, &reply1); err == nil || err.Error() != "EXISTS" {
t.Error("Unexpected result on duplication: ", err.Error())
}
}
// Test here AddTriggeredAction
func testApierAddTriggeredAction(t *testing.T) {
var reply string
attrs := &AttrAddBalance{Tenant: "cgrates.org", Account: "dan32", BalanceType: utils.MetaMonetary, Value: 1.5}
if err := rater.Call(context.Background(), utils.APIerSv1AddBalance, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.AddBalance: ", err.Error())
} else if reply != utils.OK {
t.Errorf("Calling APIerSv1.AddBalance received: %s", reply)
}
// Add balance to a previously known account
attrsAddTrigger := &AttrAddActionTrigger{Tenant: "cgrates.org", Account: "dan32", BalanceType: utils.MetaMonetary,
ThresholdType: "*min_balance", ThresholdValue: 2, BalanceDestinationIds: utils.MetaAny, Weight: 10, ActionsId: "WARN_VIA_HTTP"}
if err := rater.Call(context.Background(), utils.APIerSv1AddTriggeredAction, attrsAddTrigger, &reply); err != nil {
t.Error("Got error on APIerSv1.AddTriggeredAction: ", err.Error())
} else if reply != utils.OK {
t.Errorf("Calling APIerSv1.AddTriggeredAction received: %s", reply)
}
reply2 := utils.EmptyString
attrs2 := new(AttrAddActionTrigger)
*attrs2 = *attrsAddTrigger
attrs2.Account = "dan10" // Does not exist so it should error when adding triggers on it
// Add trigger to an account which does n exist
if err := rater.Call(context.Background(), utils.APIerSv1AddTriggeredAction, attrs2, &reply2); err == nil {
t.Error("Expecting error on APIerSv1.AddTriggeredAction.", err, reply2)
}
}
// Test here GetAccountActionTriggers
func testApierGetAccountActionTriggers(t *testing.T) {
var reply engine.ActionTriggers
req := utils.TenantAccount{Tenant: "cgrates.org", Account: "dan32"}
if err := rater.Call(context.Background(), utils.APIerSv1GetAccountActionTriggers, &req, &reply); err != nil {
t.Error("Got error on APIerSv1.GetAccountActionTimings: ", err.Error())
} else if len(reply) != 1 || reply[0].ActionsID != "WARN_VIA_HTTP" {
t.Errorf("Unexpected action triggers received %v", reply)
}
}
func testApierAddTriggeredAction2(t *testing.T) {
var reply string
// Add balance to a previously known account
attrs := &AttrAddAccountActionTriggers{ActionTriggerIDs: []string{"STANDARD_TRIGGERS"}, Tenant: "cgrates.org", Account: "dan2"}
if err := rater.Call(context.Background(), utils.APIerSv1AddAccountActionTriggers, &attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.AddAccountActionTriggers: ", err.Error())
} else if reply != utils.OK {
t.Errorf("Calling APIerSv1.AddAccountActionTriggers received: %s", reply)
}
reply2 := utils.EmptyString
attrs2 := new(AttrAddAccountActionTriggers)
*attrs2 = *attrs
attrs2.Account = "dan10" // Does not exist so it should error when adding triggers on it
// Add trigger to an account which does n exist
if err := rater.Call(context.Background(), utils.APIerSv1AddAccountActionTriggers, &attrs2, &reply2); err == nil || reply2 == utils.OK {
t.Error("Expecting error on APIerSv1.AddAccountActionTriggers.", err, reply2)
}
}
// Test here GetAccountActionTriggers
func testApierGetAccountActionTriggers2(t *testing.T) {
var reply engine.ActionTriggers
req := utils.TenantAccount{Tenant: "cgrates.org", Account: "dan2"}
if err := rater.Call(context.Background(), utils.APIerSv1GetAccountActionTriggers, &req, &reply); err != nil {
t.Error("Got error on APIerSv1.GetAccountActionTimings: ", err.Error())
} else if len(reply) != 1 || reply[0].ActionsID != "LOG_BALANCE" {
t.Errorf("Unexpected action triggers received %v", reply)
}
}
// Test here SetAccountActionTriggers
func testApierSetAccountActionTriggers(t *testing.T) {
// Test first get so we can steal the id which we need to remove
var reply engine.ActionTriggers
req := utils.TenantAccount{Tenant: "cgrates.org", Account: "dan2"}
if err := rater.Call(context.Background(), utils.APIerSv1GetAccountActionTriggers, &req, &reply); err != nil {
t.Error("Got error on APIerSv1.GetAccountActionTimings: ", err.Error())
} else if len(reply) != 1 || reply[0].ActionsID != "LOG_BALANCE" {
for _, atr := range reply {
t.Logf("ATR: %+v", atr)
}
t.Errorf("Unexpected action triggers received %v", reply)
}
var setReply string
setReq := AttrSetAccountActionTriggers{
Tenant: "cgrates.org",
Account: "dan2",
AttrSetActionTrigger: engine.AttrSetActionTrigger{
UniqueID: reply[0].UniqueID,
ActionTrigger: map[string]any{
utils.ActivationDate: "2016-02-05T18:00:00Z",
},
},
}
if err := rater.Call(context.Background(), utils.APIerSv1ResetAccountActionTriggers, setReq, &setReply); err != nil {
t.Error("Got error on APIerSv1.ResetActionTiming: ", err.Error())
} else if setReply != utils.OK {
t.Error("Unexpected answer received", setReply)
}
if err := rater.Call(context.Background(), utils.APIerSv1SetAccountActionTriggers, setReq, &setReply); err != nil {
t.Error("Got error on APIerSv1.RemoveActionTiming: ", err.Error())
} else if setReply != utils.OK {
t.Error("Unexpected answer received", setReply)
}
if err := rater.Call(context.Background(), utils.APIerSv1GetAccountActionTriggers, &req, &reply); err != nil {
t.Error("Got error on APIerSv1.GetAccountActionTriggers: ", err.Error())
} else if len(reply) != 1 || reply[0].ActivationDate != time.Date(2016, 2, 5, 18, 0, 0, 0, time.UTC) {
t.Errorf("Unexpected action triggers received %+v", reply[0])
}
}
// Test here RemAccountActionTriggers
func testApierRemAccountActionTriggers(t *testing.T) {
// Test first get so we can steal the id which we need to remove
var reply engine.ActionTriggers
req := utils.TenantAccount{Tenant: "cgrates.org", Account: "dan2"}
if err := rater.Call(context.Background(), utils.APIerSv1GetAccountActionTriggers, &req, &reply); err != nil {
t.Error("Got error on APIerSv1.GetAccountActionTimings: ", err.Error())
} else if len(reply) != 1 || reply[0].ActionsID != "LOG_BALANCE" {
for _, atr := range reply {
t.Logf("ATR: %+v", atr)
}
t.Errorf("Unexpected action triggers received %v", reply)
}
var rmReply string
rmReq := AttrRemoveAccountActionTriggers{Tenant: "cgrates.org", Account: "dan2", UniqueID: reply[0].UniqueID}
if err := rater.Call(context.Background(), utils.APIerSv1ResetAccountActionTriggers, rmReq, &rmReply); err != nil {
t.Error("Got error on APIerSv1.ResetActionTiming: ", err.Error())
} else if rmReply != utils.OK {
t.Error("Unexpected answer received", rmReply)
}
if err := rater.Call(context.Background(), utils.APIerSv1RemoveAccountActionTriggers, rmReq, &rmReply); err != nil {
t.Error("Got error on APIerSv1.RemoveActionTiming: ", err.Error())
} else if rmReply != utils.OK {
t.Error("Unexpected answer received", rmReply)
}
if err := rater.Call(context.Background(), utils.APIerSv1GetAccountActionTriggers, &req, &reply); err != nil {
t.Error("Got error on APIerSv1.GetAccountActionTriggers: ", err.Error())
} else if len(reply) != 0 {
t.Errorf("Unexpected action triggers received %+v", reply[0])
}
}
// Test here SetAccount
func testApierSetAccount(t *testing.T) {
var reply string
attrs := &utils.AttrSetAccount{Tenant: "cgrates.org", Account: "dan7", ActionPlanID: "ATMS_1", ReloadScheduler: true}
if err := rater.Call(context.Background(), utils.APIerSv1SetAccount, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.SetAccount: ", err.Error())
} else if reply != utils.OK {
t.Errorf("Calling APIerSv1.SetAccount received: %s", reply)
}
reply2 := utils.EmptyString
attrs2 := new(utils.AttrSetAccount)
*attrs2 = *attrs
attrs2.ActionPlanID = "DUMMY_DATA" // Does not exist so it should error when adding triggers on it
// Add account with actions timing which does not exist
if err := rater.Call(context.Background(), utils.APIerSv1SetAccount, attrs2, &reply2); err == nil || reply2 == utils.OK { // OK is not welcomed
t.Error("Expecting error on APIerSv1.SetAccount.", err, reply2)
}
}
// Test here GetAccountActionTimings
func testApierGetAccountActionPlan(t *testing.T) {
var reply []*AccountActionTiming
req := utils.TenantAccount{Tenant: "cgrates.org", Account: "dan7"}
if err := rater.Call(context.Background(), utils.APIerSv1GetAccountActionPlan, &req, &reply); err != nil {
t.Error("Got error on APIerSv1.GetAccountActionPlan: ", err.Error())
} else if len(reply) != 1 {
t.Error("Unexpected action plan received: ", utils.ToJSON(reply))
} else {
if reply[0].ActionPlanId != "ATMS_1" {
t.Errorf("Unexpected ActionoveAccountPlanId received")
}
}
}
func testApierGetAccountActionPlanWithoutTenant(t *testing.T) {
var reply []*AccountActionTiming
req := utils.TenantAccount{Account: "dan7"}
if err := rater.Call(context.Background(), utils.APIerSv1GetAccountActionPlan, &req, &reply); err != nil {
t.Error("Got error on APIerSv1.GetAccountActionPlan: ", err.Error())
} else if len(reply) != 1 {
t.Error("Unexpected action plan received: ", utils.ToJSON(reply))
} else {
if reply[0].ActionPlanId != "ATMS_1" {
t.Errorf("Unexpected ActionoveAccountPlanId received")
}
}
}
// Make sure we have scheduled actions
func testApierITGetScheduledActionsForAccount(t *testing.T) {
var rply []*scheduler.ScheduledAction
if err := rater.Call(context.Background(), utils.APIerSv1GetScheduledActions,
scheduler.ArgsGetScheduledActions{
Tenant: utils.StringPointer("cgrates.org"),
Account: utils.StringPointer("dan7")}, &rply); err != nil {
t.Error("Unexpected error: ", err)
} else if len(rply) == 0 {
t.Errorf("ScheduledActions: %+v", rply)
}
}
// Test here RemoveActionTiming
func testApierRemUniqueIDActionTiming(t *testing.T) {
var rmReply string
rmReq := AttrRemoveActionTiming{ActionPlanId: "ATMS_1", Tenant: "cgrates.org", Account: "dan4"}
if err := rater.Call(context.Background(), utils.APIerSv1RemoveActionTiming, &rmReq, &rmReply); err != nil {
t.Error("Got error on APIerSv1.RemoveActionTiming: ", err.Error())
} else if rmReply != utils.OK {
t.Error("Unexpected answer received", rmReply)
}
var reply []*AccountActionTiming
req := utils.TenantAccount{Tenant: "cgrates.org", Account: "dan4"}
if err := rater.Call(context.Background(), utils.APIerSv1GetAccountActionPlan, &req, &reply); err != nil {
t.Error("Got error on APIerSv1.GetAccountActionPlan: ", err.Error())
} else if len(reply) != 0 {
t.Error("Action timings was not removed")
}
}
func testApierRemUniqueIDActionTimingWithoutTenant(t *testing.T) {
var rmReply string
rmReq := AttrRemoveActionTiming{ActionPlanId: "ATMS_1", Account: "dan4"}
if err := rater.Call(context.Background(), utils.APIerSv1RemoveActionTiming, &rmReq, &rmReply); err != nil {
t.Error("Got error on APIerSv1.RemoveActionTiming: ", err.Error())
} else if rmReply != utils.OK {
t.Error("Unexpected answer received", rmReply)
}
var reply []*AccountActionTiming
req := utils.TenantAccount{Account: "dan4"}
if err := rater.Call(context.Background(), utils.APIerSv1GetAccountActionPlan, &req, &reply); err != nil {
t.Error("Got error on APIerSv1.GetAccountActionPlan: ", err.Error())
} else if len(reply) != 0 {
t.Error("Action timings was not removed")
}
}
// Test here GetAccount
func testApierGetAccount(t *testing.T) {
var reply *engine.Account
attrs := &utils.AttrGetAccount{Tenant: "cgrates.org", Account: "1001"}
if err := rater.Call(context.Background(), utils.APIerSv2GetAccount, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.GetAccount: ", err.Error())
} else if reply.BalanceMap[utils.MetaMonetary].GetTotalValue() != 11.5 { // We expect 11.5 since we have added in the previous test 1.5
t.Errorf("Calling APIerSv1.GetBalance expected: 11.5, received: %f", reply.BalanceMap[utils.MetaMonetary].GetTotalValue())
}
attrs = &utils.AttrGetAccount{Tenant: "cgrates.org", Account: "dan"}
if err := rater.Call(context.Background(), utils.APIerSv2GetAccount, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.GetAccount: ", err.Error())
} else if reply.BalanceMap[utils.MetaMonetary].GetTotalValue() != 1.5 {
t.Errorf("Calling APIerSv1.GetAccount expected: 1.5, received: %f", reply.BalanceMap[utils.MetaMonetary].GetTotalValue())
}
// The one we have topped up though executeAction
attrs = &utils.AttrGetAccount{Tenant: "cgrates.org", Account: "dan2"}
if err := rater.Call(context.Background(), utils.APIerSv2GetAccount, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.GetAccount: ", err.Error())
} else if reply.BalanceMap[utils.MetaMonetary].GetTotalValue() != 11.5 {
t.Errorf("Calling APIerSv1.GetAccount expected: 10, received: %f", reply.BalanceMap[utils.MetaMonetary].GetTotalValue())
}
attrs = &utils.AttrGetAccount{Tenant: "cgrates.org", Account: "dan3"}
if err := rater.Call(context.Background(), utils.APIerSv2GetAccount, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.GetAccount: ", err.Error())
} else if reply.BalanceMap[utils.MetaMonetary].GetTotalValue() != 3.6 {
t.Errorf("Calling APIerSv1.GetAccount expected: 3.6, received: %f", reply.BalanceMap[utils.MetaMonetary].GetTotalValue())
}
attrs = &utils.AttrGetAccount{Tenant: "cgrates.org", Account: "dan6"}
if err := rater.Call(context.Background(), utils.APIerSv2GetAccount, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.GetAccount: ", err.Error())
} else if reply.BalanceMap[utils.MetaMonetary].GetTotalValue() != 1 {
t.Errorf("Calling APIerSv1.GetAccount expected: 1, received: %f", reply.BalanceMap[utils.MetaMonetary].GetTotalValue())
}
}
// Start with initial balance, top-up to test max_balance
func testApierTriggersExecute(t *testing.T) {
var reply string
attrs := &utils.AttrSetAccount{Tenant: "cgrates.org", Account: "dan8", ReloadScheduler: true}
if err := rater.Call(context.Background(), utils.APIerSv1SetAccount, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.SetAccount: ", err.Error())
} else if reply != utils.OK {
t.Errorf("Calling APIerSv1.SetAccount received: %s", reply)
}
attrAddBlnc := &AttrAddBalance{Tenant: "cgrates.org", Account: "1008", BalanceType: utils.MetaMonetary, Value: 2}
if err := rater.Call(context.Background(), utils.APIerSv1AddBalance, attrAddBlnc, &reply); err != nil {
t.Error("Got error on APIerSv1.AddBalance: ", err.Error())
} else if reply != utils.OK {
t.Errorf("Calling APIerSv1.AddBalance received: %s", reply)
}
}
// Start fresh before loading from folder
func testApierResetDataBeforeLoadFromFolder(t *testing.T) {
testApierInitDataDb(t)
var reply string
// Simple test that command is executed without errors
if err := rater.Call(context.Background(), utils.CacheSv1Clear, &utils.AttrCacheIDsWithAPIOpts{
CacheIDs: nil,
}, &reply); err != nil {
t.Error(err)
} else if reply != utils.OK {
t.Error("Reply: ", reply)
}
var rcvStats map[string]*ltcache.CacheStats
expectedStats := engine.GetDefaultEmptyCacheStats()
err := rater.Call(context.Background(), utils.CacheSv1GetCacheStats, new(utils.AttrCacheIDsWithAPIOpts), &rcvStats)
if err != nil {
t.Error("Got error on CacheSv1.GetCacheStats: ", err.Error())
} else if !reflect.DeepEqual(rcvStats, expectedStats) {
t.Errorf("Calling CacheSv1.GetCacheStats expected: %v, received: %v", utils.ToJSON(expectedStats), utils.ToJSON(rcvStats))
}
}
// Test here LoadTariffPlanFromFolder
func testApierLoadTariffPlanFromFolder(t *testing.T) {
var reply string
attrs := &utils.AttrLoadTpFromFolder{FolderPath: utils.EmptyString}
if err := rater.Call(context.Background(), utils.APIerSv1LoadTariffPlanFromFolder, attrs, &reply); err == nil || !strings.HasPrefix(err.Error(), utils.ErrMandatoryIeMissing.Error()) {
t.Error(err)
}
attrs = &utils.AttrLoadTpFromFolder{FolderPath: "/INVALID/"}
if err := rater.Call(context.Background(), utils.APIerSv1LoadTariffPlanFromFolder, attrs, &reply); err == nil || err.Error() != utils.ErrInvalidPath.Error() {
t.Error(err)
}
// Simple test that command is executed without errors
attrs = &utils.AttrLoadTpFromFolder{FolderPath: path.Join(*utils.DataDir, "tariffplans", "testtp")}
startTime := time.Now()
if err := rater.Call(context.Background(), utils.APIerSv1LoadTariffPlanFromFolder, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.LoadTariffPlanFromFolder: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.LoadTariffPlanFromFolder got reply: ", reply)
}
elapsedTime := time.Since(startTime)
expectedDuration := 1 * time.Second
if elapsedTime < expectedDuration || elapsedTime >= 2*time.Second {
t.Errorf("Expected elapsed time of at least %v, but got %v", expectedDuration, elapsedTime)
}
time.Sleep(100 * time.Millisecond)
}
// For now just test that they execute without errors
func testApierComputeReverse(t *testing.T) {
var reply string
if err := rater.Call(context.Background(), utils.APIerSv1ComputeReverseDestinations, utils.EmptyString, &reply); err != nil {
t.Error(err)
} else if reply != utils.OK {
t.Error("Received: ", reply)
}
if err := rater.Call(context.Background(), utils.APIerSv1ComputeAccountActionPlans, new(utils.TenantWithAPIOpts), &reply); err != nil {
t.Error(err)
} else if reply != utils.OK {
t.Error("Received: ", reply)
}
}
func testApierResetDataAfterLoadFromFolder(t *testing.T) {
time.Sleep(10 * time.Millisecond)
var rcvStats map[string]*ltcache.CacheStats
expStats := engine.GetDefaultEmptyCacheStats()
// expStats[utils.CacheAccountActionPlans].Items = 3 // the items are cleared from Cache on Compute
// expStats[utils.CacheActionPlans].Items = 7
expStats[utils.CacheActions].Items = 5
expStats[utils.CacheDestinations].Items = 3
expStats[utils.CacheLoadIDs].Items = 19
expStats[utils.CacheRPCConnections].Items = 2
if err := rater.Call(context.Background(), utils.CacheSv1GetCacheStats, new(utils.AttrCacheIDsWithAPIOpts), &rcvStats); err != nil {
t.Error("Got error on CacheSv1.GetCacheStats: ", err.Error())
} else if !reflect.DeepEqual(expStats, rcvStats) {
t.Errorf("Expecting: %+v,\n received: %+v", utils.ToJSON(expStats), utils.ToJSON(rcvStats))
}
var reply string
// Simple test that command is executed without errors
if err := rater.Call(context.Background(), utils.CacheSv1LoadCache, utils.NewAttrReloadCacheWithOpts(), &reply); err != nil {
t.Error(err)
} else if reply != utils.OK {
t.Error(reply)
}
expStats[utils.CacheAccountActionPlans].Items = 3
expStats[utils.CacheActionPlans].Items = 7
expStats[utils.CacheActionTriggers].Items = 1
expStats[utils.CacheActions].Items = 13
expStats[utils.CacheAttributeProfiles].Items = 1
expStats[utils.CacheFilters].Items = 15
expStats[utils.CacheRatingPlans].Items = 5
expStats[utils.CacheRatingProfiles].Items = 5
expStats[utils.CacheResourceProfiles].Items = 3
expStats[utils.CacheResources].Items = 3
expStats[utils.CacheReverseDestinations].Items = 5
expStats[utils.CacheStatQueueProfiles].Items = 1
expStats[utils.CacheStatQueues].Items = 1
expStats[utils.CacheRouteProfiles].Items = 2
expStats[utils.CacheThresholdProfiles].Items = 1
expStats[utils.CacheThresholds].Items = 1
expStats[utils.CacheLoadIDs].Items = 38
expStats[utils.CacheTimings].Items = 12
expStats[utils.CacheThresholdFilterIndexes].Items = 5
expStats[utils.CacheThresholdFilterIndexes].Groups = 1
expStats[utils.CacheStatFilterIndexes].Items = 2
expStats[utils.CacheStatFilterIndexes].Groups = 1
expStats[utils.CacheRouteFilterIndexes].Items = 2
expStats[utils.CacheRouteFilterIndexes].Groups = 1
expStats[utils.CacheResourceFilterIndexes].Items = 5
expStats[utils.CacheResourceFilterIndexes].Groups = 1
expStats[utils.CacheAttributeFilterIndexes].Items = 4
expStats[utils.CacheAttributeFilterIndexes].Groups = 1
expStats[utils.CacheReverseFilterIndexes].Items = 10
expStats[utils.CacheReverseFilterIndexes].Groups = 7
expStats[utils.CacheRankingProfiles].Items = 1
expStats[utils.CacheRankings].Items = 1
expStats[utils.CacheTrendProfiles].Items = 1
expStats[utils.CacheTrends].Items = 1
if err := rater.Call(context.Background(), utils.CacheSv1GetCacheStats, new(utils.AttrCacheIDsWithAPIOpts), &rcvStats); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expStats, rcvStats) {
t.Errorf("Expecting: %+v, \n received: %+v", utils.ToJSON(expStats), utils.ToJSON(rcvStats))
}
}
func testApierSetChargerS(t *testing.T) {
//add a default charger
chargerProfile := &ChargerWithAPIOpts{
ChargerProfile: &engine.ChargerProfile{
Tenant: "cgrates.org",
ID: "Default",
RunID: utils.MetaDefault,
AttributeIDs: []string{"*none"},
Weight: 20,
},
}
var result string
startTime := time.Now()
if err := rater.Call(context.Background(), utils.APIerSv1SetChargerProfile, chargerProfile, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
elapsedTime := time.Since(startTime)
expectedDuration := 1 * time.Second
if elapsedTime < expectedDuration || elapsedTime >= 2*time.Second {
t.Errorf("Expected elapsed time of at least %v, but got %v", expectedDuration, elapsedTime)
}
}
// Make sure balance was topped-up
// Bug reported by DigiDaz over IRC
func testApierGetAccountAfterLoad(t *testing.T) {
var reply *engine.Account
attrs := &utils.AttrGetAccount{Tenant: "cgrates.org", Account: "1001"}
if err := rater.Call(context.Background(), utils.APIerSv2GetAccount, attrs, &reply); err != nil {
t.Error("Got error on APIerSv1.GetAccount: ", err.Error())
} else if reply.BalanceMap[utils.MetaMonetary].GetTotalValue() != 13 {
t.Errorf("Calling APIerSv1.GetBalance expected: 13, received: %v \n\n for:%s", reply.BalanceMap[utils.MetaMonetary].GetTotalValue(), utils.ToJSON(reply))
}
}
// Test here ResponderGetCost
func testApierResponderGetCost(t *testing.T) {
tStart, _ := utils.ParseTimeDetectLayout("2013-08-07T17:30:00Z", utils.EmptyString)
tEnd, _ := utils.ParseTimeDetectLayout("2013-08-07T17:31:30Z", utils.EmptyString)
cd := &engine.CallDescriptorWithAPIOpts{
CallDescriptor: &engine.CallDescriptor{
Category: "call",
Tenant: "cgrates.org",
Subject: "1001",
Account: "1001",
Destination: "+4917621621391",
DurationIndex: 90,
TimeStart: tStart,
TimeEnd: tEnd,
},
}
var cc engine.CallCost
// Simple test that command is executed without errors
if err := rater.Call(context.Background(), utils.ResponderGetCost, cd, &cc); err != nil {
t.Error("Got error on Responder.GetCost: ", err.Error())
} else if cc.Cost != 90.0 {
t.Errorf("Calling Responder.GetCost got callcost: %v", cc)
}
}
func testApierMaxDebitInexistentAcnt(t *testing.T) {
cc := &engine.CallCost{}
cd := &engine.CallDescriptorWithAPIOpts{
CallDescriptor: &engine.CallDescriptor{
Tenant: "cgrates.org",
Category: "call",
Subject: "INVALID",
Account: "INVALID",
Destination: "1002",
TimeStart: time.Date(2014, 3, 27, 10, 42, 26, 0, time.UTC),
TimeEnd: time.Date(2014, 3, 27, 10, 42, 26, 0, time.UTC).Add(10 * time.Second),
},
}
if err := rater.Call(context.Background(), utils.ResponderMaxDebit, cd, cc); err == nil {
t.Error(err)
}
if err := rater.Call(context.Background(), utils.ResponderDebit, cd, cc); err == nil {
t.Error(err)
}
}
func testApierCdrServer(t *testing.T) {
httpClient := new(http.Client)
cdrForm1 := url.Values{utils.OriginID: []string{"dsafdsaf"}, utils.OriginHost: []string{"192.168.1.1"}, utils.RequestType: []string{utils.MetaRated},
utils.Tenant: []string{"cgrates.org"}, utils.Category: []string{"call"}, utils.AccountField: []string{"1001"}, utils.Subject: []string{"1001"}, utils.Destination: []string{"1002"},
utils.SetupTime: []string{"2013-11-07T08:42:22Z"},
utils.AnswerTime: []string{"2013-11-07T08:42:26Z"}, utils.Usage: []string{"10"}, "field_extr1": []string{"val_extr1"}, "fieldextr2": []string{"valextr2"}}
cdrForm2 := url.Values{utils.OriginID: []string{"adsafdsaf"}, utils.OriginHost: []string{"192.168.1.1"}, utils.RequestType: []string{utils.MetaRated},
utils.Tenant: []string{"cgrates.org"}, utils.Category: []string{"call"}, utils.AccountField: []string{"1001"}, utils.Subject: []string{"1001"}, utils.Destination: []string{"1002"},
utils.SetupTime: []string{"2013-11-07T08:42:23Z"},
utils.AnswerTime: []string{"2013-11-07T08:42:26Z"}, utils.Usage: []string{"10"}, "field_extr1": []string{"val_extr1"}, "fieldextr2": []string{"valextr2"}}
for _, cdrForm := range []url.Values{cdrForm1, cdrForm2} {
cdrForm.Set(utils.Source, utils.TestSQL)
if _, err := httpClient.PostForm(fmt.Sprintf("http://%s/cdr_http", "127.0.0.1:2080"), cdrForm); err != nil {
t.Error(err.Error())
}
}
time.Sleep(time.Duration(*utils.WaitRater) * time.Millisecond)
}
func testApierITGetCdrs(t *testing.T) {
var reply []*engine.ExternalCDR
req := utils.AttrGetCdrs{MediationRunIds: []string{utils.MetaDefault}}
if err := rater.Call(context.Background(), utils.APIerSv1GetCDRs, &req, &reply); err != nil {
t.Error("Unexpected error: ", err.Error())
} else if len(reply) != 2 {
t.Error("Unexpected number of CDRs returned: ", len(reply))
}
}
func testApierITProcessCdr(t *testing.T) {
var reply string
cdr := &engine.CDRWithAPIOpts{
CDR: &engine.CDR{CGRID: utils.Sha1("dsafdsaf", time.Date(2013, 11, 7, 8, 42, 26, 0, time.UTC).String()), OrderID: 123, ToR: utils.MetaVoice, OriginID: "dsafdsaf",
OriginHost: "192.168.1.1", Source: "test", RequestType: utils.MetaRated, Tenant: "cgrates.org", Category: "call", Account: "1001", Subject: "1001",
Destination: "1002",
SetupTime: time.Date(2013, 11, 7, 8, 42, 26, 0, time.UTC), AnswerTime: time.Date(2013, 11, 7, 8, 42, 26, 0, time.UTC), RunID: utils.MetaDefault,
Usage: 10 * time.Second, ExtraFields: map[string]string{"field_extr1": "val_extr1", "fieldextr2": "valextr2"}, Cost: 1.01,
},
}
if err := rater.Call(context.Background(), utils.CDRsV1ProcessCDR, cdr, &reply); err != nil {
t.Error("Unexpected error: ", err.Error())
} else if reply != utils.OK {
t.Error("Unexpected reply received: ", reply)
}
var cdrs []*engine.ExternalCDR
req := utils.AttrGetCdrs{MediationRunIds: []string{utils.MetaDefault}}
if err := rater.Call(context.Background(), utils.APIerSv1GetCDRs, &req, &cdrs); err != nil {
t.Error("Unexpected error: ", err.Error())
} else if len(cdrs) != 3 {
t.Error("Unexpected number of CDRs returned: ", len(cdrs))
}
}
// Test here ResponderGetCost
func testApierGetCallCostLog(t *testing.T) {
var cc engine.EventCost
var attrs utils.AttrGetCallCost
// Simple test that command is executed without errors
if err := rater.Call(context.Background(), utils.APIerSv1GetEventCost, &attrs, &cc); err == nil {
t.Error("Failed to detect missing fields in APIerSv1.GetCallCostLog")
}
attrs.CgrId = "dummyid"
attrs.RunId = "default"
if err := rater.Call(context.Background(), utils.APIerSv1GetEventCost, &attrs, &cc); err == nil || err.Error() != utils.ErrNotFound.Error() {
t.Error("APIerSv1.GetCallCostLog: should return NOT_FOUND, got:", err)
}
tm := time.Now().Truncate(time.Millisecond).UTC()
cdr := &engine.CDRWithAPIOpts{
CDR: &engine.CDR{
CGRID: "Cdr1",
OrderID: 123,
ToR: utils.MetaVoice,
OriginID: "OriginCDR1",
OriginHost: "192.168.1.1",
Source: "test",
RequestType: utils.MetaRated,
Tenant: "cgrates.org",
Category: "call",
Account: "1001",
Subject: "1001",
Destination: "+4986517174963",
SetupTime: tm,
AnswerTime: tm,
RunID: utils.MetaDefault,
Usage: 0,
ExtraFields: map[string]string{"field_extr1": "val_extr1", "fieldextr2": "valextr2"},
Cost: 1.01,
},
}
var reply string
if err := rater.Call(context.Background(), utils.CDRsV1ProcessCDR, cdr, &reply); err != nil {
t.Error("Unexpected error: ", err.Error())
} else if reply != utils.OK {
t.Error("Unexpected reply received: ", reply)
}
time.Sleep(100 * time.Millisecond)
expected := engine.EventCost{
CGRID: "Cdr1",
RunID: utils.MetaDefault,
StartTime: tm,
Usage: utils.DurationPointer(0),
Cost: utils.Float64Pointer(0),
Charges: []*engine.ChargingInterval{{
RatingID: utils.EmptyString,
Increments: nil,
CompressFactor: 0,
}},
AccountSummary: nil,
Rating: engine.Rating{},
Accounting: engine.Accounting{},
RatingFilters: engine.RatingFilters{},
Rates: engine.ChargedRates{},
Timings: engine.ChargedTimings{},
}
if *utils.Encoding == utils.MetaGOB {
expected.Usage = nil // 0 value are encoded as nil in gob
expected.Cost = nil
}
attrs.CgrId = "Cdr1"
attrs.RunId = utils.EmptyString
if err := rater.Call(context.Background(), utils.APIerSv1GetEventCost, &attrs, &cc); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expected, cc) {
t.Errorf("Expecting %s ,received %s", utils.ToJSON(expected), utils.ToJSON(cc))
}
}
func testApierITSetDestination(t *testing.T) {
attrs := utils.AttrSetDestination{Id: "TEST_SET_DESTINATION", Prefixes: []string{"+4986517174963", "+4986517174960"}}
var reply string
if err := rater.Call(context.Background(), utils.APIerSv1SetDestination, &attrs, &reply); err != nil {
t.Error("Unexpected error", err.Error())
} else if reply != utils.OK {
t.Error("Unexpected reply returned", reply)
}
if err := rater.Call(context.Background(), utils.APIerSv1SetDestination, &attrs, &reply); err == nil || err.Error() != "EXISTS" { // Second time without overwrite should generate error
t.Error("Unexpected error", err.Error())
}
attrs = utils.AttrSetDestination{Id: "TEST_SET_DESTINATION", Prefixes: []string{"+4986517174963", "+4986517174964"}, Overwrite: true}
if err := rater.Call(context.Background(), utils.APIerSv1SetDestination, &attrs, &reply); err != nil {
t.Error("Unexpected error", err.Error())
} else if reply != utils.OK {
t.Error("Unexpected reply returned", reply)
}
eDestination := engine.Destination{Id: attrs.Id, Prefixes: attrs.Prefixes}
var rcvDestination engine.Destination
if err := rater.Call(context.Background(), utils.APIerSv1GetDestination, &attrs.Id, &rcvDestination); err != nil {
t.Error("Unexpected error", err.Error())
} else if !reflect.DeepEqual(eDestination, rcvDestination) {
t.Errorf("Expecting: %+v, received: %+v", eDestination, rcvDestination)
}
eRcvIDs := []string{attrs.Id}
var rcvIDs []string
if err := rater.Call(context.Background(), utils.APIerSv1GetReverseDestination, &attrs.Prefixes[0], &rcvIDs); err != nil {
t.Error("Unexpected error", err.Error())
} else if !reflect.DeepEqual(eRcvIDs, rcvIDs) {
t.Errorf("Expecting: %+v, received: %+v", eRcvIDs, rcvIDs)
}
}
func testApierITGetScheduledActions(t *testing.T) {
var rply []*scheduler.ScheduledAction
if err := rater.Call(context.Background(), utils.APIerSv1GetScheduledActions, scheduler.ArgsGetScheduledActions{}, &rply); err != nil {
t.Error("Unexpected error: ", err)
}
}
func testApierITGetDataCost(t *testing.T) {
attrs := AttrGetDataCost{Category: "data", Tenant: "cgrates.org",
Subject: "1001", AnswerTime: utils.MetaNow, Usage: 640113}
var rply *engine.DataCost
if err := rater.Call(context.Background(), utils.APIerSv1GetDataCost, &attrs, &rply); err != nil {
t.Error("Unexpected nil error received: ", err.Error())
} else if rply.Cost != 128.0240 {
t.Errorf("Unexpected cost received: %f", rply.Cost)
}
}
func testApierITGetCost(t *testing.T) {
attrs := AttrGetCost{Category: "data", Tenant: "cgrates.org",
Subject: "1001", AnswerTime: utils.MetaNow, Usage: "640113"}
var rply *engine.EventCost
if err := rater.Call(context.Background(), utils.APIerSv1GetCost, &attrs, &rply); err != nil {
t.Error("Unexpected nil error received: ", err.Error())
} else if *rply.Cost != 128.0240 {
t.Errorf("Unexpected cost received: %f", *rply.Cost)
}
}
// Test LoadTPFromStorDb
func testApierInitDataDb2(t *testing.T) {
if err := engine.InitDataDB(cfg); err != nil {
t.Fatal(err)
}
}
func testApierInitStorDb2(t *testing.T) {
if err := engine.InitStorDb(cfg); err != nil {
t.Fatal(err)
}
}
func testApierReloadCache2(t *testing.T) {
var reply string
// Simple test that command is executed without errors
if err := rater.Call(context.Background(), utils.CacheSv1Clear, &utils.AttrCacheIDsWithAPIOpts{
CacheIDs: nil,
}, &reply); err != nil {
t.Error("Got error on CacheSv1.ReloadCache: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling CacheSv1.ReloadCache got reply: ", reply)
}
}
func testApierReloadScheduler2(t *testing.T) {
var reply string
// Simple test that command is executed without errors
if err := rater.Call(context.Background(), utils.SchedulerSv1Reload, utils.StringWithAPIOpts{}, &reply); err != nil {
t.Error("Got error on SchedulerSv1.Reload: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling SchedulerSv1.Reload got reply: ", reply)
}
}
func testApierImportTPFromFolderPath(t *testing.T) {
var reply string
if err := rater.Call(context.Background(), utils.APIerSv1ImportTariffPlanFromFolder,
utils.AttrImportTPFromFolder{TPid: "TEST_TPID2",
FolderPath: "/usr/share/cgrates/tariffplans/oldtutorial"}, &reply); err != nil {
t.Error("Got error on APIerSv1.ImportTarrifPlanFromFolder: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.ImportTarrifPlanFromFolder got reply: ", reply)
}
time.Sleep(100 * time.Millisecond)
}
func testApierLoadTariffPlanFromStorDbDryRun(t *testing.T) {
var reply string
if err := rater.Call(context.Background(), utils.APIerSv1LoadTariffPlanFromStorDb,
&AttrLoadTpFromStorDb{TPid: "TEST_TPID2", DryRun: true}, &reply); err != nil {
t.Error("Got error on APIerSv1.LoadTariffPlanFromStorDb: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.LoadTariffPlanFromStorDb got reply: ", reply)
}
}
func testApierGetCacheStats2(t *testing.T) {
var rcvStats map[string]*ltcache.CacheStats
expectedStats := engine.GetDefaultEmptyCacheStats()
err := rater.Call(context.Background(), utils.CacheSv1GetCacheStats, new(utils.AttrCacheIDsWithAPIOpts), &rcvStats)
if err != nil {
t.Error("Got error on CacheSv1.GetCacheStats: ", err.Error())
} else if !reflect.DeepEqual(expectedStats, rcvStats) {
t.Errorf("Calling CacheSv1.GetCacheStats expected: %v, received: %v", expectedStats, rcvStats)
}
}
func testApierLoadTariffPlanFromStorDb(t *testing.T) {
var reply string
startTime := time.Now()
if err := rater.Call(context.Background(), utils.APIerSv1LoadTariffPlanFromStorDb,
&AttrLoadTpFromStorDb{TPid: "TEST_TPID2"}, &reply); err != nil {
t.Error("Got error on APIerSv1.LoadTariffPlanFromStorDb: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.LoadTariffPlanFromStorDb got reply: ", reply)
}
elapsedTime := time.Since(startTime)
expectedDuration := 1 * time.Second
if elapsedTime < expectedDuration || elapsedTime >= 2*time.Second {
t.Errorf("Expected elapsed time of at least %v, but got %v", expectedDuration, elapsedTime)
}
}
func testApierStartStopServiceStatus(t *testing.T) {
var reply string
if err := rater.Call(context.Background(), utils.ServiceManagerV1ServiceStatus, dispatchers.ArgStartServiceWithAPIOpts{ArgStartService: servmanager.ArgStartService{ServiceID: utils.MetaScheduler}},
&reply); err != nil {
t.Error(err)
} else if reply != utils.RunningCaps {
t.Errorf("Received: <%s>", reply)
}
if err := rater.Call(context.Background(), utils.ServiceManagerV1StopService, dispatchers.ArgStartServiceWithAPIOpts{ArgStartService: servmanager.ArgStartService{ServiceID: "INVALID"}},
&reply); err == nil || err.Error() != utils.UnsupportedServiceIDCaps {
t.Error(err)
}
if err := rater.Call(context.Background(), utils.ServiceManagerV1StopService, dispatchers.ArgStartServiceWithAPIOpts{ArgStartService: servmanager.ArgStartService{ServiceID: utils.MetaScheduler}},
&reply); err != nil {
t.Error(err)
} else if reply != utils.OK {
t.Errorf("Received: <%s>", reply)
}
if err := rater.Call(context.Background(), utils.ServiceManagerV1ServiceStatus, dispatchers.ArgStartServiceWithAPIOpts{ArgStartService: servmanager.ArgStartService{ServiceID: utils.MetaScheduler}},
&reply); err != nil {
t.Error(err)
} else if reply != utils.StoppedCaps {
t.Errorf("Received: <%s>", reply)
}
if err := rater.Call(context.Background(), utils.ServiceManagerV1StartService, &dispatchers.ArgStartServiceWithAPIOpts{ArgStartService: servmanager.ArgStartService{ServiceID: utils.MetaScheduler}},
&reply); err != nil {
t.Error(err)
} else if reply != utils.OK {
t.Errorf("Received: <%s>", reply)
}
if err := rater.Call(context.Background(), utils.ServiceManagerV1ServiceStatus, dispatchers.ArgStartServiceWithAPIOpts{ArgStartService: servmanager.ArgStartService{ServiceID: utils.MetaScheduler}},
&reply); err != nil {
t.Error(err)
} else if reply != utils.RunningCaps {
t.Errorf("Received: <%s>", reply)
}
if err := rater.Call(context.Background(), utils.SchedulerSv1Reload, utils.StringWithAPIOpts{}, &reply); err != nil {
t.Error("Got error on SchedulerSv1.Reload: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling SchedulerSv1.Reload got reply: ", reply)
}
}
func testApierSetRatingProfileWithoutTenant(t *testing.T) {
var reply string
rpa := &utils.TPRatingActivation{ActivationTime: "2012-01-01T00:00:00Z", RatingPlanId: "RETAIL1", FallbackSubjects: "dan4"}
rpf := &utils.AttrSetRatingProfile{Category: utils.Call, Subject: "dan3", RatingPlanActivations: []*utils.TPRatingActivation{rpa}}
if err := rater.Call(context.Background(), utils.APIerSv1SetRatingProfile, &rpf, &reply); err != nil {
t.Error("Got error on APIerSv1.SetRatingProfile: ", err.Error())
} else if reply != utils.OK {
t.Error("Calling APIerSv1.SetRatingProfile got reply: ", reply)
}
expectedID := utils.ConcatenatedKey(utils.MetaOut, "cgrates.org", utils.Call, "dan3")
var result *engine.RatingProfile
if err := rater.Call(context.Background(), utils.APIerSv1GetRatingProfile,
&utils.AttrGetRatingProfile{Category: utils.Call, Subject: "dan3"},
&result); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expectedID, result.Id) {
t.Errorf("Expected %+v, received %+v", expectedID, result.Id)
}
}
func testApierRemoveRatingProfilesWithoutTenant(t *testing.T) {
var reply string
startTime := time.Now()
if err := rater.Call(context.Background(), utils.APIerSv1RemoveRatingProfile, &AttrRemoveRatingProfile{
Category: utils.Call,
Subject: "dan3",
}, &reply); err != nil {
t.Error(err)
} else if reply != utils.OK {
t.Errorf("Expected: %s, received: %s ", utils.OK, reply)
}
elapsedTime := time.Since(startTime)
expectedDuration := 1 * time.Second
if elapsedTime < expectedDuration || elapsedTime >= 2*time.Second {
t.Errorf("Expected elapsed time of at least %v, but got %v", expectedDuration, elapsedTime)
}
var result *engine.RatingProfile
if err := rater.Call(context.Background(), utils.APIerSv1GetRatingProfile,
&utils.AttrGetRatingProfile{Category: utils.Call, Subject: "dan3"},
&result); err == nil || err.Error() != utils.ErrNotFound.Error() {
t.Error(err)
}
}
func testApierReplayFldPosts(t *testing.T) {
bev := []byte(`{"ID":"cgrates.org:1007","BalanceMap":{"*monetary":[{"Uuid":"367be35a-96ee-40a5-b609-9130661f5f12","ID":"","Value":0,"ExpirationDate":"0001-01-01T00:00:00Z","Weight":10,"DestinationIDs":{},"RatingSubject":"","Categories":{},"SharedGroups":{"SHARED_A":true},"Timings":null,"TimingIDs":{},"Disabled":false,"Factors":null,"Blocker":false}]},"UnitCounters":{"*monetary":[{"CounterType":"*event","Counters":[{"Value":0,"Filter":{"Uuid":null,"ID":"b8531413-10d5-47ad-81ad-2bc272e8f0ca","Type":"*monetary","Value":null,"ExpirationDate":null,"Weight":null,"DestinationIDs":{"FS_USERS":true},"RatingSubject":null,"Categories":null,"SharedGroups":null,"TimingIDs":null,"Timings":null,"Disabled":null,"Factors":null,"Blocker":null}}]}]},"ActionTriggers":[{"ID":"STANDARD_TRIGGERS","UniqueID":"46ac7b8c-685d-4555-bf73-fa6cfbc2fa21","ThresholdType":"*min_balance","ThresholdValue":2,"Recurrent":false,"MinSleep":0,"ExpirationDate":"0001-01-01T00:00:00Z","ActivationDate":"0001-01-01T00:00:00Z","Balance":{"Uuid":null,"ID":null,"Type":"*monetary","Value":null,"ExpirationDate":null,"Weight":null,"DestinationIDs":null,"RatingSubject":null,"Categories":null,"SharedGroups":null,"TimingIDs":null,"Timings":null,"Disabled":null,"Factors":null,"Blocker":null},"Weight":10,"ActionsID":"LOG_WARNING","MinQueuedItems":0,"Executed":true,"LastExecutionTime":"2017-01-31T14:03:57.961651647+01:00"},{"ID":"STANDARD_TRIGGERS","UniqueID":"b8531413-10d5-47ad-81ad-2bc272e8f0ca","ThresholdType":"*max_event_counter","ThresholdValue":5,"Recurrent":false,"MinSleep":0,"ExpirationDate":"0001-01-01T00:00:00Z","ActivationDate":"0001-01-01T00:00:00Z","Balance":{"Uuid":null,"ID":null,"Type":"*monetary","Value":null,"ExpirationDate":null,"Weight":null,"DestinationIDs":{"FS_USERS":true},"RatingSubject":null,"Categories":null,"SharedGroups":null,"TimingIDs":null,"Timings":null,"Disabled":null,"Factors":null,"Blocker":null},"Weight":10,"ActionsID":"LOG_WARNING","MinQueuedItems":0,"Executed":false,"LastExecutionTime":"0001-01-01T00:00:00Z"},{"ID":"STANDARD_TRIGGERS","UniqueID":"8b424186-7a31-4aef-99c5-35e12e6fed41","ThresholdType":"*max_balance","ThresholdValue":20,"Recurrent":false,"MinSleep":0,"ExpirationDate":"0001-01-01T00:00:00Z","ActivationDate":"0001-01-01T00:00:00Z","Balance":{"Uuid":null,"ID":null,"Type":"*monetary","Value":null,"ExpirationDate":null,"Weight":null,"DestinationIDs":null,"RatingSubject":null,"Categories":null,"SharedGroups":null,"TimingIDs":null,"Timings":null,"Disabled":null,"Factors":null,"Blocker":null},"Weight":10,"ActionsID":"LOG_WARNING","MinQueuedItems":0,"Executed":false,"LastExecutionTime":"0001-01-01T00:00:00Z"},{"ID":"STANDARD_TRIGGERS","UniqueID":"28557f3b-139c-4a27-9d17-bda1f54b7c19","ThresholdType":"*max_balance","ThresholdValue":100,"Recurrent":false,"MinSleep":0,"ExpirationDate":"0001-01-01T00:00:00Z","ActivationDate":"0001-01-01T00:00:00Z","Balance":{"Uuid":null,"ID":null,"Type":"*monetary","Value":null,"ExpirationDate":null,"Weight":null,"DestinationIDs":null,"RatingSubject":null,"Categories":null,"SharedGroups":null,"TimingIDs":null,"Timings":null,"Disabled":null,"Factors":null,"Blocker":null},"Weight":10,"ActionsID":"DISABLE_AND_LOG","MinQueuedItems":0,"Executed":false,"LastExecutionTime":"0001-01-01T00:00:00Z"}],"AllowNegative":false,"Disabled":false}"`)
ev := &ees.ExportEvents{
Path: "http://localhost:2081",
Type: utils.MetaHTTPjsonMap,
Attempts: 1,
Events: []any{&ees.HTTPPosterRequest{Body: bev, Header: http.Header{"Content-Type": []string{"application/json"}}}},
}
fileName := "act>*http_post|63bed4ea-615e-4096-b1f4-499f64f29b28.json"
args := ReplayFailedPostsParams{
SourcePath: t.TempDir(),
FailedPath: t.TempDir(),
}
err := ev.WriteToFile(path.Join(args.SourcePath, fileName))
if err != nil {
t.Error(err)
}
var reply string
if err := rater.Call(context.Background(), utils.APIerSv1ReplayFailedPosts, args, &reply); err != nil {
t.Error(err)
} else if reply != utils.OK {
t.Error("Unexpected reply: ", reply)
}
outPath := path.Join(args.FailedPath, fileName)
outEv, err := ees.NewExportEventsFromFile(outPath)
if err != nil {
t.Error(err)
} else if !reflect.DeepEqual(ev, outEv) {
t.Errorf("Expecting: %s, received: %s", utils.ToJSON(ev), utils.ToJSON(outEv))
}
fileName = "cdr|ae8cc4b3-5e60-4396-b82a-64b96a72a03c.json"
bev = []byte(`{"CGRID":"88ed9c38005f07576a1e1af293063833b60edcc6"}`)
fileInPath := path.Join(args.SourcePath, fileName)
ev = &ees.ExportEvents{
Path: "amqp://guest:guest@localhost:5672/",
Attempts: 1,
Opts: &config.EventExporterOpts{
AMQP: &config.AMQPOpts{
QueueID: utils.StringPointer("cgrates_cdrs")},
},
Type: utils.MetaAMQPjsonMap,
Events: []any{bev},
}
err = ev.WriteToFile(path.Join(args.SourcePath, fileName))
if err != nil {
t.Error(err)
}
if err := rater.Call(context.Background(), utils.APIerSv1ReplayFailedPosts, args, &reply); err != nil {
t.Error(err)
} else if reply != utils.OK {
t.Error("Unexpected reply: ", reply)
}
if _, err := os.Stat(fileInPath); !os.IsNotExist(err) {
t.Error("InFile still exists")
}
if _, err := os.Stat(path.Join(args.FailedPath, fileName)); !os.IsNotExist(err) {
t.Error("OutFile created")
}
// connect to RabbitMQ server and check if the content was posted there
conn, err := amqp.Dial("amqp://guest:guest@localhost:5672/")
if err != nil {
t.Fatal(err)
}
defer conn.Close()
ch, err := conn.Channel()
if err != nil {
t.Fatal(err)
}
defer ch.Close()
q, err := ch.QueueDeclare("cgrates_cdrs", true, false, false, false, nil)
if err != nil {
t.Fatal(err)
}
msgs, err := ch.Consume(q.Name, utils.EmptyString, true, false, false, false, nil)
if err != nil {
t.Fatal(err)
}
select {
case d := <-msgs:
var rcvCDR map[string]string
if err := json.Unmarshal(d.Body, &rcvCDR); err != nil {
t.Error(err)
}
if rcvCDR[utils.CGRID] != "88ed9c38005f07576a1e1af293063833b60edcc6" {
t.Errorf("Unexpected CDR received: %+v", rcvCDR)
}
case <-time.After(100 * time.Millisecond):
t.Error("No message received from RabbitMQ")
}
for _, dir := range []string{args.SourcePath, args.FailedPath} {
if err := os.RemoveAll(dir); err != nil {
t.Errorf("Error %s removing folder: %s", err, dir)
}
}
}
func testApierGetDataDBVesions(t *testing.T) {
var reply *engine.Versions
if err := rater.Call(context.Background(), utils.APIerSv1GetDataDBVersions, utils.StringPointer(utils.EmptyString), &reply); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(engine.CurrentDataDBVersions(), *reply) {
t.Errorf("Expecting : %+v, received: %+v", engine.CurrentDataDBVersions(), *reply)
}
}
func testApierGetStorDBVesions(t *testing.T) {
var reply *engine.Versions
if err := rater.Call(context.Background(), utils.APIerSv1GetStorDBVersions, utils.StringPointer(utils.EmptyString), &reply); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(engine.CurrentStorDBVersions(), *reply) {
t.Errorf("Expecting : %+v, received: %+v", engine.CurrentStorDBVersions(), *reply)
}
}
func testApierBackwardsCompatible(t *testing.T) {
var reply string
if err := rater.Call(context.Background(), "ApierV1.Ping", new(utils.CGREvent), &reply); err != nil {
t.Error(err)
} else if reply != utils.Pong {
t.Errorf("Expecting : %+v, received: %+v", utils.Pong, reply)
}
}
// Simply kill the engine after we are done with tests within this file
func testApierStopEngine(t *testing.T) {
if err := engine.KillEngine(100); err != nil {
t.Error(err)
}
}
// Start tests for caching
func testRatingProfileCachingMetaNone(t *testing.T) {
//*none option should not add profile in cache only in Datamanager
rpf := &utils.AttrSetRatingProfile{
Tenant: "cgrates.org",
Category: "call",
Subject: "dan",
RatingPlanActivations: []*utils.TPRatingActivation{{
ActivationTime: "2012-01-01T00:00:00Z",
RatingPlanId: "RETAIL1",
FallbackSubjects: "dan2",
}},
Overwrite: true,
APIOpts: map[string]any{utils.CacheOpt: utils.MetaNone},
}
// set the profile
var result string
if err := rater.Call(context.Background(), utils.APIerSv1SetRatingProfile, rpf, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
var reply bool
argsCache := utils.ArgsGetCacheItem{
CacheID: utils.CacheRatingProfiles,
ItemID: "*out:cgrates.org:call:dan",
}
if err := rater.Call(context.Background(), utils.CacheSv1HasItem, argsCache, &reply); err != nil {
t.Error(err)
} else if reply {
t.Errorf("Expected: false, received:%v", reply)
}
var rcvKeys []string
argsCache2 := utils.ArgsGetCacheItemIDs{
CacheID: utils.CacheRatingProfiles,
}
if err := rater.Call(context.Background(), utils.CacheSv1GetItemIDs, argsCache2, &rcvKeys); err == nil ||
err.Error() != utils.ErrNotFound.Error() {
t.Fatalf("Expected error: %s received error: %s and reply: %v ",
utils.ErrNotFound, err.Error(), rcvKeys)
}
//check in dataManager
expected := []string{"call:dan"}
var rcvIDs []string
if err := rater.Call(context.Background(), utils.APIerSv1GetRatingProfileIDs, utils.PaginatorWithTenant{Tenant: "cgrates.org"}, &rcvIDs); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expected, rcvIDs) {
t.Errorf("Expecting : %+v, received: %+v", expected, rcvIDs)
}
}
func testRatingProfileCachingMetaLoad(t *testing.T) {
//*load option should add profile in cache and in Datamanager
rpf := &utils.AttrSetRatingProfile{
Tenant: "cgrates.org",
Category: "call",
Subject: "dan",
RatingPlanActivations: []*utils.TPRatingActivation{{
ActivationTime: "2012-01-01T00:00:00Z",
RatingPlanId: "RETAIL1",
FallbackSubjects: "dan2",
}},
APIOpts: map[string]any{utils.CacheOpt: utils.MetaLoad},
}
// set the profile
var result string
if err := rater.Call(context.Background(), utils.APIerSv1SetRatingProfile, rpf, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
var reply bool
argsCache := utils.ArgsGetCacheItem{
CacheID: utils.CacheRatingProfiles,
ItemID: "*out:cgrates.org:call:dan",
}
if err := rater.Call(context.Background(), utils.CacheSv1HasItem, argsCache, &reply); err != nil {
t.Error(err)
} else if !reply {
t.Errorf("Expected: true, received:%v", reply)
}
var rcvKeys []string
expectedIDs := []string{"*out:cgrates.org:call:dan"}
argsCache2 := utils.ArgsGetCacheItemIDs{
CacheID: utils.CacheRatingProfiles,
}
if err := rater.Call(context.Background(), utils.CacheSv1GetItemIDs, argsCache2, &rcvKeys); err != nil {
t.Fatal(err)
} else if !reflect.DeepEqual(rcvKeys, expectedIDs) {
t.Errorf("Expecting : %+v, received: %+v", expectedIDs, rcvKeys)
}
//check in dataManager
expected := []string{"call:dan"}
var rcvIDs []string
if err := rater.Call(context.Background(), utils.APIerSv1GetRatingProfileIDs, utils.PaginatorWithTenant{Tenant: "cgrates.org"}, &rcvIDs); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expected, rcvIDs) {
t.Errorf("Expecting : %+v, received: %+v", expected, rcvIDs)
}
//remove from cache and DataManager the profile
var resp string
if err := rater.Call(context.Background(), utils.APIerSv1RemoveRatingProfile,
AttrRemoveRatingProfile{
Tenant: rpf.Tenant,
Category: rpf.Category,
Subject: rpf.Subject,
APIOpts: map[string]any{utils.CacheOpt: utils.MetaRemove}}, &resp); err != nil {
t.Error(err)
} else if resp != utils.OK {
t.Error("Unexpected reply returned", resp)
}
argsCache = utils.ArgsGetCacheItem{
CacheID: utils.CacheRatingProfiles,
ItemID: "*out:cgrates.org:call:dan",
}
if err := rater.Call(context.Background(), utils.CacheSv1HasItem, argsCache, &reply); err != nil {
t.Error(err)
} else if reply {
t.Errorf("Expected: false, received:%v", reply)
}
if err := rater.Call(context.Background(), utils.CacheSv1GetItemIDs, argsCache2, &rcvKeys); err == nil ||
err.Error() != utils.ErrNotFound.Error() {
t.Fatalf("Expected error: %s received error: %s and reply: %v ",
utils.ErrNotFound, err, rcvKeys)
}
//check in dataManager
if err := rater.Call(context.Background(), utils.APIerSv1GetRatingProfileIDs, utils.PaginatorWithTenant{Tenant: "cgrates.org"}, &rcvIDs); err == nil ||
err.Error() != utils.ErrNotFound.Error() {
t.Fatalf("Expected error: %s received error: %s and reply: %v ",
utils.ErrNotFound, err, rcvIDs)
}
}
func testRatingProfileCachingMetaReload1(t *testing.T) {
//*reload add the profile in cache if was there before
rpf := &utils.AttrSetRatingProfile{
Tenant: "cgrates.org",
Category: "call",
Subject: "dan",
RatingPlanActivations: []*utils.TPRatingActivation{{
ActivationTime: "2012-01-01T00:00:00Z",
RatingPlanId: "RETAIL1",
FallbackSubjects: "dan2",
}},
Overwrite: true,
APIOpts: map[string]any{utils.CacheOpt: utils.MetaReload},
}
// set the profile
var result string
if err := rater.Call(context.Background(), utils.APIerSv1SetRatingProfile, rpf, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
var reply bool
argsCache := utils.ArgsGetCacheItem{
CacheID: utils.CacheRatingProfiles,
ItemID: "*out:cgrates.org:call:dan",
}
if err := rater.Call(context.Background(), utils.CacheSv1HasItem, argsCache, &reply); err != nil {
t.Error(err)
} else if reply {
t.Errorf("Expected: false, received:%v", reply)
}
var rcvKeys []string
argsCache2 := utils.ArgsGetCacheItemIDs{
CacheID: utils.CacheRatingProfiles,
}
if err := rater.Call(context.Background(), utils.CacheSv1GetItemIDs, argsCache2, &rcvKeys); err == nil ||
err.Error() != utils.ErrNotFound.Error() {
t.Fatalf("Expected error: %s received error: %s and reply: %v ",
utils.ErrNotFound, err, rcvKeys)
}
//check in dataManager
expected := []string{"call:dan"}
var rcvIDs []string
if err := rater.Call(context.Background(), utils.APIerSv1GetRatingProfileIDs, utils.PaginatorWithTenant{Tenant: "cgrates.org"}, &rcvIDs); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expected, rcvIDs) {
t.Errorf("Expecting : %+v, received: %+v", expected, rcvIDs)
}
}
func testRatingProfileCachingMetaReload2(t *testing.T) {
//add cache with *load option
rpf := &utils.AttrSetRatingProfile{
Tenant: "cgrates.org",
Category: "call",
Subject: "dan",
RatingPlanActivations: []*utils.TPRatingActivation{{
ActivationTime: "2012-01-01T00:00:00Z",
RatingPlanId: "RETAIL1",
FallbackSubjects: "dan2",
}},
Overwrite: true,
APIOpts: map[string]any{utils.CacheOpt: utils.MetaLoad},
}
expected := &engine.RatingProfile{
Id: "*out:cgrates.org:call:dan",
RatingPlanActivations: engine.RatingPlanActivations{{
ActivationTime: time.Date(2012, 01, 01, 0, 0, 0, 0, time.UTC),
RatingPlanId: "RETAIL1",
FallbackKeys: []string{"*out:cgrates.org:call:dan2"},
}},
}
// set the profile
var result string
if err := rater.Call(context.Background(), utils.APIerSv1SetRatingProfile, rpf, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
var reply *engine.RatingProfile
if err := rater.Call(context.Background(), utils.APIerSv1GetRatingProfile, &utils.AttrGetRatingProfile{
Tenant: "cgrates.org", Category: "call", Subject: "dan"}, &reply); err != nil {
t.Fatal(err)
}
if !reflect.DeepEqual(expected, reply) {
t.Errorf("Expecting : %+v, received: %+v", utils.ToJSON(expected), utils.ToJSON(reply))
}
//add cache with *reload option
// should overwrite the first
rpf.RatingPlanActivations[0].FallbackSubjects = "dan3"
expected.RatingPlanActivations[0].FallbackKeys[0] = "*out:cgrates.org:call:dan3"
// set the profile
if err := rater.Call(context.Background(), utils.APIerSv1SetRatingProfile, rpf, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
if err := rater.Call(context.Background(), utils.APIerSv1GetRatingProfile, &utils.AttrGetRatingProfile{
Tenant: "cgrates.org", Category: "call", Subject: "dan"}, &reply); err != nil {
t.Fatal(err)
}
if !reflect.DeepEqual(expected, reply) {
t.Errorf("Expecting : %+v, received: %+v", utils.ToJSON(expected), utils.ToJSON(reply))
}
}
func testRatingProfileCachingMetaRemove(t *testing.T) {
//add cache with *load option
rpf := &utils.AttrSetRatingProfile{
Tenant: "cgrates.org",
Category: "call",
Subject: "dan",
RatingPlanActivations: []*utils.TPRatingActivation{{
ActivationTime: "2012-01-01T00:00:00Z",
RatingPlanId: "RETAIL1",
FallbackSubjects: "dan2",
}},
APIOpts: map[string]any{utils.CacheOpt: utils.MetaLoad},
}
// set the profile
var result string
if err := rater.Call(context.Background(), utils.APIerSv1SetRatingProfile, rpf, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
var reply bool
argsCache := utils.ArgsGetCacheItem{
CacheID: utils.CacheRatingProfiles,
ItemID: "*out:cgrates.org:call:dan",
}
if err := rater.Call(context.Background(), utils.CacheSv1HasItem, argsCache, &reply); err != nil {
t.Error(err)
} else if !reply {
t.Errorf("Expected: true, received:%v", reply)
}
var rcvKeys []string
expectedIDs := []string{"*out:cgrates.org:call:dan"}
argsCache2 := utils.ArgsGetCacheItemIDs{
CacheID: utils.CacheRatingProfiles,
}
if err := rater.Call(context.Background(), utils.CacheSv1GetItemIDs, argsCache2, &rcvKeys); err != nil {
t.Fatal(err)
} else if !reflect.DeepEqual(rcvKeys, expectedIDs) {
t.Errorf("Expecting : %+v, received: %+v", expectedIDs, rcvKeys)
}
// add with *remove cache option
// should delete it from cache
rpf = &utils.AttrSetRatingProfile{
Tenant: "cgrates.org",
Category: "call",
Subject: "dan",
RatingPlanActivations: []*utils.TPRatingActivation{{
ActivationTime: "2012-01-01T00:00:00Z",
RatingPlanId: "RETAIL1",
FallbackSubjects: "dan2",
}},
APIOpts: map[string]any{utils.CacheOpt: utils.MetaRemove},
}
// set the profile
if err := rater.Call(context.Background(), utils.APIerSv1SetRatingProfile, rpf, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
if err := rater.Call(context.Background(), utils.CacheSv1HasItem, argsCache, &reply); err != nil {
t.Error(err)
} else if reply {
t.Errorf("Expected: false, received:%v", reply)
}
if err := rater.Call(context.Background(), utils.CacheSv1GetItemIDs, argsCache2, &rcvKeys); err == nil ||
err.Error() != utils.ErrNotFound.Error() {
t.Fatalf("Expected error: %s received error: %s and reply: %v ",
utils.ErrNotFound, err.Error(), rcvKeys)
}
//check in dataManager
expected := []string{"call:dan"}
var rcvIDs []string
if err := rater.Call(context.Background(), utils.APIerSv1GetRatingProfileIDs, utils.PaginatorWithTenant{Tenant: "cgrates.org"}, &rcvIDs); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expected, rcvIDs) {
t.Errorf("Expecting : %+v, received: %+v", expected, rcvIDs)
}
}