125 lines
2.4 KiB
Go
125 lines
2.4 KiB
Go
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;`)
|
|
|
|
g := newGenus()
|
|
if err := tx.Insert(g); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return &models.Genus{g, []int64(nil)}
|
|
}
|
|
|
|
func newGenus() *models.GenusBase {
|
|
return &models.GenusBase{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()
|
|
|
|
base_genus := newGenus()
|
|
genus := models.Genus{base_genus, []int64(nil)}
|
|
|
|
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")
|
|
}
|
|
}
|