mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-25 00:58:45 +05:00
Renamed integration tests
This commit is contained in:
Edwardro22
170
engine/actions_it_test.go
Normal file
170
engine/actions_it_test.go
Normal file
@@ -0,0 +1,170 @@
|
||||
// +build integration
|
||||
|
||||
/*
|
||||
Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
|
||||
Copyright (C) ITsysCOM GmbH
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>
|
||||
*/
|
||||
package engine
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"net/rpc"
|
||||
"net/rpc/jsonrpc"
|
||||
"path"
|
||||
"strconv"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/cgrates/cgrates/config"
|
||||
"github.com/cgrates/cgrates/utils"
|
||||
)
|
||||
|
||||
var actsLclCfg *config.CGRConfig
|
||||
var actsLclRpc *rpc.Client
|
||||
var actsLclCfgPath = path.Join(*dataDir, "conf", "samples", "actions")
|
||||
|
||||
var waitRater = flag.Int("wait_rater", 100, "Number of miliseconds to wait for rater to start and cache")
|
||||
|
||||
func TestActionsitInitCfg(t *testing.T) {
|
||||
|
||||
// Init config first
|
||||
var err error
|
||||
actsLclCfg, err = config.NewCGRConfigFromFolder(actsLclCfgPath)
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
actsLclCfg.DataFolderPath = *dataDir // Share DataFolderPath through config towards StoreDb for Flush()
|
||||
config.SetCgrConfig(actsLclCfg)
|
||||
}
|
||||
|
||||
func TestActionsitInitCdrDb(t *testing.T) {
|
||||
|
||||
if err := InitStorDb(actsLclCfg); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
// Finds cgr-engine executable and starts it with default configuration
|
||||
func TestActionsitStartEngine(t *testing.T) {
|
||||
|
||||
if _, err := StartEngine(actsLclCfgPath, *waitRater); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
// Connect rpc client to rater
|
||||
func TestActionsitRpcConn(t *testing.T) {
|
||||
|
||||
var err error
|
||||
time.Sleep(500 * time.Millisecond)
|
||||
actsLclRpc, err = jsonrpc.Dial("tcp", actsLclCfg.RPCJSONListen) // We connect over JSON so we can also troubleshoot if needed
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestActionsitSetCdrlogDebit(t *testing.T) {
|
||||
|
||||
var reply string
|
||||
attrsSetAccount := &utils.AttrSetAccount{Tenant: "cgrates.org", Account: "dan2904"}
|
||||
if err := actsLclRpc.Call("ApierV1.SetAccount", attrsSetAccount, &reply); err != nil {
|
||||
t.Error("Got error on ApierV1.SetAccount: ", err.Error())
|
||||
} else if reply != utils.OK {
|
||||
t.Errorf("Calling ApierV1.SetAccount received: %s", reply)
|
||||
}
|
||||
attrsAA := &utils.AttrSetActions{ActionsId: "ACTS_1", Actions: []*utils.TPAction{
|
||||
&utils.TPAction{Identifier: DEBIT, BalanceType: utils.MONETARY, Units: "5", ExpiryTime: UNLIMITED, Weight: 20.0},
|
||||
&utils.TPAction{Identifier: CDRLOG},
|
||||
}}
|
||||
if err := actsLclRpc.Call("ApierV2.SetActions", attrsAA, &reply); err != nil && err.Error() != utils.ErrExists.Error() {
|
||||
t.Error("Got error on ApierV2.SetActions: ", err.Error())
|
||||
} else if reply != utils.OK {
|
||||
t.Errorf("Calling ApierV2.SetActions received: %s", reply)
|
||||
}
|
||||
attrsEA := &utils.AttrExecuteAction{Tenant: attrsSetAccount.Tenant, Account: attrsSetAccount.Account, ActionsId: attrsAA.ActionsId}
|
||||
if err := actsLclRpc.Call("ApierV1.ExecuteAction", attrsEA, &reply); err != nil {
|
||||
t.Error("Got error on ApierV1.ExecuteAction: ", err.Error())
|
||||
} else if reply != utils.OK {
|
||||
t.Errorf("Calling ApierV1.ExecuteAction received: %s", reply)
|
||||
}
|
||||
var rcvedCdrs []*ExternalCDR
|
||||
if err := actsLclRpc.Call("ApierV2.GetCdrs", utils.RPCCDRsFilter{Sources: []string{CDRLOG}, Accounts: []string{attrsSetAccount.Account}}, &rcvedCdrs); err != nil {
|
||||
t.Error("Unexpected error: ", err.Error())
|
||||
} else if len(rcvedCdrs) != 1 {
|
||||
t.Error("Unexpected number of CDRs returned: ", len(rcvedCdrs))
|
||||
} else if rcvedCdrs[0].ToR != utils.MONETARY ||
|
||||
rcvedCdrs[0].OriginHost != "127.0.0.1" ||
|
||||
rcvedCdrs[0].Source != CDRLOG ||
|
||||
rcvedCdrs[0].RequestType != utils.META_PREPAID ||
|
||||
rcvedCdrs[0].Tenant != "cgrates.org" ||
|
||||
rcvedCdrs[0].Account != "dan2904" ||
|
||||
rcvedCdrs[0].Subject != "dan2904" ||
|
||||
rcvedCdrs[0].Usage != "1" ||
|
||||
rcvedCdrs[0].RunID != DEBIT ||
|
||||
strconv.FormatFloat(rcvedCdrs[0].Cost, 'f', -1, 64) != attrsAA.Actions[0].Units {
|
||||
t.Errorf("Received: %+v", rcvedCdrs[0])
|
||||
}
|
||||
}
|
||||
|
||||
func TestActionsitSetCdrlogTopup(t *testing.T) {
|
||||
|
||||
var reply string
|
||||
attrsSetAccount := &utils.AttrSetAccount{Tenant: "cgrates.org", Account: "dan2905"}
|
||||
if err := actsLclRpc.Call("ApierV1.SetAccount", attrsSetAccount, &reply); err != nil {
|
||||
t.Error("Got error on ApierV1.SetAccount: ", err.Error())
|
||||
} else if reply != utils.OK {
|
||||
t.Errorf("Calling ApierV1.SetAccount received: %s", reply)
|
||||
}
|
||||
attrsAA := &utils.AttrSetActions{ActionsId: "ACTS_2", Actions: []*utils.TPAction{
|
||||
&utils.TPAction{Identifier: TOPUP, BalanceType: utils.MONETARY, Units: "5", ExpiryTime: UNLIMITED, Weight: 20.0},
|
||||
&utils.TPAction{Identifier: CDRLOG},
|
||||
}}
|
||||
if err := actsLclRpc.Call("ApierV2.SetActions", attrsAA, &reply); err != nil && err.Error() != utils.ErrExists.Error() {
|
||||
t.Error("Got error on ApierV2.SetActions: ", err.Error())
|
||||
} else if reply != utils.OK {
|
||||
t.Errorf("Calling ApierV2.SetActions received: %s", reply)
|
||||
}
|
||||
attrsEA := &utils.AttrExecuteAction{Tenant: attrsSetAccount.Tenant, Account: attrsSetAccount.Account, ActionsId: attrsAA.ActionsId}
|
||||
if err := actsLclRpc.Call("ApierV1.ExecuteAction", attrsEA, &reply); err != nil {
|
||||
t.Error("Got error on ApierV1.ExecuteAction: ", err.Error())
|
||||
} else if reply != utils.OK {
|
||||
t.Errorf("Calling ApierV1.ExecuteAction received: %s", reply)
|
||||
}
|
||||
var rcvedCdrs []*ExternalCDR
|
||||
if err := actsLclRpc.Call("ApierV2.GetCdrs", utils.RPCCDRsFilter{Sources: []string{CDRLOG}, Accounts: []string{attrsSetAccount.Account}}, &rcvedCdrs); err != nil {
|
||||
t.Error("Unexpected error: ", err.Error())
|
||||
} else if len(rcvedCdrs) != 1 {
|
||||
t.Error("Unexpected number of CDRs returned: ", len(rcvedCdrs))
|
||||
} else if rcvedCdrs[0].ToR != utils.MONETARY ||
|
||||
rcvedCdrs[0].OriginHost != "127.0.0.1" ||
|
||||
rcvedCdrs[0].Source != CDRLOG ||
|
||||
rcvedCdrs[0].RequestType != utils.META_PREPAID ||
|
||||
rcvedCdrs[0].Tenant != "cgrates.org" ||
|
||||
rcvedCdrs[0].Account != "dan2905" ||
|
||||
rcvedCdrs[0].Subject != "dan2905" ||
|
||||
rcvedCdrs[0].Usage != "1" ||
|
||||
rcvedCdrs[0].RunID != TOPUP ||
|
||||
strconv.FormatFloat(-rcvedCdrs[0].Cost, 'f', -1, 64) != attrsAA.Actions[0].Units {
|
||||
t.Errorf("Received: %+v", rcvedCdrs[0])
|
||||
}
|
||||
}
|
||||
|
||||
func TestActionsitStopCgrEngine(t *testing.T) {
|
||||
|
||||
if err := KillEngine(*waitRater); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
50
engine/cdr_it_test.go
Normal file
50
engine/cdr_it_test.go
Normal file
@@ -0,0 +1,50 @@
|
||||
// +build integration
|
||||
|
||||
/*
|
||||
Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
|
||||
Copyright (C) ITsysCOM GmbH
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>
|
||||
*/
|
||||
package engine
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"flag"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/cgrates/cgrates/utils"
|
||||
)
|
||||
|
||||
// Arguments received via test command
|
||||
var testIT = flag.Bool("integration", false, "Perform the tests only on local test environment, not by default.")
|
||||
var dataDir = flag.String("data_dir", "/usr/share/cgrates", "CGR data dir path here")
|
||||
|
||||
// Sample HttpJsonPost, more for usage purposes
|
||||
func TestHttpJsonPost(t *testing.T) {
|
||||
|
||||
cdrOut := &ExternalCDR{CGRID: utils.Sha1("dsafdsaf", time.Date(2013, 11, 7, 8, 42, 20, 0, time.UTC).String()), OrderID: 123, ToR: utils.VOICE, OriginID: "dsafdsaf",
|
||||
OriginHost: "192.168.1.1",
|
||||
Source: utils.UNIT_TEST, RequestType: utils.META_RATED, Direction: "*out", Tenant: "cgrates.org",
|
||||
Category: "call", Account: "account1", Subject: "tgooiscs0014", Destination: "1002",
|
||||
SetupTime: time.Date(2013, 11, 7, 8, 42, 20, 0, time.UTC).String(), AnswerTime: time.Date(2013, 11, 7, 8, 42, 26, 0, time.UTC).String(),
|
||||
RunID: utils.DEFAULT_RUNID,
|
||||
Usage: "0.00000001", ExtraFields: map[string]string{"field_extr1": "val_extr1", "fieldextr2": "valextr2"}, Cost: 1.01,
|
||||
}
|
||||
jsn, _ := json.Marshal(cdrOut)
|
||||
if _, err := utils.HttpJsonPost("http://localhost:8000", false, jsn); err == nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
424
engine/loader_it_test.go
Normal file
424
engine/loader_it_test.go
Normal file
@@ -0,0 +1,424 @@
|
||||
// +build integration
|
||||
|
||||
/*
|
||||
Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
|
||||
Copyright (C) ITsysCOM GmbH
|
||||
|
||||
This program is free software: you can redistribute it and/or modify
|
||||
it under the terms of the GNU General Public License as published by
|
||||
the Free Software Foundation, either version 3 of the License, or
|
||||
(at your option) any later version.
|
||||
|
||||
This program is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
GNU General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU General Public License
|
||||
along with this program. If not, see <http://www.gnu.org/licenses/>
|
||||
*/
|
||||
package engine
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"path"
|
||||
"testing"
|
||||
|
||||
"github.com/cgrates/cgrates/config"
|
||||
"github.com/cgrates/cgrates/utils"
|
||||
)
|
||||
|
||||
// Globals used
|
||||
var ratingDbCsv, ratingDbStor, ratingDbApier RatingStorage // Each ratingDb will have it's own sources to collect data
|
||||
var accountDbCsv, accountDbStor, accountDbApier AccountingStorage // Each ratingDb will have it's own sources to collect data
|
||||
var storDb LoadStorage
|
||||
var lCfg *config.CGRConfig
|
||||
|
||||
var tpCsvScenario = flag.String("tp_scenario", "testtp", "Use this scenario folder to import tp csv data from")
|
||||
|
||||
// Create connection to ratingDb
|
||||
// Will use 3 different datadbs in order to be able to see differences in data loaded
|
||||
func TestConnDataDbs(t *testing.T) {
|
||||
|
||||
lCfg, _ = config.NewDefaultCGRConfig()
|
||||
var err error
|
||||
if ratingDbCsv, err = ConfigureRatingStorage(lCfg.TpDbType, lCfg.TpDbHost, lCfg.TpDbPort, "4", lCfg.TpDbUser, lCfg.TpDbPass, lCfg.DBDataEncoding, nil, 1); err != nil {
|
||||
t.Fatal("Error on ratingDb connection: ", err.Error())
|
||||
}
|
||||
if ratingDbStor, err = ConfigureRatingStorage(lCfg.TpDbType, lCfg.TpDbHost, lCfg.TpDbPort, "5", lCfg.TpDbUser, lCfg.TpDbPass, lCfg.DBDataEncoding, nil, 1); err != nil {
|
||||
t.Fatal("Error on ratingDb connection: ", err.Error())
|
||||
}
|
||||
if ratingDbApier, err = ConfigureRatingStorage(lCfg.TpDbType, lCfg.TpDbHost, lCfg.TpDbPort, "6", lCfg.TpDbUser, lCfg.TpDbPass, lCfg.DBDataEncoding, nil, 1); err != nil {
|
||||
t.Fatal("Error on ratingDb connection: ", err.Error())
|
||||
}
|
||||
if accountDbCsv, err = ConfigureAccountingStorage(lCfg.DataDbType, lCfg.DataDbHost, lCfg.DataDbPort, "7",
|
||||
lCfg.DataDbUser, lCfg.DataDbPass, lCfg.DBDataEncoding, nil, 1); err != nil {
|
||||
t.Fatal("Error on ratingDb connection: ", err.Error())
|
||||
}
|
||||
if accountDbStor, err = ConfigureAccountingStorage(lCfg.DataDbType, lCfg.DataDbHost, lCfg.DataDbPort, "8",
|
||||
lCfg.DataDbUser, lCfg.DataDbPass, lCfg.DBDataEncoding, nil, 1); err != nil {
|
||||
t.Fatal("Error on ratingDb connection: ", err.Error())
|
||||
}
|
||||
if accountDbApier, err = ConfigureAccountingStorage(lCfg.DataDbType, lCfg.DataDbHost, lCfg.DataDbPort, "9",
|
||||
lCfg.DataDbUser, lCfg.DataDbPass, lCfg.DBDataEncoding, nil, 1); err != nil {
|
||||
t.Fatal("Error on ratingDb connection: ", err.Error())
|
||||
}
|
||||
for _, db := range []Storage{ratingDbCsv, ratingDbStor, ratingDbApier, accountDbCsv, accountDbStor, accountDbApier} {
|
||||
if err = db.Flush(""); err != nil {
|
||||
t.Fatal("Error when flushing datadb")
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
// Create/reset storage tariff plan tables, used as database connectin establishment also
|
||||
func TestCreateStorTpTables(t *testing.T) {
|
||||
|
||||
db, err := NewMySQLStorage(lCfg.StorDBHost, lCfg.StorDBPort, lCfg.StorDBName, lCfg.StorDBUser, lCfg.StorDBPass, lCfg.StorDBMaxOpenConns, lCfg.StorDBMaxIdleConns)
|
||||
if err != nil {
|
||||
t.Error("Error on opening database connection: ", err)
|
||||
return
|
||||
} else {
|
||||
storDb = db
|
||||
}
|
||||
// Creating the table serves also as reset since there is a drop prior to create
|
||||
if err := db.CreateTablesFromScript(path.Join(*dataDir, "storage", "mysql", utils.CREATE_TARIFFPLAN_TABLES_SQL)); err != nil {
|
||||
t.Error("Error on db creation: ", err.Error())
|
||||
return // No point in going further
|
||||
}
|
||||
}
|
||||
|
||||
// Loads data from csv files in tp scenario to ratingDbCsv
|
||||
func TestLoadFromCSV(t *testing.T) {
|
||||
|
||||
/*var err error
|
||||
for fn, v := range FileValidators {
|
||||
if err = ValidateCSVData(path.Join(*dataDir, "tariffplans", *tpCsvScenario, fn), v.Rule); err != nil {
|
||||
t.Error("Failed validating data: ", err.Error())
|
||||
}
|
||||
}*/
|
||||
loader := NewTpReader(ratingDbCsv, accountDbCsv, NewFileCSVStorage(utils.CSV_SEP,
|
||||
path.Join(*dataDir, "tariffplans", *tpCsvScenario, utils.DESTINATIONS_CSV),
|
||||
path.Join(*dataDir, "tariffplans", *tpCsvScenario, utils.TIMINGS_CSV),
|
||||
path.Join(*dataDir, "tariffplans", *tpCsvScenario, utils.RATES_CSV),
|
||||
path.Join(*dataDir, "tariffplans", *tpCsvScenario, utils.DESTINATION_RATES_CSV),
|
||||
path.Join(*dataDir, "tariffplans", *tpCsvScenario, utils.RATING_PLANS_CSV),
|
||||
path.Join(*dataDir, "tariffplans", *tpCsvScenario, utils.RATING_PROFILES_CSV),
|
||||
path.Join(*dataDir, "tariffplans", *tpCsvScenario, utils.SHARED_GROUPS_CSV),
|
||||
path.Join(*dataDir, "tariffplans", *tpCsvScenario, utils.LCRS_CSV),
|
||||
path.Join(*dataDir, "tariffplans", *tpCsvScenario, utils.ACTIONS_CSV),
|
||||
path.Join(*dataDir, "tariffplans", *tpCsvScenario, utils.ACTION_PLANS_CSV),
|
||||
path.Join(*dataDir, "tariffplans", *tpCsvScenario, utils.ACTION_TRIGGERS_CSV),
|
||||
path.Join(*dataDir, "tariffplans", *tpCsvScenario, utils.ACCOUNT_ACTIONS_CSV),
|
||||
path.Join(*dataDir, "tariffplans", *tpCsvScenario, utils.DERIVED_CHARGERS_CSV),
|
||||
path.Join(*dataDir, "tariffplans", *tpCsvScenario, utils.CDR_STATS_CSV),
|
||||
path.Join(*dataDir, "tariffplans", *tpCsvScenario, utils.USERS_CSV),
|
||||
path.Join(*dataDir, "tariffplans", *tpCsvScenario, utils.ALIASES_CSV),
|
||||
path.Join(*dataDir, "tariffplans", *tpCsvScenario, utils.ResourceLimitsCsv),
|
||||
), "", "")
|
||||
|
||||
if err = loader.LoadDestinations(); err != nil {
|
||||
t.Error("Failed loading destinations: ", err.Error())
|
||||
}
|
||||
if err = loader.LoadTimings(); err != nil {
|
||||
t.Error("Failed loading timings: ", err.Error())
|
||||
}
|
||||
if err = loader.LoadRates(); err != nil {
|
||||
t.Error("Failed loading rates: ", err.Error())
|
||||
}
|
||||
if err = loader.LoadDestinationRates(); err != nil {
|
||||
t.Error("Failed loading destination rates: ", err.Error())
|
||||
}
|
||||
if err = loader.LoadRatingPlans(); err != nil {
|
||||
t.Error("Failed loading rating plans: ", err.Error())
|
||||
}
|
||||
if err = loader.LoadRatingProfiles(); err != nil {
|
||||
t.Error("Failed loading rating profiles: ", err.Error())
|
||||
}
|
||||
if err = loader.LoadActions(); err != nil {
|
||||
t.Error("Failed loading actions: ", err.Error())
|
||||
}
|
||||
if err = loader.LoadActionPlans(); err != nil {
|
||||
t.Error("Failed loading action timings: ", err.Error())
|
||||
}
|
||||
if err = loader.LoadActionTriggers(); err != nil {
|
||||
t.Error("Failed loading action triggers: ", err.Error())
|
||||
}
|
||||
if err = loader.LoadAccountActions(); err != nil {
|
||||
t.Error("Failed loading account actions: ", err.Error())
|
||||
}
|
||||
if err = loader.LoadDerivedChargers(); err != nil {
|
||||
t.Error("Failed loading derived chargers: ", err.Error())
|
||||
}
|
||||
if err = loader.LoadLCRs(); err != nil {
|
||||
t.Error("Failed loading lcr rules: ", err.Error())
|
||||
}
|
||||
if err = loader.LoadUsers(); err != nil {
|
||||
t.Error("Failed loading users: ", err.Error())
|
||||
}
|
||||
if err = loader.LoadAliases(); err != nil {
|
||||
t.Error("Failed loading aliases: ", err.Error())
|
||||
}
|
||||
if err = loader.LoadResourceLimits(); err != nil {
|
||||
t.Error("Failed loading resource limits: ", err.Error())
|
||||
}
|
||||
if err := loader.WriteToDatabase(true, false, false); err != nil {
|
||||
t.Error("Could not write data into ratingDb: ", err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
// Imports data from csv files in tpScenario to storDb
|
||||
func TestImportToStorDb(t *testing.T) {
|
||||
|
||||
csvImporter := TPCSVImporter{
|
||||
TPid: utils.TEST_SQL,
|
||||
StorDb: storDb,
|
||||
DirPath: path.Join(*dataDir, "tariffplans", *tpCsvScenario),
|
||||
Sep: utils.CSV_SEP,
|
||||
Verbose: false,
|
||||
ImportId: utils.TEST_SQL}
|
||||
if err := csvImporter.Run(); err != nil {
|
||||
t.Error("Error when importing tpdata to storDb: ", err)
|
||||
}
|
||||
if tpids, err := storDb.GetTpIds(); err != nil {
|
||||
t.Error("Error when querying storDb for imported data: ", err)
|
||||
} else if len(tpids) != 1 || tpids[0] != utils.TEST_SQL {
|
||||
t.Errorf("Data in storDb is different than expected %v", tpids)
|
||||
}
|
||||
}
|
||||
|
||||
// Loads data from storDb into ratingDb
|
||||
func TestLoadFromStorDb(t *testing.T) {
|
||||
|
||||
loader := NewTpReader(ratingDbStor, accountDbStor, storDb, utils.TEST_SQL, "")
|
||||
if err := loader.LoadDestinations(); err != nil {
|
||||
t.Error("Failed loading destinations: ", err.Error())
|
||||
}
|
||||
if err := loader.LoadTimings(); err != nil {
|
||||
t.Error("Failed loading timings: ", err.Error())
|
||||
}
|
||||
if err := loader.LoadRates(); err != nil {
|
||||
t.Error("Failed loading rates: ", err.Error())
|
||||
}
|
||||
if err := loader.LoadDestinationRates(); err != nil {
|
||||
t.Error("Failed loading destination rates: ", err.Error())
|
||||
}
|
||||
if err := loader.LoadRatingPlans(); err != nil {
|
||||
t.Error("Failed loading rating plans: ", err.Error())
|
||||
}
|
||||
if err := loader.LoadRatingProfiles(); err != nil {
|
||||
t.Error("Failed loading rating profiles: ", err.Error())
|
||||
}
|
||||
if err := loader.LoadActions(); err != nil {
|
||||
t.Error("Failed loading actions: ", err.Error())
|
||||
}
|
||||
if err := loader.LoadActionPlans(); err != nil {
|
||||
t.Error("Failed loading action timings: ", err.Error())
|
||||
}
|
||||
if err := loader.LoadActionTriggers(); err != nil {
|
||||
t.Error("Failed loading action triggers: ", err.Error())
|
||||
}
|
||||
if err := loader.LoadAccountActions(); err != nil {
|
||||
t.Error("Failed loading account actions: ", err.Error())
|
||||
}
|
||||
if err := loader.LoadDerivedChargers(); err != nil {
|
||||
t.Error("Failed loading derived chargers: ", err.Error())
|
||||
}
|
||||
if err := loader.LoadLCRs(); err != nil {
|
||||
t.Error("Failed loading lcr rules: ", err.Error())
|
||||
}
|
||||
if err := loader.LoadUsers(); err != nil {
|
||||
t.Error("Failed loading users: ", err.Error())
|
||||
}
|
||||
if err := loader.LoadAliases(); err != nil {
|
||||
t.Error("Failed loading aliases: ", err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
func TestLoadIndividualProfiles(t *testing.T) {
|
||||
|
||||
loader := NewTpReader(ratingDbApier, accountDbApier, storDb, utils.TEST_SQL, "")
|
||||
// Load ratingPlans. This will also set destination keys
|
||||
if ratingPlans, err := storDb.GetTpRatingPlans(utils.TEST_SQL, "", nil); err != nil {
|
||||
t.Fatal("Could not retrieve rating plans")
|
||||
} else {
|
||||
rpls, err := TpRatingPlans(ratingPlans).GetRatingPlans()
|
||||
if err != nil {
|
||||
t.Fatal("Could not convert rating plans")
|
||||
}
|
||||
for tag := range rpls {
|
||||
if loaded, err := loader.LoadRatingPlansFiltered(tag); err != nil {
|
||||
t.Fatalf("Could not load ratingPlan for tag: %s, error: %s", tag, err.Error())
|
||||
} else if !loaded {
|
||||
t.Fatal("Cound not find ratingPLan with id:", tag)
|
||||
}
|
||||
}
|
||||
}
|
||||
// Load rating profiles
|
||||
loadId := utils.CSV_LOAD + "_" + utils.TEST_SQL
|
||||
if ratingProfiles, err := storDb.GetTpRatingProfiles(&TpRatingProfile{Tpid: utils.TEST_SQL, Loadid: loadId}); err != nil {
|
||||
t.Fatal("Could not retrieve rating profiles, error: ", err.Error())
|
||||
} else if len(ratingProfiles) == 0 {
|
||||
t.Fatal("Could not retrieve rating profiles")
|
||||
} else {
|
||||
rpfs, err := TpRatingProfiles(ratingProfiles).GetRatingProfiles()
|
||||
if err != nil {
|
||||
t.Fatal("Could not convert rating profiles")
|
||||
}
|
||||
for rpId := range rpfs {
|
||||
rp, _ := utils.NewTPRatingProfileFromKeyId(utils.TEST_SQL, loadId, rpId)
|
||||
mrp := APItoModelRatingProfile(rp)
|
||||
if err := loader.LoadRatingProfilesFiltered(&mrp[0]); err != nil {
|
||||
t.Fatalf("Could not load ratingProfile with id: %s, error: %s", rpId, err.Error())
|
||||
}
|
||||
}
|
||||
}
|
||||
// Load derived chargers
|
||||
loadId = utils.CSV_LOAD + "_" + utils.TEST_SQL
|
||||
if derivedChargers, err := storDb.GetTpDerivedChargers(&TpDerivedCharger{Tpid: utils.TEST_SQL, Loadid: loadId}); err != nil {
|
||||
t.Fatal("Could not retrieve derived chargers, error: ", err.Error())
|
||||
} else if len(derivedChargers) == 0 {
|
||||
t.Fatal("Could not retrieve derived chargers")
|
||||
} else {
|
||||
dcs, err := TpDerivedChargers(derivedChargers).GetDerivedChargers()
|
||||
if err != nil {
|
||||
t.Fatal("Could not convert derived chargers")
|
||||
}
|
||||
for dcId := range dcs {
|
||||
mdc := &TpDerivedCharger{Tpid: utils.TEST_SQL, Loadid: loadId}
|
||||
mdc.SetDerivedChargersId(dcId)
|
||||
if err := loader.LoadDerivedChargersFiltered(mdc, true); err != nil {
|
||||
t.Fatalf("Could not load derived charger with id: %s, error: %s", dcId, err.Error())
|
||||
}
|
||||
}
|
||||
}
|
||||
// Load cdr stats
|
||||
//loadId = utils.CSV_LOAD + "_" + utils.TEST_SQL
|
||||
if cdrStats, err := storDb.GetTpCdrStats(utils.TEST_SQL, ""); err != nil {
|
||||
t.Fatal("Could not retrieve cdr stats, error: ", err.Error())
|
||||
} else if len(cdrStats) == 0 {
|
||||
t.Fatal("Could not retrieve cdr stats")
|
||||
} else {
|
||||
cds, err := TpCdrStats(cdrStats).GetCdrStats()
|
||||
if err != nil {
|
||||
t.Fatal("Could not convert cdr stats")
|
||||
}
|
||||
for id := range cds {
|
||||
if err := loader.LoadCdrStatsFiltered(id, true); err != nil {
|
||||
t.Fatalf("Could not load cdr stats with id: %s, error: %s", id, err.Error())
|
||||
}
|
||||
}
|
||||
}
|
||||
// Load users
|
||||
if users, err := storDb.GetTpUsers(&TpUser{Tpid: utils.TEST_SQL}); err != nil {
|
||||
t.Fatal("Could not retrieve users, error: ", err.Error())
|
||||
} else if len(users) == 0 {
|
||||
t.Fatal("Could not retrieve users")
|
||||
} else {
|
||||
for _, usr := range users {
|
||||
if found, err := loader.LoadUsersFiltered(&usr); found && err != nil {
|
||||
t.Fatalf("Could not user with id: %s, error: %s", usr.GetId(), err.Error())
|
||||
}
|
||||
}
|
||||
}
|
||||
// Load aliases
|
||||
if aliases, err := storDb.GetTpAliases(&TpAlias{Tpid: utils.TEST_SQL}); err != nil {
|
||||
t.Fatal("Could not retrieve aliases, error: ", err.Error())
|
||||
} else if len(aliases) == 0 {
|
||||
t.Fatal("Could not retrieve aliases")
|
||||
} else {
|
||||
for _, al := range aliases {
|
||||
if found, err := loader.LoadAliasesFiltered(&al); found && err != nil {
|
||||
t.Fatalf("Could not load aliase with id: %s, error: %s", al.GetId(), err.Error())
|
||||
}
|
||||
}
|
||||
}
|
||||
// Load account actions
|
||||
if accountActions, err := storDb.GetTpAccountActions(&TpAccountAction{Tpid: utils.TEST_SQL, Loadid: loadId}); err != nil {
|
||||
t.Fatal("Could not retrieve account action profiles, error: ", err.Error())
|
||||
} else if len(accountActions) == 0 {
|
||||
t.Error("No account actions")
|
||||
} else {
|
||||
aas, err := TpAccountActions(accountActions).GetAccountActions()
|
||||
if err != nil {
|
||||
t.Fatal("Could not convert account actions")
|
||||
}
|
||||
for aaId := range aas {
|
||||
aa, _ := utils.NewTPAccountActionsFromKeyId(utils.TEST_SQL, loadId, aaId)
|
||||
maa := APItoModelAccountAction(aa)
|
||||
|
||||
if err := loader.LoadAccountActionsFiltered(maa); err != nil {
|
||||
t.Fatalf("Could not load account actions with id: %s, error: %s", aaId, err.Error())
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/*
|
||||
// Compares previously loaded data from csv and stor to be identical, redis specific tests
|
||||
func TestMatchLoadCsvWithStorRating(t *testing.T) {
|
||||
|
||||
rsCsv, redisDb := ratingDbCsv.(*RedisStorage)
|
||||
if !redisDb {
|
||||
return // We only support these tests for redis
|
||||
}
|
||||
rsStor := ratingDbStor.(*RedisStorage)
|
||||
rsApier := ratingDbApier.(*RedisStorage)
|
||||
keysCsv, err := rsCsv.db.Cmd("KEYS", "*").List()
|
||||
if err != nil {
|
||||
t.Fatal("Failed querying redis keys for csv data")
|
||||
}
|
||||
for _, key := range keysCsv {
|
||||
var refVal []byte
|
||||
for idx, rs := range []*RedisStorage{rsCsv, rsStor, rsApier} {
|
||||
if key == utils.TASKS_KEY || strings.HasPrefix(key, utils.ACTION_PLAN_PREFIX) { // action plans are not consistent
|
||||
continue
|
||||
}
|
||||
qVal, err := rs.db.Cmd("GET", key).Bytes()
|
||||
if err != nil {
|
||||
t.Fatalf("Run: %d, could not retrieve key %s, error: %s", idx, key, err.Error())
|
||||
}
|
||||
if idx == 0 { // Only compare at second iteration, first one is to set reference value
|
||||
refVal = qVal
|
||||
continue
|
||||
}
|
||||
if len(refVal) != len(qVal) {
|
||||
t.Errorf("Missmatched data for key: %s\n\t reference val: %s \n\t retrieved val: %s\n on iteration: %d", key, refVal, qVal, idx)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestMatchLoadCsvWithStorAccounting(t *testing.T) {
|
||||
|
||||
rsCsv, redisDb := accountDbCsv.(*RedisStorage)
|
||||
if !redisDb {
|
||||
return // We only support these tests for redis
|
||||
}
|
||||
rsStor := accountDbStor.(*RedisStorage)
|
||||
rsApier := accountDbApier.(*RedisStorage)
|
||||
keysCsv, err := rsCsv.db.Cmd("KEYS", "*").List()
|
||||
if err != nil {
|
||||
t.Fatal("Failed querying redis keys for csv data")
|
||||
}
|
||||
for _, key := range keysCsv {
|
||||
var refVal []byte
|
||||
if key == "load_history" {
|
||||
continue
|
||||
}
|
||||
for idx, rs := range []*RedisStorage{rsCsv, rsStor, rsApier} {
|
||||
qVal, err := rs.db.Cmd("GET", key).Bytes()
|
||||
if err != nil {
|
||||
t.Fatalf("Run: %d, could not retrieve key %s, error: %s", idx, key, err.Error())
|
||||
}
|
||||
if idx == 0 { // Only compare at second iteration, first one is to set reference value
|
||||
refVal = qVal
|
||||
continue
|
||||
}
|
||||
if len(refVal) != len(qVal) {
|
||||
t.Errorf("Missmatched data for key: %s\n\t, reference val: %s \n\t retrieved value: %s\n on iteration: %d", key, refVal, qVal, idx)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
*/
|
||||
Reference in New Issue
Block a user