From 5723f37a74c58814fa528077d7320d53c68f1ec8 Mon Sep 17 00:00:00 2001
From: Turiiya <34311583+ttytm@users.noreply.github.com>
Date: Sun, 14 Apr 2024 16:28:22 +0200
Subject: [PATCH] test custom path setup

---
 src/loglib/Logger.v     |  2 +-
 src/server/setup_test.v | 66 +++++++++++++++++++++++++++++++++++++----
 2 files changed, 62 insertions(+), 6 deletions(-)

diff --git a/src/loglib/Logger.v b/src/loglib/Logger.v
index a5e6a846..fc1ae330 100644
--- a/src/loglib/Logger.v
+++ b/src/loglib/Logger.v
@@ -15,7 +15,7 @@ pub const support_colors = term.can_show_color_on_stderr() && term.can_show_colo
 
 @[heap]
 pub struct Logger {
-mut:
+pub mut:
 	disabled   bool
 	color_mode ColorMode
 
diff --git a/src/server/setup_test.v b/src/server/setup_test.v
index a1a163c3..c15e2f74 100644
--- a/src/server/setup_test.v
+++ b/src/server/setup_test.v
@@ -1,12 +1,68 @@
 module server
 
 import os
+import lsp
+import loglib as _ // import to use __global `logger`
 
-fn test_setup_vpaths() {
+const default_vexe = @VEXE
+const default_vroot = os.dir(default_vexe)
+const default_vlib_root = os.join_path(default_vroot, 'vlib')
+const default_vmodules_root = os.vmodules_dir()
+
+fn test_setup_default_vpaths() {
 	mut ls := LanguageServer{}
 	ls.setup()
-	assert ls.paths.vexe == @VEXE
-	assert ls.paths.vroot == os.dir(ls.paths.vexe)
-	assert ls.paths.vmodules_root == os.vmodules_dir()
-	assert ls.paths.vlib_root == os.join_path(ls.paths.vroot, 'vlib')
+	assert ls.paths.vexe == default_vexe
+	assert ls.paths.vroot == default_vroot
+	assert ls.paths.vlib_root == default_vlib_root
+	assert ls.paths.vmodules_root == default_vmodules_root
+}
+
+fn test_setup_custom_vpaths() {
+	mut ls := LanguageServer{}
+
+	custom_root := os.join_path(os.vtmp_dir(), 'v-analyzer-setup-test')
+	custom_root_uri := lsp.document_uri_from_path(custom_root)
+	cfg_dir_path := os.join_path(custom_root, '.v-analyzer')
+	os.mkdir_all(cfg_dir_path)!
+	defer {
+		os.rmdir_all(cfg_dir_path) or {}
+	}
+
+	// Test custom_vroot with missing toolchain ==================================
+	mut cfg_toml := 'custom_vroot = "${custom_root}"'
+	os.write_file(os.join_path(custom_root, '.v-analyzer', 'config.toml'), cfg_toml)!
+
+	// Set output(io.Writer) for global loglib logger.
+	log_file_path := os.join_path(custom_root, 'log')
+	os.write_file(log_file_path, '')!
+	mut log_file := os.open_append(os.join_path(custom_root, 'log'))!
+	logger.out = log_file
+
+	// Run setup
+	ls.root_uri = custom_root_uri
+	ls.setup()
+
+	log_file.close()
+	mut log_out := os.read_file(log_file_path)!
+	assert log_out.contains('Find custom VROOT path')
+	assert log_out.contains('Using "${custom_root}" as toolchain')
+	assert log_out.contains('Failed to find standard library path')
+
+	// Test custom_vroot with existing toolchain =================================
+	cfg_toml = 'custom_vroot = "${default_vroot}"'
+	os.write_file(os.join_path(custom_root, '.v-analyzer', 'config.toml'), cfg_toml)!
+	os.write_file(log_file_path, '')!
+	log_file = os.open_append(os.join_path(custom_root, 'log'))!
+	logger.out = log_file
+	ls = LanguageServer{}
+	ls.root_uri = custom_root_uri
+
+	ls.setup()
+
+	log_file.close()
+	log_out = os.read_file(log_file_path)!
+	assert log_out.contains('Find custom VROOT path')
+	assert log_out.contains('Using "${default_vroot}" as toolchain')
+	assert !log_out.contains('Failed to find standard library path')
 }