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) } } }