check execution stage in tests

neutron-org · Sep 4, 2024 · 6b9ddc2 · 6b9ddc2
1 parent 584d0c8
commit 6b9ddc2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/wasmbinding/test/custom_message_test.go b/wasmbinding/test/custom_message_test.go
@@ -6,6 +6,7 @@ import (
 	"testing"
 
 	contractmanagertypes "github.com/neutron-org/neutron/v4/x/contractmanager/types"
+	types2 "github.com/neutron-org/neutron/v4/x/cron/types"
 
 	"cosmossdk.io/math"
 	wasmtypes "github.com/CosmWasm/wasmd/x/wasm/types"
@@ -721,6 +722,10 @@ func (suite *CustomMessengerTestSuite) TestAddRemoveSchedule() {
 		},
 	}
 
+	schedule, ok := suite.neutron.CronKeeper.GetSchedule(suite.ctx, "schedule1")
+	suite.True(ok)
+	suite.Equal(types2.ExecutionStage_EXECUTION_STAGE_END_BLOCKER, schedule.ExecutionStage)
+
 	// Dispatch AddSchedule message
 	_, err = suite.executeNeutronMsg(suite.contractAddress, msg)
 	suite.NoError(err)