Fix unit tests

Author: Folyd

aiscript/src/project.rs

@@ -1,6 +1,6 @@
 use std::fs;
 use std::io::Write;
-use std::path::PathBuf;
+use std::path::{PathBuf};
 
 pub struct ProjectGenerator {
     project_name: String,
@@ -35,17 +35,14 @@ impl ProjectGenerator {
 
         // Create standard directories
         self.create_directories()?;
-
+        
         // Create project.toml
         self.create_project_toml()?;
-
+        
         // Create basic example file
         self.create_example_file()?;
 
-        println!(
-            "Successfully created new AIScript project: {}",
-            self.project_name
-        );
+        println!("Successfully created new AIScript project: {}", self.project_name);
         println!("Project structure:");
         println!("{}", self.display_project_structure());
         println!("");
@@ -56,21 +53,21 @@ impl ProjectGenerator {
 
     fn create_directories(&self) -> Result<(), String> {
         let dirs = vec!["lib", "routes"];
-
+        
         for dir in dirs {
             let dir_path = self.project_path.join(dir);
             fs::create_dir_all(&dir_path).map_err(|e| {
                 format!("Failed to create directory '{}': {}", dir_path.display(), e)
             })?;
         }
-
+        
         Ok(())
     }
 
     fn create_project_toml(&self) -> Result<(), String> {
         let toml_path = self.project_path.join("project.toml");
         let username = whoami::username();
-
+        
         let toml_content = format!(
             r#"[project]
 name = "{}"
@@ -89,20 +86,22 @@ path = "/docs"
 "#,
             self.project_name, username
         );
-
-        let mut file = fs::File::create(&toml_path)
-            .map_err(|e| format!("Failed to create project.toml: {}", e))?;
-
-        file.write_all(toml_content.as_bytes())
-            .map_err(|e| format!("Failed to write to project.toml: {}", e))?;
-
+        
+        let mut file = fs::File::create(&toml_path).map_err(|e| {
+            format!("Failed to create project.toml: {}", e)
+        })?;
+        
+        file.write_all(toml_content.as_bytes()).map_err(|e| {
+            format!("Failed to write to project.toml: {}", e)
+        })?;
+        
         Ok(())
     }
 
     fn create_example_file(&self) -> Result<(), String> {
         let routes_dir = self.project_path.join("routes");
         let example_path = routes_dir.join("index.ai");
-
+        
         let example_content = r#"// Example AIScript route handler
 get /hello {
     query {
@@ -112,13 +111,15 @@ get /hello {
     return { message: f"Hello, {query.name}!" };
 }
 "#;
-
-        let mut file = fs::File::create(&example_path)
-            .map_err(|e| format!("Failed to create example file: {}", e))?;
-
-        file.write_all(example_content.as_bytes())
-            .map_err(|e| format!("Failed to write to example file: {}", e))?;
-
+        
+        let mut file = fs::File::create(&example_path).map_err(|e| {
+            format!("Failed to create example file: {}", e)
+        })?;
+        
+        file.write_all(example_content.as_bytes()).map_err(|e| {
+            format!("Failed to write to example file: {}", e)
+        })?;
+        
         Ok(())
     }
 
@@ -128,7 +129,7 @@ get /hello {
         result.push_str("├── routes/\n");
         result.push_str("│   └── index.ai\n");
         result.push_str("└── project.toml\n");
-
+        
         result
     }
 }
@@ -144,75 +145,65 @@ mod tests {
         // Use tempdir to ensure test files are cleaned up
         let temp_dir = tempdir().unwrap();
         let temp_path = temp_dir.path();
-
+        
         // Create a test project in the temp directory
         let project_name = "test_project";
+        
+        // Create an absolute path for the project
         let project_path = temp_path.join(project_name);
-
-        // Run in the temp directory
-        std::env::set_current_dir(temp_path).unwrap();
-
+        
+        // Create a generator with the project name
         let generator = ProjectGenerator::new(project_name);
+        
+        // Override the project path for testing
+        let generator = ProjectGenerator {
+            project_name: project_name.to_string(),
+            project_path: project_path.clone(),
+        };
+        
         let result = generator.generate();
-
+        
         assert!(result.is_ok(), "Project generation failed: {:?}", result);
-
+        
         // Verify project structure
         assert!(project_path.exists(), "Project directory not created");
-        assert!(
-            project_path.join("lib").exists(),
-            "lib directory not created"
-        );
-        assert!(
-            project_path.join("routes").exists(),
-            "routes directory not created"
-        );
-        assert!(
-            project_path.join("project.toml").exists(),
-            "project.toml not created"
-        );
-        assert!(
-            project_path.join("routes/index.ai").exists(),
-            "Example file not created"
-        );
-
+        assert!(project_path.join("lib").exists(), "lib directory not created");
+        assert!(project_path.join("routes").exists(), "routes directory not created");
+        assert!(project_path.join("project.toml").exists(), "project.toml not created");
+        assert!(project_path.join("routes/index.ai").exists(), "Example file not created");
+        
         // Verify project.toml content
         let toml_content = fs::read_to_string(project_path.join("project.toml")).unwrap();
         assert!(toml_content.contains(&format!("name = \"{}\"", project_name)));
         assert!(toml_content.contains("version = \"0.1.0\""));
-
+        
         // Verify example file content
         let example_content = fs::read_to_string(project_path.join("routes/index.ai")).unwrap();
-        assert!(example_content.contains("fn index(req, res)"));
+        assert!(example_content.contains("get /hello"));
     }
-
+    
     #[test]
     fn test_project_already_exists() {
         // Use tempdir to ensure test files are cleaned up
         let temp_dir = tempdir().unwrap();
         let temp_path = temp_dir.path();
-
+        
         // Create a directory that will conflict
         let project_name = "existing_project";
         let project_path = temp_path.join(project_name);
         fs::create_dir_all(&project_path).unwrap();
-
-        // Run in the temp directory
-        std::env::set_current_dir(temp_path).unwrap();
-
-        let generator = ProjectGenerator::new(project_name);
+        
+        // Create a generator with absolute path
+        let generator = ProjectGenerator {
+            project_name: project_name.to_string(),
+            project_path,
+        };
+        
         let result = generator.generate();
-
-        assert!(
-            result.is_err(),
-            "Project generation should fail for existing directory"
-        );
+        
+        assert!(result.is_err(), "Project generation should fail for existing directory");
         if let Err(err) = result {
-            assert!(
-                err.contains("already exists"),
-                "Wrong error message: {}",
-                err
-            );
+            assert!(err.contains("already exists"), "Wrong error message: {}", err);
         }
     }
 }