Function TestWriteBatchFlush_UnregistersFinalTransaction

batch_test.go:231–259

(t *testing.T)

}

func TestWriteBatchFlush_UnregistersFinalTransaction(t *testing.T) {
	db, err := Open(
		DefaultOptions,
		WithDir(t.TempDir()),
	)
	require.NoError(t, err)
	defer func() { require.NoError(t, db.Close()) }()

	const bucket = "flush_bucket"
	require.NoError(t, db.Update(func(tx *Tx) error {
		return tx.NewBucket(DataStructureBTree, bucket)
	}))

	wb, err := db.NewWriteBatch()
	require.NoError(t, err)
	require.Equal(t, int64(1), db.transactionMgr.GetActiveTxCount())

	require.NoError(t, wb.Put(bucket, []byte("key"), []byte("value"), 0))
	require.NoError(t, wb.Flush())

	require.Equal(t, int64(0), db.transactionMgr.GetActiveTxCount(), "Flush should unregister all transactions")

	require.NoError(t, db.View(func(tx *Tx) error {
		val, err := tx.Get(bucket, []byte("key"))
		require.NoError(t, err)
		require.Equal(t, []byte("value"), val)
		return nil
	}))
}

func TestWriteBatchReset_ReplacesTransaction(t *testing.T) {
	db, err := Open(