From a79096f1d2dfe03fc71a350de2bcafb287a21253 Mon Sep 17 00:00:00 2001
From: tmpolaczyk <44604217+tmpolaczyk@users.noreply.github.com>
Date: Fri, 31 Mar 2023 17:08:07 +0200
Subject: [PATCH] Allow arbitrary key-values in RelayStateSproofBuilder (#2407)

---
 cumulus/test/relay-sproof-builder/src/lib.rs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/cumulus/test/relay-sproof-builder/src/lib.rs b/cumulus/test/relay-sproof-builder/src/lib.rs
index 93da853c8d..decc6ee3aa 100644
--- a/cumulus/test/relay-sproof-builder/src/lib.rs
+++ b/cumulus/test/relay-sproof-builder/src/lib.rs
@@ -45,6 +45,7 @@ pub struct RelayStateSproofBuilder {
 	pub current_slot: relay_chain::Slot,
 	pub current_epoch: u64,
 	pub randomness: relay_chain::Hash,
+	pub additional_key_values: Vec<(Vec<u8>, Vec<u8>)>,
 }
 
 impl Default for RelayStateSproofBuilder {
@@ -71,6 +72,7 @@ impl Default for RelayStateSproofBuilder {
 			current_slot: 0.into(),
 			current_epoch: 0u64,
 			randomness: relay_chain::Hash::default(),
+			additional_key_values: vec![],
 		}
 	}
 }
@@ -163,6 +165,10 @@ impl RelayStateSproofBuilder {
 				self.randomness.encode(),
 			);
 			insert(relay_chain::well_known_keys::CURRENT_SLOT.to_vec(), self.current_slot.encode());
+
+			for (key, value) in self.additional_key_values {
+				insert(key, value);
+			}
 		}
 
 		let root = backend.root().clone();