loro 1.17.1

#[cfg(loom)]
mod loom_test {

    use loom::{explore, stop_exploring, sync::atomic::AtomicUsize, thread::yield_now};
    use loro::{ExportMode, LoroDoc, UpdateOptions};
    use std::{sync::atomic::Ordering, thread::sleep, time::Duration};

    #[test]
    fn concurrently_inserting_text_content() {
        loom::model(|| {
            let doc = LoroDoc::new();
            let doc1 = doc.clone();
            let doc2 = doc.clone();
            let h0 = loom::thread::spawn(move || {
                doc1.get_text("text").insert(0, "1").unwrap();
            });
            let h1 = loom::thread::spawn(move || {
                for _ in 0..2 {
                    doc2.get_text("text").insert(0, "2").unwrap();
                }
            });

            if let Err(e) = h0.join() {
                eprintln!("Thread h0 failed: {:?}", e);
                panic!("Thread h0 failed: {:?}", e);
            }

            if let Err(e) = h1.join() {
                eprintln!("Thread h1 failed: {:?}", e);
                panic!("Thread h1 failed: {:?}", e);
            }
            let text = doc.get_text("text");
            assert_eq!(text.len_utf8(), 3);
            dbg!("{}", text.to_string());
        });
    }

    #[test]
    fn concurrently_creating_events_with_subscriber() {
        loom::model(|| {
            let doc = LoroDoc::new();
            let count = std::sync::Arc::new(AtomicUsize::new(0));
            let count_clone = count.clone();
            let _sub = doc.subscribe_root(std::sync::Arc::new(move |e| {
                stop_exploring();
                for e in e.events {
                    let v = e.diff.as_text().unwrap();
                    for v in v {
                        match &v {
                            loro::TextDelta::Retain { .. } => unreachable!(),
                            loro::TextDelta::Delete { .. } => unreachable!(),
                            loro::TextDelta::Insert { insert, .. } => {
                                count_clone.fetch_add(insert.len(), Ordering::SeqCst);
                            }
                        }
                    }
                }
                explore();
            }));
            let doc1 = doc.clone();
            let doc2 = doc.clone();
            let h0 = loom::thread::spawn(move || {
                doc1.get_text("text").insert(0, "1").unwrap();
                doc1.commit();
            });
            let h1 = loom::thread::spawn(move || {
                doc2.get_text("text").insert(0, "2").unwrap();
                doc2.commit();
            });

            h0.join().unwrap();
            h1.join().unwrap();
            let text = doc.get_text("text");
            assert_eq!(text.len_utf8(), 2);
            assert_eq!(count.load(Ordering::SeqCst), 2);
        });
    }

    #[test]
    fn concurrent_callbacks_modifying_same_doc() {
        loom::model(|| {
            let doc = LoroDoc::new();
            let text_id = "shared_text";

            // Set up a condition variable to control thread execution
            let pair =
                std::sync::Arc::new((loom::sync::Mutex::new(false), loom::sync::Condvar::new()));
            let pair_clone1 = pair.clone();
            let pair_clone2 = pair.clone();

            // Create a counter to track callbacks
            let counter = std::sync::Arc::new(AtomicUsize::new(0));
            let counter_clone = counter.clone();

            // First subscription modifies the document in its callback
            let doc1 = doc.clone();
            let _sub1 = doc.subscribe_root(std::sync::Arc::new(move |_| {
                let count = counter_clone.fetch_add(1, Ordering::SeqCst);

                // Only react to the first few events to avoid infinite loops
                if count < 2 {
                    // This will trigger another event
                    doc1.get_text(text_id).insert(0, "A").unwrap();
                    doc1.commit();

                    // Signal the other thread
                    let (lock, cvar) = &*pair_clone1;
                    let mut started = lock.lock().unwrap();
                    *started = true;
                    cvar.notify_one();
                }
            }));

            // Second thread also has a subscription that modifies the document
            let doc2 = doc.clone();
            let doc3 = doc.clone();

            let h = loom::thread::spawn(move || {
                // Wait for the first thread's callback to trigger
                let (lock, cvar) = &*pair_clone2;
                let mut started = lock.lock().unwrap();
                while !*started {
                    // Wait until the condition variable is signaled
                    started = cvar.wait(started).unwrap();
                }

                drop(started);
                // Now both threads are modifying the document based on events
                doc2.get_text(text_id).insert(0, "B").unwrap();
                doc2.commit();

                doc2.commit();
            });

            // Start the chain of events
            doc3.get_text(text_id).insert(0, "Start").unwrap();
            doc3.commit();

            h.join().unwrap();
        });
    }

