common: Add tests for equipment & players

Author: markus-wa

Gopkg.lock

@@ -48,3 +48,7 @@
 [prune]
   go-tests = true
   unused-packages = true
+
+[[constraint]]
+  name = "github.com/stretchr/testify"
+  version = "1.2.2"

Gopkg.toml

@@ -0,0 +1,37 @@
+package common
+
+import (
+	"testing"
+
+	assert "github.com/stretchr/testify/assert"
+)
+
+func TestEquipmentElementClass(t *testing.T) {
+	assert.Equal(t, EqClassUnknown, EqUnknown.Class(), "EqUnknown should have the class EqClassUnknown")
+	assert.Equal(t, EqClassPistols, EqP2000.Class(), "EqP2000 should have the class EqClassPistols")
+	assert.Equal(t, EqClassPistols, EqRevolver.Class(), "EqRevolver should have the class EqClassPistols")
+	assert.Equal(t, EqClassRifle, EqG3SG1.Class(), "EqG3SG1 should have the class EqClassRifle")
+}
+
+func TestEquipmentClass(t *testing.T) {
+	assert.Equal(t, EqClassUnknown, NewEquipment(EqUnknown).Class(), "EqUnknown should have the class EqClassUnknown")
+	assert.Equal(t, EqClassPistols, NewEquipment(EqP2000).Class(), "EqP2000 should have the class EqClassPistols")
+	assert.Equal(t, EqClassPistols, NewEquipment(EqRevolver).Class(), "EqRevolver should have the class EqClassPistols")
+	assert.Equal(t, EqClassRifle, NewEquipment(EqG3SG1).Class(), "EqG3SG1 should have the class EqClassRifle")
+}
+
+func TestMapEquipment(t *testing.T) {
+	assert.Equal(t, EqKnife, MapEquipment("weapon_bayonet"), "'weapon_bayonet' should be mapped to EqKnife")
+	assert.Equal(t, EqKnife, MapEquipment("weapon_knife_butterfly"), "'weapon_knife_butterfly' should be mapped to EqKnife")
+	assert.Equal(t, EqM4A4, MapEquipment("weapon_m4a1"), "'weapon_m4a1' should be mapped to EqM4A4") // This is correct, weapon_m4a1 == M4A4
+	assert.Equal(t, EqM4A1, MapEquipment("weapon_m4a1_silencer"), "'weapon_m4a1_silencer' should be mapped to EqM4A1")
+	assert.Equal(t, EqUnknown, MapEquipment("asdf"), "'asdf' should be mapped to EqUnknown")
+}
+
+func TestEquipmentUniqueID(t *testing.T) {
+	assert.NotEqual(t, NewEquipment(EqAK47).UniqueID(), NewEquipment(EqAK47).UniqueID(), "UniqueIDs of different equipment instances should be different")
+}
+
+func TestEquipmentName(t *testing.T) {
+	assert.Equal(t, "Dual Barettas", EqDualBarettas.String(), "EqDualBarettas should be named correctly")
+}

common/equipment_test.go

@@ -0,0 +1,50 @@
+package common
+
+import (
+	"testing"
+
+	assert "github.com/stretchr/testify/assert"
+)
+
+func TestPlayerActiveWeapon(t *testing.T) {
+	knife := NewEquipment(EqKnife)
+	glock := NewEquipment(EqGlock)
+	ak47 := NewEquipment(EqAK47)
+
+	pl := NewPlayer()
+	pl.RawWeapons[1] = &knife
+	pl.RawWeapons[2] = &glock
+	pl.RawWeapons[3] = &ak47
+	pl.ActiveWeaponID = 3
+
+	assert.Equal(t, &ak47, pl.ActiveWeapon(), "Should have AK-47 equipped")
+}
+func TestPlayerWeapons(t *testing.T) {
+	knife := NewEquipment(EqKnife)
+	glock := NewEquipment(EqGlock)
+	ak47 := NewEquipment(EqAK47)
+
+	pl := NewPlayer()
+	pl.RawWeapons[1] = &knife
+	pl.RawWeapons[2] = &glock
+	pl.RawWeapons[3] = &ak47
+
+	expected := []*Equipment{&knife, &glock, &ak47}
+	assert.ElementsMatch(t, expected, pl.Weapons(), "Should have expected weapons")
+}
+
+func TestPlayerAlive(t *testing.T) {
+	pl := NewPlayer()
+
+	pl.Hp = 100
+	assert.Equal(t, true, pl.IsAlive(), "Should be alive")
+
+	pl.Hp = 1
+	assert.Equal(t, true, pl.IsAlive(), "Should be alive")
+
+	pl.Hp = 0
+	assert.Equal(t, false, pl.IsAlive(), "Should be dead")
+
+	pl.Hp = -10
+	assert.Equal(t, false, pl.IsAlive(), "Should be dead")
+}