Refactored tests in services/datadb_it_test.go

This commit is contained in:
andronache
2021-03-04 12:08:52 +02:00
committed by Dan Christian Bogos
parent 60f0e4aa4a
commit fd37067174
4 changed files with 16 additions and 49 deletions

View File

@@ -736,33 +736,3 @@ func TestERsProcessEvent11(t *testing.T) {
t.Fatalf("\nExpecting <%+v>,\n Received <%+v>", "MANDATORY_IE_MISSING: [connIDs]", err)
}
}
func TestERsProcessEvent12(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.ERsCfg().Readers = []*config.EventReaderCfg{
{
ID: "test",
Type: utils.MetaNone,
},
}
cfg.ERsCfg().SessionSConns = []string{rpcclient.InternalRPC}
fltrS := &engine.FilterS{}
rpcInt := map[string]chan rpcclient.ClientConnector{}
connMang := engine.NewConnManager(cfg, rpcInt)
srv := NewERService(cfg, fltrS, connMang)
rdrCfg := &config.EventReaderCfg{
Flags: map[string]utils.FlagParams{
utils.MetaMessage: map[string][]string{},
},
}
cgrEvent := &utils.CGREvent{
Opts: map[string]interface{}{
utils.OptsRoutesLimit: true,
},
}
err := srv.processEvent(cgrEvent, rdrCfg)
if err == nil || err.Error() != "UNSUPPORTED_SERVICE_METHOD" {
t.Fatalf("\nExpecting <%+v>,\n Received <%+v>", "UNSUPPORTED_SERVICE_METHOD", err)
}
}

View File

@@ -20,6 +20,19 @@ along with this program. If not, see <http://www.gnu.org/licenses/>
package ers
import (
"context"
"fmt"
"reflect"
"testing"
"time"
"github.com/cgrates/cgrates/config"
"github.com/cgrates/cgrates/engine"
"github.com/cgrates/cgrates/utils"
kafka "github.com/segmentio/kafka-go"
)
var (
rdrEvents chan *erEvent
rdrErr chan error
@@ -27,7 +40,6 @@ var (
rdr EventReader
)
/*
func TestKafkaER(t *testing.T) {
cfg, err := config.NewCGRConfigFromJSONStringWithDefaults(`{
"ers": { // EventReaderService
@@ -103,4 +115,3 @@ func TestKafkaER(t *testing.T) {
}
close(rdrExit)
}
*/

View File

@@ -148,5 +148,4 @@ func TestRegistrarcShutdown(t *testing.T) {
cfg.RegistrarCCfg().RPC.Enabled = true
regStSrv := NewRegistrarCService(cfg, nil)
regStSrv.Shutdown()
}

View File

@@ -553,27 +553,14 @@ func TestDataDBReloadVersion(t *testing.T) {
utils.Logger.SetLogLevel(7)
shdChan := utils.NewSyncedChan()
shdWg := new(sync.WaitGroup)
chS := engine.NewCacheS(cfg, nil, nil)
filterSChan := make(chan *engine.FilterS, 1)
filterSChan <- nil
close(chS.GetPrecacheChannel(utils.CacheAttributeProfiles))
close(chS.GetPrecacheChannel(utils.CacheAttributeFilterIndexes))
server := cores.NewServer(nil)
srvDep := map[string]*sync.WaitGroup{utils.DataDB: new(sync.WaitGroup)}
srvMngr := servmanager.NewServiceManager(cfg, shdChan, shdWg)
cM := engine.NewConnManager(cfg, nil)
db := NewDataDBService(cfg, cM, srvDep)
anz := NewAnalyzerService(cfg, server, filterSChan, shdChan, make(chan rpcclient.ClientConnector, 1), srvDep)
srvMngr.AddServices(NewAttributeService(cfg, db,
chS, filterSChan, server, make(chan rpcclient.ClientConnector, 1), anz, srvDep),
NewLoaderService(cfg, db, filterSChan, server, make(chan rpcclient.ClientConnector, 1), nil, anz, srvDep), db)
srvMngr.StartServices()
<-shdChan.Done()
db.dm = nil
err = db.Reload()
if err == nil || err.Error() != "can't conver DataDB of type mongo to MongoStorage" {
t.Fatal(err)
err = db.Start()
if err == nil || err.Error() != "Migration needed: please backup cgr data and run : <cgr-migrator -exec=*attributes>" {
t.Errorf("\nExpecting <%+v>,\n Received <%+v>", "Migration needed: please backup cgr data and run : <cgr-migrator -exec=*attributes>", err)
}
shdChan.CloseOnce()
time.Sleep(10 * time.Millisecond)