1 files changed, 52 insertions, 0 deletions
diff --git a/go/game/attack_test.go b/go/game/attack_test.go
new file mode 100644
index 00000000..6327b8eb
--- /dev/null
+++ b/go/game/attack_test.go
@@ -0,0 +1,52 @@
+package game
+
+import (
+	"testing"
+)
+
+func TestParseIntAttack(t *testing.T) {
+	a := parseAttack(1)
+	if a.MaxRange() != 1 {
+		t.Fatal("unexpected max range (1):", a.MaxRange())
+	}
+	if a.DamageInRange(1) != 1 {
+		t.Fatal("unexpected damage in range 1 (1):", a.DamageInRange(1))
+	}
+}
+
+func TestParseStringAttack(t *testing.T) {
+	a := parseAttack("1")
+	if a.MaxRange() != 1 {
+		t.Fatal("unexpected max range (1):", a.MaxRange())
+	}
+	if a.DamageInRange(1) != 1 {
+		t.Fatal("unexpected damage in range 1 (1):", a.DamageInRange(1))
+	}
+	if a.DamageInRange(2) != 0 {
+		t.Fatal("unexpected damage in range 2 (0):", a.DamageInRange(2))
+	}
+}
+
+func TestParseRangedAttack(t *testing.T) {
+	a := parseAttack("1 range 2")
+	if a.MaxRange() != 2 {
+		t.Fatal("unexpected max range (2):", a.MaxRange())
+	}
+	if a.DamageInRange(1) != 1 {
+		t.Fatal("unexpected damage in range 1 (1):", a.DamageInRange(1))
+	}
+	if a.DamageInRange(2) != 1 {
+		t.Fatal("unexpected damage in range 2 (1):", a.DamageInRange(2))
+	}
+	if a.DamageInRange(3) != 0 {
+		t.Fatal("unexpected damage in range 3 (0):", a.DamageInRange(3))
+	}
+}
+
+func TestAttackString(t *testing.T) {
+	aStr := "2 range 3"
+	a := parseAttack(aStr)
+	if a.String() != aStr {
+		t.Fatal("unexpected attack string:", a.String())
+	}
+}