#[macro_use]
extern crate log;
extern crate rand;
extern crate winlog;

use log::Level;
use rand::{distributions::Alphanumeric, thread_rng, Rng};
use std::{process::Command, str};
use winlog::{deregister, init, register};

#[test]
fn end_to_end() {
    let rand_string: String = thread_rng().sample_iter(&Alphanumeric).take(16).collect();
    let log_source = format!("winlog-test-{}", rand_string);

    // Add log source to Windows registry
    register(&log_source);

    // Do some logging and verification
    init(&log_source).unwrap();
    log_and_verify_one(Level::Error, &log_source, "Error", "1", "Error!!");
    log_and_verify_one(Level::Warn, &log_source, "Warning", "2", "Warning!!");
    log_and_verify_one(Level::Info, &log_source, "Information", "3", "Info!!");
    log_and_verify_one(Level::Debug, &log_source, "Information", "4", "Debug!!");
    log_and_verify_one(Level::Trace, &log_source, "Information", "5", "Trace!!");

    // Remove log source from Windows registry
    deregister(&log_source);
}

fn log_and_verify_one(level: Level, log_source: &str, entry_type: &str, entry_id: &str, msg: &str) {
    log!(level, "{}", msg);

    // Use PowerShell to extract formatted entries from the event log.
    let mut command = Command::new("powershell");
    command.arg("-Command").arg(format!(
        "Get-EventLog -Newest 1 -LogName Application -Source {} \
         | Select-Object Source, EntryType, EventID, Message \
         | foreach {{ \"$_\" }}",
        log_source
    ));
    let out = command.output().unwrap();

    assert_eq!(
        format!(
            "@{{Source={}; EntryType={}; EventID={}; Message={}}}\r\n",
            &log_source, &entry_type, entry_id, msg
        ),
        str::from_utf8(&out.stdout).unwrap()
    );
}