Skip to content

Commit

Permalink
refactor: add some test to cluster node
Browse files Browse the repository at this point in the history
  • Loading branch information
Tochemey committed Jan 21, 2024
1 parent 58c03ee commit 1a18ddc
Show file tree
Hide file tree
Showing 2 changed files with 39 additions and 9 deletions.
7 changes: 0 additions & 7 deletions cluster/node.go
Original file line number Diff line number Diff line change
Expand Up @@ -575,13 +575,6 @@ func (n *Node) consume() {
continue
}

// make sure to skip self
if n.AdvertisedAddress() == nodeLeft.NodeLeft {
// add some debug log
n.logger.Debug("skipping self")
continue
}

// TODO: need to cross check this calculation
// convert the timestamp to milliseconds
timeMilli := nodeLeft.Timestamp / int64(1e6)
Expand Down
41 changes: 39 additions & 2 deletions cluster/node_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,7 @@ import (
"github.com/tochemey/goakt/discovery"
"github.com/tochemey/goakt/discovery/nats"
internalpb "github.com/tochemey/goakt/internal/v1"
"github.com/tochemey/goakt/log"
eventspb "github.com/tochemey/goakt/pb/events/v1"
testkit "github.com/tochemey/goakt/testkit/discovery"
"github.com/travisjeffery/go-dynaport"
Expand Down Expand Up @@ -86,7 +87,8 @@ func TestSingleNode(t *testing.T) {
t.Setenv("NODE_NAME", "testNode")
t.Setenv("NODE_IP", host)

cluster, err := NewNode("test", serviceDiscovery)
logger := log.New(log.ErrorLevel, os.Stdout)
cluster, err := NewNode("test", serviceDiscovery, WithLogger(logger))
require.NotNil(t, cluster)
require.NoError(t, err)

Expand Down Expand Up @@ -285,7 +287,8 @@ func TestSingleNode(t *testing.T) {
t.Setenv("NODE_NAME", "testNode")
t.Setenv("NODE_IP", host)

cluster, err := NewNode("test", serviceDiscovery)
logger := log.New(log.WarningLevel, os.Stdout)
cluster, err := NewNode("test", serviceDiscovery, WithLogger(logger))
require.NotNil(t, cluster)
require.NoError(t, err)

Expand Down Expand Up @@ -320,6 +323,40 @@ func TestSingleNode(t *testing.T) {
require.NoError(t, cluster.Stop(ctx))
provider.AssertExpectations(t)
})
t.Run("With host node env vars not set", func(t *testing.T) {
// generate the ports for the single startNode
nodePorts := dynaport.Get(3)
gossipPort := nodePorts[0]
remotingPort := nodePorts[2]

// define discovered addresses
addrs := []string{
fmt.Sprintf("localhost:%d", gossipPort),
}

// mock the discovery provider
provider := new(testkit.Provider)
config := discovery.NewConfig()

provider.EXPECT().ID().Return("testDisco")
provider.EXPECT().Initialize().Return(nil)
provider.EXPECT().Register().Return(nil)
provider.EXPECT().Deregister().Return(nil)
provider.EXPECT().SetConfig(config).Return(nil)
provider.EXPECT().DiscoverPeers().Return(addrs, nil)
provider.EXPECT().Close().Return(nil)

// create the service discovery
serviceDiscovery := discovery.NewServiceDiscovery(provider, config)

// set the environments
t.Setenv("GOSSIP_PORT", strconv.Itoa(gossipPort))
t.Setenv("REMOTING_PORT", strconv.Itoa(remotingPort))

cluster, err := NewNode("test", serviceDiscovery)
require.Nil(t, cluster)
require.Error(t, err)
})
}

func TestMultipleNodes(t *testing.T) {
Expand Down

0 comments on commit 1a18ddc

Please sign in to comment.