Add output for ensure-availability command.

state/api/client.go

 // Copyright 2013 Canonical Ltd.
 // Licensed under the AGPLv3, see LICENCE file for details.
 
 package api
 
 import (
         "crypto/tls"
         "encoding/json"
         "fmt"
         "io"
@@ skipping 711 matching lines @@
 // APIHostPorts returns a slice of instance.HostPort for each API server.
 func (c *Client) APIHostPorts() ([][]instance.HostPort, error) {
         var result params.APIHostPortsResult
         if err := c.call("APIHostPorts", nil, &result); err != nil {
                 return nil, err
         }
         return result.Servers, nil
 }
 
 // EnsureAvailability ensures the availability of Juju state servers.
-func (c *Client) EnsureAvailability(numStateServers int, cons constraints.Value, series string) error {
+func (c *Client) EnsureAvailability(numStateServers int,
+        cons constraints.Value,
+        series string) (params.EnsureAvailabilityResult, error) {
+        var result params.EnsureAvailabilityResult
         args := params.EnsureAvailability{
                 NumStateServers: numStateServers,
                 Constraints:     cons,
                 Series:          series,
         }
-        return c.call("EnsureAvailability", args, nil)
+        if err := c.call("EnsureAvailability", args, &result); err != nil {
+                return params.EnsureAvailabilityResult{}, err
+        }
+        return result, nil
 }
 
 // AgentVersion reports the version number of the api server.
 func (c *Client) AgentVersion() (version.Number, error) {
         var result params.AgentVersionResult
         if err := c.call("AgentVersion", nil, &result); err != nil {
                 return version.Number{}, err
         }
         return result.Version, nil
 }
@@ skipping 100 matching lines @@
         var errResult params.ErrorResult
         err = json.Unmarshal(line, &errResult)
         if err != nil {
                 return nil, fmt.Errorf("unable to unmarshal initial response: %v", err)
         }
         if errResult.Error != nil {
                 return nil, errResult.Error
         }
         return connection, nil
 }