Q CLI Automation for basic chat and context

Author: shreyabhagat1

e2etests/tests/test_add_file_context.rs

@@ -0,0 +1,51 @@
+use q_cli_e2e_tests::q_chat_helper::QChatSession;
+
+#[test]
+#[cfg(feature = "context")]
+fn test_add_file_context() -> Result<(), Box<dyn std::error::Error>> {
+    println!("🔍 Testing /context add <filename> command...");
+    
+    let test_file_path = "/tmp/test_context_file.py";
+    
+    // Create a test file
+    std::fs::write(test_file_path, "# Test file for context\nprint('Hello from test file')")?;
+    println!("✅ Created test file at {}", test_file_path);
+    
+    let mut chat = QChatSession::new()?;
+    println!("✅ Q Chat session started");
+    
+    // Add file to context
+    let add_response = chat.execute_command(&format!("/context add {}", test_file_path))?;
+    
+    println!("📝 Context add response: {} bytes", add_response.len());
+    println!("📝 ADD RESPONSE:");
+    println!("{}", add_response);
+    println!("📝 END ADD RESPONSE");
+    
+    // Verify file was added successfully
+    assert!(add_response.contains("Added 1 path(s) to context"), "Missing success message for adding file");
+    println!("✅ File added to context successfully");
+    
+    // Execute /context show to confirm file is present
+    let show_response = chat.execute_command("/context show")?;
+    
+    println!("📝 Context show response: {} bytes", show_response.len());
+    println!("📝 SHOW RESPONSE:");
+    println!("{}", show_response);
+    println!("📝 END SHOW RESPONSE");
+    
+    // Verify file is present in context
+    assert!(show_response.contains(test_file_path), "File not found in context show output");
+    assert!(show_response.contains("💬 Session (temporary):"), "Missing Session section");
+    println!("✅ File confirmed present in context");
+    
+    chat.quit()?;
+    
+    // Clean up test file
+    let _ = std::fs::remove_file(test_file_path);
+    println!("✅ Cleaned up test file");
+    
+    println!("✅ Test completed successfully");
+    
+    Ok(())
+}

e2etests/tests/test_add_globe_pattern_file_context.rs

@@ -0,0 +1,59 @@
+use q_cli_e2e_tests::q_chat_helper::QChatSession;
+
+#[test]
+#[cfg(feature = "context")]
+fn test_add_glob_pattern_file_context() -> Result<(), Box<dyn std::error::Error>> {
+    println!("🔍 Testing /context add *.py glob pattern command...");
+    
+    let test_file1_path = "/tmp/test_context_file1.py";
+    let test_file2_path = "/tmp/test_context_file2.py";
+    let test_file3_path = "/tmp/test_context_file.js"; // Non-matching file
+    let glob_pattern = "/tmp/*.py";
+    
+    // Create test files
+    std::fs::write(test_file1_path, "# Test Python file 1 for context\nprint('Hello from Python file 1')")?;
+    std::fs::write(test_file2_path, "# Test Python file 2 for context\nprint('Hello from Python file 2')")?;
+    std::fs::write(test_file3_path, "// Test JavaScript file\nconsole.log('Hello from JS file');")?;
+    println!("✅ Created test files at {}, {}, {}", test_file1_path, test_file2_path, test_file3_path);
+    
+    let mut chat = QChatSession::new()?;
+    println!("✅ Q Chat session started");
+    
+    // Add glob pattern to context
+    let add_response = chat.execute_command(&format!("/context add {}", glob_pattern))?;
+    
+    println!("📝 Context add response: {} bytes", add_response.len());
+    println!("📝 ADD RESPONSE:");
+    println!("{}", add_response);
+    println!("📝 END ADD RESPONSE");
+    
+    // Verify glob pattern was added successfully
+    assert!(add_response.contains("Added 1 path(s) to context"), "Missing success message for adding glob pattern");
+    println!("✅ Glob pattern added to context successfully");
+    
+    // Execute /context show to confirm pattern matches files
+    let show_response = chat.execute_command("/context show")?;
+    
+    println!("📝 Context show response: {} bytes", show_response.len());
+    println!("📝 SHOW RESPONSE:");
+    println!("{}", show_response);
+    println!("📝 END SHOW RESPONSE");
+    
+    // Verify glob pattern is present and matches files
+    assert!(show_response.contains(glob_pattern), "Glob pattern not found in context show output");
+    assert!(show_response.contains("match"), "Missing match indicator for glob pattern");
+    assert!(show_response.contains("💬 Session (temporary):"), "Missing Session section");
+    println!("✅ Glob pattern confirmed present in context with matches");
+    
+    chat.quit()?;
+    
+    // Clean up test files
+    let _ = std::fs::remove_file(test_file1_path);
+    let _ = std::fs::remove_file(test_file2_path);
+    let _ = std::fs::remove_file(test_file3_path);
+    println!("✅ Cleaned up test files");
+    
+    println!("✅ Test completed successfully");
+    
+    Ok(())
+}

