3420 lines · 100.5 KB · Rust
RawDownload
use super::*;use crate::language_settings::{    AllLanguageSettings, AllLanguageSettingsContent, LanguageSettingsContent,};use crate::Buffer;use clock::ReplicaId;use collections::BTreeMap;use futures::FutureExt as _;use gpui::{App, AppContext as _, BorrowAppContext, Entity};use gpui::{HighlightStyle, TestAppContext};use indoc::indoc;use proto::deserialize_operation;use rand::prelude::*;use regex::RegexBuilder;use settings::SettingsStore;use std::{    env,    ops::Range,    sync::LazyLock,    time::{Duration, Instant},};use syntax_map::TreeSitterOptions;use text::network::Network;use text::{BufferId, LineEnding};use text::{Point, ToPoint};use theme::ActiveTheme;use unindent::Unindent as _;use util::test::marked_text_offsets;use util::{assert_set_eq, post_inc, test::marked_text_ranges, RandomCharIter}; pub static TRAILING_WHITESPACE_REGEX: LazyLock<regex::Regex> = LazyLock::new(|| {    RegexBuilder::new(r"[ \t]+$")        .multi_line(true)        .build()        .expect("Failed to create TRAILING_WHITESPACE_REGEX")}); #[cfg(test)]#[ctor::ctor]fn init_logger() {    if std::env::var("RUST_LOG").is_ok() {        env_logger::init();    }} #[gpui::test]fn test_line_endings(cx: &mut gpui::App) {    init_settings(cx, |_| {});     cx.new(|cx| {        let mut buffer =            Buffer::local("one\r\ntwo\rthree", cx).with_language(Arc::new(rust_lang()), cx);        assert_eq!(buffer.text(), "one\ntwo\nthree");        assert_eq!(buffer.line_ending(), LineEnding::Windows);         buffer.check_invariants();        buffer.edit(            [(buffer.len()..buffer.len(), "\r\nfour")],            Some(AutoindentMode::EachLine),            cx,        );        buffer.edit([(0..0, "zero\r\n")], None, cx);        assert_eq!(buffer.text(), "zero\none\ntwo\nthree\nfour");        assert_eq!(buffer.line_ending(), LineEnding::Windows);        buffer.check_invariants();         buffer    });} #[gpui::test]fn test_select_language(cx: &mut App) {    init_settings(cx, |_| {});     let registry = Arc::new(LanguageRegistry::test(cx.background_executor().clone()));    registry.add(Arc::new(Language::new(        LanguageConfig {            name: LanguageName::new("Rust"),            matcher: LanguageMatcher {                path_suffixes: vec!["rs".to_string()],                ..Default::default()            },            ..Default::default()        },        Some(tree_sitter_rust::LANGUAGE.into()),    )));    registry.add(Arc::new(Language::new(        LanguageConfig {            name: LanguageName::new("Make"),            matcher: LanguageMatcher {                path_suffixes: vec!["Makefile".to_string(), "mk".to_string()],                ..Default::default()            },            ..Default::default()        },        Some(tree_sitter_rust::LANGUAGE.into()),    )));     // matching file extension    assert_eq!(        registry            .language_for_file(&file("src/lib.rs"), None, cx)            .map(|l| l.name()),        Some("Rust".into())    );    assert_eq!(        registry            .language_for_file(&file("src/lib.mk"), None, cx)            .map(|l| l.name()),        Some("Make".into())    );     // matching filename    assert_eq!(        registry            .language_for_file(&file("src/Makefile"), None, cx)            .map(|l| l.name()),        Some("Make".into())    );     // matching suffix that is not the full file extension or filename    assert_eq!(        registry            .language_for_file(&file("zed/cars"), None, cx)            .map(|l| l.name()),        None    );    assert_eq!(        registry            .language_for_file(&file("zed/a.cars"), None, cx)            .map(|l| l.name()),        None    );    assert_eq!(        registry            .language_for_file(&file("zed/sumk"), None, cx)            .map(|l| l.name()),        None    );} #[gpui::test(iterations = 10)]async fn test_first_line_pattern(cx: &mut TestAppContext) {    cx.update(|cx| init_settings(cx, |_| {}));     let languages = LanguageRegistry::test(cx.executor());    let languages = Arc::new(languages);     languages.register_test_language(LanguageConfig {        name: "JavaScript".into(),        matcher: LanguageMatcher {            path_suffixes: vec!["js".into()],            first_line_pattern: Some(Regex::new(r"\bnode\b").unwrap()),        },        ..Default::default()    });     assert!(cx        .read(|cx| languages.language_for_file(&file("the/script"), None, cx))        .is_none());    assert!(cx        .read(|cx| languages.language_for_file(&file("the/script"), Some(&"nothing".into()), cx))        .is_none());     assert_eq!(        cx.read(|cx| languages.language_for_file(            &file("the/script"),            Some(&"#!/bin/env node".into()),            cx        ))        .unwrap()        .name(),        "JavaScript".into()    );} #[gpui::test]async fn test_language_for_file_with_custom_file_types(cx: &mut TestAppContext) {    cx.update(|cx| {        init_settings(cx, |settings| {            settings.file_types.extend([                ("TypeScript".into(), vec!["js".into()]),                ("C++".into(), vec!["c".into()]),                (                    "Dockerfile".into(),                    vec!["Dockerfile".into(), "Dockerfile.*".into()],                ),            ]);        })    });     let languages = Arc::new(LanguageRegistry::test(cx.executor()));     for config in [        LanguageConfig {            name: "JavaScript".into(),            matcher: LanguageMatcher {                path_suffixes: vec!["js".to_string()],                ..Default::default()            },            ..Default::default()        },        LanguageConfig {            name: "TypeScript".into(),            matcher: LanguageMatcher {                path_suffixes: vec!["js".to_string()],                ..Default::default()            },            ..Default::default()        },        LanguageConfig {            name: "C++".into(),            matcher: LanguageMatcher {                path_suffixes: vec!["cpp".to_string()],                ..Default::default()            },            ..Default::default()        },        LanguageConfig {            name: "C".into(),            matcher: LanguageMatcher {                path_suffixes: vec!["c".to_string()],                ..Default::default()            },            ..Default::default()        },        LanguageConfig {            name: "Dockerfile".into(),            matcher: LanguageMatcher {                path_suffixes: vec!["Dockerfile".to_string()],                ..Default::default()            },            ..Default::default()        },    ] {        languages.add(Arc::new(Language::new(config, None)));    }     let language = cx        .read(|cx| languages.language_for_file(&file("foo.js"), None, cx))        .unwrap();    assert_eq!(language.name(), "TypeScript".into());    let language = cx        .read(|cx| languages.language_for_file(&file("foo.c"), None, cx))        .unwrap();    assert_eq!(language.name(), "C++".into());    let language = cx        .read(|cx| languages.language_for_file(&file("Dockerfile.dev"), None, cx))        .unwrap();    assert_eq!(language.name(), "Dockerfile".into());} fn file(path: &str) -> Arc<dyn File> {    Arc::new(TestFile {        path: Path::new(path).into(),        root_name: "zed".into(),        local_root: None,    })} #[gpui::test]fn test_edit_events(cx: &mut gpui::App) {    let mut now = Instant::now();    let buffer_1_events = Arc::new(Mutex::new(Vec::new()));    let buffer_2_events = Arc::new(Mutex::new(Vec::new()));     let buffer1 = cx.new(|cx| Buffer::local("abcdef", cx));    let buffer2 = cx.new(|cx| {        Buffer::remote(            BufferId::from(cx.entity_id().as_non_zero_u64()),            1,            Capability::ReadWrite,            "abcdef",        )    });    let buffer1_ops = Arc::new(Mutex::new(Vec::new()));    buffer1.update(cx, {        let buffer1_ops = buffer1_ops.clone();        |buffer, cx| {            let buffer_1_events = buffer_1_events.clone();            cx.subscribe(&buffer1, move |_, _, event, _| match event.clone() {                BufferEvent::Operation {                    operation,                    is_local: true,                } => buffer1_ops.lock().push(operation),                event => buffer_1_events.lock().push(event),            })            .detach();            let buffer_2_events = buffer_2_events.clone();            cx.subscribe(&buffer2, move |_, _, event, _| match event.clone() {                BufferEvent::Operation {                    is_local: false, ..                } => {}                event => buffer_2_events.lock().push(event),            })            .detach();             // An edit emits an edited event, followed by a dirty changed event,            // since the buffer was previously in a clean state.            buffer.edit([(2..4, "XYZ")], None, cx);             // An empty transaction does not emit any events.            buffer.start_transaction();            buffer.end_transaction(cx);             // A transaction containing two edits emits one edited event.            now += Duration::from_secs(1);            buffer.start_transaction_at(now);            buffer.edit([(5..5, "u")], None, cx);            buffer.edit([(6..6, "w")], None, cx);            buffer.end_transaction_at(now, cx);             // Undoing a transaction emits one edited event.            buffer.undo(cx);        }    });     // Incorporating a set of remote ops emits a single edited event,    // followed by a dirty changed event.    buffer2.update(cx, |buffer, cx| {        buffer.apply_ops(buffer1_ops.lock().drain(..), cx);    });    assert_eq!(        mem::take(&mut *buffer_1_events.lock()),        vec![            BufferEvent::Edited,            BufferEvent::DirtyChanged,            BufferEvent::Edited,            BufferEvent::Edited,        ]    );    assert_eq!(        mem::take(&mut *buffer_2_events.lock()),        vec![BufferEvent::Edited, BufferEvent::DirtyChanged]    );     buffer1.update(cx, |buffer, cx| {        // Undoing the first transaction emits edited event, followed by a        // dirty changed event, since the buffer is again in a clean state.        buffer.undo(cx);    });    // Incorporating the remote ops again emits a single edited event,    // followed by a dirty changed event.    buffer2.update(cx, |buffer, cx| {        buffer.apply_ops(buffer1_ops.lock().drain(..), cx);    });    assert_eq!(        mem::take(&mut *buffer_1_events.lock()),        vec![BufferEvent::Edited, BufferEvent::DirtyChanged,]    );    assert_eq!(        mem::take(&mut *buffer_2_events.lock()),        vec![BufferEvent::Edited, BufferEvent::DirtyChanged]    );} #[gpui::test]async fn test_apply_diff(cx: &mut TestAppContext) {    let (text, offsets) = marked_text_offsets(        "one two three\nfour fiˇve six\nseven eightˇ nine\nten eleven twelve\n",    );    let buffer = cx.new(|cx| Buffer::local(text, cx));    let anchors = buffer.update(cx, |buffer, _| {        offsets            .iter()            .map(|offset| buffer.anchor_before(offset))            .collect::<Vec<_>>()    });     let (text, offsets) = marked_text_offsets(        "one two three\n{\nfour FIVEˇ six\n}\nseven AND EIGHTˇ nine\nten eleven twelve\n",    );     let diff = buffer.update(cx, |b, cx| b.diff(text.clone(), cx)).await;    buffer.update(cx, |buffer, cx| {        buffer.apply_diff(diff, cx).unwrap();        assert_eq!(buffer.text(), text);        let actual_offsets = anchors            .iter()            .map(|anchor| anchor.to_offset(buffer))            .collect::<Vec<_>>();        assert_eq!(actual_offsets, offsets);    });     let (text, offsets) =        marked_text_offsets("one two three\n{\nˇ}\nseven AND EIGHTEENˇ nine\nten eleven twelve\n");     let diff = buffer.update(cx, |b, cx| b.diff(text.clone(), cx)).await;    buffer.update(cx, |buffer, cx| {        buffer.apply_diff(diff, cx).unwrap();        assert_eq!(buffer.text(), text);        let actual_offsets = anchors            .iter()            .map(|anchor| anchor.to_offset(buffer))            .collect::<Vec<_>>();        assert_eq!(actual_offsets, offsets);    });} #[gpui::test(iterations = 10)]async fn test_normalize_whitespace(cx: &mut gpui::TestAppContext) {    let text = [        "zero",     //        "one  ",    // 2 trailing spaces        "two",      //        "three   ", // 3 trailing spaces        "four",     //        "five    ", // 4 trailing spaces    ]    .join("\n");     let buffer = cx.new(|cx| Buffer::local(text, cx));     // Spawn a task to format the buffer's whitespace.    // Pause so that the formatting task starts running.    let format = buffer.update(cx, |buffer, cx| buffer.remove_trailing_whitespace(cx));    smol::future::yield_now().await;     // Edit the buffer while the normalization task is running.    let version_before_edit = buffer.update(cx, |buffer, _| buffer.version());    buffer.update(cx, |buffer, cx| {        buffer.edit(            [                (Point::new(0, 1)..Point::new(0, 1), "EE"),                (Point::new(3, 5)..Point::new(3, 5), "EEE"),            ],            None,            cx,        );    });     let format_diff = format.await;    buffer.update(cx, |buffer, cx| {        let version_before_format = format_diff.base_version.clone();        buffer.apply_diff(format_diff, cx);         // The outcome depends on the order of concurrent tasks.        //        // If the edit occurred while searching for trailing whitespace ranges,        // then the trailing whitespace region touched by the edit is left intact.        if version_before_format == version_before_edit {            assert_eq!(                buffer.text(),                [                    "zEEero",      //                    "one",         //                    "two",         //                    "threeEEE   ", //                    "four",        //                    "five",        //                ]                .join("\n")            );        }        // Otherwise, all trailing whitespace is removed.        else {            assert_eq!(                buffer.text(),                [                    "zEEero",   //                    "one",      //                    "two",      //                    "threeEEE", //                    "four",     //                    "five",     //                ]                .join("\n")            );        }    });} #[gpui::test]async fn test_reparse(cx: &mut gpui::TestAppContext) {    let text = "fn a() {}";    let buffer = cx.new(|cx| Buffer::local(text, cx).with_language(Arc::new(rust_lang()), cx));     // Wait for the initial text to parse    cx.executor().run_until_parked();    assert!(!buffer.update(cx, |buffer, _| buffer.is_parsing()));    assert_eq!(        get_tree_sexp(&buffer, cx),        concat!(            "(source_file (function_item name: (identifier) ",            "parameters: (parameters) ",            "body: (block)))"        )    );     buffer.update(cx, |buffer, _| {        buffer.set_sync_parse_timeout(Duration::ZERO)    });     // Perform some edits (add parameter and variable reference)    // Parsing doesn't begin until the transaction is complete    buffer.update(cx, |buf, cx| {        buf.start_transaction();         let offset = buf.text().find(')').unwrap();        buf.edit([(offset..offset, "b: C")], None, cx);

Lines 1–500 of 3420.