[test] Add rom_e2e_bootstrap_write_enable_disable

Fixes #14458

Signed-off-by: Alphan Ulusoy <alphan@google.com>

sw/host/tests/rom/e2e_bootstrap_entry/src/main.rs

diff --git a/sw/host/tests/rom/e2e_bootstrap_entry/src/main.rs b/sw/host/tests/rom/e2e_bootstrap_entry/src/main.rs
index 29f1f83..4035dae 100644
--- a/sw/host/tests/rom/e2e_bootstrap_entry/src/main.rs
+++ b/sw/host/tests/rom/e2e_bootstrap_entry/src/main.rs
@@ -136,6 +136,20 @@
     Ok(())
 }
 
+fn test_write_enable_disable(transport: &TransportWrapper) -> Result<()> {
+    let spi = transport.spi("0")?;
+
+    assert_eq!(SpiFlash::read_status(&*spi)?, 0x0);
+
+    SpiFlash::set_write_enable(&*spi)?;
+    assert_eq!(SpiFlash::read_status(&*spi)?, 0x2);
+
+    SpiFlash::set_write_disable(&*spi)?;
+    assert_eq!(SpiFlash::read_status(&*spi)?, 0x0);
+
+    Ok(())
+}
+
 fn test_sfdp(opts: &Opts, transport: &TransportWrapper) -> Result<()> {
     let spi = transport.spi("0")?;
     let sfdp = SpiFlash::read_sfdp(&*spi)?;
@@ -188,5 +202,6 @@
     test_bootstrap_entry(&opts, &transport, BootstrapRequest::Yes)?;
     test_jedec_id(&opts, &transport)?;
     test_sfdp(&opts, &transport)?;
+    test_write_enable_disable(&transport)?;
     Ok(())
 }