rtc: use consistent api between stop and non-stop

1 files changed, 6 insertions, 9 deletions

diff --git a/tests/stm32/src/bin/rtc.rs b/tests/stm32/src/bin/rtc.rs
index 5c80eb250..43cf4a411 100644
--- a/tests/stm32/src/bin/rtc.rs
+++ b/tests/stm32/src/bin/rtc.rs
@@ -9,11 +9,9 @@ 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;
+use embassy_stm32::rtc::Rtc;
 #[cfg(not(feature = "stop"))]
 use embassy_stm32::rtc::{Rtc, RtcConfig};
 use embassy_time::Timer;
@@ -31,23 +29,22 @@ async fn main(_spawner: Spawner) {
         .unwrap();
 
     #[cfg(not(feature = "stop"))]
-    let mut rtc = Rtc::new(p.RTC, RtcConfig::default());
+    let (mut rtc, time_provider) = Rtc::new(p.RTC, RtcConfig::default());
 
     #[cfg(feature = "stop")]
-    let time_provider = RtcTimeProvider::new(p.RTC);
+    let (rtc, time_provider) = Rtc::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"));
+    critical_section::with(|cs| {
+        rtc.borrow_mut(cs).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();