moth/cmd/mothd/state_test.go

package main

import (
	"bytes"
	"os"
	"testing"

	"github.com/spf13/afero"
)

func TestState(t *testing.T) {
	fs := new(afero.MemMapFs)

	mustExist := func(path string) {
		_, err := fs.Stat(path)
		if os.IsNotExist(err) {
			t.Errorf("File %s does not exist", path)
		}
	}

	s := NewState(fs)
	s.Update()

	pl := s.PointsLog()
	if len(pl) != 0 {
		t.Errorf("Empty points log is not empty")
	}

	mustExist("initialized")
	mustExist("enabled")
	mustExist("hours")

	teamidsBuf, err := afero.ReadFile(fs, "teamids.txt")
	if err != nil {
		t.Errorf("Reading teamids.txt: %v", err)
	}

	teamids := bytes.Split(teamidsBuf, []byte("\n"))
	if (len(teamids) != 101) || (len(teamids[100]) > 0) {
		t.Errorf("There weren't 100 teamids, there were %d", len(teamids))
	}
	teamId := string(teamids[0])

	if err := s.SetTeamName("bad team ID", "bad team name"); err == nil {
		t.Errorf("Setting bad team ID didn't raise an error")
	}

	if err := s.SetTeamName(teamId, "My Team"); err != nil {
		t.Errorf("Setting team name: %v", err)
	}

	category := "poot"
	points := 3928
	s.AwardPoints(teamId, category, points)
	s.Update()

	pl = s.PointsLog()
	if len(pl) != 1 {
		t.Errorf("After awarding points, points log has length %d", len(pl))
	} else if (pl[0].TeamID != teamId) || (pl[0].Category != category) || (pl[0].Points != points) {
		t.Errorf("Incorrect logged award %v", pl)
	}

	fs.Remove("initialized")
	s.Update()

	pl = s.PointsLog()
	if len(pl) != 0 {
		t.Errorf("After reinitialization, points log has length %d", len(pl))
	}
}