init

tests/e2e/main_test.go

@@ -134,6 +134,12 @@ func (s *BabylonSDKTestSuite) Test3BeginBlock() {
 	s.NoError(err)
 }
 
+// TODO: trigger EndBlock via s.ConsumerChain rather than ConsumerApp
+func (s *BabylonSDKTestSuite) Test4EndBlock() {
+	_, err := s.ConsumerApp.BabylonKeeper.EndBlocker(s.ConsumerChain.GetContext())
+	s.NoError(err)
+}
+
 // TearDownSuite runs once after all the suite's tests have been run
 func (s *BabylonSDKTestSuite) TearDownSuite() {
 	// Cleanup code here

tests/testdata/version.txt

@@ -1 +1 @@
-15cbae22a2cfe7c0590b401ae885c68bb1802530
+77a165d570b7f78d315d94f7adde850cd8874547

x/babylon/contract/out_message.go

@@ -5,6 +5,7 @@ import "time"
 // SudoMsg is a message sent from the Babylon module to a smart contract
 type SudoMsg struct {
 	BeginBlockMsg *BeginBlock `json:"begin_block,omitempty"`
+	EndBlockMsg   *EndBlock   `json:"end_block,omitempty"`
 }
 
 type BeginBlock struct {
@@ -14,3 +15,11 @@ type BeginBlock struct {
 	ChainID    string    `json:"chain_id"`     // ChainId is the chain ID of the block
 	AppHashHex string    `json:"app_hash_hex"` // AppHashHex is the app hash of the block in hex
 }
+
+type EndBlock struct {
+	Height     int64     `json:"height"`       // Height is the height of the block
+	HashHex    string    `json:"hash_hex"`     // HashHex is the hash of the block in hex
+	Time       time.Time `json:"time"`         // Time is the time of the block
+	ChainID    string    `json:"chain_id"`     // ChainId is the chain ID of the block
+	AppHashHex string    `json:"app_hash_hex"` // AppHashHex is the app hash of the block in hex
+}

x/babylon/keeper/abci.go

@@ -18,5 +18,10 @@ func (k *Keeper) BeginBlocker(ctx sdk.Context) error {
 // EndBlocker is called after every block
 func (k *Keeper) EndBlocker(ctx sdk.Context) ([]abci.ValidatorUpdate, error) {
 	defer telemetry.ModuleMeasureSince(types.ModuleName, time.Now(), telemetry.MetricKeyEndBlocker)
+
+	if err := k.SendEndBlockMsg(ctx); err != nil {
+		return []abci.ValidatorUpdate{}, err
+	}
+
 	return []abci.ValidatorUpdate{}, nil
 }

x/babylon/keeper/wasm.go

@@ -35,6 +35,32 @@ func (k Keeper) SendBeginBlockMsg(ctx sdk.Context) error {
 	return k.doSudoCall(ctx, addr, msg)
 }
 
+// SendEndBlockMsg sends a EndBlock sudo message to the BTC staking contract via sudo
+func (k Keeper) SendEndBlockMsg(ctx sdk.Context) error {
+	// get address of the BTC staking contract
+	addrStr := k.GetParams(ctx).BtcStakingContractAddress
+	if len(addrStr) == 0 {
+		// the BTC staking contract address is not set yet, skip sending EndBlockMsg
+		return nil
+	}
+	addr := sdk.MustAccAddressFromBech32(addrStr)
+
+	// construct the sudo message
+	headerInfo := ctx.HeaderInfo()
+	msg := contract.SudoMsg{
+		EndBlockMsg: &contract.EndBlock{
+			Height:     headerInfo.Height,
+			HashHex:    hex.EncodeToString(headerInfo.Hash),
+			Time:       headerInfo.Time,
+			ChainID:    headerInfo.ChainID,
+			AppHashHex: hex.EncodeToString(headerInfo.AppHash),
+		},
+	}
+
+	// send the sudo call
+	return k.doSudoCall(ctx, addr, msg)
+}
+
 // caller must ensure gas limits are set proper and handle panics
 func (k Keeper) doSudoCall(ctx sdk.Context, contractAddr sdk.AccAddress, msg contract.SudoMsg) error {
 	bz, err := json.Marshal(msg)