initial testing of fsm

crates/analyzer-host/Cargo.toml

@@ -17,3 +17,6 @@ serde = "1.0.143"
 serde_json = "1.0.83"
 thiserror = "1.0.37"
 tracing-subscriber = "0.3.16"
+
+[dev-dependencies]
+async-io = "1.13.0"

crates/analyzer-host/src/fsm.rs

@@ -104,3 +104,66 @@ impl LspProtocolMachine {
 		)
 	}
 }
+
+#[cfg(test)]
+mod tests {
+	use json_rpc::message::Request;
+	use json_rpc::message::Notification;
+	use serde_json::Value;
+
+	use crate::{fs::LspEnumerableFileSystem, json_rpc};
+	use super::*;
+	#[test]
+	fn test_is_active() {
+		let rm = RequestManager::new(async_channel::unbounded::<Message>());
+		let mut lsp = LspProtocolMachine::new(None, rm.clone(),
+													Arc::new(Box::new(LspEnumerableFileSystem::new(rm.clone()))));
+		assert_eq!(lsp.is_active(), true);
+
+		lsp.current_state = LspServerState::Stopped;
+		assert_eq!(lsp.is_active(), false);
+	}
+	#[test]
+	fn test_process_message() {
+		let rm = RequestManager::new(async_channel::unbounded::<Message>());
+		let mut lsp = LspProtocolMachine::new(None, rm.clone(), 
+													Arc::new(Box::new(LspEnumerableFileSystem::new(rm.clone()))));
+		assert_eq!(lsp.current_state, LspServerState::ActiveUninitialized);
+
+		let params = serde_json::json!(analyzer_abstractions::lsp_types::InitializeParams{ ..Default::default() });
+		let message = Message::Request(Request{
+			id: 0.into(),
+			method: String::from("initialize"),
+			params: params,
+		});
+		let future = async_io::block_on(lsp.process_message(Arc::new(message)));
+		assert_eq!(future.is_ok(), true);
+		assert_eq!(lsp.current_state, LspServerState::Initializing);
+
+		let params = serde_json::json!(analyzer_abstractions::lsp_types::InitializedParams{});
+		let message = Message::Notification(Notification {
+			method: String::from("initialized"),
+			params: params,
+		});
+		let future = async_io::block_on(lsp.process_message(Arc::new(message)));
+		assert_eq!(future.is_ok(), true);
+		assert_eq!(lsp.current_state, LspServerState::ActiveInitialized);
+
+		let message = Message::Request(Request{
+			id: 0.into(),
+			method: String::from("shutdown"),
+			params: Value::Null,
+		});
+		let future = async_io::block_on(lsp.process_message(Arc::new(message)));
+		assert_eq!(future.is_ok(), true);
+		assert_eq!(lsp.current_state, LspServerState::ShuttingDown);
+
+		let message = Message::Notification(Notification{
+			method: String::from("exit"),
+			params: Value::Null,
+		});
+		let future = async_io::block_on(lsp.process_message(Arc::new(message)));
+		assert_eq!(future.is_ok(), true);
+		assert_eq!(lsp.current_state, LspServerState::Stopped);
+	}
+}