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("Fetched the wrong number of rows:", len(rDat))
	}
	if rDat[2].ipSrc != "8.8.8.8" {
		t.Fatal("contents of ip in data is incorrect")
	}
	//TODO add more tests
}

func TestCleaning(t *testing.T) {
	conf := Config{
		Interval: "10min",
		Epsilon:  0,
	}

	err := cleanData(conf, "test", "test", "", "test")
	if err != nil {
		t.Fatal(err)
	}

	db, err := sql.Open("mysql", "test:test@/test")
	if err != nil {
		t.Fatal("Failed to connect to db")
		return
	}
	defer db.Close()

	rDat, err := fetchRawData(db, time.Now())
	if err != nil {
		t.Fatal(err)
	}
	if len(rDat) > 0 {
		t.Fatal()
	}
	rows, err := db.Query("SELECT * FROM clean_data;")
	if err != nil {
		t.Fatal(err)
	}

	counter := 0
	for rows.Next() {
		counter++
	}
	if counter != 3 {
		t.Fatal("Wrong number of cleaned rows")
	}

}
*/