Add error message assert at invalid config tests

Layr-Labs · Jan 29, 2025 · e260bb9 · e260bb9
1 parent 43565a7
commit e260bb9
Showing 1 changed file with 45 additions and 0 deletions.
diff --git a/chainio/clients/elcontracts/writer_test.go b/chainio/clients/elcontracts/writer_test.go
@@ -2,6 +2,7 @@ package elcontracts_test
 
 import (
 	"context"
+	"fmt"
 	"math/big"
 	"os"
 	"testing"
@@ -1155,6 +1156,7 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		assert.Equal(t, err.Error(), elcontracts.CommonErrorMissingContract("DelegationManager"))
 	})
 
 	t.Run("update operator details", func(t *testing.T) {
@@ -1165,6 +1167,7 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		assert.Equal(t, err.Error(), elcontracts.CommonErrorMissingContract("DelegationManager"))
 	})
 
 	t.Run("update metadata URI", func(t *testing.T) {
@@ -1176,6 +1179,7 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		assert.Equal(t, err.Error(), elcontracts.CommonErrorMissingContract("DelegationManager"))
 	})
 
 	t.Run("deposit erc20 into strategy", func(t *testing.T) {
@@ -1187,6 +1191,7 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		assert.Equal(t, err.Error(), elcontracts.CommonErrorMissingContract("StrategyManager"))
 	})
 
 	t.Run("set claimer for", func(t *testing.T) {
@@ -1197,6 +1202,7 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		assert.Equal(t, err.Error(), elcontracts.CommonErrorMissingContract("RewardsCoordinator"))
 	})
 
 	t.Run("process claim and process claims", func(t *testing.T) {
@@ -1225,6 +1231,7 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		assert.Equal(t, err.Error(), elcontracts.CommonErrorMissingContract("RewardsCoordinator"))
 
 		receipt, err = chainWriter.ProcessClaims(
 			context.Background(),
@@ -1234,6 +1241,7 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		assert.Equal(t, err.Error(), elcontracts.CommonErrorMissingContract("RewardsCoordinator"))
 	})
 
 	t.Run("set operator AVS split", func(t *testing.T) {
@@ -1246,6 +1254,7 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		assert.Equal(t, err.Error(), elcontracts.CommonErrorMissingContract("RewardsCoordinator"))
 	})
 
 	t.Run("set operator PI split", func(t *testing.T) {
@@ -1257,6 +1266,7 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		assert.Equal(t, err.Error(), elcontracts.CommonErrorMissingContract("RewardsCoordinator"))
 	})
 
 	t.Run("modify allocations", func(t *testing.T) {
@@ -1285,6 +1295,7 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		assert.Equal(t, err.Error(), elcontracts.CommonErrorMissingContract("AllocationManager"))
 	})
 
 	t.Run("set allocation delay", func(t *testing.T) {
@@ -1296,6 +1307,7 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		assert.Equal(t, err.Error(), elcontracts.CommonErrorMissingContract("AllocationManager"))
 	})
 
 	t.Run("deregister from operator sets", func(t *testing.T) {
@@ -1314,6 +1326,7 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		assert.Equal(t, err.Error(), elcontracts.CommonErrorMissingContract("AllocationManager"))
 	})
 
 	t.Run("register for operator sets", func(t *testing.T) {
@@ -1340,6 +1353,7 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		assert.Equal(t, err.Error(), elcontracts.CommonErrorMissingContract("AllocationManager"))
 	})
 
 	t.Run("remove permission", func(t *testing.T) {
@@ -1363,6 +1377,12 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+
+		expectedErrMsg := fmt.Sprintf(
+			"Nested error(2) - Failed to create a new remove permission Tx: %s",
+			elcontracts.CommonErrorMissingContract("PermissionController"),
+		)
+		assert.Equal(t, err.Error(), expectedErrMsg)
 	})
 
 	t.Run("set permission", func(t *testing.T) {
@@ -1386,6 +1406,11 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		expectedErrMsg := fmt.Sprintf(
+			"Nested error(2) - Failed to create a new set permission Tx: %s",
+			elcontracts.CommonErrorMissingContract("PermissionController"),
+		)
+		assert.Equal(t, err.Error(), expectedErrMsg)
 	})
 
 	t.Run("accept admin", func(t *testing.T) {
@@ -1401,6 +1426,11 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		expectedErrMsg := fmt.Sprintf(
+			"Nested error(2) - Failed to create a new accept admin Tx: %s",
+			elcontracts.CommonErrorMissingContract("PermissionController"),
+		)
+		assert.Equal(t, err.Error(), expectedErrMsg)
 	})
 
 	t.Run("add pending admin", func(t *testing.T) {
@@ -1419,6 +1449,11 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		expectedErrMsg := fmt.Sprintf(
+			"Nested error(2) - Failed to create a new add pending admin Tx: %s",
+			elcontracts.CommonErrorMissingContract("PermissionController"),
+		)
+		assert.Equal(t, err.Error(), expectedErrMsg)
 	})
 
 	t.Run("remove admin", func(t *testing.T) {
@@ -1437,6 +1472,11 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		expectedErrMsg := fmt.Sprintf(
+			"Nested error(2) - Failed to create a new remove admin Tx: %s",
+			elcontracts.CommonErrorMissingContract("PermissionController"),
+		)
+		assert.Equal(t, err.Error(), expectedErrMsg)
 	})
 
 	t.Run("remove pending admin", func(t *testing.T) {
@@ -1453,6 +1493,11 @@ func TestInvalidConfigChainWriter(t *testing.T) {
 		)
 		assert.Error(t, err)
 		assert.Nil(t, receipt)
+		expectedErrMsg := fmt.Sprintf(
+			"Nested error(2) - Failed to create a new remove pending admin Tx: %s",
+			elcontracts.CommonErrorMissingContract("PermissionController"),
+		)
+		assert.Equal(t, err.Error(), expectedErrMsg)
 	})
 }