mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 10:06:24 +05:00
100% Coverage in Registrarc
This commit is contained in:
committed by
Dan Christian Bogos
parent
f9045d1c90
commit
981fec992e
@@ -184,38 +184,3 @@ func testDsphStopDispatcher(t *testing.T) {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRegistrarcListenAndServe(t *testing.T) {
|
||||
//cover purposes only
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.RegistrarCCfg().Dispatcher.Enabled = true
|
||||
cfg.RegistrarCCfg().RPC.Enabled = true
|
||||
regStSrv := NewRegistrarCService(cfg, nil)
|
||||
stopChan := make(chan struct{}, 1)
|
||||
rldChan := make(chan struct{}, 1)
|
||||
rldChan <- struct{}{}
|
||||
go func() {
|
||||
time.Sleep(10 * time.Millisecond)
|
||||
close(stopChan)
|
||||
}()
|
||||
regStSrv.ListenAndServe(stopChan, rldChan)
|
||||
regStSrv.Shutdown()
|
||||
}
|
||||
|
||||
func TestRegistrarcregisterRPCHostsErr(t *testing.T) {
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.RegistrarCCfg().RPC.RegistrarSConns = []string{"testConnID"}
|
||||
cfg.RegistrarCCfg().RPC.Hosts = map[string][]*config.RemoteHost{
|
||||
utils.MetaDefault: {
|
||||
{
|
||||
ID: "",
|
||||
Address: "",
|
||||
Transport: "",
|
||||
Synchronous: false,
|
||||
TLS: false,
|
||||
},
|
||||
},
|
||||
}
|
||||
regStSrv := NewRegistrarCService(cfg, nil)
|
||||
regStSrv.registerRPCHosts()
|
||||
}
|
||||
|
||||
@@ -123,3 +123,104 @@ func TestDispatcherHostsService(t *testing.T) {
|
||||
close(stopChan)
|
||||
ds.ListenAndServe(stopChan, make(chan struct{}))
|
||||
}
|
||||
|
||||
func TestRegistrarcListenAndServe(t *testing.T) {
|
||||
//cover purposes only
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.RegistrarCCfg().Dispatcher.Enabled = true
|
||||
cfg.RegistrarCCfg().RPC.Enabled = true
|
||||
regStSrv := NewRegistrarCService(cfg, nil)
|
||||
stopChan := make(chan struct{}, 1)
|
||||
rldChan := make(chan struct{}, 1)
|
||||
rldChan <- struct{}{}
|
||||
go func() {
|
||||
time.Sleep(10 * time.Millisecond)
|
||||
close(stopChan)
|
||||
}()
|
||||
regStSrv.ListenAndServe(stopChan, rldChan)
|
||||
regStSrv.Shutdown()
|
||||
}
|
||||
|
||||
func TestRegistrarcregisterRPCHostsErr(t *testing.T) {
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.RegistrarCCfg().RPC.RegistrarSConns = []string{"testConnID"}
|
||||
cfg.RegistrarCCfg().RPC.Hosts = map[string][]*config.RemoteHost{
|
||||
utils.MetaDefault: {
|
||||
{
|
||||
ID: "",
|
||||
Address: "",
|
||||
Transport: "",
|
||||
Synchronous: false,
|
||||
TLS: false,
|
||||
},
|
||||
},
|
||||
}
|
||||
regStSrv := NewRegistrarCService(cfg, nil)
|
||||
regStSrv.registerRPCHosts()
|
||||
}
|
||||
|
||||
func TestRegisterRPCHosts(t *testing.T) {
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.RegistrarCCfg().RPC.RegistrarSConns = []string{"errCon1"}
|
||||
cfg.RegistrarCCfg().RPC.Hosts = map[string][]*config.RemoteHost{
|
||||
"testHostKey": {},
|
||||
}
|
||||
cfg.RPCConns()["errCon1"] = &config.RPCConn{
|
||||
Strategy: utils.MetaFirst,
|
||||
PoolSize: 1,
|
||||
Conns: []*config.RemoteHost{
|
||||
{
|
||||
ID: "errCon1",
|
||||
Address: "127.0.0.1:9999",
|
||||
Transport: "*json",
|
||||
Synchronous: true,
|
||||
},
|
||||
},
|
||||
}
|
||||
regist := &RegistrarCService{
|
||||
cfg: cfg,
|
||||
connMgr: engine.NewConnManager(cfg, map[string]chan rpcclient.ClientConnector{}),
|
||||
}
|
||||
registCmp := &RegistrarCService{
|
||||
cfg: cfg,
|
||||
connMgr: engine.NewConnManager(cfg, map[string]chan rpcclient.ClientConnector{}),
|
||||
}
|
||||
regist.registerRPCHosts()
|
||||
if !reflect.DeepEqual(regist, registCmp) {
|
||||
t.Errorf("Expected: %+v ,received: %+v", registCmp, regist)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRegistrarcListenAndServedTmCDispatcher(t *testing.T) {
|
||||
//cover purposes only
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.RegistrarCCfg().Dispatcher.Enabled = true
|
||||
cfg.RegistrarCCfg().Dispatcher.RefreshInterval = 1
|
||||
cfg.RegistrarCCfg().RPC.Enabled = true
|
||||
regStSrv := NewRegistrarCService(cfg, nil)
|
||||
stopChan := make(chan struct{}, 1)
|
||||
rldChan := make(chan struct{}, 1)
|
||||
go func() {
|
||||
time.Sleep(20 * time.Millisecond)
|
||||
close(stopChan)
|
||||
}()
|
||||
regStSrv.ListenAndServe(stopChan, rldChan)
|
||||
regStSrv.Shutdown()
|
||||
}
|
||||
|
||||
func TestRegistrarcListenAndServedTmCRPC(t *testing.T) {
|
||||
//cover purposes only
|
||||
cfg := config.NewDefaultCGRConfig()
|
||||
cfg.RegistrarCCfg().Dispatcher.Enabled = true
|
||||
cfg.RegistrarCCfg().RPC.Enabled = true
|
||||
cfg.RegistrarCCfg().RPC.RefreshInterval = 1
|
||||
regStSrv := NewRegistrarCService(cfg, nil)
|
||||
stopChan := make(chan struct{}, 1)
|
||||
rldChan := make(chan struct{}, 1)
|
||||
go func() {
|
||||
time.Sleep(20 * time.Millisecond)
|
||||
close(stopChan)
|
||||
}()
|
||||
regStSrv.ListenAndServe(stopChan, rldChan)
|
||||
regStSrv.Shutdown()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user