mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-19 22:28:45 +05:00
0473e66555
A time.Sleep() of 10 milliseconds has been added after starting the services in order to wait for the goroutine that handles reloads to get instantiated. This solves the occasional deadlock that happens when a reload signal is sent to a subsystem that requires storDB before sending it to storDB itself. One consequence of waiting is that where we previously expected the DataDB service to not be running yet, now it has enough time.
94 lines
2.8 KiB
Go
94 lines
2.8 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 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 services
|
|
|
|
import (
|
|
"path"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/cgrates/rpcclient"
|
|
|
|
"github.com/cgrates/cgrates/config"
|
|
"github.com/cgrates/cgrates/engine"
|
|
"github.com/cgrates/cgrates/servmanager"
|
|
"github.com/cgrates/cgrates/utils"
|
|
)
|
|
|
|
func TestAttributeSReload(t *testing.T) {
|
|
cfg, err := config.NewDefaultCGRConfig()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
utils.Newlogger(utils.MetaSysLog, cfg.GeneralCfg().NodeID)
|
|
utils.Logger.SetLogLevel(7)
|
|
|
|
engineShutdown := make(chan bool, 1)
|
|
chS := engine.NewCacheS(cfg, nil)
|
|
filterSChan := make(chan *engine.FilterS, 1)
|
|
filterSChan <- nil
|
|
close(chS.GetPrecacheChannel(utils.CacheAttributeProfiles))
|
|
close(chS.GetPrecacheChannel(utils.CacheAttributeFilterIndexes))
|
|
server := utils.NewServer()
|
|
srvMngr := servmanager.NewServiceManager(cfg, engineShutdown)
|
|
db := NewDataDBService(cfg, nil)
|
|
attrS := NewAttributeService(cfg, db,
|
|
chS, filterSChan, server, make(chan rpcclient.ClientConnector, 1),
|
|
)
|
|
engine.NewConnManager(cfg, nil)
|
|
srvMngr.AddServices(attrS,
|
|
NewLoaderService(cfg, db, filterSChan, server, engineShutdown, make(chan rpcclient.ClientConnector, 1), nil), db)
|
|
if err = srvMngr.StartServices(); err != nil {
|
|
t.Error(err)
|
|
}
|
|
time.Sleep(10 * time.Millisecond)
|
|
if attrS.IsRunning() {
|
|
t.Errorf("Expected service to be down")
|
|
}
|
|
if db.IsRunning() {
|
|
t.Errorf("Expected service to be down")
|
|
}
|
|
|
|
var reply string
|
|
if err := cfg.V1ReloadConfigFromPath(&config.ConfigReloadWithArgDispatcher{
|
|
Path: path.Join("/usr", "share", "cgrates", "conf", "samples", "tutmongo"),
|
|
Section: config.ATTRIBUTE_JSN,
|
|
}, &reply); err != nil {
|
|
t.Error(err)
|
|
} else if reply != utils.OK {
|
|
t.Errorf("Expecting OK ,received %s", reply)
|
|
}
|
|
time.Sleep(10 * time.Millisecond) //need to switch to gorutine
|
|
if !attrS.IsRunning() {
|
|
t.Errorf("Expected service to be running")
|
|
}
|
|
if !db.IsRunning() {
|
|
t.Errorf("Expected service to be running")
|
|
}
|
|
cfg.AttributeSCfg().Enabled = false
|
|
cfg.GetReloadChan(config.ATTRIBUTE_JSN) <- struct{}{}
|
|
time.Sleep(10 * time.Millisecond)
|
|
if attrS.IsRunning() {
|
|
t.Errorf("Expected service to be down")
|
|
}
|
|
engineShutdown <- true
|
|
}
|