Update layout tests

[rbdr/page] / src / file_handler / file_strategies / file.rs

pub struct Strategy {}

use std::path::PathBuf;
use std::fs::{copy, create_dir_all};

use crate::file_handler::{File, FileType, FileHandlerStrategy};

impl Strategy {
    fn handle(&self, source: &PathBuf, destination: &PathBuf, file: &File) {
        let relative_path = file.path.strip_prefix(&source).unwrap();
        let complete_destination = destination.join(relative_path);
        let destination_parent = complete_destination.parent().unwrap();
        create_dir_all(destination_parent).unwrap();
        copy(&file.path, &complete_destination).unwrap();
    }
}

impl FileHandlerStrategy for Strategy {
    fn is(&self, path: &PathBuf) -> bool {
        !path.is_dir()
    }

    fn identify(&self) -> FileType {
        FileType::File
    }

    fn can_handle(&self, file_type: &FileType) -> bool {
        match file_type {
            FileType::File => true,
            _ => false,
        }
    }

    fn handle_html(&self, source: &PathBuf, destination: &PathBuf, file: &File, _l: &str) {
        return self.handle(source, destination, file);
    }

    fn handle_gemini(&self, source: &PathBuf, destination: &PathBuf, file: &File) {
        return self.handle(source, destination, file);
    }
}

#[cfg(test)]
mod tests {
    use std::fs;

    use super::*;

    use test_utilities::*;

    #[test]
    fn identifies_regular_files() {
        let test_dir = setup_test_dir();
        create_test_file(&test_dir.join("image.png"), "");
        create_test_file(&test_dir.join("style.css"), "");
        let strategy = Strategy {};
        assert!(strategy.is(&test_dir.join("image.png")));
        assert!(strategy.is(&test_dir.join("style.css")));
    }

    #[test]
    fn rejects_directories() {
        let test_dir = setup_test_dir();
        let strategy = Strategy {};
        assert!(!strategy.is(&test_dir));
    }

    #[test]
    fn identifies_file_type() {
        let strategy = Strategy {};
        assert!(matches!(strategy.identify(), FileType::File));
    }

    #[test]
    fn handles_file_type() {
        let strategy = Strategy {};
        assert!(strategy.can_handle(&FileType::File));
    }

    #[test]
    fn rejects_non_file_types() {
        let strategy = Strategy {};
        assert!(!strategy.can_handle(&FileType::Layout));
        assert!(!strategy.can_handle(&FileType::Gemini));
        assert!(!strategy.can_handle(&FileType::Unknown));
    }

    #[test]
    fn copies_single_file() {
        let test_dir = setup_test_dir();
        let source_dir = test_dir.join("source");
        let output_dir = test_dir.join("output");
        create_dir_all(&source_dir)
            .expect("Could not create source test directory");
        create_dir_all(&output_dir)
            .expect("Could not create output test directory");
        create_test_file(&source_dir.join("image.png"), "A fish playing the banjo");
        let strategy = Strategy {};

        let file = File {
            path: source_dir.join("image.png"),
            file_type: FileType::File,
        };

        strategy.handle(&source_dir, &output_dir, &file);

        let copied_path = &output_dir.join("image.png");
        assert!(copied_path.exists());

        // Verify file contents are identical
        let original = fs::read(&file.path).unwrap();
        let copied = fs::read(&copied_path).unwrap();
        assert_eq!(original, copied);
    }

    #[test]
    fn copies_nested_file() {
        let test_dir = setup_test_dir();
        let source_dir = test_dir.join("source");
        let output_dir = test_dir.join("output");
        create_dir_all(&source_dir)
            .expect("Could not create source test directory");
        create_dir_all(&output_dir)
            .expect("Could not create output test directory");
        create_dir_all(&source_dir.join("nested"))
            .expect("Could not create source test directory");
        create_test_file(&source_dir.join("nested/style.css"), "* { margin: 0; padding: 0 }");
        let strategy = Strategy {};

        let file = File {
            path: source_dir.join("nested/style.css"),
            file_type: FileType::File,
        };

        strategy.handle(&source_dir, &output_dir, &file);

        let copied_path = output_dir.join("nested/style.css");
        assert!(copied_path.exists());

        // Verify file contents are identical
        let original = fs::read(&file.path).unwrap();
        let copied = fs::read(&copied_path).unwrap();
        assert_eq!(original, copied);
    }

    #[test]
    fn handle_html_copies_file() {
        let test_dir = setup_test_dir();
        let source_dir = test_dir.join("source");
        let output_dir = test_dir.join("output");
        create_dir_all(&source_dir)
            .expect("Could not create source test directory");
        create_dir_all(&output_dir)
            .expect("Could not create output test directory");
        create_test_file(&source_dir.join("image.png"), "A fish playing the banjo");
        let strategy = Strategy {};

        let file = File {
            path: source_dir.join("image.png"),
            file_type: FileType::File,
        };

        strategy.handle_html(&source_dir, &output_dir, &file, "unused layout");

        let copied_path = &output_dir.join("image.png");
        assert!(copied_path.exists());

        // Verify file contents are identical
        let original = fs::read(&file.path).unwrap();
        let copied = fs::read(&copied_path).unwrap();
        assert_eq!(original, copied);
    }

    #[test]
    fn handle_gemini_copies_file() {
        let test_dir = setup_test_dir();
        let source_dir = test_dir.join("source");
        let output_dir = test_dir.join("output");
        create_dir_all(&source_dir)
            .expect("Could not create source test directory");
        create_dir_all(&output_dir)
            .expect("Could not create output test directory");
        create_test_file(&source_dir.join("image.png"), "A fish playing the banjo");
        let strategy = Strategy {};

        let file = File {
            path: source_dir.join("image.png"),
            file_type: FileType::File,
        };

        strategy.handle_gemini(&source_dir, &output_dir, &file);

        let copied_path = &output_dir.join("image.png");
        assert!(copied_path.exists());

        // Verify file contents are identical
        let original = fs::read(&file.path).unwrap();
        let copied = fs::read(&copied_path).unwrap();
        assert_eq!(original, copied);
    }
}