Add tests and fix error

Author: dveeden

server/stmt.go

@@ -101,7 +101,7 @@ func (c *Conn) handleStmtExecute(data []byte) (*Result, error) {
 
 	s, ok := c.stmts[id]
 	if !ok {
-		return nil, NewDefaultError(ER_UNKNOWN_STMT_HANDLER,
+		return nil, NewDefaultError(ER_UNKNOWN_STMT_HANDLER, 5,
 			strconv.FormatUint(uint64(id), 10), "stmt_execute")
 	}
 
@@ -339,7 +339,7 @@ func (c *Conn) handleStmtReset(data []byte) (*Result, error) {
 
 	s, ok := c.stmts[id]
 	if !ok {
-		return nil, NewDefaultError(ER_UNKNOWN_STMT_HANDLER,
+		return nil, NewDefaultError(ER_UNKNOWN_STMT_HANDLER, 5,
 			strconv.FormatUint(uint64(id), 10), "stmt_reset")
 	}
 

server/stmt_test.go

@@ -0,0 +1,48 @@
+package server
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestHandleStmtExecute(t *testing.T) {
+	c := Conn{}
+	c.stmts = map[uint32]*Stmt{
+		1: &Stmt{},
+	}
+	testcases := []struct {
+		data    []byte
+		errtext string
+	}{
+		{
+			[]byte{0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0, 0x0},
+			"ERROR 1243 (HY000): Unknown prepared statement handler (0) given to stmt_execute",
+		},
+		{
+			[]byte{0x1, 0x0, 0x0, 0x0, 0xff, 0x0, 0x0, 0x0, 0x0, 0x0},
+			"ERROR 1105 (HY000): unsupported flags 0xff",
+		},
+		{
+			[]byte{0x1, 0x0, 0x0, 0x0, 0x01, 0x0, 0x0, 0x0, 0x0, 0x0},
+			"ERROR 1105 (HY000): unsupported flag CURSOR_TYPE_READ_ONLY",
+		},
+		{
+			[]byte{0x1, 0x0, 0x0, 0x0, 0x02, 0x0, 0x0, 0x0, 0x0, 0x0},
+			"ERROR 1105 (HY000): unsupported flag CURSOR_TYPE_FOR_UPDATE",
+		},
+		{
+			[]byte{0x1, 0x0, 0x0, 0x0, 0x04, 0x0, 0x0, 0x0, 0x0, 0x0},
+			"ERROR 1105 (HY000): unsupported flag CURSOR_TYPE_SCROLLABLE",
+		},
+	}
+
+	for _, tc := range testcases {
+		_, err := c.handleStmtExecute(tc.data)
+		if tc.errtext == "" {
+			require.NoError(t, err)
+		} else {
+			require.ErrorContains(t, err, tc.errtext)
+		}
+	}
+}