Files
seaweedfs/weed/util/mem/slot_pool_test.go
T
Chris Lu 4a1d65939f fix(mount): bound reader cache memory across open files (#11220)
* fix(filer): bound retained reader cache buffers by bytes

* test(filer): keep in-flight downloads during cache trimming

* feat(mem): expose pooled allocation capacity for byte reservations

* fix(mount): share a configurable reader buffer budget across files

* fix(filer): release failed prefetch slots and memory reservations

* feat(mount): expose a soft Go runtime memory limit

* docs(filer): restore shared-download rationale in startCaching

The one-line comment replacing the original context.Background() explanation was too thin for readChunkAt to cross-reference shared resource semantics. Restore a concise note on why request cancellation must not abort a download shared by concurrent readers.

* test(filer): loosen reader cache test deadlines to 5s

Three tests used 1-second deadlines that can flake on CI under load:
TestReaderCacheBudgetInFlight, TestReaderCacheEvictionDoesNotHoldCacheLock,
and TestReaderCacheFailedPrefetchReleasesBudget. Increase to 5 seconds.

* test(filer): cover re-read after reader cache eviction

Add TestReaderCacheReReadAfterEviction: reads chunk 'a', reads chunk 'b'
(evicting 'a' via budget pressure), then re-reads 'a' and asserts a
fresh download returns correct data. Verifies the core correctness
property that eviction never exposes missing or stale data to readers.
2026-09-08 10:51:28 -07:00

59 lines
1.1 KiB
Go

package mem
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestAllocateFree(t *testing.T) {
buf := Allocate(12)
Free(buf)
if cap(buf) != min_size {
t.Errorf("min size error allocated capacity=%d", cap(buf))
}
if len(buf) != 12 {
t.Errorf("size error")
}
buf = Allocate(4883)
Free(buf)
if cap(buf) != 1024<<bitCount(4883) {
t.Errorf("min size error allocated capacity=%d", cap(buf))
}
if len(buf) != 4883 {
t.Errorf("size error")
}
}
func TestAllocateFreeEdgeCases(t *testing.T) {
assert.Equal(t, 1, bitCount(2048))
assert.Equal(t, 2, bitCount(2049))
buf := Allocate(2048)
Free(buf)
buf = Allocate(2049)
Free(buf)
}
func TestBitCount(t *testing.T) {
count := bitCount(12)
if count != 0 {
t.Errorf("bitCount error count=%d", count)
}
if count != bitCount(min_size) {
t.Errorf("bitCount error count=%d", count)
}
}
func TestAllocationSize(t *testing.T) {
for _, size := range []int{0, 1, 1024, 1025, 4883, 2 << 20} {
buf := Allocate(size)
assert.Equal(t, cap(buf), AllocationSize(size))
Free(buf)
}
assert.Equal(t, 257<<20, AllocationSize(257<<20))
}