this is weird

Signed-off-by: Jeff Carr <[email protected]>

1 files changed, 3 insertions, 2 deletions

diff --git a/sampleData.go b/sampleData.go
index eaf4cb7..806e1f9 100644
--- a/sampleData.go
+++ b/sampleData.go
@@ -25,7 +25,7 @@ func CreateSampleDroplet(hostname string) *Droplet {
 	return d
 }
 
-func CreateSampleHypervisor(hostname string) *Hypervisor {
+func CreateSampleHypervisor(hostname string, mem int) *Hypervisor {
 	// Generate a new UUID
 	id := uuid.New()
 	h := &Hypervisor{
@@ -35,6 +35,7 @@ func CreateSampleHypervisor(hostname string) *Hypervisor {
 		Memory: 256,
 		Comment:  "this is a fake hypervisor",
 	}
+	h.SetMemory(mem * 32)
 	return h
 }
 
@@ -53,7 +54,7 @@ func CreateSampleCluster(total int) *Cluster {
 
 	for i := 0; i < 3; i++ {
 		hostname := fmt.Sprintf("farm%d", i)
-		h := CreateSampleHypervisor(hostname)
+		h := CreateSampleHypervisor(hostname, i + 1)
 		h.Comment = fmt.Sprintf("Sample hypervisor %d", i)
 
 		c.Hypervisors = append(c.Hypervisors, h)