e2etests/tests/test_add_multiple_file_context.rs

@@ -0,0 +1,59 @@
+use q_cli_e2e_tests::q_chat_helper::QChatSession;
+
+#[test]
+#[cfg(feature = "context")]
+fn test_add_multiple_file_context() -> Result<(), Box<dyn std::error::Error>> {
+    println!("🔍 Testing /context add <filename1> <filename2> <filename3> command...");
+    
+    let test_file1_path = "/tmp/test_context_file1.py";
+    let test_file2_path = "/tmp/test_context_file2.js";
+    let test_file3_path = "/tmp/test_context_file3.txt";
+    
+    // Create multiple test files
+    std::fs::write(test_file1_path, "# Test Python file for context\nprint('Hello from Python file')")?;
+    std::fs::write(test_file2_path, "// Test JavaScript file for context\nconsole.log('Hello from JS file');")?;
+    std::fs::write(test_file3_path, "Test text file for context\nHello from text file")?;
+    println!("✅ Created test files at {}, {}, {}", test_file1_path, test_file2_path, test_file3_path);
+    
+    let mut chat = QChatSession::new()?;
+    println!("✅ Q Chat session started");
+    
+    // Add multiple files to context in one command
+    let add_response = chat.execute_command(&format!("/context add {} {} {}", test_file1_path, test_file2_path, test_file3_path))?;
+    
+    println!("📝 Context add response: {} bytes", add_response.len());
+    println!("📝 ADD RESPONSE:");
+    println!("{}", add_response);
+    println!("📝 END ADD RESPONSE");
+    
+    // Verify files were added successfully
+    assert!(add_response.contains("Added 3 path(s) to context"), "Missing success message for adding multiple files");
+    println!("✅ Multiple files added to context successfully");
+    
+    // Execute /context show to confirm files are present
+    let show_response = chat.execute_command("/context show")?;
+    
+    println!("📝 Context show response: {} bytes", show_response.len());
+    println!("📝 SHOW RESPONSE:");
+    println!("{}", show_response);
+    println!("📝 END SHOW RESPONSE");
+    
+    // Verify all files are present in context
+    assert!(show_response.contains(test_file1_path), "Python file not found in context show output");
+    assert!(show_response.contains(test_file2_path), "JavaScript file not found in context show output");
+    assert!(show_response.contains(test_file3_path), "Text file not found in context show output");
+    assert!(show_response.contains("💬 Session (temporary):"), "Missing Session section");
+    println!("✅ All files confirmed present in context");
+    
+    chat.quit()?;
+    
+    // Clean up test files
+    let _ = std::fs::remove_file(test_file1_path);
+    let _ = std::fs::remove_file(test_file2_path);
+    let _ = std::fs::remove_file(test_file3_path);
+    println!("✅ Cleaned up test files");
+    
+    println!("✅ Test completed successfully");
+    
+    Ok(())
+}

e2etests/tests/test_add_non_existing_file_context.rs

@@ -0,0 +1,31 @@
+use q_cli_e2e_tests::q_chat_helper::QChatSession;
+
+#[test]
+#[cfg(feature = "context")]
+fn test_add_non_existing_file_context() -> Result<(), Box<dyn std::error::Error>> {
+    println!("🔍 Testing /context add non-existing file command...");
+    
+    let non_existing_file_path = "/tmp/non_existing_file.py";
+    
+    let mut chat = QChatSession::new()?;
+    println!("✅ Q Chat session started");
+    
+    // Try to add non-existing file to context
+    let add_response = chat.execute_command(&format!("/context add {}", non_existing_file_path))?;
+    
+    println!("📝 Context add response: {} bytes", add_response.len());
+    println!("📝 ADD RESPONSE:");
+    println!("{}", add_response);
+    println!("📝 END ADD RESPONSE");
+    
+    // Verify error message for non-existing file
+    assert!(add_response.contains("Error:") && add_response.contains("Invalid path") && add_response.contains("does not exist"), "Missing error message for non-existing file");
+    assert!(add_response.contains("Use --force to add anyway"), "Missing --force suggestion in error message");
+    println!("✅ Found expected error message for non-existing file with --force suggestion");
+    
+    chat.quit()?;
+    
+    println!("✅ Test completed successfully");
+    
+    Ok(())
+}

