| 141 | } |
| 142 | |
| 143 | func TestPingMarkBadConnection(t *testing.T) { |
| 144 | nc := badConnection{err: errors.New("boom")} |
| 145 | mc := &mysqlConn{ |
| 146 | netConn: nc, |
| 147 | buf: newBuffer(), |
| 148 | maxAllowedPacket: defaultMaxAllowedPacket, |
| 149 | closech: make(chan struct{}), |
| 150 | cfg: NewConfig(), |
| 151 | } |
| 152 | |
| 153 | err := mc.Ping(context.Background()) |
| 154 | |
| 155 | if err != driver.ErrBadConn { |
| 156 | t.Errorf("expected driver.ErrBadConn, got %#v", err) |
| 157 | } |
| 158 | } |
| 159 | |
| 160 | func TestPingErrInvalidConn(t *testing.T) { |
| 161 | nc := badConnection{err: errors.New("failed to write"), n: 10} |