Added a unit test to verify the dao account name

pokt-network · Apr 25, 2024 · 21d70e9 · 21d70e9
1 parent 93592fa
commit 21d70e9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/x/auth/keeper/account_test.go b/x/auth/keeper/account_test.go
@@ -3,11 +3,13 @@ package keeper
 import (
 	"testing"
 
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
 	appTypes "github.com/pokt-network/pocket-core/x/apps/types"
 	"github.com/pokt-network/pocket-core/x/auth/types"
+	govTypes "github.com/pokt-network/pocket-core/x/gov/types"
 	nodeTypes "github.com/pokt-network/pocket-core/x/nodes/types"
-	"github.com/stretchr/testify/assert"
-	"github.com/stretchr/testify/require"
 )
 
 var (
@@ -56,6 +58,11 @@ func TestGetPoolAddresses(t *testing.T) {
 			pool:    appTypes.StakedPoolName,
 			address: "63533fb8f43b4883a1f37265f1561ce7b1c6c307",
 		},
+		{
+			name:    "DAO owned address",
+			pool:    govTypes.DAOAccountName,
+			address: "6386713deb27b609daad5e2e32ee6591753e5f4e",
+		},
 	}
 	ctx, keeper := createTestInput(t, false, initialPower, 0)
 	for _, tt := range tests {

diff --git a/x/gov/keeper/dao.go b/x/gov/keeper/dao.go
@@ -2,6 +2,7 @@ package keeper
 
 import (
 	"fmt"
+
 	sdk "github.com/pokt-network/pocket-core/types"
 	exported2 "github.com/pokt-network/pocket-core/x/auth/exported"
 	"github.com/pokt-network/pocket-core/x/gov/types"