use anonymize_rs::anonymizer::regex_anonymizer::RegexAnonymizer;
use anyhow::Result;

#[test]
fn test_regex_replace() -> Result<()> {
    let mut regex_anonymizer = RegexAnonymizer::new(None);
    regex_anonymizer.add_regex_pattern(r"\bapple\w*\b")?;
    regex_anonymizer.add_regex_pattern(r"\bbanana\w*\b")?;
    regex_anonymizer.add_regex_pattern(r"\bplum\w*\b")?;

    let test_cases = [
        (
            "I like to eat apples and bananas and plums",
            "I like to eat FRUIT0 and FRUIT1 and FRUIT2",
        ),
        (
            "I like to eat apples and apples and plums",
            "I like to eat FRUIT0 and FRUIT0 and FRUIT1",
        ),
        (
            "{\"content\": \"I like to eat apples and bananas and plums\"}",
            "{\"content\": \"I like to eat FRUIT0 and FRUIT1 and FRUIT2\"}",
        ),
    ];

    for test_case in test_cases {
        let text = test_case.0;

        let res = regex_anonymizer.replace_regex_matches(text, Some("FRUIT"), None)?;
        println!("{:?}", res);

        assert_eq!(res.text, test_case.1);
    }

    Ok(())
}

#[test]
fn test_regex_replace_file1() -> Result<()> {
    let mut regex_anonymizer = RegexAnonymizer::new(None);
    regex_anonymizer.add_regex_patterns_file("./tests/config/fruits_regex.txt")?;

    let text = "I like to eat apples and bananas and plums";

    let res = regex_anonymizer.replace_regex_matches(text, Some("FRUIT"), None)?;
    println!("{:?}", res);

    assert_eq!(res.text, "I like to eat FRUIT0 and FRUIT1 and FRUIT2");
    Ok(())
}

#[test]
fn test_regex_replace_file2() -> Result<()> {
    let mut regex_anonymizer = RegexAnonymizer::new(Some("FRUIT".to_string()));
    regex_anonymizer.add_regex_patterns_file("./tests/config/fruits_regex.txt")?;

    let text = "I like to eat apples and bananas and plums";

    let res = regex_anonymizer.replace_regex_matches(text, None, None)?;
    println!("{:?}", res);

    assert_eq!(res.text, "I like to eat FRUIT0 and FRUIT1 and FRUIT2");
    Ok(())
}