156 lines
2.7 KiB
Go
156 lines
2.7 KiB
Go
package schema
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/uptrace/bun"
|
|
)
|
|
|
|
func TestSelectSports(t *testing.T) {
|
|
var sports []Sport
|
|
var err error
|
|
var db *bun.DB
|
|
ctx := context.Background()
|
|
|
|
db, err = GetTestDatabase()
|
|
require.NoError(t, err)
|
|
|
|
err = db.NewSelect().Model(&sports).Scan(ctx)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 4, len(sports))
|
|
}
|
|
|
|
func TestSelectTeams(t *testing.T) {
|
|
var teams []Team
|
|
var err error
|
|
var db *bun.DB
|
|
ctx := context.Background()
|
|
|
|
db, err = GetTestDatabase()
|
|
require.NoError(t, err)
|
|
|
|
err = db.NewSelect().Model(&teams).Scan(ctx)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 122, len(teams))
|
|
}
|
|
|
|
func TestSelectFieldPositions(t *testing.T) {
|
|
var positions []FieldPosition
|
|
var err error
|
|
var db *bun.DB
|
|
ctx := context.Background()
|
|
|
|
db, err = GetTestDatabase()
|
|
require.NoError(t, err)
|
|
|
|
err = db.NewSelect().Model(&positions).Scan(ctx)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 44, len(positions))
|
|
}
|
|
|
|
func TestSelectPlayers(t *testing.T) {
|
|
var players []Player
|
|
var err error
|
|
var db *bun.DB
|
|
ctx := context.Background()
|
|
|
|
db, err = GetTestDatabase()
|
|
require.NoError(t, err)
|
|
|
|
err = db.NewSelect().Model(&players).Scan(ctx)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 38, len(players))
|
|
}
|
|
|
|
func TestSelectRoosters(t *testing.T) {
|
|
var roosters []Rooster
|
|
var err error
|
|
var db *bun.DB
|
|
ctx := context.Background()
|
|
|
|
db, err = GetTestDatabase()
|
|
require.NoError(t, err)
|
|
|
|
err = db.NewSelect().Model(&roosters).Scan(ctx)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 11, len(roosters))
|
|
}
|
|
|
|
func TestSelectVendors(t *testing.T) {
|
|
var vendors []Vendor
|
|
var err error
|
|
var db *bun.DB
|
|
ctx := context.Background()
|
|
|
|
db, err = GetTestDatabase()
|
|
require.NoError(t, err)
|
|
|
|
err = db.NewSelect().Model(&vendors).Scan(ctx)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 9, len(vendors))
|
|
}
|
|
|
|
func TestSelectCardSets(t *testing.T) {
|
|
var cardSets []CardSet
|
|
var err error
|
|
var db *bun.DB
|
|
ctx := context.Background()
|
|
|
|
db, err = GetTestDatabase()
|
|
require.NoError(t, err)
|
|
|
|
err = db.NewSelect().Model(&cardSets).Scan(ctx)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 15, len(cardSets))
|
|
}
|
|
|
|
func TestSelectCards(t *testing.T) {
|
|
var cards []Card
|
|
var err error
|
|
var db *bun.DB
|
|
ctx := context.Background()
|
|
|
|
db, err = GetTestDatabase()
|
|
require.NoError(t, err)
|
|
|
|
err = db.NewSelect().Model(&cards).Scan(ctx)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 10, len(cards))
|
|
}
|