1 files changed, 32 insertions, 0 deletions
diff --git a/sqlQueries_test.go b/sqlQueries_test.go
new file mode 100644
index 0000000..6ab904e
--- /dev/null
+++ b/sqlQueries_test.go
@@ -0,0 +1,32 @@
+package main
+
+import (
+	"database/sql"
+	_ "github.com/go-sql-driver/mysql"
+	"testing"
+	"time"
+)
+
+func TestDataFetching(t *testing.T) {
+	//initialize db
+	db, err := sql.Open("mysql", "test:test@/test")
+	if err != nil {
+		t.Fatal("Failed to connect to db")
+		return
+	}
+	defer db.Close()
+
+	//fetch the data from the db
+	rDat, err := fetchRawData(db, time.Now())
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if len(rDat) != 5 {
+		t.Fatal("did not fetch data as it should")
+	}
+	if rDat[2].ipSrc != "8.8.8.8" {
+		t.Fatal("contents of ip in data is incorrect")
+	}
+	//TODO add more tests
+}