mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 10:06:24 +05:00
Added dispatchers it tests for rankings and trends
This commit is contained in:
gezimbll
committed by
Dan Christian Bogos
Dan Christian Bogos
parent
b615f2aeba
commit
ee312b13b6
164
dispatchers/ranking_it_test.go
Normal file
164
dispatchers/ranking_it_test.go
Normal file
@@ -0,0 +1,164 @@
|
||||
//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 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 dispatchers
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/cgrates/birpc/context"
|
||||
"github.com/cgrates/cgrates/engine"
|
||||
"github.com/cgrates/cgrates/utils"
|
||||
)
|
||||
|
||||
var sTestDspRn = []func(t *testing.T){
|
||||
testDspRnPingFailover,
|
||||
testDspRnGetRankingFailover,
|
||||
testDspRnPing,
|
||||
testDspRnTestAuthKey,
|
||||
testDspRnTestAuthKey2,
|
||||
}
|
||||
|
||||
func TestDspRankingS(t *testing.T) {
|
||||
var config1, config2, config3 string
|
||||
switch *utils.DBType {
|
||||
case utils.MetaInternal:
|
||||
t.SkipNow()
|
||||
case utils.MetaMySQL:
|
||||
config1 = "trd_rnk_mysql"
|
||||
config2 = "trd_rnk2_mysql"
|
||||
config3 = "dispatchers_mysql"
|
||||
case utils.MetaMongo:
|
||||
config1 = "trd_rnk_mongo"
|
||||
config2 = "trd_rnk2_mongo"
|
||||
config3 = "dispatchers_mongo"
|
||||
case utils.MetaPostgres:
|
||||
t.SkipNow()
|
||||
default:
|
||||
t.Fatal("Unknown Database type")
|
||||
}
|
||||
testDsp(t, sTestDspRn, "TestDspRankingS", config1, config2, config3, "tutrankings", "testtp", "dispatchers")
|
||||
}
|
||||
|
||||
func testDspRnPingFailover(t *testing.T) {
|
||||
var reply string
|
||||
if err := allEngine.RPC.Call(context.Background(), utils.RankingSv1Ping, new(utils.CGREvent), &reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.Pong {
|
||||
t.Errorf("Received: %s", reply)
|
||||
}
|
||||
ev := utils.CGREvent{
|
||||
Tenant: "cgrates.org",
|
||||
|
||||
APIOpts: map[string]any{
|
||||
utils.OptsAPIKey: "rn12345",
|
||||
},
|
||||
}
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.RankingSv1Ping, &ev, &reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.Pong {
|
||||
t.Errorf("Received: %s", reply)
|
||||
}
|
||||
allEngine.stopEngine(t)
|
||||
time.Sleep(200 * time.Millisecond)
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.RankingSv1Ping, &ev, &reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.Pong {
|
||||
t.Errorf("Received: %s", reply)
|
||||
}
|
||||
allEngine2.stopEngine(t)
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.RankingSv1Ping, &ev, &reply); err == nil {
|
||||
t.Errorf("Expected error but received %v and reply %v\n", err, reply)
|
||||
}
|
||||
allEngine.startEngine(t)
|
||||
allEngine2.startEngine(t)
|
||||
}
|
||||
|
||||
func testDspRnGetRankingFailover(t *testing.T) {
|
||||
trSched := utils.ArgScheduleRankingQueries{RankingIDs: []string{"RANK2"}, TenantIDWithAPIOpts: utils.TenantIDWithAPIOpts{TenantID: &utils.TenantID{Tenant: "cgrates.org"}, APIOpts: map[string]any{utils.OptsAPIKey: "rn12345"}}}
|
||||
var scheduled int
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.RankingSv1ScheduleQueries, &trSched,
|
||||
&scheduled); err != nil {
|
||||
t.Error(err)
|
||||
} else if scheduled != 1 {
|
||||
t.Errorf("expected %d,received %d", 1, scheduled)
|
||||
}
|
||||
allEngine.stopEngine(t)
|
||||
time.Sleep(200 * time.Millisecond)
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.RankingSv1ScheduleQueries, &trSched,
|
||||
&scheduled); err == nil || err.Error() != utils.ErrPartiallyExecuted.Error() {
|
||||
t.Errorf("expected err %v,received %v", utils.ErrPartiallyExecuted, err)
|
||||
}
|
||||
allEngine.startEngine(t)
|
||||
}
|
||||
|
||||
func testDspRnPing(t *testing.T) {
|
||||
var reply string
|
||||
if err := allEngine.RPC.Call(context.Background(), utils.RankingSv1Ping, new(utils.CGREvent), &reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.Pong {
|
||||
t.Errorf("Received: %s", reply)
|
||||
}
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.RankingSv1Ping, &utils.CGREvent{
|
||||
Tenant: "cgrates.org",
|
||||
|
||||
APIOpts: map[string]any{
|
||||
utils.OptsAPIKey: "rn12345",
|
||||
},
|
||||
}, &reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.Pong {
|
||||
t.Errorf("Received: %s", reply)
|
||||
}
|
||||
}
|
||||
|
||||
func testDspRnTestAuthKey(t *testing.T) {
|
||||
var rn *engine.Ranking
|
||||
args := &utils.TenantIDWithAPIOpts{
|
||||
TenantID: &utils.TenantID{
|
||||
Tenant: "cgrates.org",
|
||||
ID: "RANK1",
|
||||
},
|
||||
APIOpts: map[string]any{
|
||||
utils.OptsAPIKey: "12345",
|
||||
},
|
||||
}
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.RankingSv1GetRanking,
|
||||
args, &rn); err == nil || err.Error() != utils.ErrUnauthorizedApi.Error() {
|
||||
t.Error(err)
|
||||
}
|
||||
rnSched := utils.ArgScheduleRankingQueries{RankingIDs: []string{"RANK1"}, TenantIDWithAPIOpts: utils.TenantIDWithAPIOpts{TenantID: &utils.TenantID{Tenant: "cgrates.org"}, APIOpts: map[string]any{utils.OptsAPIKey: "12345"}}}
|
||||
var scheduled int
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.RankingSv1ScheduleQueries,
|
||||
&rnSched, &scheduled); err == nil || err.Error() != utils.ErrUnauthorizedApi.Error() {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
func testDspRnTestAuthKey2(t *testing.T) {
|
||||
var schedRankings []utils.ScheduledRanking
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.RankingSv1GetSchedule, &utils.ArgScheduledRankings{TenantIDWithAPIOpts: utils.TenantIDWithAPIOpts{TenantID: &utils.TenantID{Tenant: "cgrates.org"}, APIOpts: map[string]any{utils.OptsAPIKey: "rn12345"}}, RankingIDPrefixes: []string{"RANK2"}}, &schedRankings); err != nil {
|
||||
t.Error(err)
|
||||
} else if len(schedRankings) != 1 {
|
||||
t.Errorf("expected 1 schedTrends, got %d", len(schedRankings))
|
||||
}
|
||||
}
|
||||
@@ -32,7 +32,7 @@ func (dS *DispatcherService) TrendSv1Ping(ctx *context.Context, args *utils.CGRE
|
||||
}
|
||||
args.Tenant = utils.FirstNonEmpty(args.Tenant, dS.cfg.GeneralCfg().DefaultTenant)
|
||||
if len(dS.cfg.DispatcherSCfg().AttributeSConns) != 0 {
|
||||
if err = dS.authorize(utils.ThresholdSv1Ping, args.Tenant,
|
||||
if err = dS.authorize(utils.TrendSv1Ping, args.Tenant,
|
||||
utils.IfaceAsString(args.APIOpts[utils.OptsAPIKey]), args.Time); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
164
dispatchers/trends_it_test.go
Normal file
164
dispatchers/trends_it_test.go
Normal file
@@ -0,0 +1,164 @@
|
||||
//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 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 dispatchers
|
||||
|
||||
import (
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/cgrates/birpc/context"
|
||||
"github.com/cgrates/cgrates/engine"
|
||||
"github.com/cgrates/cgrates/utils"
|
||||
)
|
||||
|
||||
var sTestDspTr = []func(t *testing.T){
|
||||
testDspTrPingFailover,
|
||||
testDspTrGetTrendFailover,
|
||||
testDspTrPing,
|
||||
testDspTrTestAuthKey,
|
||||
testDspTrTestAuthKey2,
|
||||
}
|
||||
|
||||
func TestDspTrendS(t *testing.T) {
|
||||
var config1, config2, config3 string
|
||||
switch *utils.DBType {
|
||||
case utils.MetaInternal:
|
||||
t.SkipNow()
|
||||
case utils.MetaMySQL:
|
||||
config1 = "trd_rnk_mysql"
|
||||
config2 = "trd_rnk2_mysql"
|
||||
config3 = "dispatchers_mysql"
|
||||
case utils.MetaMongo:
|
||||
config1 = "trd_rnk_mongo"
|
||||
config2 = "trd_rnk2_mongo"
|
||||
config3 = "dispatchers_mongo"
|
||||
case utils.MetaPostgres:
|
||||
t.SkipNow()
|
||||
default:
|
||||
t.Fatal("Unknown Database type")
|
||||
}
|
||||
testDsp(t, sTestDspTr, "TestDspTrendS", config1, config2, config3, "tuttrends", "testtp", "dispatchers")
|
||||
}
|
||||
|
||||
func testDspTrPingFailover(t *testing.T) {
|
||||
var reply string
|
||||
if err := allEngine.RPC.Call(context.Background(), utils.TrendSv1Ping, new(utils.CGREvent), &reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.Pong {
|
||||
t.Errorf("Received: %s", reply)
|
||||
}
|
||||
ev := utils.CGREvent{
|
||||
Tenant: "cgrates.org",
|
||||
|
||||
APIOpts: map[string]any{
|
||||
utils.OptsAPIKey: "tr12345",
|
||||
},
|
||||
}
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.TrendSv1Ping, &ev, &reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.Pong {
|
||||
t.Errorf("Received: %s", reply)
|
||||
}
|
||||
allEngine.stopEngine(t)
|
||||
time.Sleep(200 * time.Millisecond)
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.TrendSv1Ping, &ev, &reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.Pong {
|
||||
t.Errorf("Received: %s", reply)
|
||||
}
|
||||
allEngine2.stopEngine(t)
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.TrendSv1Ping, &ev, &reply); err == nil {
|
||||
t.Errorf("Expected error but received %v and reply %v\n", err, reply)
|
||||
}
|
||||
allEngine.startEngine(t)
|
||||
allEngine2.startEngine(t)
|
||||
}
|
||||
|
||||
func testDspTrGetTrendFailover(t *testing.T) {
|
||||
trSched := utils.ArgScheduleTrendQueries{TrendIDs: []string{"TREND_1"}, TenantIDWithAPIOpts: utils.TenantIDWithAPIOpts{TenantID: &utils.TenantID{Tenant: "cgrates.org"}, APIOpts: map[string]any{utils.OptsAPIKey: "tr12345"}}}
|
||||
var scheduled int
|
||||
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.TrendSv1ScheduleQueries, &trSched,
|
||||
&scheduled); err != nil {
|
||||
t.Error(err)
|
||||
} else if scheduled != 1 {
|
||||
t.Errorf("expected %d,received %d", 1, scheduled)
|
||||
}
|
||||
allEngine.stopEngine(t)
|
||||
time.Sleep(200 * time.Millisecond)
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.TrendSv1ScheduleQueries, &trSched,
|
||||
&scheduled); err == nil || err.Error() != utils.ErrPartiallyExecuted.Error() {
|
||||
t.Errorf("expected err %v,received %v", utils.ErrPartiallyExecuted, err)
|
||||
}
|
||||
allEngine.startEngine(t)
|
||||
}
|
||||
|
||||
func testDspTrPing(t *testing.T) {
|
||||
var reply string
|
||||
if err := allEngine.RPC.Call(context.Background(), utils.TrendSv1Ping, new(utils.CGREvent), &reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.Pong {
|
||||
t.Errorf("Received: %s", reply)
|
||||
}
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.TrendSv1Ping, &utils.CGREvent{
|
||||
Tenant: "cgrates.org",
|
||||
|
||||
APIOpts: map[string]any{
|
||||
utils.OptsAPIKey: "tr12345",
|
||||
},
|
||||
}, &reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.Pong {
|
||||
t.Errorf("Received: %s", reply)
|
||||
}
|
||||
}
|
||||
|
||||
func testDspTrTestAuthKey(t *testing.T) {
|
||||
var tr *engine.Trend
|
||||
args := &utils.ArgGetTrend{
|
||||
TenantWithAPIOpts: utils.TenantWithAPIOpts{
|
||||
Tenant: "cgrates.org",
|
||||
APIOpts: map[string]any{
|
||||
utils.OptsAPIKey: "12345",
|
||||
},
|
||||
},
|
||||
ID: "TREND_1",
|
||||
}
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.TrendSv1GetTrend,
|
||||
args, &tr); err == nil || err.Error() != utils.ErrUnauthorizedApi.Error() {
|
||||
t.Error(err)
|
||||
}
|
||||
trSched := utils.ArgScheduleTrendQueries{TrendIDs: []string{"TREND_1", "TREND_2"}, TenantIDWithAPIOpts: utils.TenantIDWithAPIOpts{TenantID: &utils.TenantID{Tenant: "cgrates.org"}, APIOpts: map[string]any{utils.OptsAPIKey: "12345"}}}
|
||||
var scheduled int
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.TrendSv1ScheduleQueries,
|
||||
&trSched, &scheduled); err == nil || err.Error() != utils.ErrUnauthorizedApi.Error() {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
func testDspTrTestAuthKey2(t *testing.T) {
|
||||
var schedTrends []utils.ScheduledTrend
|
||||
if err := dispEngine.RPC.Call(context.Background(), utils.TrendSv1GetScheduledTrends, &utils.ArgScheduledTrends{TenantIDWithAPIOpts: utils.TenantIDWithAPIOpts{TenantID: &utils.TenantID{Tenant: "cgrates.org"}, APIOpts: map[string]any{utils.OptsAPIKey: "tr12345"}}, TrendIDPrefixes: []string{"TREND_1"}}, &schedTrends); err != nil {
|
||||
t.Error(err)
|
||||
} else if len(schedTrends) != 1 {
|
||||
t.Errorf("expected 1 schedTrends, got %d", len(schedTrends))
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user