diff options
author | Daniel Langesten <daniel.langest@gmail.com> | 2015-03-26 10:26:03 +0100 |
---|---|---|
committer | Daniel Langesten <daniel.langest@gmail.com> | 2015-03-26 10:26:03 +0100 |
commit | 54f6128240711a93551d85b000766a9b57cc96cf (patch) | |
tree | 86f703dd439bc657fe24f08a277badbac396dfac | |
parent | 336a46d029a0a435655e083ccf97d84957755b58 (diff) |
added tests to see that rawdata isn't touch if not needed
-rw-r--r-- | flow-cleaner_test.go | 55 |
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") |