From eec87426e9d1e6f7a7c207e258565e7aa2f2ab29 Mon Sep 17 00:00:00 2001
From: ohkinozomu <nozomunoise@gmail.com>
Date: Tue, 6 Aug 2024 12:08:02 +0900
Subject: [PATCH] Add tests for etcd Peer

---
 cluster/raft/etcd/peer.go      |   5 +-
 cluster/raft/etcd/peer_test.go | 115 +++++++++++++++++++++++++++++++++
 2 files changed, 116 insertions(+), 4 deletions(-)
 create mode 100644 cluster/raft/etcd/peer_test.go

diff --git a/cluster/raft/etcd/peer.go b/cluster/raft/etcd/peer.go
index 5d8c1b1..0241b68 100644
--- a/cluster/raft/etcd/peer.go
+++ b/cluster/raft/etcd/peer.go
@@ -35,8 +35,6 @@ import (
 
 var (
 	ErrInvalidID = errors.New("node name must be a number")
-
-	appliedWaitDelay = 100 * time.Millisecond
 )
 
 type commit struct {
@@ -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() {
@@ -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 {
diff --git a/cluster/raft/etcd/peer_test.go b/cluster/raft/etcd/peer_test.go
new file mode 100644
index 0000000..e6a0825
--- /dev/null
+++ b/cluster/raft/etcd/peer_test.go
@@ -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)
+}