seaweedfs/weed/filesys/page_writer/dirty_page_interval_test.go

114 lines
2.3 KiB
Go
Raw Normal View History

2021-12-20 09:02:23 +00:00
package page_writer
2020-01-26 21:01:11 +00:00
import (
"bytes"
2020-08-22 23:33:00 +00:00
"math/rand"
2020-01-26 21:01:11 +00:00
"testing"
)
func TestContinuousIntervals_AddIntervalAppend(t *testing.T) {
2020-01-26 21:01:11 +00:00
c := &ContinuousIntervals{}
// 25, 25, 25
c.AddInterval(getBytes(25, 3), 0)
// _, _, 23, 23, 23, 23
c.AddInterval(getBytes(23, 4), 2)
expectedData(t, c, 0, 25, 25, 23, 23, 23, 23)
}
func TestContinuousIntervals_AddIntervalInnerOverwrite(t *testing.T) {
c := &ContinuousIntervals{}
// 25, 25, 25, 25, 25
c.AddInterval(getBytes(25, 5), 0)
// _, _, 23, 23
c.AddInterval(getBytes(23, 2), 2)
expectedData(t, c, 0, 25, 25, 23, 23, 25)
}
func TestContinuousIntervals_AddIntervalFullOverwrite(t *testing.T) {
c := &ContinuousIntervals{}
2020-03-27 05:21:52 +00:00
// 1,
c.AddInterval(getBytes(1, 1), 0)
// _, 2,
c.AddInterval(getBytes(2, 1), 1)
// _, _, 3, 3, 3
c.AddInterval(getBytes(3, 3), 2)
// _, _, _, 4, 4, 4
c.AddInterval(getBytes(4, 3), 3)
expectedData(t, c, 0, 1, 2, 3, 4, 4, 4)
}
func TestContinuousIntervals_RealCase1(t *testing.T) {
c := &ContinuousIntervals{}
// 25,
c.AddInterval(getBytes(25, 1), 0)
// _, _, _, _, 23, 23
c.AddInterval(getBytes(23, 2), 4)
// _, _, _, 24, 24, 24, 24
c.AddInterval(getBytes(24, 4), 3)
// _, 22, 22
c.AddInterval(getBytes(22, 2), 1)
expectedData(t, c, 0, 25, 22, 22, 24, 24, 24, 24)
2020-01-26 21:01:11 +00:00
}
2020-08-22 23:33:00 +00:00
func TestRandomWrites(t *testing.T) {
c := &ContinuousIntervals{}
data := make([]byte, 1024)
2020-08-30 05:28:33 +00:00
for i := 0; i < 1024; i++ {
2020-08-22 23:33:00 +00:00
start, stop := rand.Intn(len(data)), rand.Intn(len(data))
if start > stop {
2020-08-30 05:28:33 +00:00
start, stop = stop, start
2020-08-22 23:33:00 +00:00
}
2020-08-30 05:28:33 +00:00
rand.Read(data[start : stop+1])
2020-08-22 23:33:00 +00:00
c.AddInterval(data[start:stop+1], int64(start))
expectedData(t, c, 0, data...)
}
}
2020-01-26 21:01:11 +00:00
func expectedData(t *testing.T, c *ContinuousIntervals, offset int, data ...byte) {
start, stop := int64(offset), int64(offset+len(data))
for _, list := range c.lists {
nodeStart, nodeStop := max(start, list.Head.Offset), min(stop, list.Head.Offset+list.Size())
if nodeStart < nodeStop {
buf := make([]byte, nodeStop-nodeStart)
list.ReadData(buf, nodeStart, nodeStop)
if bytes.Compare(buf, data[nodeStart-start:nodeStop-start]) != 0 {
t.Errorf("expected %v actual %v", data[nodeStart-start:nodeStop-start], buf)
}
}
}
}
func getBytes(content byte, length int) []byte {
data := make([]byte, length)
for i := 0; i < length; i++ {
data[i] = content
}
return data
}