Function TestClientFromContext

context_test.go:11–31

(t *testing.T)

)

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

	ctx := context.Background()
	client := &Client[pgx.Tx]{}
	ctx = withClient(ctx, client)

	require.Equal(t, client, ClientFromContext[pgx.Tx](ctx))

	result, err := ClientFromContextSafely[pgx.Tx](ctx)
	require.NoError(t, err)
	require.Equal(t, client, result)

	require.PanicsWithError(t, errClientNotInContext.Error(), func() {
		ClientFromContext[pgx.Tx](context.Background())
	})

	result, err = ClientFromContextSafely[pgx.Tx](context.Background())
	require.ErrorIs(t, err, errClientNotInContext)
	require.Nil(t, result)
}