dioxus/packages/core/tests/lifecycle.rs

#![allow(unused, non_upper_case_globals)]
#![allow(non_snake_case)]

//! Tests for the lifecycle of components.
use dioxus::core::{ElementId, Mutation::*};
use dioxus::prelude::*;
use std::sync::{Arc, Mutex};

type Shared<T> = Arc<Mutex<T>>;

#[test]
fn manual_diffing() {
    struct AppProps {
        value: Shared<&'static str>,
    }

    static App: Component<AppProps> = |cx| {
        let val = cx.props.value.lock().unwrap();
        cx.render(rsx_without_templates! { div { "{val}" } })
    };

    let value = Arc::new(Mutex::new("Hello"));
    let mut dom = VirtualDom::new_with_props(App, AppProps { value: value.clone() });

    let _ = dom.rebuild();

    *value.lock().unwrap() = "goodbye";

    let edits = dom.rebuild();

    println!("edits: {:?}", edits);
}

#[test]
fn events_generate() {
    fn app(cx: Scope) -> Element {
        let count = cx.use_hook(|| 0);

        let inner = match *count {
            0 => {
                rsx_without_templates! {
                    div {
                        onclick: move |_| *count += 1,
                        div {
                            "nested"
                        }
                        "Click me!"
                    }
                }
            }
            _ => todo!(),
        };

        cx.render(inner)
    };

    let mut dom = VirtualDom::new(app);
    let mut channel = dom.get_scheduler_channel();
    assert!(dom.has_work());

    let edits = dom.rebuild();
    assert_eq!(
        edits.edits,
        [
            CreateElement { root: Some(1), tag: "div", children: 0 },
            NewEventListener { event_name: "click", scope: ScopeId(0), root: Some(1) },
            CreateElement { root: Some(2), tag: "div", children: 0 },
            CreateTextNode { root: Some(3), text: "nested" },
            AppendChildren { root: Some(2), children: vec![3] },
            CreateTextNode { root: Some(4), text: "Click me!" },
            AppendChildren { root: Some(1), children: vec![2, 4] },
            AppendChildren { root: Some(0), children: vec![1] }
        ]
    )
}

#[test]
fn components_generate() {
    fn app(cx: Scope) -> Element {
        let render_phase = cx.use_hook(|| 0);
        *render_phase += 1;

        cx.render(match *render_phase {
            1 => rsx_without_templates!("Text0"),
            2 => rsx_without_templates!(div {}),
            3 => rsx_without_templates!("Text2"),
            4 => rsx_without_templates!(Child {}),
            5 => rsx_without_templates!({ None as Option<()> }),
            6 => rsx_without_templates!("text 3"),
            7 => rsx_without_templates!({ (0..2).map(|f| rsx_without_templates!("text {f}")) }),
            8 => rsx_without_templates!(Child {}),
            _ => todo!(),
        })
    };

    fn Child(cx: Scope) -> Element {
        println!("Running child");
        cx.render(rsx_without_templates! {
            h1 {}
        })
    }

    let mut dom = VirtualDom::new(app);
    let edits = dom.rebuild();
    assert_eq!(
        edits.edits,
        [
            CreateTextNode { root: Some(1), text: "Text0" },
            AppendChildren { root: Some(0), children: vec![1] }
        ]
    );

    assert_eq!(
        dom.hard_diff(ScopeId(0)).edits,
        [
            CreateElement { root: Some(2), tag: "div", children: 0 },
            ReplaceWith { root: Some(1), nodes: vec![2] }
        ]
    );

    assert_eq!(
        dom.hard_diff(ScopeId(0)).edits,
        [
            CreateTextNode { root: Some(1), text: "Text2" },
            ReplaceWith { root: Some(2), nodes: vec![1] }
        ]
    );

    // child {}
    assert_eq!(
        dom.hard_diff(ScopeId(0)).edits,
        [
            CreateElement { root: Some(2), tag: "h1", children: 0 },
            ReplaceWith { root: Some(1), nodes: vec![2] }
        ]
    );

    // placeholder
    assert_eq!(
        dom.hard_diff(ScopeId(0)).edits,
        [
            CreatePlaceholder { root: Some(1) },
            ReplaceWith { root: Some(2), nodes: vec![1] }
        ]
    );

    assert_eq!(
        dom.hard_diff(ScopeId(0)).edits,
        [
            CreateTextNode { root: Some(2), text: "text 3" },
            ReplaceWith { root: Some(1), nodes: vec![2] }
        ]
    );

    assert_eq!(
        dom.hard_diff(ScopeId(0)).edits,
        [
            CreateTextNode { text: "text 0", root: Some(1) },
            CreateTextNode { text: "text 1", root: Some(3) },
            ReplaceWith { root: Some(2), nodes: vec![1, 3] },
        ]
    );

    assert_eq!(
        dom.hard_diff(ScopeId(0)).edits,
        [
            CreateElement { tag: "h1", root: Some(2), children: 0 },
            ReplaceWith { root: Some(1), nodes: vec![2] },
            Remove { root: Some(3) },
        ]
    );
}

#[test]
fn component_swap() {
    fn app(cx: Scope) -> Element {
        let render_phase = cx.use_hook(|| 0);
        *render_phase += 1;

        cx.render(match *render_phase {
            0 => rsx_without_templates!(
                div {
                    NavBar {}
                    Dashboard {}
                }
            ),
            1 => rsx_without_templates!(
                div {
                    NavBar {}
                    Results {}
                }
            ),
            2 => rsx_without_templates!(
                div {
                    NavBar {}
                    Dashboard {}
                }
            ),
            3 => rsx_without_templates!(
                div {
                    NavBar {}
                    Results {}
                }
            ),
            4 => rsx_without_templates!(
                div {
                    NavBar {}
                    Dashboard {}
                }
            ),
            _ => rsx_without_templates!("blah"),
        })
    };

    static NavBar: Component = |cx| {
        println!("running navbar");
        cx.render(rsx_without_templates! {
            h1 {
                "NavBar"
                {(0..3).map(|f| rsx_without_templates!(NavLink {}))}
            }
        })
    };

    static NavLink: Component = |cx| {
        println!("running navlink");
        cx.render(rsx_without_templates! {
            h1 {
                "NavLink"
            }
        })
    };

    static Dashboard: Component = |cx| {
        println!("running dashboard");
        cx.render(rsx_without_templates! {
            div {
                "dashboard"
            }
        })
    };

    static Results: Component = |cx| {
        println!("running results");
        cx.render(rsx_without_templates! {
            div {
                "results"
            }
        })
    };

    let mut dom = VirtualDom::new(app);
    let edits = dom.rebuild();
    dbg!(&edits);

    let edits = dom.work_with_deadline(|| false);
    dbg!(&edits);
    let edits = dom.work_with_deadline(|| false);
    dbg!(&edits);
    let edits = dom.work_with_deadline(|| false);
    dbg!(&edits);
    let edits = dom.work_with_deadline(|| false);
    dbg!(&edits);
}