diff --git a/chainio/clients/elcontracts/writer_test.go b/chainio/clients/elcontracts/writer_test.go index 499d963a..a0171426 100644 --- 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) }) }