Simplify test for client.AddCapability()

Signed-off-by: Zygmunt Krynicki <zygmunt.krynicki@canonical.com>
snapcore · Dec 3, 2015 · db9343c · db9343c
1 parent fe8ddc2
commit db9343c
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/client/client_test.go b/client/client_test.go
@@ -171,10 +171,9 @@ func (cs *clientSuite) TestClientAddCapability(c *check.C) {
 	}
 	err := cs.cli.AddCapability(cap)
 	c.Check(err, check.IsNil)
-	b, err := ioutil.ReadAll(cs.req.Body)
-	c.Check(err, check.IsNil)
 	var body map[string]interface{}
-	err = json.Unmarshal(b, &body)
+	decoder := json.NewDecoder(cs.req.Body)
+	err = decoder.Decode(&body)
 	c.Check(err, check.IsNil)
 	c.Check(body, check.DeepEquals, map[string]interface{}{
 		"name":  "n",