feat: add some mocks for solana api and unit test (#3558)

Co-authored-by: wli-pro <wli-pro>

Author: wli-pro

tee-worker/omni-executor/.env.test

@@ -6,4 +6,5 @@ OE_SENDGRID_FROM_NAME='Heima Dev Verify'
 OE_PUMPX_API_BASE_URL=http://127.0.0.1:3456/pumpx
 OE_BINANCE_API_KEY=binance_api_key
 OE_BINANCE_API_SECRET=binance_api_secret
-OE_BINANCE_API_BASE_URL=http://127.0.0.1:3456/binance
+OE_BINANCE_API_BASE_URL=http://127.0.0.1:3456/binance
+OE_SOLANA_URL=http://127.0.0.1:3456/solana

tee-worker/omni-executor/Cargo.lock

@@ -17,5 +17,8 @@ tracing = { workspace = true }
 [lints]
 workspace = true
 
+[dev-dependencies]
+mock-server = { workspace = true }
+
 [features]
 mocks = []

tee-worker/omni-executor/accounting-contract-client/Cargo.toml

@@ -229,3 +229,69 @@ pub mod mocks {
 		}
 	}
 }
+
+#[cfg(test)]
+mod tests {
+	use super::*;
+	use sp_core::{ed25519, Pair};
+
+	#[tokio::test]
+	async fn test_mock_server_execute_pay_out_request() {
+		// Start mock server and get dynamic URL
+		let mock_url = mock_server::async_run_test_only().await;
+		let param_id = "D3S1ZTrFNkfeoHaLSTAjMXZVXnRJvsNnbwh9k5mRYqqV";
+
+		// has balance
+		// address: AKnL4NNf3DGWZJS6cPknBuEGnVsV4A4m5tgebLHaRSZ9
+		let seed = [1u8; 32];
+		let pair = ed25519::Pair::from_seed(&seed);
+		let client = AccountingContractClient::new(
+			pair,
+			format!("{}/solana", mock_url),
+			param_id.to_string(),
+		);
+
+		// Test execute_pay_out_request
+		// base64 encoded tx string (sendTransaction):
+		// AaV8xKPN0GfY3XVuTiQsxFPDOFlZhXhWwbcVMBEGd2kUTgid2Ss9GUNqJ1S01iTyvDKCWea8sRMwlmlZ1A8zDAEBAAMIiojj3XQJ8ZX9UtstPLpdcspnCb8dlBIb83SIAbQPb1zG8DcMUx1vBnO2Al4wQ7nCKjwgl/c5iVbUF2HhIwZJ3ez0N0UQ/aETrzeokAeTDaVjdVNHLlFhPu0cKQjhxBso7UkoxijRwsbq6QM4kFmVYSlZJzpcY/k2NsFGFKyHN9Hx9pMmCdAce72TK7SkfatcvRdHPxk+YN0BruCENR+3BgAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAsuuu6Iwn1QSzfnYrRIa2f/i5FCzjnwDv32SN1j4fahTyq0V7Xk41kUInl1rH0tI9eWEbcpDkAbgz89/ZTaRNLMxJDpKM0uOHO7ND/JXaMxecpg9Nv0bCw26RKZ1V1Oa5AQYHAgEABwQDBRiE8aphpOG6LgEAAAAAAAAAAMqaOwAAAAA=
+		let beneficiary_seed = [3u8; 32];
+		let beneficiary_keypair =
+			Keypair::from_seed(&beneficiary_seed).expect("Failed to create beneficiary keypair");
+		let beneficiary = beneficiary_keypair.pubkey();
+		let nonce = 1u64;
+		let amount = U256::from(1000000000u64); // 1 SOL
+		let result = client.execute_pay_out_request(beneficiary, nonce, amount).await;
+		assert_eq!(result.is_ok(), true);
+	}
+
+	#[tokio::test]
+	async fn test_mock_server_get_balance() {
+		// Start mock server and get dynamic URL
+		let mock_url = mock_server::async_run_test_only().await;
+		let param_id = "D3S1ZTrFNkfeoHaLSTAjMXZVXnRJvsNnbwh9k5mRYqqV";
+
+		// has balance
+		// address: AKnL4NNf3DGWZJS6cPknBuEGnVsV4A4m5tgebLHaRSZ9
+		let seed = [1u8; 32];
+		let pair = ed25519::Pair::from_seed(&seed);
+		let client = AccountingContractClient::new(
+			pair,
+			format!("{}/solana", mock_url),
+			param_id.to_string(),
+		);
+		let balance = client.get_balance().await.expect("Fail to get balance");
+		assert_eq!(balance, U256::from(1000000000u64));
+
+		// has no balance
+		// address: 9hSR6S7WPtxmTojgo6GG3k4yDPecgJY292j7xrsUGWBu
+		let seed = [2u8; 32];
+		let pair = ed25519::Pair::from_seed(&seed);
+		let client = AccountingContractClient::new(
+			pair,
+			format!("{}/solana", mock_url),
+			param_id.to_string(),
+		);
+		let balance = client.get_balance().await.expect("Fail to get balance");
+		assert_eq!(balance, U256::from(0));
+	}
+}

