Add RAM-backed MockFlash for tests

Author: katyo

src/nor_flash.rs

@@ -379,3 +379,117 @@ where
 		Ok(())
 	}
 }
+
+/// Simple RAM-backed flash storage implementation for tests
+#[derive(Clone, Copy, Debug)]
+pub struct MockFlash<
+	const CAPACITY: usize,
+	const READ_SIZE: usize = 1,
+	const WRITE_SIZE: usize = 1,
+	const ERASE_SIZE: usize = { 1 << 10 },
+	const ERASE_BYTE: u8 = 0xff,
+> {
+	data: [u8; CAPACITY],
+}
+
+impl<
+		const CAPACITY: usize,
+		const READ_SIZE: usize,
+		const WRITE_SIZE: usize,
+		const ERASE_SIZE: usize,
+		const ERASE_BYTE: u8,
+	> Default for MockFlash<CAPACITY, READ_SIZE, WRITE_SIZE, ERASE_SIZE, ERASE_BYTE>
+{
+	fn default() -> Self {
+		Self {
+			data: [ERASE_BYTE; CAPACITY],
+		}
+	}
+}
+
+impl<
+		const CAPACITY: usize,
+		const READ_SIZE: usize,
+		const WRITE_SIZE: usize,
+		const ERASE_SIZE: usize,
+		const ERASE_BYTE: u8,
+	> core::ops::Deref for MockFlash<CAPACITY, READ_SIZE, WRITE_SIZE, ERASE_SIZE, ERASE_BYTE>
+{
+	type Target = [u8; CAPACITY];
+
+	fn deref(&self) -> &Self::Target {
+		&self.data
+	}
+}
+
+impl<
+		const CAPACITY: usize,
+		const READ_SIZE: usize,
+		const WRITE_SIZE: usize,
+		const ERASE_SIZE: usize,
+		const ERASE_BYTE: u8,
+	> core::ops::DerefMut for MockFlash<CAPACITY, READ_SIZE, WRITE_SIZE, ERASE_SIZE, ERASE_BYTE>
+{
+	fn deref_mut(&mut self) -> &mut Self::Target {
+		&mut self.data
+	}
+}
+
+impl<
+		const CAPACITY: usize,
+		const READ_SIZE: usize,
+		const WRITE_SIZE: usize,
+		const ERASE_SIZE: usize,
+		const ERASE_BYTE: u8,
+	> ErrorType for MockFlash<CAPACITY, READ_SIZE, WRITE_SIZE, ERASE_SIZE, ERASE_BYTE>
+{
+	type Error = NorFlashErrorKind;
+}
+
+impl<
+		const CAPACITY: usize,
+		const READ_SIZE: usize,
+		const WRITE_SIZE: usize,
+		const ERASE_SIZE: usize,
+		const ERASE_BYTE: u8,
+	> ReadNorFlash for MockFlash<CAPACITY, READ_SIZE, WRITE_SIZE, ERASE_SIZE, ERASE_BYTE>
+{
+	const READ_SIZE: usize = READ_SIZE;
+
+	fn read(&mut self, offset: u32, bytes: &mut [u8]) -> Result<(), Self::Error> {
+		check_read(self, offset, bytes.len())?;
+		bytes.copy_from_slice(&self.data[offset as usize..][..bytes.len()]);
+		Ok(())
+	}
+
+	fn capacity(&self) -> usize {
+		CAPACITY
+	}
+}
+
+impl<
+		const CAPACITY: usize,
+		const READ_SIZE: usize,
+		const WRITE_SIZE: usize,
+		const ERASE_SIZE: usize,
+		const ERASE_BYTE: u8,
+	> NorFlash for MockFlash<CAPACITY, READ_SIZE, WRITE_SIZE, ERASE_SIZE, ERASE_BYTE>
+{
+	const WRITE_SIZE: usize = WRITE_SIZE;
+	const ERASE_SIZE: usize = ERASE_SIZE;
+
+	fn write(&mut self, offset: u32, bytes: &[u8]) -> Result<(), Self::Error> {
+		check_write(self, offset, bytes.len())?;
+		for (dst, src) in self.data[offset as usize..].iter_mut().zip(bytes) {
+			*dst &= src;
+		}
+		Ok(())
+	}
+
+	fn erase(&mut self, from: u32, to: u32) -> Result<(), Self::Error> {
+		check_erase(self, from, to)?;
+		self.data[from as usize..to as usize].fill(ERASE_BYTE);
+		Ok(())
+	}
+}
+