Skip to content

Commit

Permalink
feat: Support Staking Module ID usage
Browse files Browse the repository at this point in the history
  • Loading branch information
AntiD2ta committed Sep 12, 2024
1 parent 182cbf0 commit 2b68d73
Show file tree
Hide file tree
Showing 3 changed files with 87 additions and 0 deletions.
4 changes: 4 additions & 0 deletions internal/lido/contracts/contractsAddress.go
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@ const (
CSAccounting = "sepolia"
CSFeeDistributor = "gnosis"
MEVBoostRelayAllowedList = "mevboostrelayallowedlist"
Vebo = "vebo"
)

var deployedAddresses = map[string]address{
Expand All @@ -45,6 +46,9 @@ var deployedAddresses = map[string]address{
configs.NetworkMainnet: "0xF95f069F9AD107938F6ba802a3da87892298610E",
configs.NetworkHolesky: "0x2d86C5855581194a386941806E38cA119E50aEA3",
},
Vebo: {
configs.NetworkHolesky: "0xffDDF7025410412deaa05E3E1cE68FE53208afcb",
},
}

func DeployedAddresses(contractName string) address {
Expand Down
35 changes: 35 additions & 0 deletions internal/lido/contracts/stakingModuleIDs.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,35 @@
/*
Copyright 2022 Nethermind
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package contracts

import (
"fmt"
"math/big"

"github.com/NethermindEth/sedge/configs"
)

var stakingModuleIDs = map[string]*big.Int{
configs.NetworkHolesky: big.NewInt(4),
}

func StakingModuleID(network string) (*big.Int, error) {
stakingModuleID, found := stakingModuleIDs[network]
if !found {
return nil, fmt.Errorf("no staking module ID found for network %s", network)
}
return stakingModuleID, nil
}
48 changes: 48 additions & 0 deletions internal/lido/contracts/stakingModuleIDs_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
/*
Copyright 2022 Nethermind
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
package contracts

import (
"math/big"
"testing"

"github.com/NethermindEth/sedge/configs"
)

func TestStakingModuleID(t *testing.T) {
tcs := []struct {
network string
want *big.Int
wantErr bool
}{
{configs.NetworkHolesky, big.NewInt(4), false},
{"unknown", nil, true},
{configs.NetworkMainnet, nil, true},
}

for _, tc := range tcs {
got, err := StakingModuleID(tc.network)
if tc.wantErr && err == nil {
t.Errorf("StakingModuleID(%s) returned no error, want error", tc.network)
}
if !tc.wantErr && err != nil {
t.Errorf("StakingModuleID(%s) returned error: %v", tc.network, err)
}
if got.Cmp(tc.want) != 0 {
t.Errorf("StakingModuleID(%s) = %v, want %v", tc.network, got, tc.want)
}
}
}

0 comments on commit 2b68d73

Please sign in to comment.