tokio-test

Tokio Test Command

This command generates comprehensive async tests for Tokio applications, including unit tests, integration tests, benchmarks, and property-based tests.

Arguments

• $1 - Target to generate tests for: file path, module name, or function name (required)
• $2 - Test type: unit, integration, benchmark, or all (optional, defaults to unit)

Usage

/rust-tokio-expert:tokio-test src/handlers/user.rs
/rust-tokio-expert:tokio-test src/service.rs integration
/rust-tokio-expert:tokio-test process_request benchmark
/rust-tokio-expert:tokio-test src/api/ all

Workflow

1. Parse Arguments

   • Validate target is provided
   • Determine test type (unit, integration, benchmark, all)
   • Identify target scope (file, module, or function)

2. Analyze Target Code

   • Read the target file(s) using Read tool
   • Identify async functions to test
   • Analyze function signatures and dependencies
   • Detect error types and return values

3. Invoke Agent

   • Use Task tool with subagent_type="rust-tokio-expert:tokio-pro"
   • Provide code context and test requirements
   • Request test generation based on type

4. Generate Unit Tests

   For each async function, create tests covering:

   Happy Path Tests

   #[tokio::test]
   async fn test_process_user_success() {
       // Arrange
       let user_id = 1;
       let expected_name = "John Doe";
   
       // Act
       let result = process_user(user_id).await;
   
       // Assert
       assert!(result.is_ok());
       let user = result.unwrap();
       assert_eq!(user.name, expected_name);
   }
   

   Error Handling Tests

   #[tokio::test]
   async fn test_process_user_not_found() {
       let result = process_user(999).await;
   
       assert!(result.is_err());
       assert!(matches!(result.unwrap_err(), Error::NotFound));
   }
   

   Timeout Tests

   #[tokio::test]
   async fn test_operation_completes_within_timeout() {
       use tokio::time::{timeout, Duration};
   
       let result = timeout(
           Duration::from_secs(5),
           slow_operation()
       ).await;
   
       assert!(result.is_ok(), "Operation timed out");
   }