From afdfcaf8f90d3410d79f511a76bfe6431c4507df Mon Sep 17 00:00:00 2001
From: Egor Kostetskiy <kosegor@gmail.com>
Date: Mon, 13 Nov 2023 19:03:06 -0300
Subject: [PATCH] intest

---
 tests/e2e/e2e_metoken_test.go          |  7 ------
 x/metoken/keeper/intest/keeper_test.go |  7 ++++++
 x/metoken/keeper/intest/price_test.go  | 30 ++++++++++++++++++++++++++
 3 files changed, 37 insertions(+), 7 deletions(-)
 create mode 100644 x/metoken/keeper/intest/price_test.go

diff --git a/tests/e2e/e2e_metoken_test.go b/tests/e2e/e2e_metoken_test.go
index 57a36d1117..6e43a5cf33 100644
--- a/tests/e2e/e2e_metoken_test.go
+++ b/tests/e2e/e2e_metoken_test.go
@@ -25,7 +25,6 @@ func (s *E2ETest) TestMetokenSwapAndRedeem() {
 				mocks.ValidToken(mocks.USDTBaseDenom, mocks.USDTSymbolDenom, 6),
 				mocks.ValidToken(mocks.USDCBaseDenom, mocks.USDCSymbolDenom, 6),
 				mocks.ValidToken(mocks.ISTBaseDenom, mocks.ISTSymbolDenom, 6),
-				mocks.ValidToken(mocks.MeUSDDenom, mocks.MeUSDDenom, 6),
 			}
 
 			err := grpc.LeverageRegistryUpdate(s.AccountClient(0), tokens, nil)
@@ -114,12 +113,6 @@ func (s *E2ETest) TestMetokenSwapAndRedeem() {
 			s.checkMetokenBalance(testAddr.String(), mocks.MeUSDDenom)
 		},
 	)
-
-	s.Run(
-		"metoken_price_on_oracle", func() {
-			s.checkMetokenPriceInOracle(mocks.MeUSDDenom)
-		},
-	)
 }
 
 func (s *E2ETest) checkMetokenPriceInOracle(denom string) {
diff --git a/x/metoken/keeper/intest/keeper_test.go b/x/metoken/keeper/intest/keeper_test.go
index 7096180e60..b5d1a9b2f9 100644
--- a/x/metoken/keeper/intest/keeper_test.go
+++ b/x/metoken/keeper/intest/keeper_test.go
@@ -52,6 +52,7 @@ func initTestSuite(t *testing.T, registry []metoken.Index, balances []metoken.In
 		AllMedianPrices(gomock.Any()).
 		Return(mocks.ValidPrices()).
 		AnyTimes()
+	oracleMock.EXPECT().SetExchangeRateWithEvent(gomock.Any(), gomock.Any(), gomock.Any()).AnyTimes()
 
 	kb := keeper.NewKeeperBuilder(
 		app.AppCodec(),
@@ -108,6 +109,12 @@ func initTestSuite(t *testing.T, registry []metoken.Index, balances []metoken.In
 			mocks.ValidToken(mocks.ETHBaseDenom, mocks.ETHSymbolDenom, 18),
 		),
 	)
+	require.NoError(
+		app.LeverageKeeper.SetTokenSettings(
+			ctx,
+			mocks.ValidToken(mocks.MeUSDDenom, mocks.MeUSDDenom, 6),
+		),
+	)
 
 	return &KeeperTestSuite{
 		ctx:                 ctx,
diff --git a/x/metoken/keeper/intest/price_test.go b/x/metoken/keeper/intest/price_test.go
new file mode 100644
index 0000000000..3a6df76d45
--- /dev/null
+++ b/x/metoken/keeper/intest/price_test.go
@@ -0,0 +1,30 @@
+package intest
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+	"github.com/umee-network/umee/v6/util/checkers"
+	"github.com/umee-network/umee/v6/x/metoken"
+	"github.com/umee-network/umee/v6/x/metoken/mocks"
+)
+
+func TestPrice_SetPricesToOracle(t *testing.T) {
+	index := mocks.StableIndex(mocks.MeUSDDenom)
+
+	s := initTestSuite(t, nil, nil)
+	msgServer, ctx, app := s.msgServer, s.ctx, s.app
+
+	_, err := msgServer.GovUpdateRegistry(
+		ctx, &metoken.MsgGovUpdateRegistry{
+			Authority:   checkers.GovModuleAddr,
+			AddIndex:    []metoken.Index{index},
+			UpdateIndex: nil,
+		},
+	)
+	require := require.New(t)
+	require.NoError(err)
+
+	err = app.MetokenKeeperB.Keeper(&ctx).SetPricesToOracle()
+	require.NoError(err)
+}