compiles

Signed-off-by: Jeff Carr <jcarr@wit.com>

droplet.proto

@@ -31,7 +31,8 @@ message Droplet {
 		int64	size    	= 2;
 	}
 
-        message StorageInfo {
-                int64 capacity = 1;  // Stores the storage capacity in bytes.
-        }
+}
+
+message StorageInfo {
+        int64 capacity = 1;  // Stores the storage capacity in bytes.
 }

helpers.go

@@ -103,7 +103,17 @@ func (c *Cluster) AddDroplet(hostname string, cpus int, mem int) *Droplet {
 		d.Cpus = 1
 	}
 	d.Memory = SetGB(mem * 32)
-	d.Testsi.Capacity = SetGB(mem * 32)
+	if d.Testsi == nil {
+		fmt.Println("Testsi is NIL")
+		fmt.Println("Testsi is NIL")
+		fmt.Println("Testsi is NIL")
+		fmt.Println("Testsi is NIL")
+		var newInfo StorageInfo
+		newInfo = StorageInfo{Capacity: 64}
+		d.Testsi = &newInfo
+	} else {
+		d.Testsi.Capacity = SetGB(mem * 32)
+	}
 	// d.Testsi = StorageInfo{Capacity: 64}
 	c.Droplets = append(c.Droplets, d)
 	return d

storageinfo.go

@@ -6,10 +6,6 @@ import (
 	"strconv"
 )
 
-type StorageInfo struct {
-	Capacity int64
-}
-
 // MarshalJSON custom marshals the StorageInfo struct to JSON
 func (s StorageInfo) MarshalJSON() ([]byte, error) {
 	capacityStr := fmt.Sprintf("%d GB", s.Capacity)