From e94cfb76c60ad173c2c09300801f09c96e50b781 Mon Sep 17 00:00:00 2001
From: Gabriel mermelstein <gabrielmermelstein@gmail.com>
Date: Thu, 28 Nov 2024 11:24:07 +0200
Subject: [PATCH] passing first test

---
 nwaku.go      | 6 ++----
 nwaku_test.go | 8 --------
 2 files changed, 2 insertions(+), 12 deletions(-)

diff --git a/nwaku.go b/nwaku.go
index 6a00750..5232deb 100644
--- a/nwaku.go
+++ b/nwaku.go
@@ -369,20 +369,16 @@ type Waku struct {
 // Start implements node.Service, starting the background data propagation thread
 // of the Waku protocol.
 func (w *Waku) Start() error {
-	fmt.Println("---------------- Start 1 -------------")
 	err := w.node.Start()
-	fmt.Println("---------------- Start 2 -------------")
 	if err != nil {
 		return fmt.Errorf("failed to start nwaku node: %v", err)
 	}
 
 	peerID, err := w.node.PeerID()
-	fmt.Println("---------------- Start 3 -------------")
 	if err != nil {
 		return err
 	}
 
-	fmt.Println("---------------- Start 4 -------------")
 	w.logger.Info("WakuV2 PeerID", zap.Stringer("id", peerID))
 
 	return nil
@@ -531,6 +527,8 @@ func New(nwakuCfg *WakuConfig, logger *zap.Logger) (*Waku, error) {
 		node:    wakunode,
 		wakuCfg: nwakuCfg,
 		logger:  logger,
+		ctx:     ctx,
+		cancel:  cancel,
 	}, nil
 }
 
diff --git a/nwaku_test.go b/nwaku_test.go
index ce89929..a1f10df 100644
--- a/nwaku_test.go
+++ b/nwaku_test.go
@@ -4,7 +4,6 @@
 package wakuv2
 
 import (
-	"fmt"
 	"testing"
 	"time"
 
@@ -28,12 +27,9 @@ func TestDial(t *testing.T) {
 		TcpPort:         60020,
 	}
 
-	fmt.Println("------------ 1 -------------")
 	dialerNode, err := New(&dialerNodeWakuConfig, logger.Named("dialerNode"))
 	require.NoError(t, err)
-	fmt.Println("------------ 2 -------------")
 	require.NoError(t, dialerNode.Start())
-	fmt.Println("------------ 3 -------------")
 	time.Sleep(1 * time.Second)
 
 	// start node that will receive the dial
@@ -47,14 +43,10 @@ func TestDial(t *testing.T) {
 		TcpPort:         60021,
 	}
 	receiverNode, err := New(&receiverNodeWakuConfig, logger.Named("receiverNode"))
-	fmt.Println("------------ 4 -------------")
 	require.NoError(t, err)
 	require.NoError(t, receiverNode.Start())
-	fmt.Println("------------ 5 -------------")
 	time.Sleep(1 * time.Second)
-	fmt.Println("------------ 6 -------------")
 	receiverMultiaddr, err := receiverNode.node.ListenAddresses()
-	fmt.Println("------------ 7 -------------")
 	require.NoError(t, err)
 	require.NotNil(t, receiverMultiaddr)
 	// Check that both nodes start with no connected peers