    #[test]
    fn concurrent_document_checkout_with_modifications() {
        let mut builder = loom::model::Builder::new();
        builder.max_branches = 3000;
        builder.check(|| {
            let doc = LoroDoc::new();
            doc.set_detached_editing(true);

            // Set up multiple frontiers to test checkout
            doc.get_text("text").insert(0, "Initial state").unwrap();
            doc.commit();
            let initial_frontier = doc.state_frontiers();

            doc.get_text("text")
                .insert(doc.get_text("text").len_utf8(), " - First update")
                .unwrap();
            doc.commit();
            let second_frontier = doc.state_frontiers();

            doc.get_text("text")
                .insert(doc.get_text("text").len_utf8(), " - Second update")
                .unwrap();
            doc.commit();
            let third_frontier = doc.state_frontiers();

            // Now create threads that will concurrently checkout and modify
            let doc1 = doc.clone();
            let doc2 = doc.clone();
            let initial_clone = initial_frontier.clone();
            let second_clone = second_frontier.clone();
            let third_clone = third_frontier.clone();

            let h1 = loom::thread::spawn(move || {
                // First checkout to an earlier state
                doc1.checkout(&initial_clone).unwrap();

                // Modify the document
                doc1.get_text("text")
                    .insert(doc1.get_text("text").len_utf8(), " - Thread 1 modification")
                    .unwrap();
                doc1.commit();

                // Sleep to increase chance of race condition
                yield_now();

                // Checkout to a later state
                doc1.checkout(&third_clone).unwrap();

                // Modify again
                doc1.get_text("text")
                    .insert(0, " - Thread 1 after checkout")
                    .unwrap();
                doc1.commit();
            });

            let h2 = loom::thread::spawn(move || {
                yield_now();

                // Checkout to the middle state
                doc2.checkout(&second_clone).unwrap();

                // Modify
                doc2.get_text("text")
                    .insert(doc2.get_text("text").len_utf8(), " - Thread 2 modification")
                    .unwrap();
                doc2.commit();

                yield_now();

                // Checkout to latest
                doc2.checkout_to_latest();

                // Modify again
                doc2.get_text("text")
                    .insert(0, " - Thread 2 after checkout")
                    .unwrap();
                doc2.commit();
            });

            h1.join().unwrap();
            h2.join().unwrap();
        });
    }

    #[test]
    fn concurrently_import_export() {
        let mut builder = loom::model::Builder::new();
        builder.max_branches = 2000;
        builder.check(|| {
            let doc1 = LoroDoc::new();
            let doc1_clone = doc1.clone();
            let doc1_clone2 = doc1.clone();
            let doc2 = LoroDoc::new();
            let doc2_clone = doc2.clone();
            let doc2_clone2 = doc2.clone();

            let mut handlers = vec![];
            handlers.push(loom::thread::spawn(move || {
                doc1_clone.get_text("text").insert(0, "1").unwrap();
                doc1_clone.commit();
                doc1_clone.get_text("text").insert(0, "1").unwrap();
                doc1_clone.commit();
            }));
            handlers.push(loom::thread::spawn(move || {
                doc2_clone.get_text("text").insert(0, "2").unwrap();
                doc2_clone.commit();
            }));
            handlers.push(loom::thread::spawn(move || {
                let e = &doc1_clone2
                    .export(ExportMode::updates(&doc2_clone2.oplog_vv()))
                    .unwrap();
                doc2_clone2.import(e).unwrap();
                yield_now();
                let e = &doc2_clone2
                    .export(ExportMode::updates(&doc1_clone2.oplog_vv()))
                    .unwrap();
                doc1_clone2.import(e).unwrap();
            }));

            for h in handlers {
                h.join().unwrap();
            }
        });
    }

