options_test.go

package wire

import (
	"context"
	"testing"

	"github.com/jackc/pgx/v5/pgtype"
	"github.com/neilotoole/slogt"
	"github.com/stretchr/testify/assert"
)

func TestParseParameters(t *testing.T) {
	type test struct {
		query      string
		parameters []uint32
	}

	tests := map[string]test{
		"positional": {
			query:      "SELECT * FROM users WHERE id = $1 AND age > $2",
			parameters: []uint32{0, 0},
		},
		"unpositional": {
			query:      "SELECT * FROM users WHERE id = ? AND age > ?",
			parameters: []uint32{0, 0},
		},
	}

	for name, test := range tests {
		t.Run(name, func(t *testing.T) {
			parameters := ParseParameters(test.query)
			assert.Equal(t, test.parameters, parameters)
		})
	}
}

func TestNilSessionHandler(t *testing.T) {
	srv, err := NewServer(nil, Logger(slogt.New(t)))
	assert.NoError(t, err)
	assert.NotNil(t, srv)

	bg := context.Background()
	ctx, err := srv.Session(bg)
	assert.NoError(t, err)
	assert.Equal(t, bg, ctx)
}

func TestSessionHandler(t *testing.T) {
	t.Parallel()

	type test []OptionFn

	type key string
	mock := key("key")
	value := "Super Secret Session ID"

	tests := map[string]test{
		"single": {
			SessionMiddleware(func(ctx context.Context) (context.Context, error) {
				return context.WithValue(ctx, mock, value), nil
			}),
		},
		"nested": {
			SessionMiddleware(func(ctx context.Context) (context.Context, error) {
				return ctx, nil
			}),
			SessionMiddleware(func(ctx context.Context) (context.Context, error) {
				return context.WithValue(ctx, mock, value), nil
			}),
		},
	}

	for name, test := range tests {
		t.Run(name, func(t *testing.T) {
			test = append(test, Logger(slogt.New(t)))
			srv, err := NewServer(nil, test...)
			assert.NoError(t, err)
			assert.NotNil(t, srv)

			ctx, err := srv.Session(context.Background())
			assert.NoError(t, err)
			assert.NotNil(t, ctx)

			result := ctx.Value(mock)
			assert.Equal(t, value, result)
		})
	}
}

func TestExtendTypes(t *testing.T) {
	extensionCalled := false

	srv, err := NewServer(nil,
		Logger(slogt.New(t)),
		ExtendTypes(func(typeMap *pgtype.Map) {
			extensionCalled = true
		}),
	)
	assert.NoError(t, err)
	assert.NotNil(t, srv)
	assert.NotNil(t, srv.typeExtension)
	assert.False(t, extensionCalled, "Extension should not be called during server creation")
}