1use std::sync::Arc;
  2
  3use crate::lsp_log::LogMenuItem;
  4
  5use super::*;
  6use futures::StreamExt;
  7use gpui::{serde_json::json, Context, TestAppContext, VisualTestContext};
  8use language::{tree_sitter_rust, FakeLspAdapter, Language, LanguageConfig, LanguageServerName};
  9use project::{FakeFs, Project};
 10use settings::SettingsStore;
 11
 12#[gpui::test]
 13async fn test_lsp_logs(cx: &mut TestAppContext) {
 14    if std::env::var("RUST_LOG").is_ok() {
 15        env_logger::init();
 16    }
 17
 18    init_test(cx);
 19
 20    let mut rust_language = Language::new(
 21        LanguageConfig {
 22            name: "Rust".into(),
 23            path_suffixes: vec!["rs".to_string()],
 24            ..Default::default()
 25        },
 26        Some(tree_sitter_rust::language()),
 27    );
 28    let mut fake_rust_servers = rust_language
 29        .set_fake_lsp_adapter(Arc::new(FakeLspAdapter {
 30            name: "the-rust-language-server",
 31            ..Default::default()
 32        }))
 33        .await;
 34
 35    let fs = FakeFs::new(cx.background_executor.clone());
 36    fs.insert_tree(
 37        "/the-root",
 38        json!({
 39            "test.rs": "",
 40            "package.json": "",
 41        }),
 42    )
 43    .await;
 44    let project = Project::test(fs.clone(), ["/the-root".as_ref()], cx).await;
 45    project.update(cx, |project, _| {
 46        project.languages().add(Arc::new(rust_language));
 47    });
 48
 49    let log_store = cx.build_model(|cx| LogStore::new(cx));
 50    log_store.update(cx, |store, cx| store.add_project(&project, cx));
 51
 52    let _rust_buffer = project
 53        .update(cx, |project, cx| {
 54            project.open_local_buffer("/the-root/test.rs", cx)
 55        })
 56        .await
 57        .unwrap();
 58
 59    let mut language_server = fake_rust_servers.next().await.unwrap();
 60    language_server
 61        .receive_notification::<lsp::notification::DidOpenTextDocument>()
 62        .await;
 63
 64    let window = cx.add_window(|cx| LspLogView::new(project.clone(), log_store.clone(), cx));
 65    let log_view = window.root(cx).unwrap();
 66    let mut cx = VisualTestContext::from_window(*window, cx);
 67
 68    language_server.notify::<lsp::notification::LogMessage>(lsp::LogMessageParams {
 69        message: "hello from the server".into(),
 70        typ: lsp::MessageType::INFO,
 71    });
 72    cx.executor().run_until_parked();
 73
 74    log_view.update(&mut cx, |view, cx| {
 75        assert_eq!(
 76            view.menu_items(cx).unwrap(),
 77            &[LogMenuItem {
 78                server_id: language_server.server.server_id(),
 79                server_name: LanguageServerName("the-rust-language-server".into()),
 80                worktree_root_name: project
 81                    .read(cx)
 82                    .worktrees()
 83                    .next()
 84                    .unwrap()
 85                    .read(cx)
 86                    .root_name()
 87                    .to_string(),
 88                rpc_trace_enabled: false,
 89                rpc_trace_selected: false,
 90                logs_selected: true,
 91            }]
 92        );
 93        assert_eq!(view.editor.read(cx).text(cx), "hello from the server\n");
 94    });
 95}
 96
 97fn init_test(cx: &mut gpui::TestAppContext) {
 98    cx.update(|cx| {
 99        let settings_store = SettingsStore::test(cx);
100        cx.set_global(settings_store);
101        theme::init(theme::LoadThemes::JustBase, cx);
102        language::init(cx);
103        client::init_settings(cx);
104        Project::init_settings(cx);
105        editor::init_settings(cx);
106    });
107}