Add tests for internal DB dump and restore

This commit is contained in:
arberkatellari
2025-03-28 19:37:27 +02:00
committed by Dan Christian Bogos
parent a168f262e2
commit ed584364d1
112 changed files with 2384 additions and 1021 deletions

View File

@@ -101,7 +101,7 @@ func TestDispatcherServiceDispatcherProfileForEventMIIDENotFound(t *testing.T) {
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, nil, connMng)
dss := NewDispatcherService(dm, cfg, nil, connMng)
ev := &utils.CGREvent{}
@@ -547,7 +547,7 @@ func TestDispatcherServiceDispatcherProfileForEventErrNil(t *testing.T) {
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, nil, connMng)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
@@ -594,7 +594,7 @@ func TestDispatcherV1GetProfileForEventReturn(t *testing.T) {
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, nil, connMng)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
@@ -647,7 +647,7 @@ func TestDispatcherServiceDispatcherProfileForEventErrNotFound(t *testing.T) {
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, nil, connMng)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
@@ -694,7 +694,7 @@ func TestDispatcherServiceDispatcherProfileForEventErrNotFound2(t *testing.T) {
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, nil, connMng)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
@@ -741,7 +741,7 @@ func TestDispatcherServiceDispatcherProfileForEventErrNotFoundTime(t *testing.T)
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, nil, connMng)
engine.Cache.Clear(nil)
dsp := &engine.DispatcherProfile{
@@ -793,7 +793,7 @@ func TestDispatcherServiceDispatcherProfileForEventErrNotFoundFilter(t *testing.
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, nil, connMng)
engine.Cache.Clear(nil)
dsp := &engine.DispatcherProfile{
@@ -841,7 +841,7 @@ func TestDispatcherServiceDispatchDspErr(t *testing.T) {
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
engine.Cache.Clear(nil)
dm := engine.NewDataManager(dataDB, nil, connMng)
dsp := &engine.DispatcherProfile{
@@ -887,7 +887,7 @@ func TestDispatcherServiceDispatchDspErrHostNotFound(t *testing.T) {
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, nil, connMng)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
@@ -940,7 +940,7 @@ func TestDispatcherServiceDispatcherProfileForEventFoundFilter(t *testing.T) {
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, nil, connMng)
engine.Cache.Clear(nil)
dsp := &engine.DispatcherProfile{
@@ -1045,7 +1045,7 @@ func TestDispatcherServiceDispatcherProfileForEventGetDispatcherError(t *testing
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, nil, connMng)
engine.Cache.Clear(nil)
dsp := &engine.DispatcherProfile{
@@ -1107,7 +1107,7 @@ func TestDispatcherServiceDispatchDspErrHostNotFound2(t *testing.T) {
cfg.DispatcherSCfg().IndexedSelects = false
rpcCl := map[string]chan birpc.ClientConnector{}
connMng := engine.NewConnManager(cfg, rpcCl)
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, nil, connMng)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
@@ -1175,7 +1175,7 @@ func TestDispatcherServiceDispatchDspErrHostNotFound3(t *testing.T) {
utils.ConcatenatedKey(utils.MetaInternal, utils.MetaReplicator): chanRPC,
}
connMgr := engine.NewConnManager(cfg, rpcInt)
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, nil, connMgr)
dsp := &engine.DispatcherProfile{
Tenant: "cgrates.org",
@@ -1321,7 +1321,7 @@ func TestDispatchersdispatcherProfileForEventAnySSfalses(t *testing.T) {
}()
cfg := config.NewDefaultCGRConfig()
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, cfg.CacheCfg(), nil)
engine.Cache = engine.NewCacheS(cfg, dm, nil)
@@ -1378,7 +1378,7 @@ func TestDispatchersdispatcherProfileForEventAnySSfalses(t *testing.T) {
} else if len(rcv) != 1 {
t.Errorf("Unexpected number of profiles:%v", len(rcv))
} else if !reflect.DeepEqual(rcv[0], dsp2) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", dsp2, rcv)
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", dsp2, rcv[0])
}
dataDB.Flush("")
@@ -1412,7 +1412,7 @@ func TestDispatchersdispatcherProfileForEventAnySSfalseFirstNotFound(t *testing.
}()
cfg := config.NewDefaultCGRConfig()
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, cfg.CacheCfg(), nil)
engine.Cache = engine.NewCacheS(cfg, dm, nil)
@@ -1469,7 +1469,7 @@ func TestDispatchersdispatcherProfileForEventAnySSfalseFirstNotFound(t *testing.
} else if len(rcv) != 1 {
t.Errorf("Unexpected number of profiles:%v", len(rcv))
} else if !reflect.DeepEqual(rcv[0], dsp2) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", dsp2, rcv)
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", dsp2, rcv[0])
}
}
@@ -1480,7 +1480,7 @@ func TestDispatchersdispatcherProfileForEventAnySSfalseFound(t *testing.T) {
}()
cfg := config.NewDefaultCGRConfig()
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, cfg.CacheCfg(), nil)
engine.Cache = engine.NewCacheS(cfg, dm, nil)
@@ -1548,7 +1548,7 @@ func TestDispatchersdispatcherProfileForEventAnySSfalseNotFound(t *testing.T) {
}()
cfg := config.NewDefaultCGRConfig()
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, cfg.CacheCfg(), nil)
engine.Cache = engine.NewCacheS(cfg, dm, nil)
@@ -1614,7 +1614,7 @@ func TestDispatchersdispatcherProfileForEventAnySStrueNotFound(t *testing.T) {
}()
cfg := config.NewDefaultCGRConfig()
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, cfg.CacheCfg(), nil)
engine.Cache = engine.NewCacheS(cfg, dm, nil)
@@ -1679,7 +1679,7 @@ func TestDispatchersdispatcherProfileForEventAnySStrueBothFound(t *testing.T) {
}()
cfg := config.NewDefaultCGRConfig()
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(dataDB, cfg.CacheCfg(), nil)
engine.Cache = engine.NewCacheS(cfg, dm, nil)
@@ -1735,7 +1735,7 @@ func TestDispatchersdispatcherProfileForEventAnySStrueBothFound(t *testing.T) {
} else if len(rcv) != 1 {
t.Errorf("Unexpected number of profiles:%v", len(rcv))
} else if !reflect.DeepEqual(rcv[0], dsp2) {
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", dsp2, rcv)
t.Errorf("\nexpected: <%+v>, \nreceived: <%+v>", dsp2, rcv[0])
}
dsp1.Weight = 30
@@ -1759,7 +1759,7 @@ func TestDispatchersdispatcherProfileForEventAnySStrueBothFound(t *testing.T) {
func TestDispatcherServiceDispatcherProfilesForEventBoolOptsErr(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
rpcCl := map[string]chan birpc.ClientConnector{}
dataDB := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
connMng := engine.NewConnManager(cfg, rpcCl)
dm := engine.NewDataManager(dataDB, nil, connMng)
fltrs := engine.NewFilterS(cfg, connMng, dm)

View File

@@ -390,7 +390,7 @@ func TestLibDispatcherNewDispatcherError(t *testing.T) {
func TestLibDispatcherSingleResultDispatcherDispatch(t *testing.T) {
wgDsp := &singleResultDispatcher{sorter: new(noSort)}
dataDB := engine.NewInternalDB(nil, nil, true, config.CgrConfig().DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, config.CgrConfig().DataDbCfg().Items)
dM := engine.NewDataManager(dataDB, config.CgrConfig().CacheCfg(), nil)
err := wgDsp.Dispatch(dM, nil, nil, "", "", &DispatcherRoute{}, "", "", "")
expected := "DSP_HOST_NOT_FOUND"
@@ -401,7 +401,7 @@ func TestLibDispatcherSingleResultDispatcherDispatch(t *testing.T) {
func TestLibDispatcherSingleResultDispatcherDispatchRouteID(t *testing.T) {
wgDsp := &singleResultDispatcher{sorter: new(roundRobinSort)}
dataDB := engine.NewInternalDB(nil, nil, true, config.CgrConfig().DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, config.CgrConfig().DataDbCfg().Items)
dM := engine.NewDataManager(dataDB, config.CgrConfig().CacheCfg(), nil)
err := wgDsp.Dispatch(dM, nil, nil, "", "routeID", &DispatcherRoute{}, "", "", "")
expected := "DSP_HOST_NOT_FOUND"
@@ -412,7 +412,7 @@ func TestLibDispatcherSingleResultDispatcherDispatchRouteID(t *testing.T) {
func TestLibDispatcherBroadcastDispatcherDispatch(t *testing.T) {
wgDsp := &broadcastDispatcher{hosts: engine.DispatcherHostProfiles{{ID: "testID"}}}
dataDB := engine.NewInternalDB(nil, nil, true, config.CgrConfig().DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, config.CgrConfig().DataDbCfg().Items)
dM := engine.NewDataManager(dataDB, config.CgrConfig().CacheCfg(), nil)
err := wgDsp.Dispatch(dM, nil, nil, "", "", &DispatcherRoute{}, "", "", "")
expected := "DSP_HOST_NOT_FOUND"
@@ -423,7 +423,7 @@ func TestLibDispatcherBroadcastDispatcherDispatch(t *testing.T) {
func TestLibDispatcherBroadcastDispatcherDispatchRouteID(t *testing.T) {
wgDsp := &broadcastDispatcher{hosts: engine.DispatcherHostProfiles{{ID: "testID"}}}
dataDB := engine.NewInternalDB(nil, nil, true, config.CgrConfig().DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, config.CgrConfig().DataDbCfg().Items)
dM := engine.NewDataManager(dataDB, config.CgrConfig().CacheCfg(), nil)
err := wgDsp.Dispatch(dM, nil, nil, "", "routeID", &DispatcherRoute{}, "", "", "")
expected := "DSP_HOST_NOT_FOUND"
@@ -434,7 +434,7 @@ func TestLibDispatcherBroadcastDispatcherDispatchRouteID(t *testing.T) {
func TestLibDispatcherLoadDispatcherDispatch(t *testing.T) {
wgDsp := &loadDispatcher{sorter: new(randomSort)}
dataDB := engine.NewInternalDB(nil, nil, true, config.CgrConfig().DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, config.CgrConfig().DataDbCfg().Items)
dM := engine.NewDataManager(dataDB, config.CgrConfig().CacheCfg(), nil)
err := wgDsp.Dispatch(dM, nil, nil, "", "", &DispatcherRoute{}, "", "", "")
expected := "DSP_HOST_NOT_FOUND"
@@ -451,7 +451,7 @@ func TestLibDispatcherLoadDispatcherDispatchHostsID(t *testing.T) {
},
sorter: new(noSort),
}
dataDB := engine.NewInternalDB(nil, nil, true, config.CgrConfig().DataDbCfg().Items)
dataDB := engine.NewInternalDB(nil, nil, true, false, config.CgrConfig().DataDbCfg().Items)
dM := engine.NewDataManager(dataDB, config.CgrConfig().CacheCfg(), nil)
err := wgDsp.Dispatch(dM, nil, nil, "", "routeID", &DispatcherRoute{}, "", "", "")
expected := "DSP_HOST_NOT_FOUND"
@@ -971,7 +971,7 @@ func TestLibDispatcherLoadStrategyDispatchCaseCallError(t *testing.T) {
func TestLibDispatcherDispatchFilterError(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
flts := engine.NewFilterS(cfg, nil, nil)
data := engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
data := engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items)
dm := engine.NewDataManager(data, cfg.CacheCfg(), nil)
var dsp Dispatcher = &singleResultDispatcher{
sorter: new(noSort),

View File

@@ -298,7 +298,7 @@ func TestDspResponderShutdownErrorNil(t *testing.T) {
func TestDspResponderGetCostOnRatingPlans(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
dm := engine.NewDataManager(engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items), cfg.CacheCfg(), nil)
dm := engine.NewDataManager(engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items), cfg.CacheCfg(), nil)
dsp := NewDispatcherService(dm, cfg, nil, nil)
args := &utils.GetCostOnRatingPlansArgs{
Account: "1002",
@@ -345,7 +345,7 @@ func TestDspResponderGetCostOnRatingPlans(t *testing.T) {
// func TestDspResponderGetMaxSessionTimeOnAccounts(t *testing.T) {
// cfg := config.NewDefaultCGRConfig()
// dm := engine.NewDataManager(engine.NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items), cfg.CacheCfg(), nil)
// dm := engine.NewDataManager(engine.NewInternalDB(nil, nil, true, false, cfg.DataDbCfg().Items), cfg.CacheCfg(), nil)
// dsp := NewDispatcherService(dm, cfg, nil, nil)
// args := &utils.GetMaxSessionTimeOnAccountsArgs{
// Subject: "1002",