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.
75 lines
1.7 KiB
75 lines
1.7 KiB
package threadTest
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"simple-kv-store/internal/nodes"
|
|
"testing"
|
|
|
|
"github.com/syndtr/goleveldb/leveldb"
|
|
)
|
|
|
|
func ExecuteNodeI(id string, isRestart bool, peerIds []string, threadTransport *nodes.ThreadTransport) (*nodes.Node, chan struct{}) {
|
|
if !isRestart {
|
|
os.RemoveAll("storage/node" + id)
|
|
}
|
|
|
|
os.RemoveAll("leveldb/simple-kv-store" + id)
|
|
|
|
db, err := leveldb.OpenFile("leveldb/simple-kv-store" + id, nil)
|
|
if err != nil {
|
|
fmt.Println("Failed to open database: ", err)
|
|
}
|
|
|
|
// 打开或创建节点数据持久化文件
|
|
storage := nodes.NewRaftStorage("storage/node" + id)
|
|
|
|
var otherIds []string
|
|
for _, ids := range peerIds {
|
|
if ids != id {
|
|
otherIds = append(otherIds, ids) // 删除目标元素
|
|
}
|
|
}
|
|
// 初始化
|
|
node, quitChan := nodes.InitThreadNode(id, otherIds, db, storage, isRestart, threadTransport)
|
|
|
|
// 开启 raft
|
|
go nodes.Start(node, quitChan)
|
|
return node, quitChan
|
|
}
|
|
|
|
func CheckOneLeader(t *testing.T, nodeCollections []* nodes.Node) {
|
|
cnt := 0
|
|
for _, node := range nodeCollections {
|
|
if node.State == nodes.Leader {
|
|
cnt++
|
|
}
|
|
}
|
|
if cnt != 1 {
|
|
t.Errorf("实际有%d个leader(!=1)", cnt)
|
|
}
|
|
}
|
|
|
|
func CheckZeroOrOneLeader(t *testing.T, nodeCollections []* nodes.Node) {
|
|
cnt := 0
|
|
for _, node := range nodeCollections {
|
|
if node.State == nodes.Leader {
|
|
cnt++
|
|
}
|
|
}
|
|
if cnt > 1 {
|
|
t.Errorf("实际有%d个leader(>1)", cnt)
|
|
}
|
|
}
|
|
|
|
func CheckIsLeader(t *testing.T, node *nodes.Node) {
|
|
if node.State != nodes.Leader {
|
|
t.Errorf("[%s]不是leader", node.SelfId)
|
|
}
|
|
}
|
|
|
|
func CheckTerm(t *testing.T, node *nodes.Node, targetTerm int) {
|
|
if node.CurrTerm != targetTerm {
|
|
t.Errorf("[%s]实际term=%d (!=%d)", node.SelfId, node.CurrTerm, targetTerm)
|
|
}
|
|
}
|