diff --git a/ers/flatstore_it_test.go b/ers/flatstore_it_test.go index f12883c2b..870bb6eb0 100644 --- a/ers/flatstore_it_test.go +++ b/ers/flatstore_it_test.go @@ -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) + } +} diff --git a/ers/sql_it_test.go b/ers/sql_it_test.go index a3ad76936..08fad1325 100644 --- a/ers/sql_it_test.go +++ b/ers/sql_it_test.go @@ -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 not supported" + if err == nil || err.Error() != expected { + t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err) + } +}