    #[test]
    fn local_edits_during_batch_import() {
        let mut builder = loom::model::Builder::new();
        builder.max_branches = 3000;
        builder.check(move || {
            let doc = LoroDoc::new();
            doc.get_text("text").insert(0, "hello").unwrap();
            let update_a = doc.export(ExportMode::all_updates()).unwrap();
            doc.get_text("text")
                .update("yo! hello", UpdateOptions::default())
                .unwrap();
            let update_b = doc.export(ExportMode::all_updates()).unwrap();

            let mut handlers = vec![];
            let doc = LoroDoc::new();
            let doc_clone = doc.clone();

            handlers.push(loom::thread::spawn(move || {
                doc_clone.get_text("text").insert(0, "1").unwrap();
                doc_clone.commit();
                doc_clone.get_text("text").insert(0, "1").unwrap();
                doc_clone.commit();
            }));

            handlers.push(loom::thread::spawn(move || {
                doc.import_batch(&[update_a, update_b]).unwrap();
            }));
            for h in handlers {
                h.join().unwrap();
            }
        });
    }

    #[test]
    fn subscribe_local_update_while_another_thread_emits() {
        loom::model(|| {
            let doc = LoroDoc::new();
            let first_calls = std::sync::Arc::new(AtomicUsize::new(0));
            let first_calls_clone = first_calls.clone();
            let _first = doc.subscribe_local_update(Box::new(move |_| {
                first_calls_clone.fetch_add(1, Ordering::SeqCst);
                true
            }));

            let doc1 = doc.clone();
            let h0 = loom::thread::spawn(move || {
                doc1.get_text("text").insert(0, "1").unwrap();
                doc1.commit();
            });
            let doc2 = doc.clone();
            let h1 = loom::thread::spawn(move || {
                let late_calls = std::sync::Arc::new(AtomicUsize::new(0));
                let late_calls_clone = late_calls.clone();
                let _late = doc2.subscribe_local_update(Box::new(move |_| {
                    late_calls_clone.fetch_add(1, Ordering::SeqCst);
                    true
                }));
                doc2.get_text("text").insert(0, "2").unwrap();
                doc2.commit();
                (late_calls, _late)
            });

            h0.join().unwrap();
            let (late_calls, _late) = h1.join().unwrap();
            let late_calls = late_calls.load(Ordering::SeqCst);
            assert!(
                late_calls >= 1,
                "late subscriber missed an update committed after it subscribed"
            );
            assert!(first_calls.load(Ordering::SeqCst) >= late_calls);
            assert_eq!(doc.get_text("text").len_utf8(), 2);
        });
    }

    /// A snapshot where peer 2 created node `0@2` under a parent that peer 1 deleted
    /// concurrently, then more history. A document loaded from it finds the parent of the
    /// node's meta only by parsing the change that created the node (loro-dev/loro#1158).
    fn deleted_parent_snapshot() -> (Vec<u8>, loro::TreeID) {
        let a = LoroDoc::new();
        a.set_peer_id(1).unwrap();
        let parent = a.get_tree("tree").create(loro::TreeParentId::Root).unwrap();
        a.commit();
        let b = LoroDoc::new();
        b.set_peer_id(2).unwrap();
        b.import(&a.export(ExportMode::all_updates()).unwrap())
            .unwrap();
        let child = b.get_tree("tree").create(parent).unwrap();
        b.commit();
        a.get_tree("tree").delete(parent).unwrap();
        a.commit();
        a.import(&b.export(ExportMode::all_updates()).unwrap())
            .unwrap();
        for i in 0..3 {
            a.get_map("m").insert("k", i).unwrap();
            a.commit();
        }
        (a.export(ExportMode::Snapshot).unwrap(), child)
    }

