Function TestAcquireTransactionLocker

transaction_test.go:49–91

(t *testing.T)

)

func TestAcquireTransactionLocker(t *testing.T) {
	t.Run("fail fast when wait timeout disabled", func(t *testing.T) {
		db, mock := redismock.NewClientMock()
		ledgerforge := &LedgerForge{
			redis: db,
			config: &config.Configuration{
				Transaction: config.TransactionConfig{
					LockDuration: 5 * time.Second,
				},
			},
		}
		locker := redlock.NewMultiLocker(db, []string{"key-b", "key-a"}, "test-value")

		mock.ExpectSetNX("key-a", "test-value", 5*time.Second).SetVal(true)
		mock.ExpectSetNX("key-b", "test-value", 5*time.Second).SetVal(true)

		err := ledgerforge.acquireTransactionLocker(context.Background(), locker)
		require.NoError(t, err)
		require.NoError(t, mock.ExpectationsWereMet())
	})

	t.Run("waits when configured", func(t *testing.T) {
		db, mock := redismock.NewClientMock()
		ledgerforge := &LedgerForge{
			redis: db,
			config: &config.Configuration{
				Transaction: config.TransactionConfig{
					LockDuration:    5 * time.Second,
					LockWaitTimeout: time.Second,
				},
			},
		}
		locker := redlock.NewMultiLocker(db, []string{"key-b", "key-a"}, "test-value")

		mock.ExpectSetNX("key-a", "test-value", 5*time.Second).SetVal(false)
		mock.ExpectSetNX("key-a", "test-value", 5*time.Second).SetVal(true)
		mock.ExpectSetNX("key-b", "test-value", 5*time.Second).SetVal(true)

		err := ledgerforge.acquireTransactionLocker(context.Background(), locker)
		require.NoError(t, err)
		require.NoError(t, mock.ExpectationsWereMet())
	})
}

// pollForTransactionStatus polls for a transaction until it reaches the expected status or a timeout occurs.
func pollForTransactionStatus(ctx context.Context, ds database.IDataSource, transactionRef string, expectedStatus string, pollInterval time.Duration, timeoutDuration time.Duration) (*model.Transaction, error) {