fix: Nil session tree logger

pokt-network · Dec 12, 2024 · 8f541cd · 8f541cd
1 parent 5d8c6dc
commit 8f541cd
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 3 deletions.
diff --git a/api/poktroll/application/tx.pulsar.go b/api/poktroll/application/tx.pulsar.go
diff --git a/pkg/relayer/session/session.go b/pkg/relayer/session/session.go
@@ -192,7 +192,7 @@ func (rs *relayerSessionsManager) ensureSessionTree(
 	// If the sessionTree does not exist, create and assign it to the
 	// sessionTreeWithSessionId map for the given supplier operator address.
 	if !ok {
-		sessionTree, err = NewSessionTree(sessionHeader, &supplierOperatorAccAddress, rs.storesDirectory)
+		sessionTree, err = NewSessionTree(sessionHeader, &supplierOperatorAccAddress, rs.storesDirectory, rs.logger)
 		if err != nil {
 			return nil, err
 		}

diff --git a/pkg/relayer/session/sessiontree.go b/pkg/relayer/session/sessiontree.go
@@ -72,6 +72,7 @@ func NewSessionTree(
 	sessionHeader *sessiontypes.SessionHeader,
 	supplierOperatorAddress *cosmostypes.AccAddress,
 	storesDirectory string,
+	logger polylog.Logger,
 ) (relayer.SessionTree, error) {
 	// Join the storePrefix and the session.sessionId and supplier's operator address to
 	// create a unique storePath.
@@ -102,6 +103,7 @@ func NewSessionTree(
 		sessionSMT:              trie,
 		sessionMu:               &sync.Mutex{},
 		supplierOperatorAddress: supplierOperatorAddress,
+		logger:                  logger,
 	}
 
 	return sessionTree, nil

diff --git a/testutil/testtree/tree.go b/testutil/testtree/tree.go
@@ -10,6 +10,7 @@ import (
 	"github.com/stretchr/testify/require"
 
 	"github.com/pokt-network/poktroll/pkg/crypto"
+	"github.com/pokt-network/poktroll/pkg/polylog"
 	"github.com/pokt-network/poktroll/pkg/relayer"
 	"github.com/pokt-network/poktroll/pkg/relayer/session"
 	"github.com/pokt-network/poktroll/testutil/testrelayer"
@@ -31,7 +32,7 @@ func NewFilledSessionTree(
 	t.Helper()
 
 	// Initialize an empty session tree with the given session header.
-	sessionTree := NewEmptySessionTree(t, sessionTreeHeader, supplierOperatorAddr)
+	sessionTree := NewEmptySessionTree(t, sessionTreeHeader, supplierOperatorAddr, ctx)
 
 	// Add numRelays of relays to the session tree.
 	FillSessionTree(
@@ -52,6 +53,7 @@ func NewEmptySessionTree(
 	t *testing.T,
 	sessionTreeHeader *sessiontypes.SessionHeader,
 	supplierOperatorAddr string,
+	ctx context.Context,
 ) relayer.SessionTree {
 	t.Helper()
 
@@ -71,6 +73,7 @@ func NewEmptySessionTree(
 		sessionTreeHeader,
 		&accAddress,
 		testSessionTreeStoreDir,
+		polylog.Ctx(ctx),
 	)
 	require.NoError(t, err)