1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859 |
- package chunk_cache
- import (
- "bytes"
- "fmt"
- "io/ioutil"
- "math/rand"
- "os"
- "testing"
- )
- func TestOnDisk(t *testing.T) {
- tmpDir, _ := ioutil.TempDir("", "c")
- defer os.RemoveAll(tmpDir)
- totalDiskSizeMb := int64(32)
- cache := NewChunkCache(0, tmpDir, totalDiskSizeMb)
- writeCount := 5
- type test_data struct {
- data []byte
- fileId string
- size uint64
- }
- testData := make([]*test_data, writeCount)
- for i := 0; i < writeCount; i++ {
- buff := make([]byte, 1024*1024)
- rand.Read(buff)
- testData[i] = &test_data{
- data: buff,
- fileId: fmt.Sprintf("1,%daabbccdd", i+1),
- size: uint64(len(buff)),
- }
- cache.SetChunk(testData[i].fileId, testData[i].data)
- }
- for i := 0; i < writeCount; i++ {
- data := cache.GetChunk(testData[i].fileId, testData[i].size)
- if bytes.Compare(data, testData[i].data) != 0 {
- t.Errorf("failed to write to and read from cache: %d", i)
- }
- }
- cache.Shutdown()
- cache = NewChunkCache(0, tmpDir, totalDiskSizeMb)
- for i := 0; i < writeCount; i++ {
- data := cache.GetChunk(testData[i].fileId, testData[i].size)
- if bytes.Compare(data, testData[i].data) != 0 {
- t.Errorf("failed to write to and read from cache: %d", i)
- }
- }
- cache.Shutdown()
- }
|