Skip to content

Commit 803dd56

Browse files
committed
Revert "re-implement TestReadPacketFail"
This reverts commit e849989.
1 parent 8c82535 commit 803dd56

File tree

2 files changed

+42
-51
lines changed

2 files changed

+42
-51
lines changed

packets.go

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -38,12 +38,9 @@ func (mc *mysqlConn) readPacket(ctx context.Context) (*packet, error) {
3838
// read packet header
3939
err := mc.readFull(ctx, mc.data[:4])
4040
if err != nil {
41-
if err == context.Canceled || err == context.DeadlineExceeded {
42-
return nil, err
43-
}
4441
mc.cfg.Logger.Print(err)
4542
mc.closeContext(ctx)
46-
return nil, ErrInvalidConn
43+
return nil, err
4744
}
4845

4946
// packet length [24 bit]
@@ -78,9 +75,6 @@ func (mc *mysqlConn) readPacket(ctx context.Context) (*packet, error) {
7875
data := make([]byte, pktLen)
7976
err = mc.readFull(ctx, data)
8077
if err != nil {
81-
if err == context.Canceled || err == context.DeadlineExceeded {
82-
return nil, err
83-
}
8478
mc.cfg.Logger.Print(err)
8579
mc.closeContext(ctx)
8680
return nil, ErrInvalidConn

packets_test.go

Lines changed: 41 additions & 44 deletions
Original file line numberDiff line numberDiff line change
@@ -228,50 +228,47 @@ func TestReadPacketSplit(t *testing.T) {
228228
})
229229
}
230230

231-
func TestReadPacketFail(t *testing.T) {
232-
t.Run("illegal empty (stand-alone) packet", func(t *testing.T) {
233-
conn, mc := newRWMockConn(t, 0)
234-
go func() {
235-
conn.Write([]byte{0x00, 0x00, 0x00, 0x00})
236-
}()
237-
go func() {
238-
io.Copy(io.Discard, conn)
239-
}()
240-
241-
_, err := mc.readPacket(context.Background())
242-
if err != ErrInvalidConn {
243-
t.Errorf("expected ErrInvalidConn, got %v", err)
244-
}
245-
})
246-
247-
t.Run("fail to read header", func(t *testing.T) {
248-
conn, mc := newRWMockConn(t, 0)
249-
go func() {
250-
conn.Close()
251-
}()
252-
253-
_, err := mc.readPacket(context.Background())
254-
if err != ErrInvalidConn {
255-
t.Errorf("expected ErrInvalidConn, got %v", err)
256-
}
257-
})
258-
259-
t.Run("fail to read body", func(t *testing.T) {
260-
conn, mc := newRWMockConn(t, 0)
261-
go func() {
262-
conn.Write([]byte{0x01, 0x00, 0x00, 0x00})
263-
conn.Close()
264-
}()
265-
go func() {
266-
io.Copy(io.Discard, conn)
267-
}()
268-
269-
_, err := mc.readPacket(context.Background())
270-
if err != ErrInvalidConn {
271-
t.Errorf("expected ErrInvalidConn, got %v", err)
272-
}
273-
})
274-
}
231+
// func TestReadPacketFail(t *testing.T) {
232+
// conn := new(mockConn)
233+
// mc := &mysqlConn{
234+
// buf: newBuffer(conn),
235+
// closech: make(chan struct{}),
236+
// cfg: NewConfig(),
237+
// }
238+
239+
// // illegal empty (stand-alone) packet
240+
// conn.data = []byte{0x00, 0x00, 0x00, 0x00}
241+
// conn.maxReads = 1
242+
// _, err := mc.readPacket()
243+
// if err != ErrInvalidConn {
244+
// t.Errorf("expected ErrInvalidConn, got %v", err)
245+
// }
246+
247+
// // reset
248+
// conn.reads = 0
249+
// mc.sequence = 0
250+
// mc.buf = newBuffer(conn)
251+
252+
// // fail to read header
253+
// conn.closed = true
254+
// _, err = mc.readPacket()
255+
// if err != ErrInvalidConn {
256+
// t.Errorf("expected ErrInvalidConn, got %v", err)
257+
// }
258+
259+
// // reset
260+
// conn.closed = false
261+
// conn.reads = 0
262+
// mc.sequence = 0
263+
// mc.buf = newBuffer(conn)
264+
265+
// // fail to read body
266+
// conn.maxReads = 1
267+
// _, err = mc.readPacket()
268+
// if err != ErrInvalidConn {
269+
// t.Errorf("expected ErrInvalidConn, got %v", err)
270+
// }
271+
// }
275272

276273
// https://github.com/go-sql-driver/mysql/pull/801
277274
// not-NUL terminated plugin_name in init packet

0 commit comments

Comments
 (0)