From 65320aa4ffeef0a158aacdf1adacc09f8faf1a75 Mon Sep 17 00:00:00 2001
From: andronache <andrei.andronache@itsyscom.com>
Date: Thu, 25 Mar 2021 15:50:29 +0200
Subject: [PATCH] Tests for sql.go and partial_csv.go in ers

---
 ers/partial_csv_it_test.go |  3 ---
 ers/sql_it_test.go         | 50 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 50 insertions(+), 3 deletions(-)

diff --git a/ers/partial_csv_it_test.go b/ers/partial_csv_it_test.go
index 5daf22f5c..a22fb627e 100644
--- a/ers/partial_csv_it_test.go
+++ b/ers/partial_csv_it_test.go
@@ -525,8 +525,6 @@ func TestPartialCSVProcessFile2(t *testing.T) {
 	}
 }
 
-/*
-
 func TestPartialCSVServe2(t *testing.T) {
 	cfg := config.NewDefaultCGRConfig()
 	cfg.ERsCfg().Readers = []*config.EventReaderCfg{
@@ -568,4 +566,3 @@ func TestPartialCSVServe2(t *testing.T) {
 		t.Errorf("\nExpected <%+v>, \nReceived <%+v>", "no such file or directory", err)
 	}
 }
-*/
diff --git a/ers/sql_it_test.go b/ers/sql_it_test.go
index e1f57625a..c20406cad 100644
--- a/ers/sql_it_test.go
+++ b/ers/sql_it_test.go
@@ -585,3 +585,53 @@ func testSQLStop2(t *testing.T) {
 	}
 
 }
+
+func TestSQLProcessMessageError(t *testing.T) {
+	cfg := config.NewDefaultCGRConfig()
+	testSQLEventReader := &SQLEventReader{
+		cgrCfg:        cfg,
+		cfgIdx:        0,
+		fltrS:         &engine.FilterS{},
+		connString:    "",
+		connType:      "",
+		tableName:     "testName",
+		expConnString: "",
+		expConnType:   utils.Postgres,
+		expTableName:  "",
+		rdrEvents:     nil,
+		rdrExit:       nil,
+		rdrErr:        nil,
+		cap:           nil,
+	}
+
+	msgTest := map[string]interface{}{}
+	err := testSQLEventReader.processMessage(msgTest)
+	expected := "NOT_FOUND:ToR"
+	if err == nil || err.Error() != expected {
+		t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
+	}
+}
+
+func TestSQLSetURLError(t *testing.T) {
+	cfg := config.NewDefaultCGRConfig()
+	testSQLEventReader := &SQLEventReader{
+		cgrCfg:        cfg,
+		cfgIdx:        0,
+		fltrS:         &engine.FilterS{},
+		connString:    "",
+		connType:      "",
+		tableName:     "testName",
+		expConnString: "",
+		expConnType:   utils.Postgres,
+		expTableName:  "",
+		rdrEvents:     nil,
+		rdrExit:       nil,
+		rdrErr:        nil,
+		cap:           nil,
+	}
+	err := testSQLEventReader.setURL("http://user^:passwo^rd@foo.com/", "", nil)
+	expected := `parse "http://user^:passwo^rd@foo.com/": net/url: invalid userinfo`
+	if err == nil || err.Error() != expected {
+		t.Errorf("\nExpected <%+v>, \nReceived <%+v>", expected, err)
+	}
+}