mirror of
https://github.com/logos-messaging/logos-messaging-go-bindings.git
synced 2026-01-10 01:43:08 +00:00
Add new wrappers in temp branch
This commit is contained in:
aya
parent
9c78421cd0
commit
5ef7d03a49
348
testlibs/main-wrappers.go
Normal file
348
testlibs/main-wrappers.go
Normal file
@ -0,0 +1,348 @@
|
||||
package testlibs
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
"github.com/libp2p/go-libp2p/core/peer"
|
||||
|
||||
utilities "github.com/waku-org/waku-go-bindings/testlibs/utilities"
|
||||
"github.com/waku-org/waku-go-bindings/waku"
|
||||
"go.uber.org/zap"
|
||||
)
|
||||
|
||||
type WakuNodeWrapper struct {
|
||||
*waku.WakuNode
|
||||
config *waku.WakuConfig
|
||||
}
|
||||
|
||||
// This function create waku node from config and start it
|
||||
func Wrappers_StartWakuNode(customCfg *waku.WakuConfig, logger *zap.Logger) (*WakuNodeWrapper, error) {
|
||||
var nodeCfg waku.WakuConfig
|
||||
|
||||
if customCfg == nil {
|
||||
nodeCfg = *utilities.DefaultWakuConfig
|
||||
} else {
|
||||
nodeCfg = *customCfg
|
||||
}
|
||||
|
||||
nodeCfg.Discv5UdpPort = utilities.GenerateUniquePort()
|
||||
nodeCfg.TcpPort = utilities.GenerateUniquePort()
|
||||
|
||||
utilities.Debug("Create node with default config")
|
||||
node, err := waku.NewWakuNode(&nodeCfg, logger)
|
||||
if err != nil {
|
||||
utilities.Error("Can't create node")
|
||||
return nil, err
|
||||
}
|
||||
|
||||
utilities.Debug("Attempting to start WakuNode")
|
||||
wrapper := &WakuNodeWrapper{WakuNode: node, config: &nodeCfg} // Store config in wrapper
|
||||
|
||||
if err := utilities.CheckWakuNodeNull(logger, wrapper.WakuNode); err != nil {
|
||||
utilities.Error("Failed to start WakuNode", zap.Error(err))
|
||||
return nil, err
|
||||
}
|
||||
|
||||
err = wrapper.WakuNode.Start()
|
||||
if err != nil {
|
||||
utilities.Error("Failed to start WakuNode", zap.Error(err))
|
||||
return nil, err
|
||||
}
|
||||
utilities.Debug("Successfully started WakuNode")
|
||||
|
||||
return wrapper, nil
|
||||
}
|
||||
|
||||
// Stops the WakuNode.
|
||||
func (node *WakuNodeWrapper) Wrappers_Stop() error {
|
||||
if err := utilities.CheckWakuNodeNull(nil, node.WakuNode); err != nil {
|
||||
utilities.Error("Failed to stop WakuNode", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
utilities.Debug("Attempting to stop WakuNode")
|
||||
err := node.WakuNode.Stop()
|
||||
if err != nil {
|
||||
utilities.Error("Failed to stop WakuNode", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
utilities.Debug("Successfully stopped WakuNode")
|
||||
return nil
|
||||
}
|
||||
|
||||
// Destroys the WakuNode.
|
||||
func (node *WakuNodeWrapper) Wrappers_Destroy() error {
|
||||
if err := utilities.CheckWakuNodeNull(nil, node.WakuNode); err != nil {
|
||||
utilities.Error("Failed to destroy WakuNode", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
utilities.Debug("Attempting to destroy WakuNode")
|
||||
err := node.WakuNode.Destroy()
|
||||
if err != nil {
|
||||
utilities.Error("Failed to destroy WakuNode", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
utilities.Debug("Successfully destroyed WakuNode")
|
||||
return nil
|
||||
}
|
||||
|
||||
func (wrapper *WakuNodeWrapper) Wrappers_StopAndDestroy() error {
|
||||
if err := utilities.CheckWakuNodeNull(nil, wrapper.WakuNode); err != nil {
|
||||
utilities.Error("Failed to stop or destroy WakuNode", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
utilities.Debug("Attempting to stop WakuNode")
|
||||
err := wrapper.Stop()
|
||||
if err != nil {
|
||||
utilities.Error("Failed to stop WakuNode", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
utilities.Debug("Attempting to destroy WakuNode")
|
||||
err = wrapper.Destroy()
|
||||
if err != nil {
|
||||
utilities.Error("Failed to destroy WakuNode", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
utilities.Debug("Successfully stopped and destroyed WakuNode")
|
||||
return nil
|
||||
}
|
||||
|
||||
func (wrapper *WakuNodeWrapper) Wrappers_GetConnectedPeers() ([]peer.ID, error) {
|
||||
if err := utilities.CheckWakuNodeNull(nil, wrapper.WakuNode); err != nil {
|
||||
utilities.Error("Cannot proceed; node is nil", zap.Error(err))
|
||||
return nil, err
|
||||
}
|
||||
|
||||
peerID, err := wrapper.WakuNode.PeerID()
|
||||
if err != nil {
|
||||
utilities.Error("Failed to get PeerID of node", zap.Error(err))
|
||||
return nil, err
|
||||
}
|
||||
|
||||
utilities.Debug("Getting number of connected peers to node", zap.String("node", peerID.String()))
|
||||
|
||||
peers, err := wrapper.WakuNode.GetConnectedPeers()
|
||||
if err != nil {
|
||||
utilities.Error("Failed to get connected peers", zap.Error(err))
|
||||
return nil, err
|
||||
}
|
||||
|
||||
utilities.Debug("Successfully fetched connected peers",
|
||||
zap.Int("count", len(peers)),
|
||||
)
|
||||
return peers, nil
|
||||
}
|
||||
|
||||
func (wrapper *WakuNodeWrapper) Wrappers_GetNumConnectedRelayPeers(optPubsubTopic ...string) (int, error) {
|
||||
utilities.Debug("Wrappers_GetNumConnectedRelayPeers called")
|
||||
|
||||
if err := utilities.CheckWakuNodeNull(nil, wrapper.WakuNode); err != nil {
|
||||
utilities.Error("Cannot proceed; node is nil", zap.Error(err))
|
||||
return 0, err
|
||||
}
|
||||
|
||||
numPeers, err := wrapper.WakuNode.GetNumConnectedRelayPeers(optPubsubTopic...)
|
||||
if err != nil {
|
||||
utilities.Error("Failed to get number of connected relay peers", zap.Error(err))
|
||||
return 0, err
|
||||
}
|
||||
|
||||
utilities.Debug("Successfully fetched number of connected relay peers",
|
||||
zap.Int("count", numPeers),
|
||||
)
|
||||
return numPeers, nil
|
||||
}
|
||||
|
||||
func (wrapper *WakuNodeWrapper) Wrappers_ConnectPeer(targetNode *WakuNodeWrapper) error {
|
||||
|
||||
utilities.Debug("Connect node to peer")
|
||||
if err := utilities.CheckWakuNodeNull(nil, wrapper.WakuNode); err != nil {
|
||||
utilities.Error("Cannot call Connect; caller node is nil", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
if err := utilities.CheckWakuNodeNull(nil, targetNode.WakuNode); err != nil {
|
||||
utilities.Error("Cannot connect; target node is nil", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
targetPeerID, err := targetNode.WakuNode.PeerID()
|
||||
if err != nil {
|
||||
utilities.Error("Failed to get PeerID of target node", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
utilities.Debug("Get connected peers before attempting to connect")
|
||||
|
||||
connectedPeersBefore, err := wrapper.Wrappers_GetConnectedPeers()
|
||||
if err != nil {
|
||||
utilities.Debug("Could not fetch connected peers before connecting (might be none yet)", zap.Error(err))
|
||||
} else {
|
||||
utilities.Debug("Connected peers before connecting", zap.Int("count", len(connectedPeersBefore)))
|
||||
}
|
||||
|
||||
utilities.Debug("Attempt to connect to the target node")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), utilities.ConnectPeerTimeout)
|
||||
defer cancel()
|
||||
|
||||
targetAddr, err := targetNode.WakuNode.ListenAddresses()
|
||||
if err != nil || len(targetAddr) == 0 {
|
||||
utilities.Error("Failed to get listen addresses for target node", zap.Error(err))
|
||||
return errors.New("target node has no listen addresses")
|
||||
}
|
||||
|
||||
utilities.Debug("Connecting to peer", zap.String("address", targetAddr[0].String()))
|
||||
err = wrapper.WakuNode.Connect(ctx, targetAddr[0])
|
||||
if err != nil {
|
||||
utilities.Error("Failed to connect to peer", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
utilities.Debug("Get connected peers after attempting to connect")
|
||||
connectedPeersAfter, err := wrapper.Wrappers_GetConnectedPeers()
|
||||
if err != nil {
|
||||
utilities.Error("Failed to get connected peers after connecting", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
utilities.Debug("Connected peers after connecting", zap.Int("count", len(connectedPeersAfter)))
|
||||
|
||||
utilities.Debug("Check if the target peer is now connected")
|
||||
isConnected := false
|
||||
for _, peerID := range connectedPeersAfter {
|
||||
if peerID == targetPeerID {
|
||||
isConnected = true
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if !isConnected {
|
||||
err := errors.New("failed to connect; target peer is not in connected peers list")
|
||||
utilities.Error("Connect operation failed", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
utilities.Debug("Successfully connected to target peer", zap.String("targetPeerID", targetPeerID.String()))
|
||||
return nil
|
||||
}
|
||||
|
||||
func (wrapper *WakuNodeWrapper) Wrappers_DisconnectPeer(target *WakuNodeWrapper) error {
|
||||
|
||||
if err := utilities.CheckWakuNodeNull(nil, wrapper.WakuNode); err != nil {
|
||||
utilities.Error("Cannot call Disconnect; caller node is nil", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
if err := utilities.CheckWakuNodeNull(nil, target.WakuNode); err != nil {
|
||||
utilities.Error("Cannot disconnect; target node is nil", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
utilities.Debug("Check if nodes are peers first")
|
||||
|
||||
peerID, err := target.WakuNode.PeerID()
|
||||
if err != nil {
|
||||
utilities.Error("Failed to get PeerID of target node", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
connectedPeers, err := wrapper.Wrappers_GetConnectedPeers()
|
||||
if err != nil {
|
||||
utilities.Error("Failed to get connected peers", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
isPeer := false
|
||||
for _, connectedPeerID := range connectedPeers {
|
||||
if connectedPeerID == peerID {
|
||||
isPeer = true
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if !isPeer {
|
||||
err = errors.New("nodes are not connected as peers")
|
||||
utilities.Error("Cannot disconnect; nodes are not peers", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
utilities.Debug("Nodes are peers.. attempting to disconnect")
|
||||
err = wrapper.WakuNode.DisconnectPeerByID(peerID)
|
||||
if err != nil {
|
||||
utilities.Error("Failed to disconnect peer", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
utilities.Debug("Successfully disconnected peer", zap.String("peerID", peerID.String()))
|
||||
return nil
|
||||
}
|
||||
|
||||
// Wrapper for Peer Exchange Request
|
||||
func (wrapper *WakuNodeWrapper) Wrappers_PeerExchangeRequest(numPeers uint64) (uint64, error) {
|
||||
if err := utilities.CheckWakuNodeNull(nil, wrapper.WakuNode); err != nil {
|
||||
utilities.Error("Cannot perform peer exchange; node is nil", zap.Error(err))
|
||||
return 0, err
|
||||
}
|
||||
|
||||
utilities.Debug("Attempting peer exchange request", zap.Uint64("numPeers", numPeers))
|
||||
numRecvPeers, err := wrapper.WakuNode.PeerExchangeRequest(numPeers)
|
||||
if err != nil {
|
||||
utilities.Error("Failed to perform peer exchange request", zap.Error(err))
|
||||
return 0, err
|
||||
}
|
||||
|
||||
utilities.Debug("Successfully received peers from peer exchange", zap.Uint64("numPeersReceived", numRecvPeers))
|
||||
return numRecvPeers, nil
|
||||
}
|
||||
|
||||
// Wrapper for Start Discovery V5
|
||||
func (wrapper *WakuNodeWrapper) Wrappers_StartDiscV5() error {
|
||||
if err := utilities.CheckWakuNodeNull(nil, wrapper.WakuNode); err != nil {
|
||||
utilities.Error("Cannot start Discovery V5; node is nil", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
if wrapper.config == nil || !wrapper.config.Discv5Discovery {
|
||||
err := errors.New("discovery V5 is not enabled in the node configuration")
|
||||
utilities.Error("Failed to start Discovery V5", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
utilities.Debug("Attempting to start Discovery V5")
|
||||
err := wrapper.WakuNode.StartDiscV5()
|
||||
if err != nil {
|
||||
utilities.Error("Failed to start Discovery V5", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
utilities.Debug("Successfully started Discovery V5")
|
||||
return nil
|
||||
}
|
||||
|
||||
// Wrapper for Stop Discovery V5
|
||||
func (wrapper *WakuNodeWrapper) Wrappers_StopDiscV5() error {
|
||||
if err := utilities.CheckWakuNodeNull(nil, wrapper.WakuNode); err != nil {
|
||||
utilities.Error("Cannot stop Discovery V5; node is nil", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
if wrapper.config == nil || !wrapper.config.Discv5Discovery {
|
||||
err := errors.New("discovery V5 is not enabled in the node configuration")
|
||||
utilities.Error("Failed to stop Discovery V5", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
utilities.Debug("Attempting to stop Discovery V5")
|
||||
err := wrapper.WakuNode.StopDiscV5()
|
||||
if err != nil {
|
||||
utilities.Error("Failed to stop Discovery V5", zap.Error(err))
|
||||
return err
|
||||
}
|
||||
|
||||
utilities.Debug("Successfully stopped Discovery V5")
|
||||
return nil
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user