payments/internal/storage/pools_test.go

package storage

import (
	"context"
	"encoding/json"
	"testing"

	"github.com/formancehq/go-libs/v3/bun/bunpaginate"
	"github.com/formancehq/go-libs/v3/logging"
	"github.com/formancehq/go-libs/v3/query"
	"github.com/formancehq/go-libs/v3/time"
	"github.com/formancehq/payments/internal/models"
	"github.com/formancehq/payments/pkg/events"
	"github.com/google/uuid"
	"github.com/pkg/errors"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

var (
	poolID1 = uuid.New()
	poolID2 = uuid.New()
	poolID3 = uuid.New()
)

func defaultPools() []models.Pool {
	defaultAccounts := defaultAccounts()
	return []models.Pool{
		{
			ID:           poolID1,
			Name:         "test1",
			CreatedAt:    now.Add(-60 * time.Minute).UTC().Time,
			Type:         models.POOL_TYPE_STATIC,
			PoolAccounts: []models.AccountID{defaultAccounts[0].ID, defaultAccounts[1].ID},
		},
		{
			ID:           poolID2,
			Name:         "test2",
			CreatedAt:    now.Add(-30 * time.Minute).UTC().Time,
			Type:         models.POOL_TYPE_STATIC,
			PoolAccounts: []models.AccountID{defaultAccounts[2].ID},
		},
		{
			ID:        poolID3,
			Name:      "test3",
			CreatedAt: now.Add(-55 * time.Minute).UTC().Time,
			Type:      models.POOL_TYPE_DYNAMIC,
			Query: map[string]any{
				"$match": map[string]any{
					"account_id": "test3",
				},
			},
		},
	}
}

func upsertPool(t *testing.T, ctx context.Context, storage Storage, pool models.Pool) {
	require.NoError(t, storage.PoolsUpsert(ctx, pool))
}

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

	ctx := logging.TestingContext()
	store := newStore(t)
	t.Cleanup(func() {
		store.Close()
	})

	upsertConnector(t, ctx, store, defaultConnector)
	upsertAccounts(t, ctx, store, defaultAccounts())
	upsertPool(t, ctx, store, defaultPools()[0])
	upsertPool(t, ctx, store, defaultPools()[1])

	t.Run("upsert with same name", func(t *testing.T) {
		poolID3 := uuid.New()
		p := models.Pool{
			ID:           poolID3,
			Name:         "test1",
			Type:         models.POOL_TYPE_STATIC,
			CreatedAt:    now.Add(-30 * time.Minute).UTC().Time,
			PoolAccounts: []models.AccountID{defaultAccounts()[2].ID},
		}

		err := store.PoolsUpsert(ctx, p)
		require.Error(t, err)
	})

	t.Run("upsert with same id", func(t *testing.T) {
		upsertPool(t, ctx, store, defaultPools()[1])

		actual, err := store.PoolsGet(ctx, defaultPools()[1].ID)
		require.NoError(t, err)
		require.Equal(t, defaultPools()[1], *actual)
	})

	t.Run("upsert with same id but more related accounts", func(t *testing.T) {
		p := defaultPools()[0]
		p.PoolAccounts = append(p.PoolAccounts, defaultAccounts()[2].ID)

		upsertPool(t, ctx, store, p)

		actual, err := store.PoolsGet(ctx, defaultPools()[0].ID)
		require.NoError(t, err)
		require.Equal(t, p, *actual)
	})

	t.Run("upsert but account does not exist", func(t *testing.T) {
		p := defaultPools()[0]
		p.PoolAccounts = append(p.PoolAccounts, models.AccountID{
			Reference:   "unknown",
			ConnectorID: defaultConnector.ID,
		})

		err := store.PoolsUpsert(ctx, p)
		require.Error(t, err)
	})

	t.Run("outbox event created for new pool", func(t *testing.T) {
		// Create a new pool for this test
		newPool := models.Pool{
			ID:           uuid.New(),
			Name:         "outbox-test-pool",
			CreatedAt:    now.Add(-10 * time.Minute).UTC().Time,
			PoolAccounts: []models.AccountID{defaultAccounts()[0].ID},
		}

		expectedKey := newPool.IdempotencyKey()

		require.NoError(t, store.PoolsUpsert(ctx, newPool))

		// Verify outbox event was created
		pendingEvents, err := store.OutboxEventsPollPending(ctx, 100)
		require.NoError(t, err)

		// Find our event
		var ourEvent *models.OutboxEvent
		for i := range pendingEvents {
			if pendingEvents[i].EventType == events.EventTypeSavedPool &&
				pendingEvents[i].EntityID == newPool.ID.String() &&
				pendingEvents[i].ID.EventIdempotencyKey == expectedKey {
				ourEvent = &pendingEvents[i]
				break
			}
		}
		require.NotNil(t, ourEvent, "expected outbox event for pool saved")

		// Verify event details
		assert.Equal(t, events.EventTypeSavedPool, ourEvent.EventType)
		assert.Equal(t, models.OUTBOX_STATUS_PENDING, ourEvent.Status)
		assert.Equal(t, newPool.ID.String(), ourEvent.EntityID)
		assert.Nil(t, ourEvent.ConnectorID) // Pools don't have connector ID
		assert.Equal(t, 0, ourEvent.RetryCount)
		assert.Equal(t, expectedKey, ourEvent.ID.EventIdempotencyKey)

		// Verify payload
		var payload map[string]interface{}
		err = json.Unmarshal(ourEvent.Payload, &payload)
		require.NoError(t, err)
		assert.Equal(t, newPool.ID.String(), payload["id"])
		assert.Equal(t, newPool.Name, payload["name"])
		assert.NotNil(t, payload["accountIDs"])
		assert.NotNil(t, payload["createdAt"])
	})

	t.Run("no outbox event for existing pool update", func(t *testing.T) {
		// Count events before
		eventsBefore, err := store.OutboxEventsPollPending(ctx, 1000)
		require.NoError(t, err)
		countBefore := len(eventsBefore)

		// Update existing pool (should not create event)
		existingPool := defaultPools()[1]
		existingPool.PoolAccounts = append(existingPool.PoolAccounts, defaultAccounts()[0].ID)
		require.NoError(t, store.PoolsUpsert(ctx, existingPool))

		// Verify no new outbox event was created
		eventsAfter, err := store.OutboxEventsPollPending(ctx, 1000)
		require.NoError(t, err)
		countAfter := len(eventsAfter)

		// Should have same number of events (no new pool saved event)
		assert.Equal(t, countBefore, countAfter, "updating existing pool should not create saved event")
	})
}

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

	ctx := logging.TestingContext()
	store := newStore(t)
	defer store.Close()

	upsertConnector(t, ctx, store, defaultConnector)
	upsertAccounts(t, ctx, store, defaultAccounts())
	upsertPool(t, ctx, store, defaultPools()[0])
	upsertPool(t, ctx, store, defaultPools()[1])
	upsertPool(t, ctx, store, defaultPools()[2])

	t.Run("update query of existing pool", func(t *testing.T) {
		err := store.PoolsUpdateQuery(ctx, defaultPools()[2], map[string]any{
			"$match": map[string]any{
				"account_id": "newtest",
			},
		})
		require.NoError(t, err)

		updated := defaultPools()[2]
		updated.Query = map[string]any{
			"$match": map[string]any{
				"account_id": "newtest",
			},
		}

		actual, err := store.PoolsGet(ctx, defaultPools()[2].ID)
		require.NoError(t, err)
		require.Equal(t, updated, *actual)
	})
}

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

	ctx := logging.TestingContext()
	store := newStore(t)
	defer store.Close()

	upsertConnector(t, ctx, store, defaultConnector)
	upsertAccounts(t, ctx, store, defaultAccounts())
	upsertPool(t, ctx, store, defaultPools()[0])
	upsertPool(t, ctx, store, defaultPools()[1])
	upsertPool(t, ctx, store, defaultPools()[2])

	t.Run("get existing pool", func(t *testing.T) {
		for _, p := range defaultPools() {
			actual, err := store.PoolsGet(ctx, p.ID)
			require.NoError(t, err)
			require.Equal(t, p, *actual)
		}
	})

	t.Run("get non-existing pool", func(t *testing.T) {
		p, err := store.PoolsGet(ctx, uuid.New())
		require.Error(t, err)
		require.ErrorIs(t, err, ErrNotFound)
		require.Nil(t, p)
	})
}

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

	ctx := logging.TestingContext()
	store := newStore(t)
	t.Cleanup(func() {
		store.Close()
	})

	upsertConnector(t, ctx, store, defaultConnector)
	upsertAccounts(t, ctx, store, defaultAccounts())
	upsertPool(t, ctx, store, defaultPools()[0])
	upsertPool(t, ctx, store, defaultPools()[1])
	upsertPool(t, ctx, store, defaultPools()[2])

	t.Run("delete unknown pool", func(t *testing.T) {
		deleted, err := store.PoolsDelete(ctx, uuid.New())
		require.NoError(t, err)
		require.False(t, deleted)
		for _, p := range defaultPools() {
			actual, err := store.PoolsGet(ctx, p.ID)
			require.NoError(t, err)
			require.Equal(t, p, *actual)
		}
	})

	t.Run("delete existing pool", func(t *testing.T) {
		deleted, err := store.PoolsDelete(ctx, defaultPools()[0].ID)
		require.NoError(t, err)
		require.True(t, deleted)

		_, err = store.PoolsGet(ctx, defaultPools()[0].ID)
		require.Error(t, err)
		require.ErrorIs(t, err, ErrNotFound)

		actual, err := store.PoolsGet(ctx, defaultPools()[1].ID)
		require.NoError(t, err)
		require.Equal(t, defaultPools()[1], *actual)
	})

	t.Run("outbox event created for pool deletion", func(t *testing.T) {
		// Create a new pool for this test
		deleteTestPool := models.Pool{
			ID:           uuid.New(),
			Name:         "delete-test-pool",
			CreatedAt:    now.Add(-5 * time.Minute).UTC().Time,
			PoolAccounts: []models.AccountID{defaultAccounts()[0].ID},
		}
		require.NoError(t, store.PoolsUpsert(ctx, deleteTestPool))

		// Delete the pool
		deleted, err := store.PoolsDelete(ctx, deleteTestPool.ID)
		require.NoError(t, err)
		require.True(t, deleted)

		// Verify outbox event was created
		pendingEvents, err := store.OutboxEventsPollPending(ctx, 100)
		require.NoError(t, err)

		// Find our event
		var ourEvent *models.OutboxEvent
		for i := range pendingEvents {
			if pendingEvents[i].EventType == events.EventTypeDeletePool &&
				pendingEvents[i].EntityID == deleteTestPool.ID.String() {
				ourEvent = &pendingEvents[i]
				break
			}
		}
		require.NotNil(t, ourEvent, "expected outbox event for pool deleted")

		// Verify event details
		assert.Equal(t, events.EventTypeDeletePool, ourEvent.EventType)
		assert.Equal(t, models.OUTBOX_STATUS_PENDING, ourEvent.Status)
		assert.Equal(t, deleteTestPool.ID.String(), ourEvent.EntityID)
		assert.Nil(t, ourEvent.ConnectorID) // Pools don't have connector ID
		assert.Equal(t, 0, ourEvent.RetryCount)
		assert.Equal(t, deleteTestPool.ID.String(), ourEvent.ID.EventIdempotencyKey)

		// Verify payload
		var payload map[string]interface{}
		err = json.Unmarshal(ourEvent.Payload, &payload)
		require.NoError(t, err)
		assert.Equal(t, deleteTestPool.ID.String(), payload["id"])
		assert.NotNil(t, payload["createdAt"])
	})

	t.Run("no outbox event for non-existent pool deletion", func(t *testing.T) {
		// Count events before
		eventsBefore, err := store.OutboxEventsPollPending(ctx, 1000)
		require.NoError(t, err)
		countBefore := len(eventsBefore)

		// Try to delete non-existent pool
		deleted, err := store.PoolsDelete(ctx, uuid.New())
		require.NoError(t, err)
		require.False(t, deleted)

		// Verify no new outbox event was created
		eventsAfter, err := store.OutboxEventsPollPending(ctx, 1000)
		require.NoError(t, err)
		countAfter := len(eventsAfter)

		// Should have same number of events (no new pool deleted event)
		assert.Equal(t, countBefore, countAfter, "deleting non-existent pool should not create deleted event")
	})
}

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

	ctx := logging.TestingContext()
	store := newStore(t)
	t.Cleanup(func() {
		store.Close()
	})

	upsertConnector(t, ctx, store, defaultConnector)
	upsertAccounts(t, ctx, store, defaultAccounts())
	upsertPool(t, ctx, store, defaultPools()[0])
	upsertPool(t, ctx, store, defaultPools()[1])

	t.Run("add account to unknown pool", func(t *testing.T) {
		err := store.PoolsAddAccount(ctx, uuid.New(), defaultAccounts()[0].ID)
		require.Error(t, err)
	})

	t.Run("add account to pool", func(t *testing.T) {
		require.NoError(t, store.PoolsAddAccount(ctx, defaultPools()[0].ID, defaultAccounts()[2].ID))

		p := defaultPools()[0]
		p.PoolAccounts = append(p.PoolAccounts, defaultAccounts()[2].ID)

		actual, err := store.PoolsGet(ctx, defaultPools()[0].ID)
		require.NoError(t, err)
		require.Equal(t, p, *actual)
	})

	t.Run("add account to pool but account does not exist", func(t *testing.T) {
		err := store.PoolsAddAccount(ctx, defaultPools()[0].ID, models.AccountID{
			Reference:   "unknown",
			ConnectorID: defaultConnector.ID,
		})
		require.Error(t, err)
	})

	t.Run("outbox event created when adding account to pool", func(t *testing.T) {
		// Create a new pool for this test
		testPool := models.Pool{
			ID:           uuid.New(),
			Name:         "add-account-test-pool",
			CreatedAt:    now.Add(-10 * time.Minute).UTC().Time,
			Type:         models.POOL_TYPE_STATIC,
			PoolAccounts: []models.AccountID{defaultAccounts()[0].ID},
		}
		require.NoError(t, store.PoolsUpsert(ctx, testPool))

		// Clear any existing events
		_, _ = store.OutboxEventsPollPending(ctx, 1000)

		// Add account to pool
		require.NoError(t, store.PoolsAddAccount(ctx, testPool.ID, defaultAccounts()[1].ID))

		// Expected pool state after adding account
		expectedPool := testPool
		expectedPool.PoolAccounts = append(expectedPool.PoolAccounts, defaultAccounts()[1].ID)
		expectedKey := expectedPool.IdempotencyKey()

		// Verify outbox event was created
		pendingEvents, err := store.OutboxEventsPollPending(ctx, 100)
		require.NoError(t, err)

		// Find our event
		var ourEvent *models.OutboxEvent
		for i := range pendingEvents {
			if pendingEvents[i].EventType == events.EventTypeSavedPool &&
				pendingEvents[i].EntityID == testPool.ID.String() &&
				pendingEvents[i].ID.EventIdempotencyKey == expectedKey {
				ourEvent = &pendingEvents[i]
				break
			}
		}
		require.NotNil(t, ourEvent, "expected outbox event for pool saved after adding account")

		// Verify event details
		assert.Equal(t, events.EventTypeSavedPool, ourEvent.EventType)
		assert.Equal(t, models.OUTBOX_STATUS_PENDING, ourEvent.Status)
		assert.Equal(t, testPool.ID.String(), ourEvent.EntityID)
		assert.Nil(t, ourEvent.ConnectorID) // Pools don't have connector ID
		assert.Equal(t, 0, ourEvent.RetryCount)
		assert.Equal(t, expectedKey, ourEvent.ID.EventIdempotencyKey)

		// Verify payload
		var payload map[string]interface{}
		err = json.Unmarshal(ourEvent.Payload, &payload)
		require.NoError(t, err)
		assert.Equal(t, testPool.ID.String(), payload["id"])
		assert.Equal(t, testPool.Name, payload["name"])
		accountIDs, ok := payload["accountIDs"].([]interface{})
		require.True(t, ok)
		require.Len(t, accountIDs, 2)
		assert.NotNil(t, payload["createdAt"])
	})

	t.Run("no outbox event when adding account already in pool", func(t *testing.T) {
		// Create a new pool for this test
		testPool := models.Pool{
			ID:           uuid.New(),
			Name:         "add-duplicate-test-pool",
			CreatedAt:    now.Add(-10 * time.Minute).UTC().Time,
			Type:         models.POOL_TYPE_STATIC,
			PoolAccounts: []models.AccountID{defaultAccounts()[0].ID},
		}
		require.NoError(t, store.PoolsUpsert(ctx, testPool))

		// Clear any existing events
		_, _ = store.OutboxEventsPollPending(ctx, 1000)

		// Count events before
		eventsBefore, err := store.OutboxEventsPollPending(ctx, 1000)
		require.NoError(t, err)
		countBefore := len(eventsBefore)

		// Try to add account that's already in the pool (should not create event)
		require.NoError(t, store.PoolsAddAccount(ctx, testPool.ID, defaultAccounts()[0].ID))

		// Verify no new outbox event was created
		eventsAfter, err := store.OutboxEventsPollPending(ctx, 1000)
		require.NoError(t, err)
		countAfter := len(eventsAfter)

		// Should have same number of events (no new pool saved event)
		assert.Equal(t, countBefore, countAfter, "adding duplicate account should not create saved event")
	})
}

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

	ctx := logging.TestingContext()
	store := newStore(t)
	t.Cleanup(func() {
		store.Close()
	})

	upsertConnector(t, ctx, store, defaultConnector)
	upsertAccounts(t, ctx, store, defaultAccounts())
	upsertPool(t, ctx, store, defaultPools()[0])
	upsertPool(t, ctx, store, defaultPools()[1])

	t.Run("remove unknown account from pool", func(t *testing.T) {
		require.NoError(t, store.PoolsRemoveAccount(ctx, defaultPools()[0].ID, models.AccountID{
			Reference:   "unknown",
			ConnectorID: defaultConnector.ID,
		}))
	})

	t.Run("remove account from unknown pool", func(t *testing.T) {
		require.NoError(t, store.PoolsRemoveAccount(ctx, uuid.New(), defaultAccounts()[0].ID))
	})

	t.Run("remove account from pool", func(t *testing.T) {
		require.NoError(t, store.PoolsRemoveAccount(ctx, defaultPools()[0].ID, defaultAccounts()[1].ID))

		p := defaultPools()[0]
		p.PoolAccounts = p.PoolAccounts[:1]

		actual, err := store.PoolsGet(ctx, defaultPools()[0].ID)
		require.NoError(t, err)
		require.Equal(t, p, *actual)
	})

	t.Run("outbox event created when removing account from pool", func(t *testing.T) {
		// Create a new pool for this test
		testPool := models.Pool{
			ID:           uuid.New(),
			Name:         "remove-account-test-pool",
			CreatedAt:    now.Add(-10 * time.Minute).UTC().Time,
			Type:         models.POOL_TYPE_STATIC,
			PoolAccounts: []models.AccountID{defaultAccounts()[0].ID, defaultAccounts()[1].ID},
		}
		require.NoError(t, store.PoolsUpsert(ctx, testPool))

		// Clear any existing events
		_, _ = store.OutboxEventsPollPending(ctx, 1000)

		// Remove account from pool
		require.NoError(t, store.PoolsRemoveAccount(ctx, testPool.ID, defaultAccounts()[1].ID))

		// Expected pool state after removing account
		expectedPool := testPool
		expectedPool.PoolAccounts = []models.AccountID{defaultAccounts()[0].ID}
		expectedKey := expectedPool.IdempotencyKey()

		// Verify outbox event was created
		pendingEvents, err := store.OutboxEventsPollPending(ctx, 100)
		require.NoError(t, err)

		// Find our event
		var ourEvent *models.OutboxEvent
		for i := range pendingEvents {
			if pendingEvents[i].EventType == events.EventTypeSavedPool &&
				pendingEvents[i].EntityID == testPool.ID.String() &&
				pendingEvents[i].ID.EventIdempotencyKey == expectedKey {
				ourEvent = &pendingEvents[i]
				break
			}
		}
		require.NotNil(t, ourEvent, "expected outbox event for pool saved after removing account")

		// Verify event details
		assert.Equal(t, events.EventTypeSavedPool, ourEvent.EventType)
		assert.Equal(t, models.OUTBOX_STATUS_PENDING, ourEvent.Status)
		assert.Equal(t, testPool.ID.String(), ourEvent.EntityID)
		assert.Nil(t, ourEvent.ConnectorID) // Pools don't have connector ID
		assert.Equal(t, 0, ourEvent.RetryCount)
		assert.Equal(t, expectedKey, ourEvent.ID.EventIdempotencyKey)

		// Verify payload
		var payload map[string]interface{}
		err = json.Unmarshal(ourEvent.Payload, &payload)
		require.NoError(t, err)
		assert.Equal(t, testPool.ID.String(), payload["id"])
		assert.Equal(t, testPool.Name, payload["name"])
		accountIDs, ok := payload["accountIDs"].([]interface{})
		require.True(t, ok)
		require.Len(t, accountIDs, 1)
		assert.NotNil(t, payload["createdAt"])
	})

	t.Run("no outbox event when removing account not in pool", func(t *testing.T) {
		// Create a new pool for this test
		testPool := models.Pool{
			ID:           uuid.New(),
			Name:         "remove-unknown-test-pool",
			CreatedAt:    now.Add(-10 * time.Minute).UTC().Time,
			Type:         models.POOL_TYPE_STATIC,
			PoolAccounts: []models.AccountID{defaultAccounts()[0].ID},
		}
		require.NoError(t, store.PoolsUpsert(ctx, testPool))

		// Clear any existing events
		_, _ = store.OutboxEventsPollPending(ctx, 1000)

		// Count events before
		eventsBefore, err := store.OutboxEventsPollPending(ctx, 1000)
		require.NoError(t, err)
		countBefore := len(eventsBefore)

		// Try to remove account that's not in the pool (should not create event)
		require.NoError(t, store.PoolsRemoveAccount(ctx, testPool.ID, defaultAccounts()[1].ID))

		// Verify no new outbox event was created
		eventsAfter, err := store.OutboxEventsPollPending(ctx, 1000)
		require.NoError(t, err)
		countAfter := len(eventsAfter)

		// Should have same number of events (no new pool saved event)
		assert.Equal(t, countBefore, countAfter, "removing non-existent account should not create saved event")
	})
}

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

	ctx := logging.TestingContext()
	store := newStore(t)
	t.Cleanup(func() {
		store.Close()
	})

	upsertConnector(t, ctx, store, defaultConnector)
	upsertAccounts(t, ctx, store, defaultAccounts())
	upsertPool(t, ctx, store, defaultPools()[0])

	t.Run("remove accounts from unknown connector", func(t *testing.T) {
		require.NoError(t, store.PoolsRemoveAccountsFromConnectorID(ctx, models.ConnectorID{
			Reference: uuid.New(),
			Provider:  "unknown",
		}))

		actual, err := store.PoolsGet(ctx, defaultPools()[0].ID)
		require.NoError(t, err)
		require.Equal(t, defaultPools()[0], *actual)
	})

	t.Run("remove accounts from connector", func(t *testing.T) {
		require.NoError(t, store.PoolsRemoveAccountsFromConnectorID(ctx, defaultConnector.ID))

		actual, err := store.PoolsGet(ctx, defaultPools()[0].ID)
		require.NoError(t, err)
		require.Len(t, actual.PoolAccounts, 0)
	})
}

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

	ctx := logging.TestingContext()
	store := newStore(t)
	t.Cleanup(func() {
		store.Close()
	})

	upsertConnector(t, ctx, store, defaultConnector)
	upsertAccounts(t, ctx, store, defaultAccounts())
	upsertPool(t, ctx, store, defaultPools()[0])
	upsertPool(t, ctx, store, defaultPools()[1])
	upsertPool(t, ctx, store, defaultPools()[2])

	t.Run("wrong query builder operator when listing by name", func(t *testing.T) {
		q := NewListPoolsQuery(
			bunpaginate.NewPaginatedQueryOptions(PoolQuery{}).
				WithPageSize(15).
				WithQueryBuilder(query.Lt("name", "test1")),
		)

		cursor, err := store.PoolsList(ctx, q)
		require.Error(t, err)
		require.Nil(t, cursor)
		assert.True(t, errors.Is(err, ErrValidation))
		assert.Regexp(t, "name", err.Error())
	})

	t.Run("list pools by name", func(t *testing.T) {
		q := NewListPoolsQuery(
			bunpaginate.NewPaginatedQueryOptions(PoolQuery{}).
				WithPageSize(15).
				WithQueryBuilder(query.Match("name", "test1")),
		)

		cursor, err := store.PoolsList(ctx, q)
		require.NoError(t, err)
		require.Len(t, cursor.Data, 1)
		require.False(t, cursor.HasMore)
		require.Empty(t, cursor.Previous)
		require.Empty(t, cursor.Next)
		require.Equal(t, []models.Pool{defaultPools()[0]}, cursor.Data)
	})

	t.Run("list pools by unknown name", func(t *testing.T) {
		q := NewListPoolsQuery(
			bunpaginate.NewPaginatedQueryOptions(PoolQuery{}).
				WithPageSize(15).
				WithQueryBuilder(query.Match("name", "unknown")),
		)

		cursor, err := store.PoolsList(ctx, q)
		require.NoError(t, err)
		require.Empty(t, cursor.Data)
		require.False(t, cursor.HasMore)
		require.Empty(t, cursor.Previous)
		require.Empty(t, cursor.Next)
	})

	t.Run("list pools by id", func(t *testing.T) {
		q := NewListPoolsQuery(
			bunpaginate.NewPaginatedQueryOptions(PoolQuery{}).
				WithPageSize(15).
				WithQueryBuilder(query.Match("id", poolID1.String())),
		)

		cursor, err := store.PoolsList(ctx, q)
		require.NoError(t, err)
		require.Len(t, cursor.Data, 1)
		require.False(t, cursor.HasMore)
		require.Empty(t, cursor.Previous)
		require.Empty(t, cursor.Next)
		require.Equal(t, []models.Pool{defaultPools()[0]}, cursor.Data)
	})

	t.Run("list pools by unknown id", func(t *testing.T) {
		q := NewListPoolsQuery(
			bunpaginate.NewPaginatedQueryOptions(PoolQuery{}).
				WithPageSize(15).
				WithQueryBuilder(query.Match("id", uuid.New().String())),
		)

		cursor, err := store.PoolsList(ctx, q)
		require.NoError(t, err)
		require.Empty(t, cursor.Data)
		require.False(t, cursor.HasMore)
		require.Empty(t, cursor.Previous)
		require.Empty(t, cursor.Next)
	})

	t.Run("list pools by account id 1", func(t *testing.T) {
		q := NewListPoolsQuery(
			bunpaginate.NewPaginatedQueryOptions(PoolQuery{}).
				WithPageSize(15).
				WithQueryBuilder(query.Match("account_id", defaultAccounts()[0].ID.String())),
		)

		cursor, err := store.PoolsList(ctx, q)
		require.NoError(t, err)
		require.Len(t, cursor.Data, 1)
		require.False(t, cursor.HasMore)
		require.Empty(t, cursor.Previous)
		require.Empty(t, cursor.Next)
		require.Equal(t, []models.Pool{defaultPools()[0]}, cursor.Data)
	})

	t.Run("list pools by account id 2", func(t *testing.T) {
		q := NewListPoolsQuery(
			bunpaginate.NewPaginatedQueryOptions(PoolQuery{}).
				WithPageSize(15).
				WithQueryBuilder(query.Match("account_id", defaultAccounts()[2].ID.String())),
		)

		cursor, err := store.PoolsList(ctx, q)
		require.NoError(t, err)
		require.Len(t, cursor.Data, 1)
		require.False(t, cursor.HasMore)
		require.Empty(t, cursor.Previous)
		require.Empty(t, cursor.Next)
		require.Equal(t, []models.Pool{defaultPools()[1]}, cursor.Data)
	})

	t.Run("list pools by unknown account id", func(t *testing.T) {
		q := NewListPoolsQuery(
			bunpaginate.NewPaginatedQueryOptions(PoolQuery{}).
				WithPageSize(15).
				WithQueryBuilder(query.Match("account_id", uuid.New().String())),
		)

		cursor, err := store.PoolsList(ctx, q)
		require.NoError(t, err)
		require.Empty(t, cursor.Data)
		require.False(t, cursor.HasMore)
		require.Empty(t, cursor.Previous)
		require.Empty(t, cursor.Next)
	})

	t.Run("unknown query builder key when listing", func(t *testing.T) {
		q := NewListPoolsQuery(
			bunpaginate.NewPaginatedQueryOptions(PoolQuery{}).
				WithPageSize(15).
				WithQueryBuilder(query.Match("unknown", "test1")),
		)

		cursor, err := store.PoolsList(ctx, q)
		require.Error(t, err)
		require.Nil(t, cursor)
	})

	t.Run("list pools test cursor", func(t *testing.T) {
		q := NewListPoolsQuery(
			bunpaginate.NewPaginatedQueryOptions(PoolQuery{}).
				WithPageSize(1),
		)
		defaultPools := defaultPools()

		cursor, err := store.PoolsList(ctx, q)
		require.NoError(t, err)
		require.Len(t, cursor.Data, 1)
		require.True(t, cursor.HasMore)
		require.Empty(t, cursor.Previous)
		require.NotEmpty(t, cursor.Next)
		require.Equal(t, []models.Pool{defaultPools[1]}, cursor.Data)

		err = bunpaginate.UnmarshalCursor(cursor.Next, &q)
		require.NoError(t, err)
		cursor, err = store.PoolsList(ctx, q)
		require.NoError(t, err)
		require.Len(t, cursor.Data, 1)
		require.True(t, cursor.HasMore)
		require.NotEmpty(t, cursor.Previous)
		require.NotEmpty(t, cursor.Next)
		require.Equal(t, []models.Pool{defaultPools[2]}, cursor.Data)

		err = bunpaginate.UnmarshalCursor(cursor.Next, &q)
		require.NoError(t, err)
		cursor, err = store.PoolsList(ctx, q)
		require.NoError(t, err)
		require.Len(t, cursor.Data, 1)
		require.False(t, cursor.HasMore)
		require.NotEmpty(t, cursor.Previous)
		require.Empty(t, cursor.Next)
		require.Equal(t, []models.Pool{defaultPools[0]}, cursor.Data)

		err = bunpaginate.UnmarshalCursor(cursor.Previous, &q)
		require.NoError(t, err)
		cursor, err = store.PoolsList(ctx, q)
		require.NoError(t, err)
		require.Len(t, cursor.Data, 1)
		require.True(t, cursor.HasMore)
		require.NotEmpty(t, cursor.Previous)
		require.NotEmpty(t, cursor.Next)
		require.Equal(t, []models.Pool{defaultPools[2]}, cursor.Data)

		err = bunpaginate.UnmarshalCursor(cursor.Previous, &q)
		require.NoError(t, err)
		cursor, err = store.PoolsList(ctx, q)
		require.NoError(t, err)
		require.Len(t, cursor.Data, 1)
		require.True(t, cursor.HasMore)
		require.Empty(t, cursor.Previous)
		require.NotEmpty(t, cursor.Next)
		require.Equal(t, []models.Pool{defaultPools[1]}, cursor.Data)
	})

	t.Run("list pools with validation issues in filter", func(t *testing.T) {
		q := NewListPoolsQuery(bunpaginate.PaginatedQueryOptions[PoolQuery]{
			QueryBuilder: query.And(query.Match("id", "not a valid uuid")),
			PageSize:     uint64(5),
		})

		_, err := store.PoolsList(ctx, q)
		require.Error(t, err)
		assert.ErrorIs(t, err, ErrValidation)
	})
}