Populate ScheduledEnclave upon registration for dev worker (#2514)

* Populate ScheduledEnclave entry in dev mode

* add test

* comment

pallets/teebag/src/lib.rs

@@ -222,7 +222,7 @@ pub mod pallet {
 	#[pallet::storage]
 	#[pallet::getter(fn scheduled_enclave)]
 	pub type ScheduledEnclave<T: Config> =
-		StorageMap<_, Blake2_128Concat, (WorkerType, SidechainBlockNumber), MrEnclave>;
+		StorageMap<_, Blake2_128Concat, (WorkerType, SidechainBlockNumber), MrEnclave, OptionQuery>;
 
 	#[pallet::storage]
 	#[pallet::getter(fn latest_sidechain_block_confirmation)]
@@ -468,7 +468,12 @@ pub mod pallet {
 						Error::<T>::EnclaveNotInSchedule
 					);
 				},
-				OperationalMode::Development => (),
+				OperationalMode::Development => {
+					// populate the registry if the entry doesn't exist
+					if !ScheduledEnclave::<T>::contains_key((worker_type, 0)) {
+						ScheduledEnclave::<T>::insert((worker_type, 0), enclave.mrenclave);
+					}
+				},
 			};
 			Self::add_enclave(&sender, &enclave)?;
 			Ok(().into())

pallets/teebag/src/tests.rs

@@ -103,6 +103,10 @@ fn register_enclave_dev_works_with_no_scheduled_enclave() {
 		assert_eq!(Teebag::enclave_count(WorkerType::Identity), 1);
 		assert_eq!(Teebag::enclave_count(WorkerType::BitAcross), 0);
 		assert_eq!(EnclaveRegistry::<Test>::get(alice()).unwrap(), enclave);
+		assert_eq!(
+			ScheduledEnclave::<Test>::get((WorkerType::default(), 0)).unwrap().to_vec(),
+			TEST4_MRENCLAVE.to_vec()
+		);
 	})
 }