use rust_sensitive::model::{AcSensitiveWordMap, MyTrait};
#[test]
fn test_filter_case1() {
let sensitive_word_map = AcSensitiveWordMap::init_ac_dic_from_file("./data.txt");
let text = "你真敏感";
let filtered_text = sensitive_word_map.filter(text);
assert_eq!(filtered_text, "你真");
}
#[test]
fn test_filter_case2() {
let sensitive_word_map = AcSensitiveWordMap::init_ac_dic_from_file("./data.txt");
let text = "你真敏感测试asss832";
let filtered_text = sensitive_word_map.filter(text);
assert_eq!(filtered_text, "你真ss");
}
#[test]
fn test_replace_case1() {
let sensitive_word_map = AcSensitiveWordMap::init_ac_dic_from_file("./data.txt");
let text = "你真敏感测试asss832";
let replaced_text = sensitive_word_map.replace(text, "*");
assert_eq!(replaced_text, "你真******ss***");
}
#[test]
fn test_replace_case2() {
let sensitive_word_map = AcSensitiveWordMap::init_ac_dic_from_file("./data.txt");
let text = "敏感测试832as";
let replaced_text = sensitive_word_map.replace(text, "!");
assert_eq!(replaced_text, "!!!!!!!!!");
}
#[test]
fn test_is_sensitive_case1() {
let sensitive_word_map = AcSensitiveWordMap::init_ac_dic_from_file("./data.txt");
let text = "你真敏感测试asss832";
let is_sensitive = sensitive_word_map.is_sensitive(text);
assert_eq!(is_sensitive, true);
}
#[test]
fn test_is_sensitive_case2() {
let sensitive_word_map = AcSensitiveWordMap::init_ac_dic_from_file("./data.txt");
let text = "这是个测案例试";
let is_sensitive = sensitive_word_map.is_sensitive(text);
assert_eq!(is_sensitive, false);
}
#[test]
fn test_find_one_case1() {
let sensitive_word_map = AcSensitiveWordMap::init_ac_dic_from_file("./data.txt");
let text = "你真敏感测试asss832";
let result = sensitive_word_map.find_one(text);
assert_eq!(result, "敏感".to_string());
}
#[test]
fn test_find_one_case2() {
let sensitive_word_map = AcSensitiveWordMap::init_ac_dic_from_file("./data.txt");
let text = "这是个测案例试";
let result = sensitive_word_map.find_one(text);
assert_eq!(result, "".to_string());
}
#[test]
fn test_find_all_case1() {
let sensitive_word_map = AcSensitiveWordMap::init_ac_dic_from_file("./data.txt");
let text = "你真敏感测试asss832";
let result = sensitive_word_map.find_all(text);
assert_eq!(
result,
vec![
"敏感".to_string(),
"测试".to_string(),
"as".to_string(),
"832".to_string()
]
);
}
#[test]
fn test_find_all_case2() {
let sensitive_word_map = AcSensitiveWordMap::init_ac_dic_from_file("./data.txt");
let text = "这是个测案例试";
let result = sensitive_word_map.find_all(text);
assert_eq!(result, Vec::<String>::new());
}
#[test]
fn test_is_sensitive_case3() {
let sensitive_word_map = AcSensitiveWordMap::init_ac_dic_from_file("./data.txt");
let m = AcSensitiveWordMap::get_ac_dic();
let text = "你真敏感测试asss832";
let is_sensitive = m.is_sensitive(text);
assert_eq!(is_sensitive, true);
}