tee-worker/omni-executor/accounting-contract-client/src/solana.rs

@@ -4,7 +4,7 @@ name = "mock-server"
 version = "0.1.0"
 
 [dependencies]
-env_logger = "0.10"
+env_logger = { workspace = true }
 hex = { workspace = true, features = ["std"] }
 log = { workspace = true, features = ["std"] }
 serde_json = { workspace = true, features = ["std"] }

tee-worker/omni-executor/mock-server/Cargo.toml

@@ -14,6 +14,7 @@ use warp::Filter;
 mod binance;
 mod pumpx;
 mod sendgrid;
+mod solana;
 
 // It should only works on UNIX.
 async fn shutdown_signal() {
@@ -32,19 +33,31 @@ async fn shutdown_signal() {
 	log::info!("Shutdown signal received, stopping server...");
 }
 
-pub fn run(port: u16) -> Result<String, Box<dyn std::error::Error>> {
+pub fn run(port: u16) -> Result<String, Box<dyn std::error::Error + Send + Sync>> {
+	run_with_shutdown_control(port, true)
+}
+
+pub fn run_with_shutdown_control(
+	port: u16,
+	wait_for_shutdown: bool,
+) -> Result<String, Box<dyn std::error::Error + Send + Sync>> {
 	let (result_in, result_out) = channel();
 	let (shutdown_in, shutdown_out) = channel();
 
 	thread::spawn(move || {
 		let runtime = Builder::new_current_thread().enable_all().build().unwrap();
 		LocalSet::new().block_on(&runtime, async {
-			let (addr, srv) =
-				warp::serve(binance::handle().or(pumpx::handle()).or(sendgrid::handle()).boxed())
-					.bind_with_graceful_shutdown(([0, 0, 0, 0], port), async {
-						shutdown_signal().await;
-						let _ = shutdown_in.send(());
-					});
+			let (addr, srv) = warp::serve(
+				binance::handle()
+					.or(pumpx::handle())
+					.or(sendgrid::handle())
+					.or(solana::handle())
+					.boxed(),
+			)
+			.bind_with_graceful_shutdown(([0, 0, 0, 0], port), async {
+				shutdown_signal().await;
+				let _ = shutdown_in.send(());
+			});
 
 			log::info!("mock-server listen on addr:{:?}", addr);
 			let _ = result_in.send(format!("http://{:?}", addr));
@@ -57,7 +70,19 @@ pub fn run(port: u16) -> Result<String, Box<dyn std::error::Error>> {
 
 	let url = result_out.blocking_recv()?;
 
-	let _ = shutdown_out.blocking_recv();
+	if wait_for_shutdown {
+		let _ = shutdown_out.blocking_recv();
+	}
 
 	Ok(url)
 }
+
+/// Helper function to start a mock server for testing purposes.
+/// Returns the server URL as a String.
+/// The server will be started on a random available port.
+pub async fn async_run_test_only() -> String {
+	tokio::task::spawn_blocking(move || run_with_shutdown_control(0, false))
+		.await
+		.expect("Fail to start mock server")
+		.expect("Failed to get server URL")
+}