    /// The creator resolver reaches the change store without the op log lock (here under the
    /// state lock, from `is_deleted` and `has_container`), while another thread reads the
    /// store under the op log lock. Both must take the store's locks in the same order.
    #[test]
    fn resolving_a_meta_parent_while_another_thread_reads_the_history() {
        for reader in 0..3 {
            loom::model(move || {
                let (snapshot, child) = deleted_parent_snapshot();
                let doc = LoroDoc::new();
                doc.import(&snapshot).unwrap();
                let meta = doc.get_tree("tree").get_meta(child).unwrap();
                let meta_id = loro::ContainerTrait::id(&meta);
                let doc2 = doc.clone();
                let h0 = loom::thread::spawn(move || {
                    if reader == 2 {
                        assert!(doc.has_container(&meta_id));
                    } else {
                        assert!(loro::ContainerTrait::is_deleted(&meta));
                    }
                });
                let h1 = loom::thread::spawn(move || {
                    if reader == 0 {
                        assert!(doc2.len_changes() > 0);
                    } else {
                        doc2.export(ExportMode::all_updates()).unwrap();
                    }
                });
                h0.join().unwrap();
                h1.join().unwrap();
            });
        }
    }

    /// An import that fails before its rollback scope begins (here while decoding JSON) rolls
    /// the arena back to where it started. A block that the resolver parsed on another thread
    /// in the meantime must not keep indices that the rollback drops.
    #[test]
    fn resolving_a_meta_parent_while_an_import_fails() {
        struct Fixture {
            snapshot: Vec<u8>,
            meta: loro::ContainerID,
            bad_json: String,
            vv: loro::VersionVector,
            history: String,
        }
        // Built once, in a single-threaded model of its own, so the racing model does not
        // explore it.
        static FIXTURE: std::sync::Mutex<Option<std::sync::Arc<Fixture>>> =
            std::sync::Mutex::new(None);
        let mut builder = loom::model::Builder::new();
        builder.max_branches = 100_000;
        builder.check(|| {
            let (snapshot, child) = deleted_parent_snapshot();
            // A separate document, so that exporting parses nothing in the tested one.
            let reference = LoroDoc::new();
            reference.import(&snapshot).unwrap();
            let vv = reference.oplog_vv();
            let history = serde_json::to_string(
                &reference
                    .export_json_updates(&Default::default(), &vv)
                    .unwrap(),
            )
            .unwrap();
            // Another peer creates a container, so decoding its change registers one. The
            // empty change after it fails the decode.
            let e = LoroDoc::new();
            e.set_peer_id(9).unwrap();
            e.import(&snapshot).unwrap();
            e.get_map("x")
                .insert_container("c", loro::LoroMap::new())
                .unwrap()
                .insert("q", 1)
                .unwrap();
            e.commit();
            let mut json =
                serde_json::to_value(e.export_json_updates(&vv, &e.oplog_vv()).unwrap()).unwrap();
            let changes = json["changes"].as_array_mut().unwrap();
            let mut empty = changes.last().unwrap().clone();
            empty["ops"] = serde_json::json!([]);
            changes.push(empty);
            *FIXTURE.lock().unwrap() = Some(std::sync::Arc::new(Fixture {
                snapshot,
                meta: child.associated_meta_container(),
                bad_json: serde_json::to_string(&json).unwrap(),
                vv,
                history,
            }));
        });
        let f = FIXTURE.lock().unwrap().clone().unwrap();

        let mut builder = loom::model::Builder::new();
        builder.max_branches = 100_000;
        builder.check(move || {
            let doc = LoroDoc::new();
            doc.import(&f.snapshot).unwrap();
            let (doc1, doc2) = (doc.clone(), doc.clone());
            let (f1, f2) = (f.clone(), f.clone());
            let h0 = loom::thread::spawn(move || {
                doc1.has_container(&f1.meta);
            });
            let h1 = loom::thread::spawn(move || {
                assert!(doc2.import_json_updates(f2.bad_json.as_str()).is_err());
            });
            h0.join().unwrap();
            h1.join().unwrap();
            assert!(doc.has_container(&f.meta));
            let history = serde_json::to_string(
                &doc.export_json_updates(&Default::default(), &f.vv).unwrap(),
            )
            .unwrap();
            assert_eq!(history, f.history);
        });
    }
}