Cover tests in ers

This commit is contained in:
andronache
2021-03-10 16:12:53 +02:00
committed by Dan Christian Bogos
parent 0b229e5c3c
commit 1b580e3c1a
2 changed files with 92 additions and 0 deletions

View File

@@ -25,6 +25,7 @@ import (
"net/rpc"
"os"
"path"
"reflect"
"testing"
"time"
@@ -209,3 +210,69 @@ func testFlatstoreITKillEngine(t *testing.T) {
t.Error(err)
}
}
func TestNewFlatstoreER(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
expected := &FlatstoreER{
cgrCfg: cfg,
}
cfg.ERsCfg().Readers[0].SourcePath = "/"
result, err := NewFlatstoreER(cfg, 0, nil, nil, nil, nil)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
result.(*FlatstoreER).cache = nil
result.(*FlatstoreER).conReqs = nil
if !reflect.DeepEqual(result, expected) {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, result)
}
}
func TestFlatstoreConfig(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
cfg.ERsCfg().Readers = []*config.EventReaderCfg{
{
ID: "file_reader1",
Type: utils.MetaFileCSV,
RowLength: 5,
FieldSep: ",",
HeaderDefineChar: ":",
RunDelay: -1,
ConcurrentReqs: 1024,
SourcePath: "/tmp/ers/in",
ProcessedPath: "/tmp/ers/out",
XMLRootPath: utils.HierarchyPath{utils.EmptyString},
Tenant: nil,
Timezone: utils.EmptyString,
Filters: []string{},
Flags: utils.FlagsWithParams{},
Opts: make(map[string]interface{}),
},
{
ID: "file_reader2",
Type: utils.MetaFileCSV,
RowLength: 5,
FieldSep: ",",
HeaderDefineChar: ":",
RunDelay: -1,
ConcurrentReqs: 1024,
SourcePath: "/tmp/ers/in",
ProcessedPath: "/tmp/ers/out",
XMLRootPath: utils.HierarchyPath{utils.EmptyString},
Tenant: nil,
Timezone: utils.EmptyString,
Filters: []string{},
Flags: utils.FlagsWithParams{},
Opts: make(map[string]interface{}),
},
}
expected := cfg.ERsCfg().Readers[0]
rdr, err := NewFlatstoreER(cfg, 0, nil, nil, nil, nil)
if err != nil {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", nil, err)
}
result := rdr.Config()
if !reflect.DeepEqual(result, expected) {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, result)
}
}

View File

@@ -325,3 +325,28 @@ func testSQLStop(t *testing.T) {
}
}
func TestSQLPostCDR(t *testing.T) {
cfg := config.NewDefaultCGRConfig()
rdr := &SQLEventReader{
cgrCfg: cfg,
cfgIdx: 1,
fltrS: &engine.FilterS{},
connString: "testString",
connType: "testType",
tableName: "testName",
expConnString: "testExpConnString",
expConnType: "testExpConnType",
expTableName: "testExpTableName",
rdrEvents: nil,
rdrExit: nil,
rdrErr: nil,
cap: nil,
}
in := make([]interface{}, 2)
err := rdr.postCDR(in)
expected := "db type <testExpConnType> not supported"
if err == nil || err.Error() != expected {
t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
}
}