e2etests/tests/test_clear_context_command.rs

@@ -0,0 +1,82 @@
+use q_cli_e2e_tests::q_chat_helper::QChatSession;
+
+#[test]
+#[cfg(feature = "context")]
+fn test_clear_context_command() -> Result<(), Box<dyn std::error::Error>> {
+    println!("🔍 Testing /context clear command...");
+    
+    let test_file1_path = "/tmp/test_context_file1.py";
+    let test_file2_path = "/tmp/test_context_file2.js";
+    
+    // Create multiple test files
+    std::fs::write(test_file1_path, "# Test Python file for context\nprint('Hello from Python file')")?;
+    std::fs::write(test_file2_path, "// Test JavaScript file for context\nconsole.log('Hello from JS file');")?;
+    println!("✅ Created test files at {}, {}", test_file1_path, test_file2_path);
+    
+    let mut chat = QChatSession::new()?;
+    println!("✅ Q Chat session started");
+    
+    // Add multiple files to context
+    let add_response = chat.execute_command(&format!("/context add {} {}", test_file1_path, test_file2_path))?;
+    
+    println!("📝 Context add response: {} bytes", add_response.len());
+    println!("📝 ADD RESPONSE:");
+    println!("{}", add_response);
+    println!("📝 END ADD RESPONSE");
+    
+    // Verify files were added successfully
+    assert!(add_response.contains("Added 2 path(s) to context"), "Missing success message for adding files");
+    println!("✅ Files added to context successfully");
+    
+    // Execute /context show to confirm files are present
+    let show_response = chat.execute_command("/context show")?;
+    
+    println!("📝 Context show response: {} bytes", show_response.len());
+    println!("📝 SHOW RESPONSE:");
+    println!("{}", show_response);
+    println!("📝 END SHOW RESPONSE");
+    
+    // Verify files are present in context
+    assert!(show_response.contains(test_file1_path), "Python file not found in context show output");
+    assert!(show_response.contains(test_file2_path), "JavaScript file not found in context show output");
+    println!("✅ Files confirmed present in context");
+    
+    // Execute /context clear to remove all files
+    let clear_response = chat.execute_command("/context clear")?;
+    
+    println!("📝 Context clear response: {} bytes", clear_response.len());
+    println!("📝 CLEAR RESPONSE:");
+    println!("{}", clear_response);
+    println!("📝 END CLEAR RESPONSE");
+    
+    // Verify context was cleared successfully
+    assert!(clear_response.contains("Cleared context"), "Missing success message for clearing context");
+    println!("✅ Context cleared successfully");
+    
+    // Execute /context show to confirm no files remain
+    let final_show_response = chat.execute_command("/context show")?;
+    
+    println!("📝 Final context show response: {} bytes", final_show_response.len());
+    println!("📝 FINAL SHOW RESPONSE:");
+    println!("{}", final_show_response);
+    println!("📝 END FINAL SHOW RESPONSE");
+    
+    // Verify no files remain in context
+    assert!(!final_show_response.contains(test_file1_path), "Python file still found in context after clear");
+    assert!(!final_show_response.contains(test_file2_path), "JavaScript file still found in context after clear");
+    assert!(final_show_response.contains("👤 Agent (q_cli_default):"), "Missing Agent section");
+    assert!(final_show_response.contains("💬 Session (temporary):"), "Missing Session section");
+    assert!(final_show_response.contains("<none>"), "Missing <none> indicator for cleared context");
+    println!("✅ All files confirmed removed from context and <none> sections present");
+    
+    chat.quit()?;
+    
+    // Clean up test files
+    let _ = std::fs::remove_file(test_file1_path);
+    let _ = std::fs::remove_file(test_file2_path);
+    println!("✅ Cleaned up test files");
+    
+    println!("✅ Test completed successfully");
+    
+    Ok(())
+}

e2etests/tests/test_load_command.rs

