// Copyright (C) 2025-2026 Michael S. Klishin and Contributors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

mod test_helpers;

use rlqt_lib::Severity;
use rlqt_lib::entry_metadata::Annotator;
use rlqt_lib::entry_metadata::label_annotators::{LabelAnnotator, QueuesAnnotator};
use rlqt_lib::entry_metadata::labels::LogEntryLabels;
use test_helpers::create_test_entry;

#[test]
fn test_queues_annotator_matches_queue_pattern() {
    let entry = create_test_entry("Declaring queue 'my-queue' in vhost '/'", Severity::Info);
    let annotator = QueuesAnnotator;
    assert!(annotator.does_match(&entry));
}

#[test]
fn test_queues_annotator_matches_finished_rebuilding_index() {
    let entry = create_test_entry("Finished rebuilding index for queue", Severity::Info);
    let annotator = QueuesAnnotator;
    assert!(annotator.does_match(&entry));
}

#[test]
fn test_queues_annotator_matches_rebuilding_message_location_index() {
    let entry = create_test_entry(
        "Rebuilding message location index from disk",
        Severity::Info,
    );
    let annotator = QueuesAnnotator;
    assert!(annotator.does_match(&entry));
}

#[test]
fn test_queues_annotator_no_match() {
    let entry = create_test_entry("Unrelated message", Severity::Info);
    let annotator = QueuesAnnotator;
    assert!(!annotator.does_match(&entry));
}

#[test]
fn test_queues_annotator_annotates() {
    let annotator = QueuesAnnotator;
    let mut labels = LogEntryLabels::default();
    annotator.annotate(&mut labels);
    assert!(labels.contains(LogEntryLabels::QUEUES));
}
