Files
cheesy-arena-lite/model/test_helpers.go

37 lines
1.1 KiB
Go
Executable File

// Copyright 2017 Team 254. All Rights Reserved.
// Author: pat@patfairbank.com (Patrick Fairbank)
//
// Helper methods for use in tests in this package and others.
package model
import (
"fmt"
"github.com/Team254/cheesy-arena-lite/game"
"github.com/stretchr/testify/assert"
"os"
"path/filepath"
"testing"
)
func SetupTestDb(t *testing.T, uniqueName string) *Database {
BaseDir = ".."
dbPath := filepath.Join(BaseDir, fmt.Sprintf("%s_test.db", uniqueName))
os.Remove(dbPath)
database, err := OpenDatabase(dbPath)
assert.Nil(t, err)
return database
}
func BuildTestMatchResult(matchId int, playNumber int) *MatchResult {
matchResult := &MatchResult{MatchId: matchId, PlayNumber: playNumber, MatchType: "qualification"}
matchResult.RedScore = game.TestScore1()
matchResult.BlueScore = game.TestScore2()
return matchResult
}
func BuildTestAlliances(database *Database) {
database.CreateAlliance(&Alliance{Id: 2, TeamIds: []int{1718, 2451, 1619}, Lineup: [3]int{2451, 1718, 1619}})
database.CreateAlliance(&Alliance{Id: 1, TeamIds: []int{254, 469, 2848, 74, 3175}, Lineup: [3]int{469, 254, 2848}})
}