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
}