func createMockNode(id int, myConf *Config, sm *State_Machine, cl *mock.MockCluster) cluster.Server {
initNode(id, myConf, sm)
// Give each raftNode its own "Server" from the cluster.
server, err := cl.AddServer(id)
if err != nil {
panic(err)
}
return server
}
func TestBasic(t *testing.T) {
cnt = 0
//Initialization.
var myRaft Raft
cluster := new(mock.MockCluster)
//cluster = new(mock.MockCluster)
cleanDB() //Clear the old database.
myRaft, cluster = myRaft.makeMockRafts() //make mock cluster
//Simple cluster can also be created using myraft.makeRafts() method.
leaderId := myRaft.GetLeader() //Get current leader.
//Appending Entries And check for replication of entry.
for j := 1; j <= 5; j++ {
str := "test - " + strconv.Itoa(j)
myRaft.Cluster[leaderId].Append([]byte(str))
}
//Checking whether Entries are replicated.
for i := 0; i < PEERS; i++ {
for j := 1; j <= 5; j++ {
str := "test - " + strconv.Itoa(j)
ci := <-myRaft.Cluster[i].SM.CommMedium.CommitCh
tmp := ci.(CommitInfo)
expect(t, tmp, str, myRaft.Cluster[i].SM.id)
}
}
//Find non leader server and send timeout signal to it.
//for checking whether term increases or not.
leaderId = myRaft.GetLeader()
for i := 0; i < PEERS; i++ {
if i != leaderId {
time.Sleep(1 * time.Second)
myRaft.Cluster[i].SM.CommMedium.timeoutCh <- nil
break
}
}
//time.Sleep(1 * time.Second)
// myRaft.Cluster[leaderId].SM.CommMedium.timeoutCh <- nil
//Creating partion.
L := myRaft.GetLeader()
fmt.Println("Leader-", L)
switch L {
case 1, 2:
fmt.Println("11")
cluster.Partition([]int{1, 2}, []int{3, 4, 5})
case 3, 4:
fmt.Println("22")
cluster.Partition([]int{1, 2, 5}, []int{3, 4})
case 5:
fmt.Println("33")
cluster.Partition([]int{1, 2, 3}, []int{4, 5})
}
time.Sleep(2 * time.Second)
//time.Sleep(1 * time.Second)
leaderId = myRaft.GetLeader()
leaderId = myRaft.GetLeader()
//Appending Entries And check for replication of entry.
for j := 6; j <= 10; j++ {
str := "test - " + strconv.Itoa(j)
myRaft.Cluster[leaderId].Append([]byte(str))
}
//time.Sleep(1 * time.Second)
//Merging the cluster back.
cluster.Heal()
leaderId = myRaft.GetLeader()
//Appending Entries And check for replication of entry.
for j := 11; j <= 13; j++ {
str := "test - " + strconv.Itoa(j)
myRaft.Cluster[leaderId].Append([]byte(str))
}
//Printing the commit index of all nodes in cluster.
for k := 0; k < PEERS; k++ {
fmt.Println(myRaft.Cluster[k].SM.CommitIndex)
}
time.Sleep(2 * time.Second)
//Printing database entries of all nodes in cluster.
var j int64
for j = 0; j < 13; j++ {
res, err1 := myRaft.Cluster[0].Conf.lg.Get(j)
if err1 != nil {
fmt.Println(err1)
}
fmt.Print(res)
fmt.Print("\t")
res, err1 = myRaft.Cluster[1].Conf.lg.Get(j)
if err1 != nil {
fmt.Print(err1)
}
fmt.Print(res)
fmt.Print("\t")
res, err1 = myRaft.Cluster[2].Conf.lg.Get(j)
if err1 != nil {
fmt.Print(err1)
}
fmt.Print(res)
fmt.Print("\t")
res, err1 = myRaft.Cluster[3].Conf.lg.Get(j)
if err1 != nil {
fmt.Print(err1)
}
fmt.Print(res)
fmt.Print("\t")
res, err1 = myRaft.Cluster[4].Conf.lg.Get(j)
if err1 != nil {
fmt.Print(err1)
}
fmt.Print(res)
fmt.Print("\n")
}
/*
fileDB, err := leveldb.OpenFile("./Log_1", nil)
if err != nil {
fmt.Println(err)
}
iter := fileDB.NewIterator(nil, nil)
for iter.Next() {
key := iter.Key()
fmt.Println(string(key))
}
*/
_ = cluster
_ = leaderId
}