diff --git a/services/cores_it_test.go b/services/cores_it_test.go new file mode 100644 index 000000000..cae932128 --- /dev/null +++ b/services/cores_it_test.go @@ -0,0 +1,105 @@ +// +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 +*/ +package services + +import ( + "path" + "sync" + "testing" + "time" + + "github.com/cgrates/rpcclient" + + "github.com/cgrates/cgrates/config" + "github.com/cgrates/cgrates/cores" + "github.com/cgrates/cgrates/engine" + "github.com/cgrates/cgrates/servmanager" + "github.com/cgrates/cgrates/utils" +) + +func TestCoreSReload(t *testing.T) { + cfg := config.NewDefaultCGRConfig() + + utils.Logger, _ = utils.Newlogger(utils.MetaSysLog, cfg.GeneralCfg().NodeID) + utils.Logger.SetLogLevel(7) + + shdChan := utils.NewSyncedChan() + shdWg := new(sync.WaitGroup) + filterSChan := make(chan *engine.FilterS, 1) + filterSChan <- nil + server := cores.NewServer(nil) + srvMngr := servmanager.NewServiceManager(cfg, shdChan, shdWg) + srvDep := map[string]*sync.WaitGroup{utils.DataDB: new(sync.WaitGroup)} + db := NewDataDBService(cfg, nil, srvDep) + coreRPC := make(chan rpcclient.ClientConnector, 1) + anz := NewAnalyzerService(cfg, server, filterSChan, shdChan, make(chan rpcclient.ClientConnector, 1), srvDep) + caps := engine.NewCaps(1, "test_caps") + coreS := NewCoreService(cfg, caps, server, coreRPC, anz, srvDep) + engine.NewConnManager(cfg, nil) + srvMngr.AddServices(coreS, + NewLoaderService(cfg, db, filterSChan, server, make(chan rpcclient.ClientConnector, 1), nil, anz, srvDep), db) + if err := srvMngr.StartServices(); err != nil { + t.Error(err) + } + if coreS.IsRunning() { + t.Errorf("Expected service to be down") + } + + var reply string + if err := cfg.V1ReloadConfig(&config.ReloadArgs{ + Path: path.Join("/usr", "share", "cgrates", "conf", "samples", "caps_queue"), + Section: config.CoreSCfgJson, + }, &reply); err != nil { + t.Error(err) + } else if reply != utils.OK { + t.Errorf("Expecting OK ,received %s", reply) + } + select { + case d := <-coreRPC: + coreRPC <- d + case <-time.After(time.Second): + t.Fatal("It took to long to reload the cache") + } + if !coreS.IsRunning() { + t.Errorf("Expected service to be running") + } + err := coreS.Start() + if err == nil || err != utils.ErrServiceAlreadyRunning { + t.Errorf("\nExpecting <%+v>,\n Received <%+v>", utils.ErrServiceAlreadyRunning, err) + } + err = coreS.Reload() + if err != nil { + t.Errorf("\nExpecting ,\n Received <%+v>", err) + } + err = coreS.Shutdown() + if err != nil { + t.Errorf("\nExpecting ,\n Received <%+v>", err) + } + cfg.GetReloadChan(config.CoreSCfgJson) <- struct{}{} + time.Sleep(10 * time.Millisecond) + + if coreS.IsRunning() { + t.Errorf("Expected service to be down") + } + + shdChan.CloseOnce() + time.Sleep(10 * time.Millisecond) + +}