Improve unit tests around Hasresultset and WriteValue

mysql/result.go

@@ -36,6 +36,9 @@ func (r *Result) Close() {
 }
 
 func (r *Result) HasResultset() bool {
+if r == nil {
+return false
+}
 if r.Resultset != nil && len(r.Resultset.Fields) > 0 {
 return true
 }

mysql/result_test.go

@@ -17,3 +17,17 @@ func TestHasResultset_true(t *testing.T) {
 b := r.HasResultset()
 require.Equal(t, true, b)
 }
+
+// this shouldn't happen after d02e79a, but test just in case
+func TestHasResultset_nilset(t *testing.T) {
+r := NewResultReserveResultset(0)
+r.Resultset = nil
+b := r.HasResultset()
+require.Equal(t, false, b)
+}
+
+func TestHasResultset_nil(t *testing.T) {
+var r *Result
+b := r.HasResultset()
+require.Equal(t, false, b)
+}

server/resp_test.go

@@ -14,10 +14,9 @@ func TestConnWriteOK(t *testing.T) {
 clientConn := &mockconn.MockConn{}
 conn := &Conn{Conn: packet.NewConn(clientConn)}
 
-result := &mysql.Result{
-AffectedRows: 1,
-InsertId: 2,
-}
+result := mysql.NewResultReserveResultset(0)
+result.AffectedRows = 1
+result.InsertId = 2
 
 // write ok with insertid and affectedrows set
 err := conn.writeOK(result)
@@ -229,3 +228,36 @@ func TestConnWriteFieldValues(t *testing.T) {
 // EOF
 require.Equal(t, []byte{1, 0, 0, 4, mysql.EOF_HEADER}, clientConn.WriteBuffered[43:])
 }
+
+func TestWriteValue(t *testing.T) {
+clientConn := &mockconn.MockConn{MultiWrite: true}
+conn := &Conn{Conn: packet.NewConn(clientConn)}
+
+// simple OK
+err := conn.WriteValue(mysql.NewResultReserveResultset(0))
+require.NoError(t, err)
+expected := []byte{3, 0, 0, 0, mysql.OK_HEADER, 0, 0}
+require.Equal(t, expected, clientConn.WriteBuffered)
+
+// reset write buffer
+clientConn.WriteBuffered = []byte{}
+
+// resultset with no rows
+rs := mysql.NewResultReserveResultset(1)
+rs.Fields = []*mysql.Field{{Name: []byte("a")}}
+err = conn.WriteValue(rs)
+require.NoError(t, err)
+expected = []byte{1, 0, 0, 1, mysql.MORE_DATE_HEADER}
+require.Equal(t, expected, clientConn.WriteBuffered[:5])
+
+// reset write buffer
+clientConn.WriteBuffered = []byte{}
+
+// resultset with rows
+rs.Fields = []*mysql.Field{{Name: []byte("a")}}
+rs.RowDatas = []mysql.RowData{[]byte{1, 2, 3}}
+err = conn.WriteValue(rs)
+require.NoError(t, err)
+expected = []byte{1, 0, 0, 5, mysql.MORE_DATE_HEADER}
+require.Equal(t, expected, clientConn.WriteBuffered[:5])
+}

server/server_test.go

@@ -244,30 +244,16 @@ func (h *testHandler) handleQuery(query string, binary bool) (*mysql.Result, err
 if err != nil {
 return nil, errors.Trace(err)
 } else {
-return &mysql.Result{
-Status: 0,
-Warnings: 0,
-InsertId: 0,
-AffectedRows: 0,
-Resultset: r,
-}, nil
+return mysql.NewResult(r), nil
 }
 case "insert":
-return &mysql.Result{
-Status: 0,
-Warnings: 0,
-InsertId: 1,
-AffectedRows: 0,
-Resultset: nil,
-}, nil
+res := mysql.NewResultReserveResultset(0)
+res.InsertId = 1
+return res, nil
 case "delete", "update", "replace":
-return &mysql.Result{
-Status: 0,
-Warnings: 0,
-InsertId: 0,
-AffectedRows: 1,
-Resultset: nil,
-}, nil
+res := mysql.NewResultReserveResultset(0)
+res.AffectedRows = 1
+return res, nil
 default:
 return nil, fmt.Errorf("invalid query %s", query)
 }