package datastore

import (
	"database/sql"
	"reflect"
	"testing"

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

func insertStrain(t *testing.T, tx *modl.Transaction) *models.Strain {
	// clean up our target table
	tx.Exec(`DELETE FROM strains;`)
	s := newStrain(t, tx)
	if err := tx.Insert(s); err != nil {
		t.Fatal(err)
	}
	return &models.Strain{s, []int64(nil)}
}

func newStrain(t *testing.T, tx *modl.Transaction) *models.StrainBase {
	// we want to create and insert a species (and genus) record too
	species := insertSpecies(t, tx)
	return &models.StrainBase{
		SpeciesId:  species.Id,
		StrainName: "Test Strain",
		StrainType: "Test Type",
		Etymology: models.NullString{
			sql.NullString{
				String: "Test Etymology",
				Valid:  true,
			},
		},
		AccessionBanks: "Test Bank",
		GenbankEmblDdb: models.NullString{
			sql.NullString{
				String: "Test Genbank",
				Valid:  true,
			},
		},
	}
}

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

	want := insertStrain(t, tx)

	d := NewDatastore(tx)

	strain, err := d.Strains.Get(want.Id)
	if err != nil {
		t.Fatal(err)
	}

	normalizeTime(&want.CreatedAt, &want.UpdatedAt, &want.DeletedAt)
	normalizeTime(&strain.CreatedAt, &strain.UpdatedAt, &strain.DeletedAt)

	if !reflect.DeepEqual(strain, want) {
		t.Errorf("got strain %+v, want %+v", strain, want)
	}
}

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

	base_strain := newStrain(t, tx)
	strain := models.Strain{base_strain, []int64(nil)}

	d := NewDatastore(tx)

	created, err := d.Strains.Create(&strain)
	if err != nil {
		t.Fatal(err)
	}
	if !created {
		t.Error("!created")
	}
	if strain.Id == 0 {
		t.Error("want nonzero strain.Id after submitting")
	}
}

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

	want_strain := insertStrain(t, tx)
	want := []*models.Strain{want_strain}

	d := NewDatastore(tx)

	strains, err := d.Strains.List(&models.StrainListOptions{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(&strains[i].CreatedAt, &strains[i].UpdatedAt, &strains[i].DeletedAt)
	}
	if !reflect.DeepEqual(strains, want) {
		t.Errorf("got strains %+v, want %+v", strains, want)
	}
}

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

	strain := insertStrain(t, tx)

	d := NewDatastore(tx)

	// Tweak it
	strain.StrainName = "Updated Strain"
	updated, err := d.Strains.Update(strain.Id, strain)
	if err != nil {
		t.Fatal(err)
	}

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

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

	strain := insertStrain(t, tx)

	d := NewDatastore(tx)

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

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