Merge pull request #4838 from embassy-rs/lp2

stm32/low-power: cleanup and improve

1 files changed, 17 insertions, 1 deletions

diff --git a/tests/stm32/src/bin/rtc.rs b/tests/stm32/src/bin/rtc.rs
index 5fe98d807..5c80eb250 100644
--- a/tests/stm32/src/bin/rtc.rs
+++ b/tests/stm32/src/bin/rtc.rs
@@ -9,10 +9,14 @@ use chrono::{NaiveDate, NaiveDateTime};
 use common::*;
 use defmt::assert;
 use embassy_executor::Spawner;
+#[cfg(feature = "stop")]
+use embassy_stm32::low_power::reconfigure_rtc;
 use embassy_stm32::rcc::LsConfig;
+#[cfg(feature = "stop")]
+use embassy_stm32::rtc::RtcTimeProvider;
+#[cfg(not(feature = "stop"))]
 use embassy_stm32::rtc::{Rtc, RtcConfig};
 use embassy_time::Timer;
-
 #[embassy_executor::main]
 async fn main(_spawner: Spawner) {
     let mut config = config();
@@ -26,14 +30,26 @@ async fn main(_spawner: Spawner) {
         .and_hms_opt(10, 30, 15)
         .unwrap();
 
+    #[cfg(not(feature = "stop"))]
     let mut rtc = Rtc::new(p.RTC, RtcConfig::default());
 
+    #[cfg(feature = "stop")]
+    let time_provider = RtcTimeProvider::new(p.RTC);
+
+    #[cfg(not(feature = "stop"))]
     rtc.set_datetime(now.into()).expect("datetime not set");
 
+    #[cfg(feature = "stop")]
+    reconfigure_rtc(|rtc| rtc.set_datetime(now.into()).expect("datetime not set"));
+
     info!("Waiting 5 seconds");
     Timer::after_millis(5000).await;
 
+    #[cfg(not(feature = "stop"))]
     let then: NaiveDateTime = rtc.now().unwrap().into();
+    #[cfg(feature = "stop")]
+    let then: NaiveDateTime = time_provider.now().unwrap().into();
+
     let seconds = (then - now).num_seconds();
 
     info!("measured = {}", seconds);