Refactor database models into separate package.

model/match_test.go

@@ -0,0 +1,119 @@
+// Copyright 2014 Team 254. All Rights Reserved.
+// Author: pat@patfairbank.com (Patrick Fairbank)
+
+package model
+
+import (
+	"github.com/stretchr/testify/assert"
+	"testing"
+	"time"
+)
+
+func TestGetNonexistentMatch(t *testing.T) {
+	db := setupTestDb(t)
+
+	match, err := db.GetMatchById(1114)
+	assert.Nil(t, err)
+	assert.Nil(t, match)
+}
+
+func TestMatchCrud(t *testing.T) {
+	db := setupTestDb(t)
+
+	match := Match{0, "qualification", "254", time.Now().UTC(), 0, 0, 0, 1, false, 2, false, 3, false, 4, false,
+		5, false, 6, false, "", time.Now().UTC(), ""}
+	db.CreateMatch(&match)
+	match2, err := db.GetMatchById(1)
+	assert.Nil(t, err)
+	assert.Equal(t, match, *match2)
+	match3, err := db.GetMatchByName("qualification", "254")
+	assert.Nil(t, err)
+	assert.Equal(t, match, *match3)
+
+	match.Status = "started"
+	db.SaveMatch(&match)
+	match2, err = db.GetMatchById(1)
+	assert.Nil(t, err)
+	assert.Equal(t, match.Status, match2.Status)
+
+	db.DeleteMatch(&match)
+	match2, err = db.GetMatchById(1)
+	assert.Nil(t, err)
+	assert.Nil(t, match2)
+}
+
+func TestTruncateMatches(t *testing.T) {
+	db := setupTestDb(t)
+
+	match := Match{0, "qualification", "254", time.Now().UTC(), 0, 0, 0, 1, false, 2, false, 3, false, 4, false,
+		5, false, 6, false, "", time.Now().UTC(), ""}
+	db.CreateMatch(&match)
+	db.TruncateMatches()
+	match2, err := db.GetMatchById(1)
+	assert.Nil(t, err)
+	assert.Nil(t, match2)
+}
+
+func TestGetMatchesByElimRoundGroup(t *testing.T) {
+	db := setupTestDb(t)
+
+	match := Match{Type: "elimination", DisplayName: "SF1-1", ElimRound: 2, ElimGroup: 1, ElimInstance: 1}
+	db.CreateMatch(&match)
+	match2 := Match{Type: "elimination", DisplayName: "SF2-2", ElimRound: 2, ElimGroup: 2, ElimInstance: 2}
+	db.CreateMatch(&match2)
+	match3 := Match{Type: "elimination", DisplayName: "SF2-1", ElimRound: 2, ElimGroup: 2, ElimInstance: 1}
+	db.CreateMatch(&match3)
+	match4 := Match{Type: "elimination", DisplayName: "QF2-1", ElimRound: 4, ElimGroup: 2, ElimInstance: 1}
+	db.CreateMatch(&match4)
+	match5 := Match{Type: "practice", DisplayName: "1"}
+	db.CreateMatch(&match5)
+
+	matches, err := db.GetMatchesByElimRoundGroup(4, 1)
+	assert.Nil(t, err)
+	assert.Empty(t, matches)
+	matches, err = db.GetMatchesByElimRoundGroup(2, 2)
+	assert.Nil(t, err)
+	if assert.Equal(t, 2, len(matches)) {
+		assert.Equal(t, "SF2-1", matches[0].DisplayName)
+		assert.Equal(t, "SF2-2", matches[1].DisplayName)
+	}
+}
+
+func TestGetMatchesByType(t *testing.T) {
+	db := setupTestDb(t)
+
+	match := Match{0, "qualification", "1", time.Now().UTC(), 0, 0, 0, 1, false, 2, false, 3, false, 4, false,
+		5, false, 6, false, "", time.Now().UTC(), ""}
+	db.CreateMatch(&match)
+	match2 := Match{0, "practice", "1", time.Now().UTC(), 0, 0, 0, 1, false, 2, false, 3, false, 4, false, 5,
+		false, 6, false, "", time.Now().UTC(), ""}
+	db.CreateMatch(&match2)
+	match3 := Match{0, "practice", "2", time.Now().UTC(), 0, 0, 0, 1, false, 2, false, 3, false, 4, false, 5,
+		false, 6, false, "", time.Now().UTC(), ""}
+	db.CreateMatch(&match3)
+
+	matches, err := db.GetMatchesByType("test")
+	assert.Nil(t, err)
+	assert.Empty(t, matches)
+	matches, err = db.GetMatchesByType("practice")
+	assert.Nil(t, err)
+	assert.Equal(t, 2, len(matches))
+	matches, err = db.GetMatchesByType("qualification")
+	assert.Nil(t, err)
+	assert.Equal(t, 1, len(matches))
+}
+
+func TestTbaCode(t *testing.T) {
+	match := Match{Type: "practice", DisplayName: "3"}
+	assert.Equal(t, "", match.TbaCode())
+	match = Match{Type: "qualification", DisplayName: "26"}
+	assert.Equal(t, "qm26", match.TbaCode())
+	match = Match{Type: "elimination", DisplayName: "EF2-1", ElimRound: 8, ElimGroup: 2, ElimInstance: 1}
+	assert.Equal(t, "ef2m1", match.TbaCode())
+	match = Match{Type: "elimination", DisplayName: "QF3-2", ElimRound: 4, ElimGroup: 3, ElimInstance: 2}
+	assert.Equal(t, "qf3m2", match.TbaCode())
+	match = Match{Type: "elimination", DisplayName: "SF1-3", ElimRound: 2, ElimGroup: 1, ElimInstance: 3}
+	assert.Equal(t, "sf1m3", match.TbaCode())
+	match = Match{Type: "elimination", DisplayName: "F2", ElimRound: 1, ElimGroup: 1, ElimInstance: 2}
+	assert.Equal(t, "f1m2", match.TbaCode())
+}