From db5fdc6eff757de0a311d31d7002ca6b75f7b2c3 Mon Sep 17 00:00:00 2001
From: Amar Singh <asinghchrony@protonmail.com>
Date: Thu, 28 Jul 2022 18:41:23 -0400
Subject: [PATCH] Add epoch index to RelayStateSproofBuilder (for testing)
 (#1419)

* insert zero epoch index into relay sproof

* fix
---
 cumulus/test/relay-sproof-builder/src/lib.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/cumulus/test/relay-sproof-builder/src/lib.rs b/cumulus/test/relay-sproof-builder/src/lib.rs
index 179b817dbc..36b091eb7e 100644
--- a/cumulus/test/relay-sproof-builder/src/lib.rs
+++ b/cumulus/test/relay-sproof-builder/src/lib.rs
@@ -43,6 +43,7 @@ pub struct RelayStateSproofBuilder {
 	pub hrmp_egress_channel_index: Option<Vec<ParaId>>,
 	pub hrmp_channels: BTreeMap<relay_chain::v2::HrmpChannelId, AbridgedHrmpChannel>,
 	pub current_slot: relay_chain::v2::Slot,
+	pub current_epoch: u64,
 }
 
 impl Default for RelayStateSproofBuilder {
@@ -67,6 +68,7 @@ impl Default for RelayStateSproofBuilder {
 			hrmp_egress_channel_index: None,
 			hrmp_channels: BTreeMap::new(),
 			current_slot: 0.into(),
+			current_epoch: 0u64,
 		}
 	}
 }
@@ -153,7 +155,7 @@ impl RelayStateSproofBuilder {
 			for (channel, metadata) in self.hrmp_channels {
 				insert(relay_chain::well_known_keys::hrmp_channels(channel), metadata.encode());
 			}
-
+			insert(relay_chain::well_known_keys::EPOCH_INDEX.to_vec(), self.current_epoch.encode());
 			insert(relay_chain::well_known_keys::CURRENT_SLOT.to_vec(), self.current_slot.encode());
 		}