More test fixes

PeggyJV · Jul 2, 2024 · 5da5f7b · 5da5f7b
1 parent c0872d3
commit 5da5f7b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/integration_tests/cork_test.go b/integration_tests/cork_test.go
@@ -19,7 +19,7 @@ import (
 	"github.com/ethereum/go-ethereum/crypto"
 	"github.com/ethereum/go-ethereum/ethclient"
 	gravityTypes "github.com/peggyjv/gravity-bridge/module/v4/x/gravity/types"
-	corktypesv1 "github.com/peggyjv/sommelier/v7/x/cork/types/v1"
+	corktypesv2 "github.com/peggyjv/sommelier/v7/x/cork/types/v2"
 	"github.com/peggyjv/steward/steward_proto_go/steward_proto"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials"
@@ -790,7 +790,7 @@ func (s *IntegrationTestSuite) checkCellarExists(cellar common.Address) {
 	queryClient, err := s.chain.validators[0].GetQueryClient()
 	s.Require().NoError(err, "error getting query client")
 	s.Require().Eventuallyf(func() bool {
-		res, err := queryClient.QueryCellarIDs(context.Background(), &corktypesv1.QueryCellarIDsRequest{})
+		res, err := queryClient.QueryCellarIDs(context.Background(), &corktypesv2.QueryCellarIDsRequest{})
 		if err != nil || res == nil {
 			return false
 		}

diff --git a/integration_tests/validator.go b/integration_tests/validator.go
@@ -26,7 +26,7 @@ import (
 	"github.com/ethereum/go-ethereum/crypto"
 	gravitytypes "github.com/peggyjv/gravity-bridge/module/v4/x/gravity/types"
 	"github.com/peggyjv/sommelier/v7/app"
-	typesv1 "github.com/peggyjv/sommelier/v7/x/cork/types/v1"
+	typesv2 "github.com/peggyjv/sommelier/v7/x/cork/types/v2"
 	tmcfg "github.com/tendermint/tendermint/config"
 	tmos "github.com/tendermint/tendermint/libs/os"
 	"github.com/tendermint/tendermint/p2p"
@@ -52,13 +52,13 @@ type ethereumKey struct {
 	address    string
 }
 
-func (v *validator) GetQueryClient() (typesv1.QueryClient, error) {
+func (v *validator) GetQueryClient() (typesv2.QueryClient, error) {
 	clientCtx, err := v.clientContext("tcp://localhost:26657")
 	if err != nil {
 		return nil, err
 	}
 
-	return typesv1.NewQueryClient(clientCtx), nil
+	return typesv2.NewQueryClient(clientCtx), nil
 }
 
 func (v *validator) instanceName() string {