diff --git a/client/client_test.go b/client/client_test.go
index a317a40..04f82bc 100644
--- a/client/client_test.go
+++ b/client/client_test.go
@@ -13,11 +13,21 @@ func TestClient(t *testing.T) {
 	cli, err := NewWheatClient("127.0.0.1:5891", middle.WithUnaryColonyClient)
 	require.NoError(t, err)
 	ctx := context.Background()
-	_, err = cli.Set(ctx, &proto.SetRequest{
-		Key: &proto.BaseKey{
-			Key: "apple",
-		},
+	bKey := &proto.BaseKey{
+		Key: "apple",
+	}
+	resp, err := cli.Set(ctx, &proto.SetRequest{
+		Key: bKey,
 		Val: "yyyy",
 	})
+
 	require.NoError(t, err)
+	require.Equal(t, resp.Result, "yyyy")
+
+	getResp, err := cli.Get(ctx, &proto.GetRequest{
+		Key: bKey,
+	})
+	require.NoError(t, err)
+	require.Equal(t, getResp.Result, "yyyy")
+
 }