use vkopt_message_parser::filter::Filter;

mod test_helper;
use test_helper::*;

pub fn read_events_filtered(fixture: &str, filter: Filter) -> Vec<String> {
    read_events_skipping(fixture, |e| filter.filter_event(e).is_some())
}

#[test]
fn it_does_not_apply_filters_by_default() {
    let filtered = read_events_filtered("messages.html", Default::default());
    let unfiltered = read_events("messages.html");
    assert_eq!(filtered, unfiltered);
}

#[test]
fn it_filters_by_min_date() {
    let date =
        chrono::NaiveDateTime::parse_from_str("2018.01.22 00:00:00", "%Y.%m.%d %H:%M:%S").unwrap();
    let filter = Filter {
        since_date: Some(date),
        ..Default::default()
    };
    let filtered = read_events_filtered("messages.html", filter);
    assert_events!(&filtered,
        "Start(0)",
        "FullNameExtracted(\"Sota\")",
        "ShortNameExtracted(\"sota\")",
        "DateExtracted(\"2018.01.21 13:53:59\")",
        "Start(0)",
        "FullNameExtracted(\"Sota\")",
        "ShortNameExtracted(\"sota\")",
        "DateExtracted(\"2018.01.21 11:05:13\")",
        "Start(0)",
        "FullNameExtracted(\"Denko\")",
        "ShortNameExtracted(\"denko\")",
        "DateExtracted(\"2018.01.21 17:02:54\")",
        "Start(0)",
        "FullNameExtracted(\"Sota\")",
        "ShortNameExtracted(\"sota\")",
        "DateExtracted(\"2018.01.22 10:03:04\")",
        "AttachmentExtracted { kind: Photo, url: \"photo\", vk_obj: \"photo420_13\", description: \"(420x6)\" }",
        "Start(0)",
        "FullNameExtracted(\"Sota\")",
        "ShortNameExtracted(\"sota\")",
        "DateExtracted(\"2018.01.22 10:05:13\")",
        "BodyPartExtracted(\"W-what do you think? I hope you like it (´･ω･`) here\\\'s my pixiv to crash the parser https://t.co/aaaaaa&lt;/a&gt\")"
    );
}

#[test]
fn it_filters_by_short_name_blacklist() {
    let mut blacklist = std::collections::BTreeSet::new();
    blacklist.insert("sota");
    let filter = Filter {
        short_name_blacklist: Some(blacklist),
        ..Default::default()
    };
    let filtered = read_events_filtered("messages.html", filter);
    assert_events!(&filtered,
        "Start(0)",
        "FullNameExtracted(\"Sota\")",
        "ShortNameExtracted(\"sota\")",
        "Start(0)",
        "FullNameExtracted(\"Sota\")",
        "ShortNameExtracted(\"sota\")",
        "Start(0)",
        "FullNameExtracted(\"Denko\")",
        "ShortNameExtracted(\"denko\")",
        "DateExtracted(\"2018.01.21 17:02:54\")",
        "BodyPartExtracted(\"🤔\")",
        "BodyPartExtracted(\"🤔\")",
        "BodyPartExtracted(\"🤔\")",
        "Start(0)",
        "FullNameExtracted(\"Sota\")",
        "ShortNameExtracted(\"sota\")",
        "Start(0)",
        "FullNameExtracted(\"Sota\")",
        "ShortNameExtracted(\"sota\")"
    );
}

#[test]
fn it_filters_by_short_name_whitelist() {
    let mut whitelist = std::collections::BTreeSet::new();
    whitelist.insert("sota");
    let filter = Filter {
        short_name_whitelist: Some(whitelist),
        ..Default::default()
    };
    let filtered = read_events_filtered("messages.html", filter);
    assert_events!(&filtered,
        "Start(0)",
        "FullNameExtracted(\"Sota\")",
        "ShortNameExtracted(\"sota\")",
        "DateExtracted(\"2018.01.21 13:53:59\")",
        "Start(0)",
        "FullNameExtracted(\"Sota\")",
        "ShortNameExtracted(\"sota\")",
        "DateExtracted(\"2018.01.21 11:05:13\")",
        "BodyPartExtracted(\"Hi Denko\")",
        "BodyPartExtracted(\"\\n\")",
        "BodyPartExtracted(\"\\n\")",
        "BodyPartExtracted(\"I’m drinking \\\"jasmine\\\" tea right now, thinking about what to have for dinner (´･ω･`)\")",
        "Start(0)",
        "FullNameExtracted(\"Denko\")",
        "ShortNameExtracted(\"denko\")",
        "Start(0)",
        "FullNameExtracted(\"Sota\")",
        "ShortNameExtracted(\"sota\")",
        "DateExtracted(\"2018.01.22 10:03:04\")",
        "AttachmentExtracted { kind: Photo, url: \"photo\", vk_obj: \"photo420_13\", description: \"(420x6)\" }",
        "Start(0)",
        "FullNameExtracted(\"Sota\")",
        "ShortNameExtracted(\"sota\")",
        "DateExtracted(\"2018.01.22 10:05:13\")",
        "BodyPartExtracted(\"W-what do you think? I hope you like it (´･ω･`) here\\\'s my pixiv to crash the parser https://t.co/aaaaaa&lt;/a&gt\")"
    );
}