package datastore

import (
	"reflect"
	"testing"

	"github.com/jmoiron/modl"
	"github.com/thermokarst/bactdb/models"
)

func insertGenus(t *testing.T, tx *modl.Transaction) *models.Genus {
	// Test on a clean database
	tx.Exec(`DELETE FROM genera;`)

	genus := newGenus()
	if err := tx.Insert(genus); err != nil {
		t.Fatal(err)
	}
	return genus
}

func newGenus() *models.Genus {
	return &models.Genus{GenusName: "Test Genus"}
}

func TestGeneraStore_Get_db(t *testing.T) {
	tx, _ := DB.Begin()
	defer tx.Rollback()

	want := insertGenus(t, tx)

	d := NewDatastore(tx)
	genus, err := d.Genera.Get(want.Id)
	if err != nil {
		t.Fatal(err)
	}

	normalizeTime(&want.CreatedAt, &want.UpdatedAt, &want.DeletedAt)
	normalizeTime(&genus.CreatedAt, &genus.UpdatedAt, &genus.DeletedAt)
	if !reflect.DeepEqual(genus, want) {
		t.Errorf("got genus %+v, want %+v", genus, want)
	}
}

func TestGeneraStore_Create_db(t *testing.T) {
	tx, _ := DB.Begin()
	defer tx.Rollback()

	genus := newGenus()

	d := NewDatastore(tx)
	created, err := d.Genera.Create(genus)
	if err != nil {
		t.Fatal(err)
	}

	if !created {
		t.Error("!created")
	}
	if genus.Id == 0 {
		t.Error("want nonzero genus.Id after submitting")
	}
}

func TestGeneraStore_List_db(t *testing.T) {
	tx, _ := DB.Begin()
	defer tx.Rollback()

	genus := insertGenus(t, tx)
	want := []*models.Genus{genus}

	d := NewDatastore(tx)
	genera, err := d.Genera.List(&models.GenusListOptions{ListOptions: models.ListOptions{Page: 1, PerPage: 10}})
	if err != nil {
		t.Fatal(err)
	}

	for i := range want {
		normalizeTime(&want[i].CreatedAt, &want[i].UpdatedAt, &want[i].DeletedAt)
		normalizeTime(&genera[i].CreatedAt, &genera[i].UpdatedAt, &genera[i].DeletedAt)
	}
	if !reflect.DeepEqual(genera, want) {
		t.Errorf("got genera %+v, want %+v", genera, want)
	}
}

func TestGeneraStore_Update_db(t *testing.T) {
	tx, _ := DB.Begin()
	defer tx.Rollback()

	genus := insertGenus(t, tx)

	d := NewDatastore(tx)

	// Tweak it
	genus.GenusName = "Updated Genus"
	updated, err := d.Genera.Update(genus.Id, genus)
	if err != nil {
		t.Fatal(err)
	}

	if !updated {
		t.Error("!updated")
	}
}

func TestGeneraStore_Delete_db(t *testing.T) {
	tx, _ := DB.Begin()
	defer tx.Rollback()

	genus := insertGenus(t, tx)

	d := NewDatastore(tx)

	// Delete it
	deleted, err := d.Genera.Delete(genus.Id)
	if err != nil {
		t.Fatal(err)
	}

	if !deleted {
		t.Error("!delete")
	}
}