web3infra-foundation
diff --git a/‎.gitignore‎
Lines changed: 3 additions & 0 deletions b/‎.gitignore‎
Lines changed: 3 additions & 0 deletions
diff --git a/‎libra/src/command/remove.rs‎
Lines changed: 13 additions & 13 deletions b/‎libra/src/command/remove.rs‎
Lines changed: 13 additions & 13 deletions
diff --git a/‎libra/tests/command/add_test.rs‎
Lines changed: 253 additions & 0 deletions b/‎libra/tests/command/add_test.rs‎
Lines changed: 253 additions & 0 deletions
@@ -21,6 +21,9 @@ Cargo.lock
 .DS_Store
 **/*.dylib
 
+.cursor
+.md
+
 # See https://help.github.com/articles/ignoring-files/ for more about ignoring files.
 .VSCodeCounter
 
 
@@ -33,17 +33,15 @@ pub fn execute(args: RemoveArgs) -> Result<(), GitError> {
     let mut index = Index::load(&idx_file)?;
 
     // check if pathspec is all in index (skip if force is enabled)
-    if !args.force && !validate_pathspec(&args.pathspec, &index) {
-        return Ok(());
+    if !args.force {
+        validate_pathspec(&args.pathspec, &index)?
     }
 
     let dirs = get_dirs(&args.pathspec, &index, args.force);
     if !dirs.is_empty() && !args.recursive {
-        println!(
-            "fatal: not removing '{}' recursively without -r",
-            dirs[0].bright_blue()
-        ); // Git print first
-        return Ok(());
+        let error_msg = format!("not removing '{}' recursively without -r", dirs[0]);
+        println!("fatal: {error_msg}");
+        return Err(GitError::CustomError(error_msg));
     }
 
     for path_str in args.pathspec.iter() {
@@ -87,10 +85,11 @@ pub fn execute(args: RemoveArgs) -> Result<(), GitError> {
 
 /// check if pathspec is all valid(in index)
 /// - if path is a dir, check if any file in the dir is in index
-fn validate_pathspec(pathspec: &[String], index: &Index) -> bool {
+fn validate_pathspec(pathspec: &[String], index: &Index) -> Result<(), GitError> {
     if pathspec.is_empty() {
-        println!("fatal: No pathspec was given. Which files should I remove?");
-        return false;
+        let error_msg = "No pathspec was given. Which files should I remove?".to_string();
+        println!("fatal: {error_msg}");
+        return Err(GitError::CustomError(error_msg));
     }
     for path_str in pathspec.iter() {
         let path = PathBuf::from(path_str);
@@ -99,12 +98,13 @@ fn validate_pathspec(pathspec: &[String], index: &Index) -> bool {
             // not tracked, but path may be a directory
             // check if any tracked file in the directory
             if !index.contains_dir_file(&path_wd) {
-                println!("fatal: pathspec '{path_str}' did not match any files");
-                return false;
+                let error_msg = format!("pathspec '{path_str}' did not match any files");
+                println!("fatal: {error_msg}");
+                return Err(GitError::CustomError(error_msg));
             }
         }
     }
-    true
+    Ok(())
 }
 
 /// run after `validate_pathspec`
 
@@ -1 +1,254 @@
+use super::*;
+use std::fs;
+use std::io::Write;
 
+#[tokio::test]
+#[serial]
+/// Tests the basic functionality of add command by adding a single file
+async fn test_add_single_file() {
+    let test_dir = tempdir().unwrap();
+    test::setup_with_new_libra_in(test_dir.path()).await;
+    let _guard = test::ChangeDirGuard::new(test_dir.path());
+
+    // Create a new file
+    let file_content = "Hello, World!";
+    let file_path = "test_file.txt";
+    let mut file = fs::File::create(file_path).unwrap();
+    file.write_all(file_content.as_bytes()).unwrap();
+
+    // Execute add command
+    add::execute(AddArgs {
+        pathspec: vec![String::from(file_path)],
+        all: false,
+        update: false,
+        refresh: false,
+        verbose: false,
+        dry_run: false,
+        ignore_errors: false,
+    })
+    .await;
+
+    // Verify the file was added to index.
+    let changes = changes_to_be_committed().await;
+
+    assert!(changes.new.iter().any(|x| x.to_str().unwrap() == file_path));
+}
+
+#[tokio::test]
+#[serial]
+/// Tests adding multiple files at once
+async fn test_add_multiple_files() {
+    let test_dir = tempdir().unwrap();
+    test::setup_with_new_libra_in(test_dir.path()).await;
+    let _guard = test::ChangeDirGuard::new(test_dir.path());
+
+    // Create multiple files
+    for i in 1..=3 {
+        let file_content = format!("File content {i}");
+        let file_path = format!("test_file_{i}.txt");
+        let mut file = fs::File::create(&file_path).unwrap();
+        file.write_all(file_content.as_bytes()).unwrap();
+    }
+
+    // Execute add command
+    add::execute(AddArgs {
+        pathspec: vec![
+            String::from("test_file_1.txt"),
+            String::from("test_file_2.txt"),
+            String::from("test_file_3.txt"),
+        ],
+        all: false,
+        update: false,
+        refresh: false,
+        verbose: false,
+        dry_run: false,
+        ignore_errors: false,
+    })
+    .await;
+
+    // Verify all files were added to index
+    let changes = changes_to_be_committed().await;
+    assert!(changes.new.iter().any(|x| x.to_str().unwrap() == "test_file_1.txt"));
+    assert!(changes.new.iter().any(|x| x.to_str().unwrap() == "test_file_2.txt"));
+    assert!(changes.new.iter().any(|x| x.to_str().unwrap() == "test_file_3.txt"));
+}
+
+#[tokio::test]
+#[serial]
+/// Tests the --all flag which adds all files in the working tree
+async fn test_add_all_flag() {
+    let test_dir = tempdir().unwrap();
+    test::setup_with_new_libra_in(test_dir.path()).await;
+    let _guard = test::ChangeDirGuard::new(test_dir.path());
+
+    // Create multiple files
+    for i in 1..=3 {
+        let file_content = format!("File content {i}");
+        let file_path = format!("test_file_{i}.txt");
+        let mut file = fs::File::create(&file_path).unwrap();
+        file.write_all(file_content.as_bytes()).unwrap();
+    }
+
+    // Execute add command with --all flag
+    add::execute(AddArgs {
+        pathspec: vec![],
+        all: true,
+        update: false,
+        refresh: false,
+        verbose: false,
+        dry_run: false,
+        ignore_errors: false,
+    })
+    .await;
+
+    // Verify all files were added to index
+    let changes = changes_to_be_committed().await;
+    assert!(changes.new.iter().any(|x| x.to_str().unwrap() == "test_file_1.txt"));
+    assert!(changes.new.iter().any(|x| x.to_str().unwrap() == "test_file_2.txt"));
+    assert!(changes.new.iter().any(|x| x.to_str().unwrap() == "test_file_3.txt"));
+}
+
+#[tokio::test]
+#[serial]
+/// Tests the --update flag which only updates files already in the index
+async fn test_add_update_flag() {
+    let test_dir = tempdir().unwrap();
+    test::setup_with_new_libra_in(test_dir.path()).await;
+    let _guard = test::ChangeDirGuard::new(test_dir.path());
+
+    // Create files and add one to the index
+    let tracked_file = "tracked_file.txt";
+    let untracked_file = "untracked_file.txt";
+    
+    // Create and write initial content
+    let mut file1 = fs::File::create(tracked_file).unwrap();
+    file1.write_all(b"Initial content").unwrap();
+    
+    let mut file2 = fs::File::create(untracked_file).unwrap();
+    file2.write_all(b"Initial content").unwrap();
+
+    // Add only one file to the index
+    add::execute(AddArgs {
+        pathspec: vec![String::from(tracked_file)],
+        all: false,
+        update: false,
+        refresh: false,
+        verbose: false,
+        dry_run: false,
+        ignore_errors: false,
+    })
+    .await;
+
+    // Modify both files
+    let mut file1 = fs::OpenOptions::new().write(true).truncate(true).open(tracked_file).unwrap();
+    file1.write_all(b" - Modified").unwrap();
+    
+    let mut file2 = fs::OpenOptions::new().write(true).truncate(true).open(untracked_file).unwrap();
+    file2.write_all(b" - Modified").unwrap();
+
+    // Execute add command with --update flag
+    add::execute(AddArgs {
+        pathspec: vec![String::from(".")],
+        all: false,
+        update: true,
+        refresh: false,
+        verbose: false,
+        dry_run: false,
+        ignore_errors: false,
+    })
+    .await;
+
+    // Verify only tracked file was updated
+    let changes = changes_to_be_staged();
+    // Tracked file should not appear in changes (because it was updated in index)
+    assert!(!changes.modified.iter().any(|x| x.to_str().unwrap() == tracked_file));
+    // Untracked file should still be untracked and show as new
+    assert!(changes.new.iter().any(|x| x.to_str().unwrap() == untracked_file));
+}
+
+#[tokio::test]
+#[serial]
+/// Tests adding files with respect to ignore patterns in .libraignore
+async fn test_add_with_ignore_patterns() {
+    let test_dir = tempdir().unwrap();
+    test::setup_with_new_libra_in(test_dir.path()).await;
+    let _guard = test::ChangeDirGuard::new(test_dir.path());
+
+    // Create .libraignore file
+    let mut ignore_file = fs::File::create(".libraignore").unwrap();
+    ignore_file.write_all(b"ignored_*.txt\nignore_dir/**").unwrap();
+
+    // Create files that should be ignored and not ignored
+    let ignored_file = "ignored_file.txt";
+    let tracked_file = "tracked_file.txt";
+    
+    // Create directory that should be ignored
+    fs::create_dir("ignore_dir").unwrap();
+    let ignored_dir_file = "ignore_dir/file.txt";
+
+    // Create and write content
+    let mut file1 = fs::File::create(ignored_file).unwrap();
+    file1.write_all(b"Should be ignored").unwrap();
+    
+    let mut file2 = fs::File::create(tracked_file).unwrap();
+    file2.write_all(b"Should be tracked").unwrap();
+    
+    let mut file3 = fs::File::create(ignored_dir_file).unwrap();
+    file3.write_all(b"Should be ignored").unwrap();
+
+    // Execute add command with all files
+    add::execute(AddArgs {
+        pathspec: vec![String::from(".")],
+        all: true,
+        update: false,
+        refresh: false,
+        verbose: false,
+        dry_run: false,
+        ignore_errors: false,
+    })
+    .await;
+
+    // Verify only non-ignored files were added
+    let changes_staged = changes_to_be_staged();
+    let changes_committed = changes_to_be_committed().await;
+    
+    // Ignored files should not appear in any status (they are ignored)
+    assert!(!changes_staged.new.iter().any(|x| x.to_str().unwrap() == ignored_file));
+    assert!(!changes_staged.new.iter().any(|x| x.to_str().unwrap() == ignored_dir_file));
+    assert!(!changes_committed.new.iter().any(|x| x.to_str().unwrap() == ignored_file));
+    assert!(!changes_committed.new.iter().any(|x| x.to_str().unwrap() == ignored_dir_file));
+    
+    // Non-ignored file should not show as new in staged (was added) but should show in committed
+    assert!(!changes_staged.new.iter().any(|x| x.to_str().unwrap() == tracked_file));
+    assert!(changes_committed.new.iter().any(|x| x.to_str().unwrap() == tracked_file));
+}
+
+#[tokio::test]
+#[serial]
+/// Tests the dry-run flag which should not actually add files
+async fn test_add_dry_run() {
+    let test_dir = tempdir().unwrap();
+    test::setup_with_new_libra_in(test_dir.path()).await;
+    let _guard = test::ChangeDirGuard::new(test_dir.path());
+
+    // Create a file.
+    let file_path = "test_file.txt";
+    let mut file = fs::File::create(file_path).unwrap();
+    file.write_all(b"Test content").unwrap();
+
+    // Execute add command with dry-run
+    add::execute(AddArgs {
+        pathspec: vec![String::from(file_path)],
+        all: false,
+        update: false,
+        refresh: false,
+        verbose: false,
+        dry_run: true,
+        ignore_errors: false,
+    })
+    .await;
+
+    // Verify the file was not actually added to index
+    let changes = changes_to_be_staged();
+    assert!(changes.new.iter().any(|x| x.to_str().unwrap() == file_path));
+}