tools/functional-tester/etcd-tester: report agent status

tools/functional-tester/etcd-tester/cluster.go

@@ -16,6 +16,7 @@ package main
 
 import (
 	"fmt"
+	"log"
 	"math/rand"
 	"net"
 	"strings"
@@ -37,6 +38,10 @@ type cluster struct {
 	ClientURLs []string
 }
 
+type ClusterStatus struct {
+	AgentStatuses map[string]client.Status
+}
+
 // newCluster starts and returns a new cluster. The caller should call Terminate when finished, to shut it down.
 func newCluster(agentEndpoints []string, datadir string) (*cluster, error) {
 	c := &cluster{
@@ -160,6 +165,24 @@ func (c *cluster) Terminate() {
 	}
 }
 
+func (c *cluster) Status() ClusterStatus {
+	cs := ClusterStatus{
+		AgentStatuses: make(map[string]client.Status),
+	}
+
+	for i, a := range c.Agents {
+		s, err := a.Status()
+		// TODO: add a.Desc() as a key of the map
+		desc := c.agentEndpoints[i]
+		if err != nil {
+			cs.AgentStatuses[desc] = client.Status{State: "unknown"}
+			log.Printf("etcd-tester: failed to get the status of agent [%s]", desc)
+		}
+		cs.AgentStatuses[desc] = s
+	}
+	return cs
+}
+
 // setHealthKey sets health key on all given urls.
 func setHealthKey(us []string) error {
 	for _, u := range us {