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