From 6c424bf6242408722f1c652a963c7943ba9e9755 Mon Sep 17 00:00:00 2001
From: Alexey <4681325+qqrm@users.noreply.github.com>
Date: Fri, 22 Aug 2025 21:48:55 +0300
Subject: [PATCH] Add interval switch stream test

---
 src/app.rs                | 48 ++++++++++++++++++++++++++------
 tests/interval_restart.rs | 58 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 97 insertions(+), 9 deletions(-)
 create mode 100644 tests/interval_restart.rs

diff --git a/src/app.rs b/src/app.rs
index 94473eb..fc78278 100644
--- a/src/app.rs
+++ b/src/app.rs
@@ -917,7 +917,7 @@ fn ChartContainer() -> impl IntoView {
             <div style="display:flex;justify-content:space-between;margin-bottom:8px;width:800px;">
                 <AssetSelector set_status=set_status />
                 <div style="display:flex;gap:6px;">
-                    <TimeframeSelector chart=chart() />
+                    <TimeframeSelector chart=chart() set_status=set_status />
                 </div>
             </div>
 
@@ -1049,7 +1049,7 @@ fn ChartTooltip() -> impl IntoView {
 }
 
 #[component]
-fn TimeframeSelector(chart: RwSignal<Chart>) -> impl IntoView {
+fn TimeframeSelector(chart: RwSignal<Chart>, set_status: WriteSignal<String>) -> impl IntoView {
     let options = vec![
         TimeInterval::TwoSeconds,
         TimeInterval::OneMinute,
@@ -1066,11 +1066,24 @@ fn TimeframeSelector(chart: RwSignal<Chart>) -> impl IntoView {
                 children=move |interval| {
                     let label = interval.as_ref().to_string();
                     let chart_signal = chart;
+                    let status_signal = set_status;
                     view! {
                         <button
                             style="padding:4px 6px;border:none;border-radius:4px;background:#74c787;color:black;"
                             on:click=move |_| {
                                 current_interval().set(interval);
+                                if let Some(handle) = stream_abort_handles()
+                                    .with(|m| m.get(&current_symbol().get_untracked()).cloned())
+                                {
+                                    handle.abort();
+                                    stream_abort_handles().update(|m| {
+                                        m.remove(&current_symbol().get_untracked());
+                                    });
+                                }
+                                let status = status_signal;
+                                let _ = spawn_local_with_current_owner(async move {
+                                    start_websocket_stream(status).await;
+                                });
                                 chart_signal.update(|c| c.update_viewport_for_data());
                                 chart_signal.with_untracked(|c| {
                                     if c.get_candle_count() > 0 && with_global_renderer(|r| {
@@ -1196,10 +1209,12 @@ pub async fn start_websocket_stream(set_status: WriteSignal<String>) {
     ensure_chart(&symbol);
     let chart = get_chart_signal(&symbol).unwrap();
 
-    if let Some(_handle) = stream_abort_handles().with(|m| m.get(&symbol).cloned()) {
-        // Already streaming for this symbol
-        set_status.set("🔄 Using existing stream".to_string());
-        return;
+    if let Some(handle) = stream_abort_handles().with(|m| m.get(&symbol).cloned()) {
+        handle.abort();
+        stream_abort_handles().update(|m| {
+            m.remove(&symbol);
+        });
+        set_status.set("🔄 Restarting stream".to_string());
     }
 
     let interval = current_interval().get_untracked();
@@ -1441,26 +1456,41 @@ mod tests {
             .map_err(|_| format!("element with id {id} is not an HtmlInputElement"))
     }
 
-    #[wasm_bindgen_test]
-    fn timeframe_buttons_update_interval() {
+    #[wasm_bindgen_test(async)]
+    async fn timeframe_buttons_update_interval() {
+        use gloo_timers::future::sleep;
+        use std::time::Duration;
+
         let container = setup_container();
         let chart = create_rw_signal(Chart::new("test".to_string(), ChartType::Candlestick, 100));
-        leptos::mount_to(container.clone(), move || view! { <TimeframeSelector chart=chart /> });
+        let (_, set_status) = create_signal(String::new());
+        leptos::mount_to(
+            container.clone(),
+            move || view! { <TimeframeSelector chart=chart set_status=set_status /> },
+        );
 
         let two_sec = find_button(&container, "2s").expect("2s button not found");
         two_sec.click();
+        sleep(Duration::from_millis(10)).await;
+        abort_other_streams(&current_symbol().get_untracked());
         assert_eq!(current_interval().get(), TimeInterval::TwoSeconds);
 
         let five = find_button(&container, "5m").expect("5m button not found");
         five.click();
+        sleep(Duration::from_millis(10)).await;
+        abort_other_streams(&current_symbol().get_untracked());
         assert_eq!(current_interval().get(), TimeInterval::FiveMinutes);
 
         let fifteen = find_button(&container, "15m").expect("15m button not found");
         fifteen.click();
+        sleep(Duration::from_millis(10)).await;
+        abort_other_streams(&current_symbol().get_untracked());
         assert_eq!(current_interval().get(), TimeInterval::FifteenMinutes);
 
         let one_hour = find_button(&container, "1h").expect("1h button not found");
         one_hour.click();
+        sleep(Duration::from_millis(10)).await;
+        abort_other_streams(&current_symbol().get_untracked());
         assert_eq!(current_interval().get(), TimeInterval::OneHour);
     }
 
diff --git a/tests/interval_restart.rs b/tests/interval_restart.rs
new file mode 100644
index 0000000..cc3b42b
--- /dev/null
+++ b/tests/interval_restart.rs
@@ -0,0 +1,58 @@
+use gloo_timers::future::sleep;
+use leptos::*;
+use price_chart_wasm::app::{
+    current_interval, current_symbol, start_websocket_stream, stream_abort_handles,
+};
+use price_chart_wasm::domain::market_data::{
+    Candle, OHLCV, Price, TimeInterval, Timestamp, Volume, value_objects::Symbol,
+};
+use price_chart_wasm::global_state::{ensure_chart, get_chart_signal, push_realtime_candle};
+use std::time::Duration;
+use wasm_bindgen_test::*;
+
+#[wasm_bindgen_test(async)]
+async fn candle_after_interval_change() {
+    current_symbol().set(Symbol::from("BTCUSDT"));
+    ensure_chart(&current_symbol().get_untracked());
+    current_interval().set(TimeInterval::OneMinute);
+    let (_, set_status) = create_signal(String::new());
+    start_websocket_stream(set_status).await;
+    sleep(Duration::from_millis(10)).await;
+    let chart_signal = get_chart_signal(&current_symbol().get_untracked()).unwrap();
+    let before = chart_signal.with(|c| c.get_candle_count());
+
+    current_interval().set(TimeInterval::TwoSeconds);
+    if let Some(handle) =
+        stream_abort_handles().with(|m| m.get(&current_symbol().get_untracked()).cloned())
+    {
+        handle.abort();
+        stream_abort_handles().update(|m| {
+            m.remove(&current_symbol().get_untracked());
+        });
+    }
+    start_websocket_stream(set_status).await;
+    sleep(Duration::from_millis(10)).await;
+
+    let candle = Candle::new(
+        Timestamp::from_millis(1),
+        OHLCV::new(
+            Price::from(1.0),
+            Price::from(1.0),
+            Price::from(1.0),
+            Price::from(1.0),
+            Volume::from(1.0),
+        ),
+    );
+    push_realtime_candle(candle);
+    sleep(Duration::from_millis(10)).await;
+    assert!(chart_signal.with(|c| c.get_candle_count()) > before);
+
+    if let Some(handle) =
+        stream_abort_handles().with(|m| m.get(&current_symbol().get_untracked()).cloned())
+    {
+        handle.abort();
+        stream_abort_handles().update(|m| {
+            m.remove(&current_symbol().get_untracked());
+        });
+    }
+}