mirror of https://github.com/hak5/bolt.git
99 lines
2.9 KiB
Go
99 lines
2.9 KiB
Go
package bolt
|
|
|
|
import (
|
|
"testing"
|
|
"unsafe"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// Ensure that a page is added to a transaction's freelist.
|
|
func TestFreelist_free(t *testing.T) {
|
|
f := &freelist{pending: make(map[txid][]pgid)}
|
|
f.free(100, &page{id: 12})
|
|
assert.Equal(t, f.pending[100], []pgid{12})
|
|
}
|
|
|
|
// Ensure that a page and its overflow is added to a transaction's freelist.
|
|
func TestFreelist_free_overflow(t *testing.T) {
|
|
f := &freelist{pending: make(map[txid][]pgid)}
|
|
f.free(100, &page{id: 12, overflow: 3})
|
|
assert.Equal(t, f.pending[100], []pgid{12, 13, 14, 15})
|
|
}
|
|
|
|
// Ensure that a transaction's free pages can be released.
|
|
func TestFreelist_release(t *testing.T) {
|
|
f := &freelist{pending: make(map[txid][]pgid)}
|
|
f.free(100, &page{id: 12, overflow: 1})
|
|
f.free(100, &page{id: 9})
|
|
f.free(102, &page{id: 39})
|
|
f.release(100)
|
|
f.release(101)
|
|
assert.Equal(t, []pgid{9, 12, 13}, f.ids)
|
|
f.release(102)
|
|
assert.Equal(t, []pgid{9, 12, 13, 39}, f.ids)
|
|
}
|
|
|
|
// Ensure that a freelist can find contiguous blocks of pages.
|
|
func TestFreelist_allocate(t *testing.T) {
|
|
f := &freelist{ids: []pgid{3, 4, 5, 6, 7, 9, 12, 13, 18}}
|
|
assert.Equal(t, 3, int(f.allocate(3)))
|
|
assert.Equal(t, 6, int(f.allocate(1)))
|
|
assert.Equal(t, 0, int(f.allocate(3)))
|
|
assert.Equal(t, 12, int(f.allocate(2)))
|
|
assert.Equal(t, 7, int(f.allocate(1)))
|
|
assert.Equal(t, 0, int(f.allocate(0)))
|
|
assert.Equal(t, []pgid{9, 18}, f.ids)
|
|
assert.Equal(t, 9, int(f.allocate(1)))
|
|
assert.Equal(t, 18, int(f.allocate(1)))
|
|
assert.Equal(t, 0, int(f.allocate(1)))
|
|
assert.Equal(t, []pgid{}, f.ids)
|
|
}
|
|
|
|
// Ensure that a freelist can deserialize from a freelist page.
|
|
func TestFreelist_read(t *testing.T) {
|
|
// Create a page.
|
|
var buf [4096]byte
|
|
page := (*page)(unsafe.Pointer(&buf[0]))
|
|
page.flags = freelistPageFlag
|
|
page.count = 2
|
|
|
|
// Insert 2 page ids.
|
|
ids := (*[3]pgid)(unsafe.Pointer(&page.ptr))
|
|
ids[0] = 23
|
|
ids[1] = 50
|
|
|
|
// Deserialize page into a freelist.
|
|
f := &freelist{pending: make(map[txid][]pgid)}
|
|
f.read(page)
|
|
|
|
// Ensure that there are two page ids in the freelist.
|
|
assert.Equal(t, len(f.ids), 2)
|
|
assert.Equal(t, f.ids[0], pgid(23))
|
|
assert.Equal(t, f.ids[1], pgid(50))
|
|
}
|
|
|
|
// Ensure that a freelist can serialize into a freelist page.
|
|
func TestFreelist_write(t *testing.T) {
|
|
// Create a freelist and write it to a page.
|
|
var buf [4096]byte
|
|
f := &freelist{ids: []pgid{12, 39}, pending: make(map[txid][]pgid)}
|
|
f.pending[100] = []pgid{28, 11}
|
|
f.pending[101] = []pgid{3}
|
|
p := (*page)(unsafe.Pointer(&buf[0]))
|
|
f.write(p)
|
|
|
|
// Read the page back out.
|
|
f2 := &freelist{pending: make(map[txid][]pgid)}
|
|
f2.read(p)
|
|
|
|
// Ensure that the freelist is correct.
|
|
// All pages should be present and in reverse order.
|
|
assert.Equal(t, len(f2.ids), 5)
|
|
assert.Equal(t, f2.ids[0], pgid(3))
|
|
assert.Equal(t, f2.ids[1], pgid(11))
|
|
assert.Equal(t, f2.ids[2], pgid(12))
|
|
assert.Equal(t, f2.ids[3], pgid(28))
|
|
assert.Equal(t, f2.ids[4], pgid(39))
|
|
}
|