Updated services tests

This commit is contained in:
Trial97
2019-09-19 17:05:57 +03:00
committed by Dan Christian Bogos
parent d93a0917a8
commit 1cffc4ca90
8 changed files with 36 additions and 8 deletions

View File

@@ -35,6 +35,8 @@ func TestAttributeSReload(t *testing.T) {
if err != nil {
t.Fatal(err)
}
utils.Newlogger(utils.MetaSysLog, cfg.GeneralCfg().NodeID)
utils.Logger.SetLogLevel(7)
filterSChan := make(chan *engine.FilterS, 1)
filterSChan <- nil
engineShutdown := make(chan bool, 1)

View File

@@ -36,6 +36,8 @@ func TestChargerSReload(t *testing.T) {
if err != nil {
t.Fatal(err)
}
utils.Newlogger(utils.MetaSysLog, cfg.GeneralCfg().NodeID)
utils.Logger.SetLogLevel(7)
cfg.AttributeSCfg().Enabled = true
filterSChan := make(chan *engine.FilterS, 1)
filterSChan <- nil

View File

@@ -36,6 +36,8 @@ func TestResourceSReload(t *testing.T) {
if err != nil {
t.Fatal(err)
}
utils.Newlogger(utils.MetaSysLog, cfg.GeneralCfg().NodeID)
utils.Logger.SetLogLevel(7)
cfg.ThresholdSCfg().Enabled = true
filterSChan := make(chan *engine.FilterS, 1)
filterSChan <- nil

View File

@@ -28,6 +28,7 @@ import (
"github.com/cgrates/cgrates/engine"
"github.com/cgrates/cgrates/servmanager"
"github.com/cgrates/cgrates/utils"
"github.com/cgrates/rpcclient"
)
func TestSchedulerSReload(t *testing.T) {
@@ -56,7 +57,9 @@ func TestSchedulerSReload(t *testing.T) {
/*loadStorage*/ nil, filterSChan,
server, nil, engineShutdown)
schS := NewSchedulerService()
srvMngr.AddService(schS)
internalCdrSChan := make(chan rpcclient.RpcClientConnection, 1)
internalCdrSChan <- nil
srvMngr.AddService(schS, NewCDRServer(internalCdrSChan))
if err = srvMngr.StartServices(); err != nil {
t.Error(err)
}
@@ -65,7 +68,7 @@ func TestSchedulerSReload(t *testing.T) {
}
var reply string
if err := cfg.V1ReloadConfig(&config.ConfigReloadWithArgDispatcher{
Path: path.Join("/usr", "share", "cgrates", "conf", "samples", "tutmongo"),
Path: path.Join("/usr", "share", "cgrates", "conf", "samples", "tutmongonew"),
Section: config.SCHEDULER_JSN,
}, &reply); err != nil {
t.Error(err)

View File

@@ -36,6 +36,8 @@ func TestStatSReload(t *testing.T) {
if err != nil {
t.Fatal(err)
}
utils.Newlogger(utils.MetaSysLog, cfg.GeneralCfg().NodeID)
utils.Logger.SetLogLevel(7)
cfg.ThresholdSCfg().Enabled = true
filterSChan := make(chan *engine.FilterS, 1)
filterSChan <- nil

View File

@@ -35,6 +35,8 @@ func TestSupplierSReload(t *testing.T) {
if err != nil {
t.Fatal(err)
}
utils.Newlogger(utils.MetaSysLog, cfg.GeneralCfg().NodeID)
utils.Logger.SetLogLevel(7)
cfg.StatSCfg().Enabled = true
filterSChan := make(chan *engine.FilterS, 1)
filterSChan <- nil

View File

@@ -36,6 +36,8 @@ func TestThresholdSReload(t *testing.T) {
if err != nil {
t.Fatal(err)
}
utils.Newlogger(utils.MetaSysLog, cfg.GeneralCfg().NodeID)
utils.Logger.SetLogLevel(7)
filterSChan := make(chan *engine.FilterS, 1)
filterSChan <- nil
engineShutdown := make(chan bool, 1)

View File

@@ -60,12 +60,11 @@ type ServiceManager struct {
dm *engine.DataManager
engineShutdown chan bool
cacheS *engine.CacheS
cdrStorage engine.CdrStorage
loadStorage engine.LoadStorage
filterS chan *engine.FilterS
server *utils.Server
subsystems map[string]Service
cdrStorage engine.CdrStorage
loadStorage engine.LoadStorage
filterS chan *engine.FilterS
server *utils.Server
subsystems map[string]Service
dispatcherSChan chan rpcclient.RpcClientConnection
}
@@ -104,6 +103,12 @@ type ArgStartService struct {
// ShutdownService shuts-down a service with ID
func (srvMngr *ServiceManager) V1StartService(args ArgStartService, reply *string) (err error) {
switch args.ServiceID {
case utils.MetaScheduler:
// stop the service using the config
srvMngr.Lock()
srvMngr.cfg.SchedulerCfg().Enabled = true
srvMngr.Unlock()
srvMngr.cfg.GetReloadChan(config.SCHEDULER_JSN) <- struct{}{}
default:
err = errors.New(utils.UnsupportedServiceIDCaps)
}
@@ -117,6 +122,12 @@ func (srvMngr *ServiceManager) V1StartService(args ArgStartService, reply *strin
// ShutdownService shuts-down a service with ID
func (srvMngr *ServiceManager) V1StopService(args ArgStartService, reply *string) (err error) {
switch args.ServiceID {
case utils.MetaScheduler:
// stop the service using the config
srvMngr.Lock()
srvMngr.cfg.SchedulerCfg().Enabled = false
srvMngr.Unlock()
srvMngr.cfg.GetReloadChan(config.SCHEDULER_JSN) <- struct{}{}
default:
err = errors.New(utils.UnsupportedServiceIDCaps)
}
@@ -133,6 +144,8 @@ func (srvMngr *ServiceManager) V1ServiceStatus(args ArgStartService, reply *stri
defer srvMngr.RUnlock()
var running bool
switch args.ServiceID {
case utils.MetaScheduler:
running = srvMngr.subsystems[utils.SchedulerS].IsRunning()
default:
return errors.New(utils.UnsupportedServiceIDCaps)
}