func getUpdateStreamError(stream vtgateconn.UpdateStreamReader, err error) error {
if err != nil {
return err
}
for {
_, _, err := stream.Recv()
switch err {
case nil:
// keep going
case io.EOF:
return nil
default:
return err
}
}
}
func testEchoUpdateStream(t *testing.T, conn *vtgateconn.VTGateConn) {
var stream vtgateconn.UpdateStreamReader
var err error
ctx := callerid.NewContext(context.Background(), callerID, nil)
stream, err = conn.UpdateStream(ctx, echoPrefix+query, keyRanges[0], tabletType, 0, eventToken)
if err != nil {
t.Fatal(err)
}
se, _, err := stream.Recv()
if err != nil {
t.Fatal(err)
}
if se.EventToken.Position != updateStreamEcho {
t.Errorf("UpdateStream(0) = %v, want %v", se.EventToken.Position, updateStreamEcho)
}
}