mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-22 15:48:44 +05:00
Implement start/stop/status methods for servmanager + tests
This commit is contained in:
ionutboangiu
committed by
Dan Christian Bogos
Dan Christian Bogos
parent
ad7bd111d1
commit
60a4da69e0
212
apis/servmanager_it_test.go
Normal file
212
apis/servmanager_it_test.go
Normal file
@@ -0,0 +1,212 @@
|
||||
//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 apis
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"path"
|
||||
"testing"
|
||||
|
||||
"github.com/cgrates/birpc"
|
||||
"github.com/cgrates/birpc/context"
|
||||
"github.com/cgrates/cgrates/config"
|
||||
"github.com/cgrates/cgrates/engine"
|
||||
"github.com/cgrates/cgrates/servmanager"
|
||||
"github.com/cgrates/cgrates/utils"
|
||||
)
|
||||
|
||||
var (
|
||||
srvMngCfgPath string
|
||||
srvMngCfg *config.CGRConfig
|
||||
srvMngRPC *birpc.Client
|
||||
srvMngConfigDIR string //run tests for specific configuration
|
||||
|
||||
sTestsServManager = []func(t *testing.T){
|
||||
testSrvMngInitCfg,
|
||||
testSrvMngInitDataDb,
|
||||
testSrvMngStartEngine,
|
||||
testSrvMngSRPCConn,
|
||||
|
||||
testSrvMngPing,
|
||||
|
||||
testSrvMngSKillEngine,
|
||||
}
|
||||
)
|
||||
|
||||
func TestServManagerIT(t *testing.T) {
|
||||
switch *dbType {
|
||||
case utils.MetaInternal:
|
||||
srvMngConfigDIR = "apis_srvmng_internal"
|
||||
case utils.MetaMongo:
|
||||
srvMngConfigDIR = "apis_srvmng_mongo"
|
||||
case utils.MetaMySQL:
|
||||
srvMngConfigDIR = "apis_srvmng_mysql"
|
||||
case utils.MetaPostgres:
|
||||
t.SkipNow()
|
||||
default:
|
||||
t.Fatal("Unknown Database type")
|
||||
}
|
||||
for _, stest := range sTestsServManager {
|
||||
t.Run(srvMngConfigDIR, stest)
|
||||
}
|
||||
}
|
||||
|
||||
func testSrvMngInitCfg(t *testing.T) {
|
||||
var err error
|
||||
srvMngCfgPath = path.Join(*dataDir, "conf", "samples", srvMngConfigDIR)
|
||||
srvMngCfg, err = config.NewCGRConfigFromPath(context.Background(), srvMngCfgPath)
|
||||
if err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
func testSrvMngInitDataDb(t *testing.T) {
|
||||
if err := engine.InitDataDB(srvMngCfg); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
// Start CGR Engine
|
||||
func testSrvMngStartEngine(t *testing.T) {
|
||||
if _, err := engine.StopStartEngine(srvMngCfgPath, *waitRater); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func testSrvMngSRPCConn(t *testing.T) {
|
||||
var err error
|
||||
srvMngRPC, err = newRPCClient(srvMngCfg.ListenCfg()) // We connect over JSON so we can also troubleshoot if needed
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
// Kill the engine when it is about to be finished
|
||||
func testSrvMngSKillEngine(t *testing.T) {
|
||||
if err := engine.KillEngine(100); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
func testSrvMngPing(t *testing.T) {
|
||||
var reply string
|
||||
if err := srvMngRPC.Call(context.Background(), utils.ServiceManagerV1Ping, nil,
|
||||
&reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.Pong {
|
||||
t.Errorf("Unexpected reply: %s", reply)
|
||||
}
|
||||
|
||||
serviceToMethod := map[string]string{
|
||||
utils.AdminS: utils.AdminSv1Ping,
|
||||
utils.AccountS: utils.AccountSv1Ping,
|
||||
utils.ActionS: utils.ActionSv1Ping,
|
||||
utils.AnalyzerS: utils.AnalyzerSv1Ping,
|
||||
utils.AttributeS: utils.AttributeSv1Ping,
|
||||
utils.CDRServer: utils.CDRsV1Ping,
|
||||
utils.ChargerS: utils.ChargerSv1Ping,
|
||||
// utils.DispatcherS: utils.DispatcherSv1Ping,
|
||||
utils.EEs: utils.EeSv1Ping,
|
||||
utils.EFs: utils.EfSv1Ping,
|
||||
utils.RateS: utils.RateSv1Ping,
|
||||
utils.ResourceS: utils.ResourceSv1Ping,
|
||||
utils.RouteS: utils.RouteSv1Ping,
|
||||
utils.SessionS: utils.SessionSv1Ping,
|
||||
utils.StatS: utils.StatSv1Ping,
|
||||
utils.ThresholdS: utils.ThresholdSv1Ping,
|
||||
utils.TPeS: utils.TPeSv1Ping,
|
||||
}
|
||||
|
||||
/*
|
||||
Run the following tests for each service:
|
||||
|
||||
- ping before enabling service (expect can't find service error)
|
||||
- query for service status (expect "STOPPED" reply)
|
||||
- start the service
|
||||
- query for service status (expect "RUNNING" reply)
|
||||
- ping (expect "Pong")
|
||||
- stop service
|
||||
- query for service status (expect "STOPPED" reply)
|
||||
- ping after stopping service (expect "can't find service" error)
|
||||
*/
|
||||
|
||||
for serviceID, serviceMethod := range serviceToMethod {
|
||||
t.Run(fmt.Sprintf("test for service %s", serviceID), func(t *testing.T) {
|
||||
rpcErr := fmt.Sprintf("rpc: can't find service %s", serviceMethod)
|
||||
if err := srvMngRPC.Call(context.Background(), serviceMethod, nil,
|
||||
&reply); err == nil || err.Error() != rpcErr {
|
||||
t.Errorf("expected: <%+v>,\nreceived: <%+v>", rpcErr, err)
|
||||
}
|
||||
|
||||
args := servmanager.ArgsServiceID{
|
||||
ServiceID: serviceID,
|
||||
}
|
||||
|
||||
if err := srvMngRPC.Call(context.Background(), utils.ServiceManagerV1ServiceStatus, args,
|
||||
&reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.StoppedCaps {
|
||||
t.Errorf("Unexpected reply: %s", reply)
|
||||
}
|
||||
|
||||
if err := srvMngRPC.Call(context.Background(), utils.ServiceManagerV1StartService, args,
|
||||
&reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.OK {
|
||||
t.Errorf("Unexpected reply: %s", reply)
|
||||
}
|
||||
|
||||
if err := srvMngRPC.Call(context.Background(), utils.ServiceManagerV1ServiceStatus, args,
|
||||
&reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.RunningCaps {
|
||||
t.Errorf("Unexpected reply: %s", reply)
|
||||
}
|
||||
|
||||
if err := srvMngRPC.Call(context.Background(), serviceMethod, nil,
|
||||
&reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.Pong {
|
||||
t.Errorf("Unexpected reply: %s", reply)
|
||||
}
|
||||
|
||||
if err := srvMngRPC.Call(context.Background(), utils.ServiceManagerV1StopService, args,
|
||||
&reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.OK {
|
||||
t.Errorf("Unexpected reply: %s", reply)
|
||||
}
|
||||
|
||||
if err := srvMngRPC.Call(context.Background(), utils.ServiceManagerV1ServiceStatus, args,
|
||||
&reply); err != nil {
|
||||
t.Error(err)
|
||||
} else if reply != utils.StoppedCaps {
|
||||
t.Errorf("Unexpected reply: %s", reply)
|
||||
}
|
||||
|
||||
if err := srvMngRPC.Call(context.Background(), serviceMethod, nil,
|
||||
&reply); err == nil || err.Error() != rpcErr {
|
||||
t.Errorf("expected: <%+v>,\nreceived: <%+v>", rpcErr, err)
|
||||
}
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user