roll_test.go

package main

import "testing"

// This should be set to any number that is at least two times
// the total number of values that can be generated by the roll.
const executions int = 1000

func testResults(spec RollSpec, t *testing.T) {
	totalDiceRolled := spec.DieCount
	if spec.BestOf != 0 {
		totalDiceRolled = spec.BestOf
	}
	minimumResult := int(totalDiceRolled + spec.Modifier)
	maximumResult := int(totalDiceRolled * (spec.Sides + spec.Modifier))

	for i := 0; i < executions; i++ {
		result := DoRolls(spec)
		var e int64
		for e = 0; e < spec.Times; e++ {
			total := result.Rolls[e].Total
			if total < minimumResult {
				t.Fatalf("Roll %d was below minimum of %d", total, minimumResult)
			}
			if total > maximumResult {
				t.Fatalf("Roll %d was greater than maximum result of %d", total, maximumResult)
			}
		}
	}
}

func TestD10(t *testing.T) {
	rollSpec, err := Parse("d10")
	if err != nil {
		t.Fatalf("Error parsing 'd10': %v", err)
	}
	testResults(*rollSpec, t)
}

func TestD10Plus1(t *testing.T) {
	rollSpec, err := Parse("d10+1")
	if err != nil {
		t.Fatalf("Error parsing 'd10+1': %v", err)
	}
	testResults(*rollSpec, t)
}

func Test1D10Plus2(t *testing.T) {
	rollSpec, err := Parse("1d10+2")
	if err != nil {
		t.Fatalf("Error parsing '1d10+2': %v", err)
	}
	testResults(*rollSpec, t)
}

func TestD6Times6(t *testing.T) {
	rollSpec, err := Parse("3d6x6")
	if err != nil {
		t.Fatalf("Error parsing '3d6x6': %v", err)
	}
	testResults(*rollSpec, t)
}

func TestBest3of4D6Times6(t *testing.T) {
	rollSpec, err := Parse("3,4d6x6")
	if err != nil {
		t.Fatalf("Error parsing '3,4d6x6': %v", err)
	}
	testResults(*rollSpec, t)
}

func TestBest3of4D6(t *testing.T) {
	rollSpec, err := Parse("3,4d6")
	if err != nil {
		t.Fatalf("Error parsing '3,4d6': %v", err)
	}
	testResults(*rollSpec, t)
}

func TestBest2of3D12Plus6(t *testing.T) {
	rollSpec, err := Parse("2,3d12+6")
	if err != nil {
		t.Fatalf("Error parsing '2,3d12+6': %v", err)
	}
	testResults(*rollSpec, t)
}

func TestBest2of3D100Plus5TenTimes(t *testing.T) {
	rollSpec, err := Parse("2,3d100+5x10")
	if err != nil {
		t.Fatalf("Error parsing '2,3d100+5x10': %v", err)
	}
	testResults(*rollSpec, t)
}