test: add ut

cloudwego · Nov 29, 2023 · d3a9273 · d3a9273
1 parent a95b095
commit d3a9273
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pkg/protocol/http1/server_test.go b/pkg/protocol/http1/server_test.go
@@ -443,3 +443,13 @@ type mockErrorWriter struct {
 func (errorWriter *mockErrorWriter) Flush() error {
 	return errors.New("error")
 }
+
+func TestShouldRecordInTraceError(t *testing.T) {
+	assert.False(t, shouldRecordInTraceError(nil))
+	assert.False(t, shouldRecordInTraceError(errHijacked))
+	assert.False(t, shouldRecordInTraceError(errIdleTimeout))
+	assert.False(t, shouldRecordInTraceError(errShortConnection))
+
+	assert.True(t, shouldRecordInTraceError(errTimeout))
+	assert.True(t, shouldRecordInTraceError(errors.New("foo error")))
+}