Adding test for nonHijacker

Signed-off-by: rehanpfmr <111350825+rehanpfmr@users.noreply.github.com>
open-telemetry · Aug 27, 2024 · 2179b86 · 2179b86
1 parent 76a7bb9
commit 2179b86
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/instrumentation/github.com/gorilla/mux/otelmux/test/mux_test.go b/instrumentation/github.com/gorilla/mux/otelmux/test/mux_test.go
@@ -66,6 +66,10 @@ func (m *mockHijacker) Hijack() (net.Conn, *bufio.ReadWriter, error) {
     return conn, bufio.NewReadWriter(bufio.NewReader(rw), bufio.NewWriter(rw)), nil
 }
 
+type mockNonHijacker struct {
+        http.ResponseWriter
+}
+
 func TestRecordingResponseWriterHijack(t *testing.T) {
     mockWriter := &mockHijacker{httptest.NewRecorder()}
     recWriter := &recordingResponseWriter{writer: mockWriter}