| 421 | } |
| 422 | |
| 423 | func (m mockServerSentWriter) Flush() { |
| 424 | b := m.buffer.Bytes() |
| 425 | _, err := m.serverConn.Write(b) |
| 426 | require.NoError(m.testContext, err) |
| 427 | m.buffer.Reset() |
| 428 | |
| 429 | // Must close server connection to indicate EOF for any reads from the |
| 430 | // client connection; otherwise reads block forever. This is a testing |
| 431 | // limitation compared to the one-way websockets, since we have no way to |
| 432 | // frame the data and auto-indicate EOF for each message |
| 433 | err = m.serverConn.Close() |
| 434 | require.NoError(m.testContext, err) |
| 435 | } |
| 436 | |
| 437 | func (m mockServerSentWriter) Header() http.Header { |
| 438 | return m.serverRecorder.Header() |