mock fee state

ava-labs · Dec 5, 2024 · 6e77fd5 · 6e77fd5
1 parent 6854b47
commit 6e77fd5
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/service/backend/pchain/construction_test.go b/service/backend/pchain/construction_test.go
@@ -222,7 +222,6 @@ func TestExportTxConstruction(t *testing.T) {
 	require.NoError(t, err)
 
 	t.Run("preprocess endpoint", func(t *testing.T) {
-		shouldMockGetFeeState(clientMock)
 		resp, err := backend.ConstructionPreprocess(
 			ctx,
 			&types.ConstructionPreprocessRequest{
@@ -236,6 +235,7 @@ func TestExportTxConstruction(t *testing.T) {
 	})
 
 	t.Run("metadata endpoint", func(t *testing.T) {
+		shouldMockGetFeeState(clientMock)
 		clientMock.EXPECT().GetBlockchainID(ctx, constants.PChain.String()).Return(pChainID, nil)
 		clientMock.EXPECT().GetBlockchainID(ctx, constants.CChain.String()).Return(cChainID, nil)
 
@@ -436,7 +436,6 @@ func TestImportTxConstruction(t *testing.T) {
 	require.NoError(t, err)
 
 	t.Run("preprocess endpoint", func(t *testing.T) {
-		shouldMockGetFeeState(clientMock)
 		resp, err := backend.ConstructionPreprocess(
 			ctx,
 			&types.ConstructionPreprocessRequest{
@@ -450,6 +449,7 @@ func TestImportTxConstruction(t *testing.T) {
 	})
 
 	t.Run("metadata endpoint", func(t *testing.T) {
+		shouldMockGetFeeState(clientMock)
 		clientMock.EXPECT().GetBlockchainID(ctx, constants.PChain.String()).Return(pChainID, nil)
 		clientMock.EXPECT().GetBlockchainID(ctx, constants.CChain.String()).Return(cChainID, nil)
 
@@ -687,7 +687,6 @@ func TestAddValidatorTxConstruction(t *testing.T) {
 	require.NoError(t, err)
 
 	t.Run("preprocess endpoint", func(t *testing.T) {
-		shouldMockGetFeeState(clientMock)
 		resp, err := backend.ConstructionPreprocess(
 			ctx,
 			&types.ConstructionPreprocessRequest{
@@ -927,7 +926,6 @@ func TestAddDelegatorTxConstruction(t *testing.T) {
 	require.NoError(t, err)
 
 	t.Run("preprocess endpoint", func(t *testing.T) {
-		shouldMockGetFeeState(clientMock)
 		resp, err := backend.ConstructionPreprocess(
 			ctx,
 			&types.ConstructionPreprocessRequest{