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