crdt_ttl_test.go

package bmdb

import (
	"testing"
	"time"

	"github.com/stretchr/testify/require"
)

// Test that CRDT replay does not resurrect keys that were already expired on disk
// when the DB is reopened. This mirrors the regression we fixed earlier.
func TestCRDTReplayDoesNotResurrectExpired(t *testing.T) {
	opts := DefaultOptions
	// enable debug logs to observe replay logging
	opts.DebugLogs = true

	runBMDBTest(t, &opts, func(t *testing.T, db *DB) {
		bucket := "bucket-crdt"
		txCreateBucket(t, db, DataStructureBTree, bucket, nil)

		// key0 expires quickly; key1 persists
		txPut(t, db, bucket, GetTestBytes(0), GetTestBytes(0), 1, nil, nil)
		txPut(t, db, bucket, GetTestBytes(1), GetTestBytes(1), Persistent, nil, nil)

		// ensure key0 expires
		waitFor(t, func() bool {
			err := db.View(func(tx *Tx) error {
				_, err := tx.Get(bucket, GetTestBytes(0))
				return err
			})
			return err == ErrKeyNotFound
		}, 2*time.Second)

		require.NoError(t, db.Close())

		// reopen and ensure key0 is not resurrected by CRDT
		var reopened *DB
		waitFor(t, func() bool {
			d, err := Open(opts)
			if err != nil {
				return false
			}
			err = d.View(func(tx *Tx) error {
				_, err := tx.Get(bucket, GetTestBytes(0))
				return err
			})
			if err == ErrKeyNotFound {
				reopened = d
				return true
			}
			_ = d.Close()
			return false
		}, 5*time.Second)

		require.NotNil(t, reopened)
		db = reopened

		// key1 should still be present
		txGet(t, db, bucket, GetTestBytes(1), GetTestBytes(1), nil)
	})
}

// Basic CRDT/TTL interplay test ensuring CRDT replay does not conflict with
// TTL-driven deletes on open.
func TestCRDTAndTTLInterplay(t *testing.T) {
	opts := DefaultOptions
	opts.DebugLogs = true

	runBMDBTest(t, &opts, func(t *testing.T, db *DB) {
		bucket := "bucket-crdt-ttl"
		txCreateBucket(t, db, DataStructureBTree, bucket, nil)

		// Add several keys with different TTLs
		txPut(t, db, bucket, GetTestBytes(0), GetTestBytes(0), 1, nil, nil)
		txPut(t, db, bucket, GetTestBytes(1), GetTestBytes(1), 3, nil, nil)
		txPut(t, db, bucket, GetTestBytes(2), GetTestBytes(2), Persistent, nil, nil)

		// wait for key0 to expire but key1 still exists
		waitFor(t, func() bool {
			var got0, got1 error
			_ = db.View(func(tx *Tx) error {
				_, got0 = tx.Get(bucket, GetTestBytes(0))
				return nil
			})
			_ = db.View(func(tx *Tx) error {
				_, got1 = tx.Get(bucket, GetTestBytes(1))
				return nil
			})
			return got0 == ErrKeyNotFound && got1 == nil
		}, 3*time.Second)

		// Close and reopen; ensure behavior persists and no resurrection
		require.NoError(t, db.Close())
		d2, err := Open(opts)
		require.NoError(t, err)
		defer d2.Close()

		// key0 should still be gone; key1 may expire later
		txGet(t, d2, bucket, GetTestBytes(0), nil, ErrKeyNotFound)
		txGet(t, d2, bucket, GetTestBytes(2), GetTestBytes(2), nil)
	})
}