add test

Author: fernandez14

extra/pgotel/pgotel.go

@@ -126,7 +126,7 @@ func (h *TracingHook) AfterQuery(ctx context.Context, evt *pg.QueryEvent) error
 			span.RecordError(evt.Err)
 			span.SetStatus(codes.Error, evt.Err.Error())
 		}
-	} else if evt.Result != nil && (reflect.ValueOf(evt.Result).Kind() != reflect.Ptr || !reflect.ValueOf(evt.Result).IsNil()) {
+	} else if hasResults(evt) {
 		numRow := evt.Result.RowsAffected()
 		if numRow == 0 {
 			numRow = evt.Result.RowsReturned()
@@ -139,6 +139,10 @@ func (h *TracingHook) AfterQuery(ctx context.Context, evt *pg.QueryEvent) error
 	return nil
 }
 
+func hasResults(evt *pg.QueryEvent) bool {
+	return evt.Result != nil && (reflect.ValueOf(evt.Result).Kind() != reflect.Ptr || !reflect.ValueOf(evt.Result).IsNil())
+}
+
 func funcFileLine(pkg string) (string, string, int) {
 	const depth = 16
 	var pcs [depth]uintptr

extra/pgotel/pgotel_test.go

@@ -0,0 +1,62 @@
+package pgotel
+
+import (
+	"testing"
+
+	"github.com/go-pg/pg/v10"
+	"github.com/go-pg/pg/v10/orm"
+)
+
+// mockResult is a mock implementation of the Result interface
+type mockResult struct {
+	model        orm.Model
+	rowsAffected int
+	rowsReturned int
+}
+
+func (m mockResult) Model() orm.Model {
+	return m.model
+}
+
+func (m mockResult) RowsAffected() int {
+	return m.rowsAffected
+}
+
+func (m mockResult) RowsReturned() int {
+	return m.rowsReturned
+}
+
+func TestHasResults(t *testing.T) {
+	// Define test cases
+	testCases := []struct {
+		name     string
+		event    *pg.QueryEvent
+		expected bool
+	}{
+		{
+			name:     "Nil Result",
+			event:    &pg.QueryEvent{Result: nil},
+			expected: false,
+		},
+		{
+			name:     "Nil Pointer Result",
+			event:    &pg.QueryEvent{Result: (*mockResult)(nil)},
+			expected: false,
+		},
+		{
+			name:     "Non-Nil Result",
+			event:    &pg.QueryEvent{Result: mockResult{rowsAffected: 1, rowsReturned: 1}},
+			expected: true,
+		},
+	}
+
+	// Run test cases
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			result := hasResults(tc.event)
+			if result != tc.expected {
+				t.Errorf("hasResults(%v) = %v, want %v", tc.event, result, tc.expected)
+			}
+		})
+	}
+}