feat: unmarshal int to Entity (#369)

## Issue
N/A

## Description
Allow unmarshalling entities from ints as well as strings.

---------

Signed-off-by: Tyler Gillson <tyler.gillson@gmail.com>

Author: TylerGillson

api/vcenter/entity/entities.go

@@ -82,7 +82,14 @@ func (e Entity) MarshalYAML() (interface{}, error) {
 
 // UnmarshalYAML implements the yaml.Unmarshaler interface.
 func (e *Entity) UnmarshalYAML(unmarshal func(interface{}) error) error {
-	var entityStr string
+	var (
+		entityInt int
+		entityStr string
+	)
+	if err := unmarshal(&entityInt); err == nil {
+		*e = Entity(entityInt)
+		return nil
+	}
 	if err := unmarshal(&entityStr); err != nil {
 		return err
 	}

api/vcenter/entity/entities_test.go

@@ -40,8 +40,8 @@ func TestMarshalYAML(t *testing.T) {
 }
 
 func TestUnMarshalYAML(t *testing.T) {
-	// string: correct casing
-	in := []byte("Resource Pool\n")
+	// int
+	in := []byte("8\n")
 	expected := ResourcePool
 
 	var e Entity
@@ -52,6 +52,16 @@ func TestUnMarshalYAML(t *testing.T) {
 		t.Errorf("got %v != expected %v", e, expected)
 	}
 
+	// string: correct casing
+	in = []byte("Resource Pool\n")
+
+	if err := yaml.Unmarshal(in, &e); err != nil {
+		t.Errorf("failed to unmarshal ResourcePool: %v", err)
+	}
+	if !reflect.DeepEqual(e, expected) {
+		t.Errorf("got %v != expected %v", e, expected)
+	}
+
 	// struct + case-insensitive
 	in = []byte("entityType: resource pool\n")
 	expectedS := estruct{EntityType: ResourcePool}