Fix tmsp-cli to conform to go-wire new TypeByte behavior; Rename tmsp_cli to tmsp-cli

This commit is contained in:
Jae Kwon 2015-12-21 16:18:41 -08:00
parent 2d9abb06ca
commit 844c4a519d
2 changed files with 9 additions and 5 deletions

View File

@ -3,10 +3,12 @@ package main
import (
"bufio"
"encoding/hex"
"errors"
"fmt"
"io"
"net"
"os"
"reflect"
"strings"
. "github.com/tendermint/go-common"
@ -241,13 +243,13 @@ func makeRequest(conn net.Conn, req types.Request) (types.Response, error) {
var err error
// Write desired request
wire.WriteBinaryLengthPrefixed(req, conn, &n, &err)
wire.WriteBinaryLengthPrefixed(struct{ types.Request }{req}, conn, &n, &err)
if err != nil {
return nil, err
}
// Write flush request
wire.WriteBinaryLengthPrefixed(types.RequestFlush{}, conn, &n, &err)
wire.WriteBinaryLengthPrefixed(struct{ types.Request }{types.RequestFlush{}}, conn, &n, &err)
if err != nil {
return nil, err
}
@ -260,11 +262,14 @@ func makeRequest(conn net.Conn, req types.Request) (types.Response, error) {
}
// Read flush response
var resFlush types.ResponseFlush
var resFlush types.Response
wire.ReadBinaryPtrLengthPrefixed(&resFlush, conn, 0, &n, &err)
if err != nil {
return nil, err
}
if _, ok := resFlush.(types.ResponseFlush); !ok {
return nil, errors.New(Fmt("Expected types.ResponseFlush but got %v instead", reflect.TypeOf(resFlush)))
}
return res, nil
}

View File

@ -35,8 +35,7 @@ func TestStream(t *testing.T) {
var n int
var err error
var res types.Response
wire.ReadVarint(conn, &n, &err) // ignore
wire.ReadBinaryPtr(&res, conn, 0, &n, &err)
wire.ReadBinaryPtrLengthPrefixed(&res, conn, 0, &n, &err)
if err != nil {
Exit(err.Error())
}