added tests to see that rawdata isn't touch if not needed

1 files changed, 51 insertions, 4 deletions

diff --git a/flow-cleaner_test.go b/flow-cleaner_test.go
index 51283a7..9be5d69 100644
--- a/flow-cleaner_test.go
+++ b/flow-cleaner_test.go
@@ -22,6 +22,7 @@ func TestCleaningFromJSON(t *testing.T) {
 
 		DBConn:     "",
 		DBName:     "test",
+		RawTable:   "test_raw",
 		CleanTable: "test_clean",
 		DBUser:     "flowcleaner",
 		DBPass:     "nil",
@@ -31,7 +32,8 @@ func TestCleaningFromJSON(t *testing.T) {
 	prepareDB(t, cfg)
 	testProcessFromStdin(t, cfg)
 	time.Sleep(15 * time.Second)
-	controlDB(t, cfg)
+	controlCleanDB(t, cfg)
+	controlRawDBStdin(t, cfg)
 	fmt.Println("== Finished testing to process from stdin ==")
 }
 
@@ -53,7 +55,8 @@ func TestCleaningFromDB(t *testing.T) {
 	fmt.Println("== Testing to process from DB ==")
 	prepareDB(t, cfg)
 	processFromDB(cfg)
-	controlDB(t, cfg)
+	controlCleanDB(t, cfg)
+	controlRawDBMySQL(t, cfg)
 	fmt.Println("== Finished testing to process from DB ==")
 }
 
@@ -136,7 +139,51 @@ func fakeStdin(t *testing.T, wr chan<- []byte) {
 	}
 }
 
-func controlDB(t *testing.T, cfg *Config) {
+func controlRawDBStdin(t *testing.T, cfg *Config) {
+	db, err := sql.Open("mysql", cfg.DBUser+":"+cfg.DBPass+"@"+cfg.DBConn+"/"+cfg.DBName)
+	if err != nil {
+		t.Fatal("Failed to connect to db:", err)
+	}
+	defer db.Close()
+
+	rows, err := db.Query("SELECT * FROM " + cfg.RawTable)
+	if err != nil {
+		t.Fatal("Failed to select cleaned data:", err)
+	}
+	defer rows.Close()
+
+	numRows := 0
+	for rows.Next() {
+		numRows++
+	}
+	if numRows != 33 {
+		t.Fatal("Wrong number of rows found in db, should be 33 found:", numRows)
+	}
+}
+
+func controlRawDBMySQL(t *testing.T, cfg *Config) {
+	db, err := sql.Open("mysql", cfg.DBUser+":"+cfg.DBPass+"@"+cfg.DBConn+"/"+cfg.DBName)
+	if err != nil {
+		t.Fatal("Failed to connect to db:", err)
+	}
+	defer db.Close()
+
+	rows, err := db.Query("SELECT * FROM " + cfg.RawTable)
+	if err != nil {
+		t.Fatal("Failed to select cleaned data:", err)
+	}
+	defer rows.Close()
+
+	numRows := 0
+	for rows.Next() {
+		numRows++
+	}
+	if numRows != 3 {
+		t.Fatal("Wrong number of rows found in db, should be 3 found:", numRows)
+	}
+}
+
+func controlCleanDB(t *testing.T, cfg *Config) {
 	db, err := sql.Open("mysql", cfg.DBUser+":"+cfg.DBPass+"@"+cfg.DBConn+"/"+cfg.DBName)
 	if err != nil {
 		t.Fatal("Failed to connect to db:", err)
@@ -154,7 +201,7 @@ func controlDB(t *testing.T, cfg *Config) {
 		numRows++
 	}
 	if numRows != 14 {
-		t.Fatal("Wrong number of rows found in db")
+		t.Fatal("Wrong number of rows found in db, should be 14 found:", numRows)
 	}
 
 	check1, err := db.Query("SELECT occurences, volume FROM " + cfg.CleanTable + " ORDER BY occurences DESC")