fix compilation errors from prev rebase

Layr-Labs · Aug 6, 2024 · 5616d83 · 5616d83
1 parent 8583945
commit 5616d83
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 7 deletions.
diff --git a/chainio/clients/wallet/privatekey_wallet_test.go b/chainio/clients/wallet/privatekey_wallet_test.go
@@ -6,11 +6,11 @@ import (
 	"testing"
 	"time"
 
-	"github.com/Layr-Labs/eigensdk-go/chainio/clients/eth"
 	"github.com/Layr-Labs/eigensdk-go/signerv2"
 	"github.com/Layr-Labs/eigensdk-go/testutils"
 	"github.com/ethereum/go-ethereum/core/types"
 	"github.com/ethereum/go-ethereum/crypto"
+	"github.com/ethereum/go-ethereum/ethclient"
 	"github.com/stretchr/testify/require"
 )
 
@@ -28,7 +28,7 @@ func TestPrivateKeyWallet(t *testing.T) {
 		defer cancel()
 		anvilHttpEndpoint, err := anvilC.Endpoint(ctxWithTimeout, "http")
 		require.NoError(t, err)
-		ethClient, err := eth.NewClient(anvilHttpEndpoint)
+		ethClient, err := ethclient.Dial(anvilHttpEndpoint)
 		require.NoError(t, err)
 
 		ecdsaPrivKeyHex := "ac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80"

diff --git a/chainio/txmgr/geometric/geometric_example_test.go b/chainio/txmgr/geometric/geometric_example_test.go
@@ -15,6 +15,7 @@ import (
 	"github.com/ethereum/go-ethereum/common"
 	"github.com/ethereum/go-ethereum/core/types"
 	"github.com/ethereum/go-ethereum/crypto"
+	"github.com/ethereum/go-ethereum/ethclient"
 	"github.com/prometheus/client_golang/prometheus"
 )
 
@@ -52,7 +53,7 @@ func ExampleGeometricTxManager() {
 	// Output: Tx sent
 }
 
-func createTx(client eth.Client, address common.Address) *types.Transaction {
+func createTx(client eth.HttpBackend, address common.Address) *types.Transaction {
 	zeroAddr := common.HexToAddress("0x0")
 	nonce, err := client.PendingNonceAt(context.TODO(), address)
 	if err != nil {
@@ -64,9 +65,9 @@ func createTx(client eth.Client, address common.Address) *types.Transaction {
 	})
 }
 
-func createTxMgr(rpcUrl string, ecdsaPrivateKey *ecdsa.PrivateKey) (eth.Client, *GeometricTxManager) {
+func createTxMgr(rpcUrl string, ecdsaPrivateKey *ecdsa.PrivateKey) (eth.HttpBackend, *GeometricTxManager) {
 	logger := logging.NewTextSLogger(os.Stdout, &logging.SLoggerOptions{})
-	client, err := eth.NewClient(rpcUrl)
+	client, err := ethclient.Dial(rpcUrl)
 	if err != nil {
 		panic(err)
 	}

diff --git a/chainio/txmgr/geometric/geometric_test.go b/chainio/txmgr/geometric/geometric_test.go
@@ -8,7 +8,6 @@ import (
 	"testing"
 	"time"
 
-	"github.com/Layr-Labs/eigensdk-go/chainio/clients/eth"
 	"github.com/Layr-Labs/eigensdk-go/chainio/clients/wallet"
 	"github.com/Layr-Labs/eigensdk-go/crypto/ecdsa"
 	"github.com/Layr-Labs/eigensdk-go/logging"
@@ -17,6 +16,7 @@ import (
 	"github.com/ethereum/go-ethereum"
 	"github.com/ethereum/go-ethereum/common"
 	"github.com/ethereum/go-ethereum/core/types"
+	"github.com/ethereum/go-ethereum/ethclient"
 	"github.com/stretchr/testify/require"
 	"golang.org/x/sync/errgroup"
 )
@@ -327,7 +327,7 @@ func newIntegrationTestHarness(t *testing.T) *integrationTestHarness {
 	defer cancel()
 	anvilHttpEndpoint, err := anvilC.Endpoint(ctxWithTimeout, "http")
 	require.NoError(t, err)
-	anvilHttpClient, err := eth.NewClient(anvilHttpEndpoint)
+	anvilHttpClient, err := ethclient.Dial(anvilHttpEndpoint)
 	require.NoError(t, err)
 
 	ecdsaSk, ecdsaAddr, err := ecdsa.KeyAndAddressFromHexKey("0xac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80")