Fix govmomi #192: HostSystem doesn't seem to be returning the correct…

… host

compute_resource.go

@@ -16,12 +16,35 @@ limitations under the License.
 
 package govmomi
 
-import "github.com/vmware/govmomi/vim25/types"
+import (
+	"github.com/vmware/govmomi/vim25/mo"
+	"github.com/vmware/govmomi/vim25/types"
+)
 
 type ComputeResource struct {
 	types.ManagedObjectReference
+
+	c *Client
+}
+
+func NewComputeResource(c *Client, ref types.ManagedObjectReference) *ComputeResource {
+	return &ComputeResource{
+		ManagedObjectReference: ref,
+		c: c,
+	}
 }
 
 func (c ComputeResource) Reference() types.ManagedObjectReference {
 	return c.ManagedObjectReference
 }
+
+func (c ComputeResource) Hosts() ([]types.ManagedObjectReference, error) {
+	var cr mo.ComputeResource
+	ps := []string{"host"}
+
+	err := c.c.Properties(c.Reference(), ps, &cr)
+	if err != nil {
+		return nil, err
+	}
+	return cr.Host, nil
+}

find/finder.go

@@ -313,10 +313,16 @@ func (f *Finder) HostSystemList(path ...string) ([]*govmomi.HostSystem, error) {
 	var hss []*govmomi.HostSystem
 	for _, e := range es {
 		switch o := e.Object.(type) {
-		case mo.HostSystem:
-			hs := govmomi.NewHostSystem(f.Client, o.Reference())
+		case mo.ComputeResource:
+			cr := govmomi.NewComputeResource(f.Client, o.Reference())
+			hosts, err := cr.Hosts()
+			if err != nil {
+				return nil, err
+			}
+			hs := govmomi.NewHostSystem(f.Client, hosts[0])
 			hss = append(hss, hs)
 		}
+
 	}
 
 	return hss, nil