mirror of
https://git.proxmox.com/git/pmg-log-tracker
synced 2025-04-29 11:15:31 +00:00
2912583794
Because of the relay before lmtp issue we now add and remove QEntrys based on their message-id. But if the cleanup line containing the message-id appears twice in the log for the same QEntry it keeps a weak reference to itself which leads to a panic because of a mutable borrow while it is already mutably borrowed in the 'finalize' function. To circumvent this we check after the lookup if it is the same QEntry and if so, insert it again with that message-id in the lookup table and don't add the weak reference. Signed-off-by: Mira Limbeck <m.limbeck@proxmox.com> Tested-By: Aaron Lauterer <a.lauterer@proxmox.com>
198 lines
5.9 KiB
Rust
198 lines
5.9 KiB
Rust
use std::fs::File;
|
|
use std::io::BufReader;
|
|
use std::process::Command;
|
|
|
|
mod utils;
|
|
|
|
#[test]
|
|
fn after_queue_start_end_time_string() {
|
|
let output = Command::new("faketime")
|
|
.arg("2020-12-31 23:59:59")
|
|
.arg(utils::log_tracker_path())
|
|
.arg("-vv")
|
|
.arg("-s")
|
|
.arg("2020-12-18 15:40:00")
|
|
.arg("-e")
|
|
.arg("2020-12-18 16:00:00")
|
|
.arg("-i")
|
|
.arg("tests/test_input_mixed")
|
|
.output()
|
|
.expect("failed to execute pmg-log-tracker");
|
|
|
|
let expected_file =
|
|
File::open("tests/test_output_after_queue").expect("failed to open test_output");
|
|
|
|
let expected_output = BufReader::new(&expected_file);
|
|
let output_reader = BufReader::new(&output.stdout[..]);
|
|
utils::compare_output(output_reader, expected_output);
|
|
}
|
|
|
|
#[test]
|
|
fn after_queue_start_end_timestamp() {
|
|
let output = Command::new("faketime")
|
|
.arg("2020-12-31 23:59:59")
|
|
.arg(utils::log_tracker_path())
|
|
.arg("-vv")
|
|
.arg("-s")
|
|
.arg("1608302400")
|
|
.arg("-e")
|
|
.arg("1608303600")
|
|
.arg("-i")
|
|
.arg("tests/test_input_mixed")
|
|
.output()
|
|
.expect("failed to execute pmg-log-tracker");
|
|
|
|
let expected_file =
|
|
File::open("tests/test_output_after_queue").expect("failed to open test_output");
|
|
|
|
let expected_output = BufReader::new(&expected_file);
|
|
let output_reader = BufReader::new(&output.stdout[..]);
|
|
utils::compare_output(output_reader, expected_output);
|
|
}
|
|
|
|
#[test]
|
|
fn after_queue_qid() {
|
|
let output = Command::new("faketime")
|
|
.arg("2020-12-31 23:59:59")
|
|
.arg(utils::log_tracker_path())
|
|
.arg("-vv")
|
|
.arg("-s")
|
|
.arg("1608302400")
|
|
.arg("-e")
|
|
.arg("1608303600")
|
|
.arg("-i")
|
|
.arg("tests/test_input_mixed")
|
|
.arg("-q")
|
|
.arg("0022C3801B5")
|
|
.output()
|
|
.expect("failed to execute pmg-log-tracker");
|
|
|
|
let expected_file =
|
|
File::open("tests/test_output_after_queue_qid").expect("failed to open test_output");
|
|
|
|
let expected_output = BufReader::new(&expected_file);
|
|
let output_reader = BufReader::new(&output.stdout[..]);
|
|
utils::compare_output(output_reader, expected_output);
|
|
}
|
|
|
|
#[test]
|
|
fn after_queue_host() {
|
|
let output = Command::new("faketime")
|
|
.arg("2020-12-31 23:59:59")
|
|
.arg(utils::log_tracker_path())
|
|
.arg("-vv")
|
|
.arg("-s")
|
|
.arg("1608302400")
|
|
.arg("-e")
|
|
.arg("1608303600")
|
|
.arg("-i")
|
|
.arg("tests/test_input_mixed")
|
|
.arg("-h")
|
|
.arg("localhost")
|
|
.output()
|
|
.expect("failed to execute pmg-log-tracker");
|
|
|
|
let expected_file =
|
|
File::open("tests/test_output_after_queue_host").expect("failed to open test_output");
|
|
|
|
let expected_output = BufReader::new(&expected_file);
|
|
let output_reader = BufReader::new(&output.stdout[..]);
|
|
utils::compare_output(output_reader, expected_output);
|
|
}
|
|
|
|
#[test]
|
|
fn after_queue_search_string() {
|
|
let output = Command::new("faketime")
|
|
.arg("2020-12-31 23:59:59")
|
|
.arg(utils::log_tracker_path())
|
|
.arg("-vv")
|
|
.arg("-s")
|
|
.arg("1608302400")
|
|
.arg("-e")
|
|
.arg("1608303600")
|
|
.arg("-i")
|
|
.arg("tests/test_input_mixed")
|
|
.arg("-x")
|
|
.arg("reject")
|
|
.output()
|
|
.expect("failed to execute pmg-log-tracker");
|
|
|
|
let expected_file = File::open("tests/test_output_after_queue_search_string")
|
|
.expect("failed to open test_output");
|
|
|
|
let expected_output = BufReader::new(&expected_file);
|
|
let output_reader = BufReader::new(&output.stdout[..]);
|
|
utils::compare_output(output_reader, expected_output);
|
|
}
|
|
|
|
#[test]
|
|
fn after_queue_search_string_case_insensitive() {
|
|
let output = Command::new("faketime")
|
|
.arg("2020-12-31 23:59:59")
|
|
.arg(utils::log_tracker_path())
|
|
.arg("-vv")
|
|
.arg("-s")
|
|
.arg("1608302400")
|
|
.arg("-e")
|
|
.arg("1608303600")
|
|
.arg("-i")
|
|
.arg("tests/test_input_mixed")
|
|
.arg("-x")
|
|
.arg("reJECT")
|
|
.output()
|
|
.expect("failed to execute pmg-log-tracker");
|
|
|
|
let expected_file = File::open("tests/test_output_after_queue_search_string")
|
|
.expect("failed to open test_output");
|
|
|
|
let expected_output = BufReader::new(&expected_file);
|
|
let output_reader = BufReader::new(&output.stdout[..]);
|
|
utils::compare_output(output_reader, expected_output);
|
|
}
|
|
|
|
#[test]
|
|
fn after_queue_relay_before_lmtp() {
|
|
let output = Command::new("faketime")
|
|
.arg("2020-12-31 23:59:59")
|
|
.arg(utils::log_tracker_path())
|
|
.arg("-vv")
|
|
.arg("-s")
|
|
.arg("2020-01-21 07:30:00")
|
|
.arg("-e")
|
|
.arg("2020-01-21 07:35:00")
|
|
.arg("-i")
|
|
.arg("tests/test_input_after_queue_relay_before_lmtp")
|
|
.output()
|
|
.expect("failed to execute pmg-log-tracker");
|
|
|
|
let expected_file = File::open("tests/test_output_after_queue_relay_before_lmtp")
|
|
.expect("failed to open test_output");
|
|
|
|
let expected_output = BufReader::new(&expected_file);
|
|
let output_reader = BufReader::new(&output.stdout[..]);
|
|
utils::compare_output(output_reader, expected_output);
|
|
}
|
|
|
|
#[test]
|
|
fn after_queue_duplicate_msgid() {
|
|
let output = Command::new("faketime")
|
|
.arg("2020-12-31 23:59:59")
|
|
.arg(utils::log_tracker_path())
|
|
.arg("-vv")
|
|
.arg("-s")
|
|
.arg("2020-01-21 07:30:00")
|
|
.arg("-e")
|
|
.arg("2020-01-21 07:35:00")
|
|
.arg("-i")
|
|
.arg("tests/test_input_after_queue_duplicate_msgid")
|
|
.output()
|
|
.expect("failed to execute pmg-log-tracker");
|
|
|
|
let expected_file = File::open("tests/test_output_after_queue_duplicate_msgid")
|
|
.expect("failed to open test_output");
|
|
|
|
let expected_output = BufReader::new(&expected_file);
|
|
let output_reader = BufReader::new(&output.stdout[..]);
|
|
utils::compare_output(output_reader, expected_output);
|
|
}
|