You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
76 lines
2.3 KiB
76 lines
2.3 KiB
package consensus
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
var testTxt = `{"time":"2016-01-16T04:42:00.390Z","msg":[1,{"height":28219,"round":0,"step":"RoundStepPrevote"}]}
|
|
{"time":"2016-01-16T04:42:00.390Z","msg":[2,{"msg":[20,{"ValidatorIndex":0,"Vote":{"height":28219,"round":0,"type":1,"block_hash":"67F9689F15BEC30BF311FB4C0C80C5E661AA44E0","block_parts_header":{"total":1,"hash":"DFFD4409A1E273ED61AC27CAF975F446020D5676"},"signature":"4CC6845A128E723A299B470CCBB2A158612AA51321447F6492F3DA57D135C27FCF4124B3B19446A248252BDA45B152819C76AAA5FD35E1C07091885CE6955E05"}}],"peer_key":""}]}
|
|
{"time":"2016-01-16T04:42:00.392Z","msg":[1,{"height":28219,"round":0,"step":"RoundStepPrecommit"}]}
|
|
{"time":"2016-01-16T04:42:00.392Z","msg":[2,{"msg":[20,{"ValidatorIndex":0,"Vote":{"height":28219,"round":0,"type":2,"block_hash":"67F9689F15BEC30BF311FB4C0C80C5E661AA44E0","block_parts_header":{"total":1,"hash":"DFFD4409A1E273ED61AC27CAF975F446020D5676"},"signature":"1B9924E010F47E0817695DFE462C531196E5A12632434DE12180BBA3EFDAD6B3960FDB9357AFF085EB61729A7D4A6AD8408555D7569C87D9028F280192FD4E05"}}],"peer_key":""}]}
|
|
{"time":"2016-01-16T04:42:00.393Z","msg":[1,{"height":28219,"round":0,"step":"RoundStepCommit"}]}
|
|
{"time":"2016-01-16T04:42:00.395Z","msg":[1,{"height":28220,"round":0,"step":"RoundStepNewHeight"}]}`
|
|
|
|
func TestSeek(t *testing.T) {
|
|
f, err := ioutil.TempFile(os.TempDir(), "seek_test_")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
stat, _ := f.Stat()
|
|
name := stat.Name()
|
|
|
|
_, err = f.WriteString(testTxt)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
f.Close()
|
|
|
|
wal, err := NewWAL(path.Join(os.TempDir(), name))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
keyWord := "Precommit"
|
|
n, err := wal.SeekFromEnd(func(b []byte) bool {
|
|
if strings.Contains(string(b), keyWord) {
|
|
return true
|
|
}
|
|
return false
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// confirm n
|
|
spl := strings.Split(testTxt, "\n")
|
|
var i int
|
|
var s string
|
|
for i, s = range spl {
|
|
if strings.Contains(s, keyWord) {
|
|
break
|
|
}
|
|
}
|
|
// n is lines from the end.
|
|
spl = spl[i:]
|
|
if n != len(spl) {
|
|
t.Fatalf("Wrong nLines. Got %d, expected %d", n, len(spl))
|
|
}
|
|
|
|
b, err := ioutil.ReadAll(wal.fp)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// first char is a \n
|
|
spl2 := strings.Split(strings.Trim(string(b), "\n"), "\n")
|
|
for i, s := range spl {
|
|
if s != spl2[i] {
|
|
t.Fatalf("Mismatch. Got %s, expected %s", spl2[i], s)
|
|
}
|
|
}
|
|
|
|
}
|