querier_test.go

package db_test

import (
	"github.com/drborges/datastore-model"
	"github.com/drborges/goexpect"
	"testing"
)

var (
	diego  = &Person{Name: "Diego", Country: "Brazil"}
	munjal = &Person{Name: "Munjal", Country: "USA"}
)

type Person struct {
	db.Model `db:"People"`
	Name     string `db:"id"`
	Country  string
}

func TestQuerierEntityAtSlicePtr(t *testing.T) {
	t.Parallel()
	people := []*Person{diego, munjal}
	entity := db.EntityAt(&people, 1)

	expect := goexpect.New(t)
	expect(entity).ToBe(munjal)
}

func TestQuerierEntityAtSlice(t *testing.T) {
	t.Parallel()
	people := []*Person{diego, munjal}
	entity := db.EntityAt(people, 0)

	expect := goexpect.New(t)
	expect(entity).ToBe(diego)
}

func TestQuerierEntityAtPanicsWhenInvalidParameterIsProvided(t *testing.T) {
	t.Parallel()
	defer func() {
		expect := goexpect.New(t)
		err := recover()
		expect(err).ToBe(db.ErrInvalidType)
	}()

	db.EntityAt(123, 0) // Should panic
	panic("Should not reach here")
}