src/sys/component_manager/tests/attributed_logging/components/archive-reader.rs

// Copyright 2020 the Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

use diagnostics_data::ExtendedMoniker;
use diagnostics_reader::ArchiveReader;
use futures::stream::StreamExt;
use log::info;
use std::collections::HashMap;

#[fuchsia::main(logging_tags = ["archive-reader"])]
async fn main() {
    let reader = ArchiveReader::logs();
    let mut non_matching_logs = vec![];

    type Fingerprint = Vec<&'static str>;
    let mut treasure = HashMap::<ExtendedMoniker, Vec<Fingerprint>>::new();
    treasure.insert(
        "routing-tests/offers-to-children-unavailable/child-for-offer-from-parent"
            .try_into()
            .unwrap(),
        vec![vec![
            "protocol `fidl.test.components.Trigger`",
            "not available for target `child-for-offer-from-parent`",
            "`fidl.test.components.Trigger` was not offered to",
            "`root/routing-tests/offers-to-children-unavailable` by parent",
        ]],
    );
    treasure.insert(
        "routing-tests/offers-to-children-unavailable-but-optional/child-for-offer-from-parent"
            .try_into()
            .unwrap(),
        vec![vec![
            "Optional",
            "protocol `fidl.test.components.Trigger`",
            "not available for target `child-for-offer-from-parent`",
            "`fidl.test.components.Trigger` was not offered to",
            "`root/routing-tests/offers-to-children-unavailable-but-optional` by parent",
        ]],
    );
    treasure.insert(
        "routing-tests/child".try_into().unwrap(),
        vec![vec![
            "protocol `fidl.test.components.Trigger`",
            "not available for target `child`",
            "`fidl.test.components.Trigger` was not offered to",
            "`root/routing-tests/child` by parent",
        ]],
    );
    treasure.insert(
        "routing-tests/child-with-optional-use".try_into().unwrap(),
        vec![vec![
            "Optional",
            "protocol `fidl.test.components.Trigger`",
            "not available for target `child-with-optional-use`",
            "`fidl.test.components.Trigger` was not offered to",
            "`root/routing-tests/child-with-optional-use` by parent",
        ]],
    );
    treasure.insert(
        "routing-tests/offers-to-children-unavailable/child-for-offer-from-sibling".try_into().unwrap(),
        vec![vec![
            "protocol `fidl.test.components.Trigger`",
            "not available for target `child-for-offer-from-sibling`",
            "`fidl.test.components.Trigger` was not exposed to `root/routing-tests/offers-to-children-unavailable`",
            "from `#child-that-doesnt-expose`"
        ]],
    );
    treasure.insert(
        "routing-tests/offers-to-children-unavailable-but-optional/child-for-offer-from-sibling".try_into().unwrap(),
        vec![vec![
            "Optional",
            "protocol `fidl.test.components.Trigger`",
            "not available for target `child-for-offer-from-sibling`",
            "`fidl.test.components.Trigger` was not exposed to `root/routing-tests/offers-to-children-unavailable-but-optional`",
            "from `#child-that-doesnt-expose`"
        ]],
    );
    treasure.insert(
        "routing-tests/offers-to-children-unavailable/child-open-unrequested".try_into().unwrap(),
        vec![vec![
            "No capability available",
            "fidl.test.components.Trigger",
            "root/routing-tests/offers-to-children-unavailable/child-open-unrequested",
            "`use` declaration",
        ]],
    );
    treasure.insert(
        "routing-tests/offers-to-children-unavailable-but-optional/child-open-unrequested"
            .try_into()
            .unwrap(),
        vec![vec![
            "No capability available",
            "fidl.test.components.Trigger",
            "root/routing-tests/offers-to-children-unavailable-but-optional/child-open-unrequested",
            "`use` declaration",
        ]],
    );

    if let Ok(mut results) = reader.snapshot_then_subscribe() {
        while let Some(Ok(log_record)) = results.next().await {
            if let Some(log_str) = log_record.msg() {
                info!("Log from {}: {}", log_record.moniker, log_str);
                match treasure.get_mut(&log_record.moniker) {
                    None => non_matching_logs.push(log_record),
                    Some(log_fingerprints) => {
                        let removed = {
                            let print_count = log_fingerprints.len();
                            log_fingerprints.retain(|fingerprint| {
                                // If all the part of the fingerprint match, remove
                                // the fingerprint, otherwise keep it.
                                let has_all_features =
                                    fingerprint.iter().all(|feature| log_str.contains(feature));
                                !has_all_features
                            });

                            print_count != log_fingerprints.len()
                        };

                        // If there are no more fingerprint sets for this
                        // component, remove it
                        if log_fingerprints.is_empty() {
                            treasure.remove(&log_record.moniker);
                        }
                        // If we didn't remove any fingerprints, this log didn't
                        // match anything, so push it into the non-matching logs.
                        if !removed {
                            non_matching_logs.push(log_record);
                        }
                        if treasure.is_empty() {
                            return;
                        }
                    }
                }
            }
        }
    }

    // Assert that no logs were received for optional-use-from-void.
    for log in &non_matching_logs {
        if log.moniker == "routing-tests/optional-use-from-void".try_into().unwrap() {
            panic!("Received unexpected log for optional-use-from-void: {:?}", log);
        }
    }

    panic!(
        "One or more logs were not found, remaining fingerprints: {:?}\n\n
        These log records were read, but did not match any fingerprints: {:?}",
        treasure, non_matching_logs
    );
}