client.Populate assign client to proxy

CHANGELOG.md

@@ -3,6 +3,7 @@
 * Store all the executable `main` packages in `cmd` folder. (#335, @miry)
 * Extract common test helpers to own files. (#336, @miry)
 * Client: Allow HTTPS endpoints. (#338, @chen-anders)
+* client.Populate assign client to proxy. (#291, @hellodudu)
 
 # [2.2.0]
 

api_test.go

@@ -369,6 +369,36 @@ func TestPopulateProxyWithBadDataShouldReturnError(t *testing.T) {
 	})
 }
 
+func TestPopulateAddToxic(t *testing.T) {
+	WithServer(t, func(addr string) {
+		testProxies, err := client.Populate([]tclient.Proxy{
+			{
+				Name:     "one",
+				Listen:   "localhost:7070",
+				Upstream: "localhost:7171",
+				Enabled:  true,
+			},
+		})
+
+		if err != nil {
+			t.Fatal("Unable to populate:", err)
+		}
+
+		if len(testProxies) != 1 {
+			t.Fatalf("Wrong number of proxies returned: %d != %d", len(testProxies), 1)
+		}
+
+		if testProxies[0].Name != "one" {
+			t.Fatalf("Wrong proxy name returned: %s != one", testProxies[0].Name)
+		}
+
+		_, err = testProxies[0].AddToxic("", "latency", "downstream", 1, nil)
+		if err != nil {
+			t.Fatal("Failed to AddToxic.")
+		}
+	})
+}
+
 func TestListingProxies(t *testing.T) {
 	WithServer(t, func(addr string) {
 		_, err := client.CreateProxy("mysql_master", "localhost:3310", "localhost:20001")

client/client.go

@@ -161,6 +161,13 @@ func (client *Client) Populate(config []Proxy) ([]*Proxy, error) {
 
 	resp.Body = ioutil.NopCloser(&body)
 	err = checkError(resp, http.StatusCreated, "Populate")
+	if err != nil {
+		return proxies.Proxies, err
+	}
+
+	for _, proxy := range proxies.Proxies {
+		proxy.client = client
+	}
 	return proxies.Proxies, err
 }