mirror of https://github.com/Llewellynvdm/fzf.git
81 lines
1.8 KiB
Go
81 lines
1.8 KiB
Go
package fzf
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/junegunn/fzf/src/util"
|
|
)
|
|
|
|
func TestChunkList(t *testing.T) {
|
|
// FIXME global
|
|
sortCriteria = []criterion{byScore, byLength}
|
|
|
|
cl := NewChunkList(func(item *Item, s []byte) bool {
|
|
item.text = util.ToChars(s)
|
|
return true
|
|
})
|
|
|
|
// Snapshot
|
|
snapshot, count := cl.Snapshot()
|
|
if len(snapshot) > 0 || count > 0 {
|
|
t.Error("Snapshot should be empty now")
|
|
}
|
|
|
|
// Add some data
|
|
cl.Push([]byte("hello"))
|
|
cl.Push([]byte("world"))
|
|
|
|
// Previously created snapshot should remain the same
|
|
if len(snapshot) > 0 {
|
|
t.Error("Snapshot should not have changed")
|
|
}
|
|
|
|
// But the new snapshot should contain the added items
|
|
snapshot, count = cl.Snapshot()
|
|
if len(snapshot) != 1 && count != 2 {
|
|
t.Error("Snapshot should not be empty now")
|
|
}
|
|
|
|
// Check the content of the ChunkList
|
|
chunk1 := snapshot[0]
|
|
if chunk1.count != 2 {
|
|
t.Error("Snapshot should contain only two items")
|
|
}
|
|
if chunk1.items[0].text.ToString() != "hello" ||
|
|
chunk1.items[1].text.ToString() != "world" {
|
|
t.Error("Invalid data")
|
|
}
|
|
if chunk1.IsFull() {
|
|
t.Error("Chunk should not have been marked full yet")
|
|
}
|
|
|
|
// Add more data
|
|
for i := 0; i < chunkSize*2; i++ {
|
|
cl.Push([]byte(fmt.Sprintf("item %d", i)))
|
|
}
|
|
|
|
// Previous snapshot should remain the same
|
|
if len(snapshot) != 1 {
|
|
t.Error("Snapshot should stay the same")
|
|
}
|
|
|
|
// New snapshot
|
|
snapshot, count = cl.Snapshot()
|
|
if len(snapshot) != 3 || !snapshot[0].IsFull() ||
|
|
!snapshot[1].IsFull() || snapshot[2].IsFull() || count != chunkSize*2+2 {
|
|
t.Error("Expected two full chunks and one more chunk")
|
|
}
|
|
if snapshot[2].count != 2 {
|
|
t.Error("Unexpected number of items")
|
|
}
|
|
|
|
cl.Push([]byte("hello"))
|
|
cl.Push([]byte("world"))
|
|
|
|
lastChunkCount := snapshot[len(snapshot)-1].count
|
|
if lastChunkCount != 2 {
|
|
t.Error("Unexpected number of items:", lastChunkCount)
|
|
}
|
|
}
|