1 files changed, 30 insertions, 0 deletions
diff --git a/internal/combat/types_test.go b/internal/combat/types_test.go
new file mode 100644
index 0000000..a999eb6
--- /dev/null
+++ b/internal/combat/types_test.go
@@ -0,0 +1,30 @@
+package combat
+
+import "testing"
+
+func TestIsMeleeType(t *testing.T) {
+	melee := []string{AttackStab, AttackSlash, AttackCrush}
+	for _, m := range melee {
+		if !IsMeleeType(m) {
+			t.Errorf("IsMeleeType(%q) = false, want true", m)
+		}
+	}
+	for _, nm := range []string{AttackRanged, AttackScience, "", "magic"} {
+		if IsMeleeType(nm) {
+			t.Errorf("IsMeleeType(%q) = true, want false", nm)
+		}
+	}
+}
+
+func TestIsValidAttackType(t *testing.T) {
+	for _, v := range AllAttackTypes {
+		if !IsValidAttackType(v) {
+			t.Errorf("IsValidAttackType(%q) = false, want true", v)
+		}
+	}
+	for _, iv := range []string{"", "magic", "melee"} {
+		if IsValidAttackType(iv) {
+			t.Errorf("IsValidAttackType(%q) = true, want false", iv)
+		}
+	}
+}