Fix API test

orchardup · Feb 17, 2014 · 1ca341c · 1ca341c
1 parent 94a6972
commit 1ca341c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/api/api_test.go b/api/api_test.go
@@ -50,13 +50,17 @@ func TestCreateHost(t *testing.T) {
 		}
 
 		body, _ := ioutil.ReadAll(r.Body)
-		var data map[string]string
+		var data map[string]interface{}
 		json.Unmarshal(body, &data)
 
 		if data["name"] != "newhost" {
 			t.Errorf("expected 'newhost', got '%s'", data["name"])
 		}
 
+		if int(data["size"].(float64)) != 512 {
+			t.Errorf("expected 512, got %#v", data["size"])
+		}
+
 		w.WriteHeader(201)
 		fmt.Fprintln(w, `{
       "id": "14dff6d8-3b9a-41be-9ffd-d0d054a17492",
@@ -67,7 +71,7 @@ func TestCreateHost(t *testing.T) {
 
 	client := HTTPClient{ts.URL, "dummy_token"}
 
-	host, err := client.CreateHost("newhost")
+	host, err := client.CreateHost("newhost", 512)
 	if err != nil {
 		t.Error(err)
 	}