@@ -0,0 +1,65 @@
+use q_cli_e2e_tests::q_chat_helper::QChatSession;
+
+struct FileCleanup<'a> {
+    path: &'a str,
+}
+
+impl<'a> Drop for FileCleanup<'a> {
+    fn drop(&mut self) {
+        if std::path::Path::new(self.path).exists() {
+            let _ = std::fs::remove_file(self.path);
+            println!("✅ Cleaned up test file");
+        }
+    }
+}
+
+#[test]
+#[cfg(feature = "save_load")]
+fn test_load_command() -> Result<(), Box<dyn std::error::Error>> {
+    println!("🔍 Testing /load command...");
+    
+    let save_path = "/tmp/qcli_test_load.json";
+    let _cleanup = FileCleanup { path: save_path };
+    
+    let mut chat = QChatSession::new()?;
+    println!("✅ Q Chat session started");
+    
+    // Create actual conversation content
+    let _help_response = chat.execute_command("/help")?;
+    let _tools_response = chat.execute_command("/tools")?;
+    println!("✅ Created conversation content with /help and /tools commands");
+    
+    // Execute /save command to create a file to load
+    let save_response = chat.execute_command(&format!("/save {}", save_path))?;
+    
+    println!("📝 Save response: {} bytes", save_response.len());
+    println!("📝 SAVE OUTPUT:");
+    println!("{}", save_response);
+    println!("📝 END SAVE OUTPUT");
+    
+    // Verify save was successful
+    assert!(save_response.contains("Exported conversation state to") && save_response.contains(save_path), "Missing export confirmation message");
+    println!("✅ Save completed successfully");
+    
+    // Verify file was created
+    assert!(std::path::Path::new(save_path).exists(), "Save file was not created");
+    println!("✅ Save file created at {}", save_path);
+    
+    // Execute /load command to load the saved conversation
+    let load_response = chat.execute_command(&format!("/load {}", save_path))?;
+    
+    println!("📝 Load response: {} bytes", load_response.len());
+    println!("📝 LOAD OUTPUT:");
+    println!("{}", load_response);
+    println!("📝 END LOAD OUTPUT");
+    
+    // Verify load was successful
+    assert!(!load_response.is_empty(), "Load command should return non-empty response");
+    assert!(load_response.contains("Imported conversation state from") && load_response.contains(save_path), "Missing import confirmation message");
+    println!("✅ Load command executed successfully and imported conversation state");
+    
+    chat.quit()?;
+    println!("✅ Test completed successfully");
+    
+    Ok(())
+}

e2etests/tests/test_load_command_argument_validation.rs

@@ -0,0 +1,44 @@
+use q_cli_e2e_tests::q_chat_helper::QChatSession;
+
+#[test]
+#[cfg(feature = "save_load")]
+fn test_load_command_argument_validation() -> Result<(), Box<dyn std::error::Error>> {
+    println!("🔍 Testing /load command...");
+    
+    let mut chat = QChatSession::new()?;
+    println!("✅ Q Chat session started");
+    
+    let response = chat.execute_command("/load")?;
+    
+    println!("📝 Help response: {} bytes", response.len());
+    println!("📝 FULL OUTPUT:");
+    println!("{}", response);
+    println!("📝 END OUTPUT");
+    
+    // Verify load error message
+    assert!(response.contains("error:") && response.contains("the following required arguments were not provided:"), "Missing load error message");
+    println!("✅ Found load error message");
+    
+    assert!(response.contains("Usage:"), "Missing Usage section");
+    assert!(response.contains("/load"), "Missing /load command in usage");
+    println!("✅ Found Usage section with /load command");
+    
+    assert!(response.contains("Arguments:"), "Missing Arguments section");
+    assert!(response.contains("<PATH>"), "Missing PATH argument");
+    println!("✅ Found Arguments section with PATH parameter");
+    
+    assert!(response.contains("Options:"), "Missing Options section");
+    assert!(response.contains("-h"), "Missing -h flag");
+    assert!(response.contains("--help") || response.contains("—help"), "Missing --help flag");
+    println!("✅ Found Options section with -h, --help flags");
+    
+    assert!(response.contains("Print help"), "Missing help description");
+    println!("✅ Found help flag description");
+    
+    println!("✅ All help content verified!");
+    
+    chat.quit()?;
+    println!("✅ Test completed successfully");
+    
+    Ok(())
+}