Skip to content

Commit

Permalink
Merge pull request #113 from ohkinozomu/etcd-test
Browse files Browse the repository at this point in the history
Add tests for etcd Peer
  • Loading branch information
wind-c authored Aug 6, 2024
2 parents 2ea6541 + eec8742 commit 7c0cd02
Show file tree
Hide file tree
Showing 2 changed files with 116 additions and 4 deletions.
5 changes: 1 addition & 4 deletions cluster/raft/etcd/peer.go
Original file line number Diff line number Diff line change
Expand Up @@ -35,8 +35,6 @@ import (

var (
ErrInvalidID = errors.New("node name must be a number")

appliedWaitDelay = 100 * time.Millisecond
)

type commit struct {
Expand Down Expand Up @@ -285,7 +283,6 @@ func (p *Peer) serveRaft() {
default:
log.Fatal("[raft] failed to serve rafthttp", "error", err)
}
close(p.httpStopC)
}

func (p *Peer) serveChannels() {
Expand Down Expand Up @@ -622,7 +619,7 @@ func (p *Peer) IsApplyRight() bool {
}

func (p *Peer) GetLeader() (addr, id string) {
return
return "", strconv.FormatUint(p.node.Status().SoftState.Lead, 10)
}

func (p *Peer) GenPeersFile(file string) error {
Expand Down
115 changes: 115 additions & 0 deletions cluster/raft/etcd/peer_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,115 @@
package etcd

import (
"net"
"strconv"
"testing"
"time"

"github.com/stretchr/testify/require"
"github.com/wind-c/comqtt/v2/cluster/log"
"github.com/wind-c/comqtt/v2/cluster/message"
"github.com/wind-c/comqtt/v2/config"
"github.com/wind-c/comqtt/v2/mqtt/packets"
)

func createTestPeer(t *testing.T) *Peer {
conf := &config.Cluster{
NodeName: "1",
BindAddr: "127.0.0.1",
RaftImpl: config.RaftImplEtcd,
RaftPort: 8948,
RaftDir: t.TempDir(),
RaftBootstrap: true,
}
notifyCh := make(chan *message.Message, 1)
peer, err := Setup(conf, notifyCh)
require.Nil(t, err)
return peer
}

func TestJoinAndLeave(t *testing.T) {
log.Init(log.DefaultOptions())
peer := createTestPeer(t)
defer peer.Stop()

node2ID := "2"
node2Host := "127.0.0.1"
node2Port := 8949
conf2 := &config.Cluster{
NodeName: node2ID,
BindAddr: node2Host,
RaftImpl: config.RaftImplEtcd,
RaftPort: node2Port,
RaftDir: t.TempDir(),
RaftBootstrap: true,
}
notifyCh := make(chan *message.Message, 1)
_, err := Setup(conf2, notifyCh)
require.NoError(t, err)

node2IDUint, err := strconv.ParseUint(node2ID, 10, 64)
require.NoError(t, err)

// Test Join
err = peer.Join(node2ID, net.JoinHostPort(node2Host, strconv.Itoa(node2Port)))
require.NoError(t, err)

time.Sleep(2 * time.Second)

found := false
for _, majorityConfig := range peer.node.Status().Config.Voters {
if _, ok := majorityConfig[node2IDUint]; ok {
found = true
break
}
}
require.True(t, found)

err = peer.Leave(node2ID)
require.NoError(t, err)

time.Sleep(2 * time.Second)

found = false
for _, c := range peer.node.Status().Config.Voters {
t.Log(found)
if _, ok := c[node2IDUint]; ok {
found = true
break
}
}
require.False(t, found)
}

func TestProposeAndLookup(t *testing.T) {
peer := createTestPeer(t)
defer peer.Stop()

msg := &message.Message{
Type: packets.Subscribe,
NodeID: "1",
Payload: []byte("filter"),
}

err := peer.Propose(msg)
require.NoError(t, err)

key := "filter"
expectedValue := "1"

time.Sleep(2 * time.Second)

result := peer.Lookup(key)
require.Equal(t, []string{expectedValue}, result)
}

func TestGetLeader(t *testing.T) {
peer := createTestPeer(t)
defer peer.Stop()

time.Sleep(2 * time.Second)

_, id := peer.GetLeader()
require.Equal(t, "1", id)
}

0 comments on commit 7c0cd02

Please sign in to comment.