use base64::display::Base64Display;
use fidl_fuchsia_diagnostics::{
PropertySelector, Selector, StringSelector, StringSelectorUnknown, SubtreeSelector,
TreeSelector,
};
use num_derive::{FromPrimitive, ToPrimitive};
use num_traits::bounds::Bounded;
use selectors::ValidateExt;
use serde::{Deserialize, Serialize};
use std::borrow::{Borrow, Cow};
use std::cmp::Ordering;
use std::collections::BTreeMap;
use std::fmt::{Display, Formatter, Result as FmtResult};
use std::ops::{Add, AddAssign, MulAssign};
use thiserror::Error;
pub mod macros;
pub mod serialization;
pub const LINEAR_HISTOGRAM_EXTRA_SLOTS: usize = 4;
pub const EXPONENTIAL_HISTOGRAM_EXTRA_SLOTS: usize = 5;
#[derive(Clone, Debug, PartialEq, Eq, FromPrimitive, ToPrimitive)]
#[repr(u8)]
pub enum ArrayFormat {
Default = 0,
LinearHistogram = 1,
ExponentialHistogram = 2,
}
#[derive(Clone, Debug, PartialEq)]
pub struct DiagnosticsHierarchy<Key = String> {
pub name: String,
pub properties: Vec<Property<Key>>,
pub children: Vec<DiagnosticsHierarchy<Key>>,
pub missing: Vec<MissingValue>,
}
#[derive(Clone, Debug, PartialEq)]
pub struct MissingValue {
pub reason: MissingValueReason,
pub name: String,
}
#[derive(Clone, Debug, PartialEq)]
pub enum MissingValueReason {
LinkNotFound,
LinkParseFailure,
LinkInvalid,
LinkNeverExpanded,
Timeout,
}
fn name_partial_cmp(a: &str, b: &str) -> Ordering {
match (a.parse::<u64>(), b.parse::<u64>()) {
(Ok(n), Ok(m)) => n.partial_cmp(&m).unwrap(),
_ => a.partial_cmp(b).unwrap(),
}
}
impl<Key> DiagnosticsHierarchy<Key>
where
Key: AsRef<str>,
{
pub fn sort(&mut self) {
self.properties.sort_by(|p1, p2| name_partial_cmp(p1.name(), p2.name()));
self.children.sort_by(|c1, c2| name_partial_cmp(&c1.name, &c2.name));
for child in self.children.iter_mut() {
child.sort();
}
}
pub fn new_root() -> Self {
DiagnosticsHierarchy::new("root", vec![], vec![])
}
pub fn new(
name: impl Into<String>,
properties: Vec<Property<Key>>,
children: Vec<DiagnosticsHierarchy<Key>>,
) -> Self {
Self { name: name.into(), properties, children, missing: vec![] }
}
pub fn get_or_add_child_mut<T>(&mut self, name: T) -> &mut DiagnosticsHierarchy<Key>
where
T: AsRef<str>,
{
match (0..self.children.len()).find(|&i| self.children[i].name == name.as_ref()) {
Some(matching_index) => &mut self.children[matching_index],
None => {
self.children.push(DiagnosticsHierarchy::new(name.as_ref(), vec![], vec![]));
self.children
.last_mut()
.expect("We just added an entry so we cannot get None here.")
}
}
}
pub fn add_child(&mut self, insert: DiagnosticsHierarchy<Key>) {
self.children.push(insert);
}
pub fn get_or_add_node<T>(&mut self, node_path: &[T]) -> &mut DiagnosticsHierarchy<Key>
where
T: AsRef<str>,
{
assert!(!node_path.is_empty());
let mut iter = node_path.iter();
let first_path_string = iter.next().unwrap().as_ref();
assert_eq!(first_path_string, &self.name);
let mut curr_node = self;
for node_path_entry in iter {
curr_node = curr_node.get_or_add_child_mut(node_path_entry);
}
curr_node
}
pub fn add_property(&mut self, property: Property<Key>) {
self.properties.push(property);
}
pub fn add_property_at_path<T>(&mut self, node_path: &[T], property: Property<Key>)
where
T: AsRef<str>,
{
self.get_or_add_node(node_path).properties.push(property);
}
pub fn property_iter(&self) -> DiagnosticsHierarchyIterator<'_, Key> {
DiagnosticsHierarchyIterator::new(self)
}
pub fn add_missing(&mut self, reason: MissingValueReason, name: String) {
self.missing.push(MissingValue { reason, name });
}
pub fn get_property(&self, name: &str) -> Option<&Property<Key>> {
self.properties.iter().find(|prop| prop.name() == name)
}
pub fn get_child(&self, name: &str) -> Option<&DiagnosticsHierarchy<Key>> {
self.children.iter().find(|node| node.name == name)
}
pub fn get_child_mut(&mut self, name: &str) -> Option<&mut DiagnosticsHierarchy<Key>> {
self.children.iter_mut().find(|node| node.name == name)
}
pub fn get_child_by_path(&self, path: &[&str]) -> Option<&DiagnosticsHierarchy<Key>> {
let mut result = Some(self);
for name in path {
result = result.and_then(|node| node.get_child(name));
}
result
}
pub fn get_child_by_path_mut(
&mut self,
path: &[&str],
) -> Option<&mut DiagnosticsHierarchy<Key>> {
let mut result = Some(self);
for name in path {
result = result.and_then(|node| node.get_child_mut(name));
}
result
}
pub fn get_property_by_path(&self, path: &[&str]) -> Option<&Property<Key>> {
let node = self.get_child_by_path(&path[..path.len() - 1]);
node.and_then(|node| node.get_property(path[path.len() - 1]))
}
}
macro_rules! property_type_getters_ref {
($([$variant:ident, $fn_name:ident, $type:ty]),*) => {
paste::item! {
impl<Key> Property<Key> {
$(
#[doc = "Returns the " $variant " value or `None` if the property isn't of that type"]
pub fn $fn_name(&self) -> Option<&$type> {
match self {
Property::$variant(_, value) => Some(value),
_ => None,
}
}
)*
}
}
}
}
macro_rules! property_type_getters_copy {
($([$variant:ident, $fn_name:ident, $type:ty]),*) => {
paste::item! {
impl<Key> Property<Key> {
$(
#[doc = "Returns the " $variant " value or `None` if the property isn't of that type"]
pub fn $fn_name(&self) -> Option<$type> {
match self {
Property::$variant(_, value) => Some(*value),
_ => None,
}
}
)*
}
}
}
}
property_type_getters_copy!(
[Int, int, i64],
[Uint, uint, u64],
[Double, double, f64],
[Bool, boolean, bool]
);
property_type_getters_ref!(
[String, string, str],
[Bytes, bytes, [u8]],
[DoubleArray, double_array, ArrayContent<f64>],
[IntArray, int_array, ArrayContent<i64>],
[UintArray, uint_array, ArrayContent<u64>],
[StringList, string_list, [String]]
);
struct WorkStackEntry<'a, Key> {
node: &'a DiagnosticsHierarchy<Key>,
key: Vec<&'a str>,
}
pub struct DiagnosticsHierarchyIterator<'a, Key> {
work_stack: Vec<WorkStackEntry<'a, Key>>,
current_key: Vec<&'a str>,
current_node: Option<&'a DiagnosticsHierarchy<Key>>,
current_property_index: usize,
}
impl<'a, Key> DiagnosticsHierarchyIterator<'a, Key> {
fn new(hierarchy: &'a DiagnosticsHierarchy<Key>) -> Self {
DiagnosticsHierarchyIterator {
work_stack: vec![WorkStackEntry { node: hierarchy, key: vec![&hierarchy.name] }],
current_key: vec![],
current_node: None,
current_property_index: 0,
}
}
}
impl<'a, Key> Iterator for DiagnosticsHierarchyIterator<'a, Key> {
type Item = (Vec<&'a str>, Option<&'a Property<Key>>);
fn next(&mut self) -> Option<Self::Item> {
loop {
let node = match self.current_node {
Some(node) => node,
None => {
let WorkStackEntry { node, key } = match self.work_stack.pop() {
None => return None,
Some(entry) => entry,
};
for child in node.children.iter() {
let mut child_key = key.clone();
child_key.push(&child.name);
self.work_stack.push(WorkStackEntry { node: child, key: child_key })
}
if node.properties.is_empty() {
return Some((key.clone(), None));
}
self.current_property_index = 0;
self.current_key = key;
node
}
};
if self.current_property_index == node.properties.len() {
self.current_node = None;
continue;
}
let property = &node.properties[self.current_property_index];
self.current_property_index += 1;
self.current_node = Some(node);
return Some((self.current_key.clone(), Some(property)));
}
}
}
#[derive(Debug, PartialEq, Clone)]
pub enum Property<Key = String> {
String(Key, String),
Bytes(Key, Vec<u8>),
Int(Key, i64),
Uint(Key, u64),
Double(Key, f64),
Bool(Key, bool),
DoubleArray(Key, ArrayContent<f64>),
IntArray(Key, ArrayContent<i64>),
UintArray(Key, ArrayContent<u64>),
StringList(Key, Vec<String>),
}
impl<K> Property<K> {
pub fn key(&self) -> &K {
match self {
Property::String(k, _) => k,
Property::Bytes(k, _) => k,
Property::Int(k, _) => k,
Property::Uint(k, _) => k,
Property::Double(k, _) => k,
Property::Bool(k, _) => k,
Property::DoubleArray(k, _) => k,
Property::IntArray(k, _) => k,
Property::UintArray(k, _) => k,
Property::StringList(k, _) => k,
}
}
pub fn discriminant_name(&self) -> &'static str {
match self {
Property::String(_, _) => "String",
Property::Bytes(_, _) => "Bytes",
Property::Int(_, _) => "Int",
Property::IntArray(_, _) => "IntArray",
Property::Uint(_, _) => "Uint",
Property::UintArray(_, _) => "UintArray",
Property::Double(_, _) => "Double",
Property::DoubleArray(_, _) => "DoubleArray",
Property::Bool(_, _) => "Bool",
Property::StringList(_, _) => "StringList",
}
}
pub fn number_as_int(&self) -> Option<i64> {
match self {
Property::Int(_, i) => Some(*i),
Property::Uint(_, u) => i64::try_from(*u).ok(),
Property::String(..)
| Property::Bytes(..)
| Property::Double(..)
| Property::Bool(..)
| Property::DoubleArray(..)
| Property::IntArray(..)
| Property::UintArray(..)
| Property::StringList(..) => None,
}
}
}
impl<K> Display for Property<K>
where
K: AsRef<str>,
{
fn fmt(&self, f: &mut Formatter<'_>) -> FmtResult {
macro_rules! pair {
($fmt:literal, $val:expr) => {
write!(f, "{}={}", self.key().as_ref(), format_args!($fmt, $val))
};
}
match self {
Property::String(_, v) => pair!("{}", v),
Property::Bytes(_, v) => {
pair!("b64:{}", Base64Display::new(v, &base64::engine::general_purpose::STANDARD))
}
Property::Int(_, v) => pair!("{}", v),
Property::Uint(_, v) => pair!("{}", v),
Property::Double(_, v) => pair!("{}", v),
Property::Bool(_, v) => pair!("{}", v),
Property::DoubleArray(_, v) => pair!("{:?}", v),
Property::IntArray(_, v) => pair!("{:?}", v),
Property::UintArray(_, v) => pair!("{:?}", v),
Property::StringList(_, v) => pair!("{:?}", v),
}
}
}
#[derive(Debug, Error)]
pub enum Error {
#[error(
"Missing elements for {histogram_type:?} histogram. Expected {expected}, got {actual}"
)]
MissingHistogramElements { histogram_type: ArrayFormat, expected: usize, actual: usize },
#[error("TreeSelector only supports property and subtree selection.")]
InvalidTreeSelector,
#[error(transparent)]
Selectors(#[from] selectors::Error),
#[error(transparent)]
InvalidSelector(#[from] selectors::ValidationError),
}
impl Error {
fn missing_histogram_elements(
histogram_type: ArrayFormat,
actual: usize,
expected: usize,
) -> Self {
Self::MissingHistogramElements { histogram_type, actual, expected }
}
}
#[derive(Debug, Serialize, Deserialize, PartialEq, Clone)]
pub struct LinearHistogram<T> {
pub size: usize,
pub floor: T,
pub step: T,
pub counts: Vec<T>,
#[serde(skip_serializing_if = "Option::is_none")]
pub indexes: Option<Vec<usize>>,
}
#[derive(Debug, Serialize, Deserialize, PartialEq, Clone)]
pub struct ExponentialHistogram<T> {
pub size: usize,
pub floor: T,
pub initial_step: T,
pub step_multiplier: T,
pub counts: Vec<T>,
#[serde(skip_serializing_if = "Option::is_none")]
pub indexes: Option<Vec<usize>>,
}
#[derive(Debug, PartialEq, Clone)]
pub enum ArrayContent<T> {
Values(Vec<T>),
LinearHistogram(LinearHistogram<T>),
ExponentialHistogram(ExponentialHistogram<T>),
}
impl<T> ArrayContent<T>
where
T: Add<Output = T> + num_traits::Zero + AddAssign + Copy + MulAssign + PartialEq + Bounded,
{
pub fn new(values: Vec<T>, format: ArrayFormat) -> Result<Self, Error> {
match format {
ArrayFormat::Default => Ok(Self::Values(values)),
ArrayFormat::LinearHistogram => {
if values.len() < 5 {
return Err(Error::missing_histogram_elements(
ArrayFormat::LinearHistogram,
values.len(),
5,
));
}
let original_counts = &values[2..];
let (counts, indexes) =
match serialization::maybe_condense_histogram(original_counts, &None) {
None => (original_counts.to_vec(), None),
Some((counts, indexes)) => (counts, Some(indexes)),
};
Ok(Self::LinearHistogram(LinearHistogram {
floor: values[0],
step: values[1],
counts,
indexes,
size: values.len() - 2,
}))
}
ArrayFormat::ExponentialHistogram => {
if values.len() < 6 {
return Err(Error::missing_histogram_elements(
ArrayFormat::LinearHistogram,
values.len(),
5,
));
}
let original_counts = &values[3..];
let (counts, indexes) =
match serialization::maybe_condense_histogram(original_counts, &None) {
None => (original_counts.to_vec(), None),
Some((counts, indexes)) => (counts, Some(indexes)),
};
Ok(Self::ExponentialHistogram(ExponentialHistogram {
floor: values[0],
initial_step: values[1],
step_multiplier: values[2],
counts,
indexes,
size: values.len() - 3,
}))
}
}
}
pub fn len(&self) -> usize {
match self {
Self::Values(vals) => vals.len(),
Self::LinearHistogram(LinearHistogram { size, .. })
| Self::ExponentialHistogram(ExponentialHistogram { size, .. }) => *size,
}
}
pub fn is_empty(&self) -> bool {
self.len() == 0
}
pub fn raw_values(&self) -> Cow<'_, Vec<T>> {
match self {
Self::Values(values) => Cow::Borrowed(values),
Self::LinearHistogram(LinearHistogram { size, counts, indexes, .. })
| Self::ExponentialHistogram(ExponentialHistogram { size, counts, indexes, .. }) => {
if let Some(indexes) = indexes {
let mut values = vec![T::zero(); *size];
for (count, index) in counts.iter().zip(indexes.iter()) {
if index <= size {
values[*index] = *count;
}
}
Cow::Owned(values)
} else {
Cow::Borrowed(counts)
}
}
}
}
}
pub mod testing {
use crate::ArrayContent;
use num_traits::bounds::Bounded;
use std::ops::{Add, AddAssign, MulAssign};
pub trait CondensableOnDemand {
fn condense_histogram(&mut self);
}
fn condense_counts<T: num_traits::Zero + Copy + PartialEq>(
counts: &[T],
) -> (Vec<T>, Vec<usize>) {
let mut condensed_counts = vec![];
let mut indexes = vec![];
for (index, count) in counts.iter().enumerate() {
if *count != T::zero() {
condensed_counts.push(*count);
indexes.push(index);
}
}
(condensed_counts, indexes)
}
impl<T> CondensableOnDemand for ArrayContent<T>
where
T: Add<Output = T> + num_traits::Zero + AddAssign + Copy + MulAssign + PartialEq + Bounded,
{
fn condense_histogram(&mut self) {
match self {
Self::Values(_) => (),
Self::LinearHistogram(histogram) => {
if histogram.indexes.is_some() {
return;
}
let (counts, indexes) = condense_counts(&histogram.counts);
histogram.counts = counts;
histogram.indexes = Some(indexes);
}
Self::ExponentialHistogram(histogram) => {
if histogram.indexes.is_some() {
return;
}
let (counts, indexes) = condense_counts(&histogram.counts);
histogram.counts = counts;
histogram.indexes = Some(indexes);
}
}
}
}
}
impl<Key> Property<Key>
where
Key: AsRef<str>,
{
pub fn name(&self) -> &str {
match self {
Property::String(name, _)
| Property::Bytes(name, _)
| Property::Int(name, _)
| Property::IntArray(name, _)
| Property::Uint(name, _)
| Property::UintArray(name, _)
| Property::Double(name, _)
| Property::Bool(name, _)
| Property::DoubleArray(name, _)
| Property::StringList(name, _) => name.as_ref(),
}
}
}
impl<T: Borrow<Selector>> TryFrom<&[T]> for HierarchyMatcher {
type Error = Error;
fn try_from(selectors: &[T]) -> Result<Self, Self::Error> {
let mut matcher = HierarchyMatcher::default();
for selector in selectors {
let selector = selector.borrow();
selector.validate().map_err(|e| Error::Selectors(e.into()))?;
match selector.tree_selector.clone().unwrap() {
TreeSelector::SubtreeSelector(subtree_selector) => {
matcher.insert_subtree(subtree_selector.clone());
}
TreeSelector::PropertySelector(property_selector) => {
matcher.insert_property(property_selector.clone());
}
_ => return Err(Error::Selectors(selectors::Error::InvalidTreeSelector)),
}
}
Ok(matcher)
}
}
impl<T: Borrow<Selector>> TryFrom<Vec<T>> for HierarchyMatcher {
type Error = Error;
fn try_from(selectors: Vec<T>) -> Result<Self, Self::Error> {
selectors[..].try_into()
}
}
#[derive(Debug)]
struct OrdStringSelector(StringSelector);
impl From<StringSelector> for OrdStringSelector {
fn from(selector: StringSelector) -> Self {
Self(selector)
}
}
impl Ord for OrdStringSelector {
fn cmp(&self, other: &OrdStringSelector) -> Ordering {
match (&self.0, &other.0) {
(StringSelector::ExactMatch(s), StringSelector::ExactMatch(o)) => s.cmp(o),
(StringSelector::StringPattern(s), StringSelector::StringPattern(o)) => s.cmp(o),
(StringSelector::ExactMatch(_), StringSelector::StringPattern(_)) => Ordering::Less,
(StringSelector::StringPattern(_), StringSelector::ExactMatch(_)) => Ordering::Greater,
(StringSelectorUnknown!(), StringSelector::ExactMatch(_)) => Ordering::Less,
(StringSelectorUnknown!(), StringSelector::StringPattern(_)) => Ordering::Less,
(StringSelectorUnknown!(), StringSelectorUnknown!()) => Ordering::Equal,
}
}
}
impl PartialOrd for OrdStringSelector {
fn partial_cmp(&self, other: &OrdStringSelector) -> Option<Ordering> {
Some(self.cmp(other))
}
}
impl PartialEq for OrdStringSelector {
fn eq(&self, other: &OrdStringSelector) -> bool {
match (&self.0, &other.0) {
(StringSelector::ExactMatch(s), StringSelector::ExactMatch(o)) => s.eq(o),
(StringSelector::StringPattern(s), StringSelector::StringPattern(o)) => s.eq(o),
(StringSelector::ExactMatch(_), StringSelector::StringPattern(_)) => false,
(StringSelector::StringPattern(_), StringSelector::ExactMatch(_)) => false,
(StringSelectorUnknown!(), StringSelectorUnknown!()) => true,
}
}
}
impl Eq for OrdStringSelector {}
#[derive(Default, Debug)]
pub struct HierarchyMatcher {
nodes: BTreeMap<OrdStringSelector, HierarchyMatcher>,
properties: Vec<OrdStringSelector>,
subtree: bool,
}
impl HierarchyMatcher {
pub fn new<I>(selectors: I) -> Result<Self, Error>
where
I: Iterator<Item = Selector>,
{
let mut matcher = HierarchyMatcher::default();
for selector in selectors {
selector.validate().map_err(|e| Error::Selectors(e.into()))?;
match selector.tree_selector.unwrap() {
TreeSelector::SubtreeSelector(subtree_selector) => {
matcher.insert_subtree(subtree_selector);
}
TreeSelector::PropertySelector(property_selector) => {
matcher.insert_property(property_selector);
}
_ => return Err(Error::Selectors(selectors::Error::InvalidTreeSelector)),
}
}
Ok(matcher)
}
fn insert_subtree(&mut self, selector: SubtreeSelector) {
self.insert(selector.node_path, None);
}
fn insert_property(&mut self, selector: PropertySelector) {
self.insert(selector.node_path, Some(selector.target_properties));
}
fn insert(&mut self, node_path: Vec<StringSelector>, property: Option<StringSelector>) {
let mut matcher = self;
for node in node_path {
matcher = matcher.nodes.entry(node.into()).or_default();
}
match property {
Some(property) => {
matcher.properties.push(property.into());
}
None => matcher.subtree = true,
}
}
}
pub fn select_from_hierarchy<'a, 'b, Key>(
root_node: &'a DiagnosticsHierarchy<Key>,
selector: &'b Selector,
) -> Result<Vec<&'a Property<Key>>, Error>
where
Key: AsRef<str>,
'a: 'b,
{
selector.validate()?;
struct StackEntry<'a, Key> {
node: &'a DiagnosticsHierarchy<Key>,
node_path_index: usize,
explored_path: Vec<&'a str>,
}
let (node_path, property_selector, stack_entry) = match selector.tree_selector.as_ref().unwrap()
{
TreeSelector::SubtreeSelector(ref subtree_selector) => (
&subtree_selector.node_path,
None,
StackEntry { node: root_node, node_path_index: 0, explored_path: vec![] },
),
TreeSelector::PropertySelector(ref property_selector) => (
&property_selector.node_path,
Some(&property_selector.target_properties),
StackEntry { node: root_node, node_path_index: 0, explored_path: vec![] },
),
_ => return Err(Error::InvalidTreeSelector),
};
let mut stack = vec![stack_entry];
let mut result = vec![];
while let Some(StackEntry { node, node_path_index, mut explored_path }) = stack.pop() {
if !selectors::match_string(&node_path[node_path_index], &node.name) {
continue;
}
explored_path.push(&node.name);
if node_path_index != node_path.len() - 1 {
for child in node.children.iter() {
stack.push(StackEntry {
node: child,
node_path_index: node_path_index + 1,
explored_path: explored_path.clone(),
});
}
} else if let Some(s) = property_selector {
for property in &node.properties {
if selectors::match_string(s, property.key()) {
result.push(property);
}
}
} else {
for (_path, property) in node.property_iter() {
if let Some(property) = property {
result.push(property);
}
}
}
}
Ok(result)
}
pub fn filter_tree<Key>(
root_node: DiagnosticsHierarchy<Key>,
selectors: &[TreeSelector],
) -> Option<DiagnosticsHierarchy<Key>>
where
Key: AsRef<str>,
{
let mut matcher = HierarchyMatcher::default();
for selector in selectors {
match selector {
TreeSelector::SubtreeSelector(subtree_selector) => {
matcher.insert_subtree(subtree_selector.clone());
}
TreeSelector::PropertySelector(property_selector) => {
matcher.insert_property(property_selector.clone());
}
_ => {}
}
}
filter_hierarchy(root_node, &matcher)
}
pub fn filter_hierarchy<Key>(
mut root_node: DiagnosticsHierarchy<Key>,
hierarchy_matcher: &HierarchyMatcher,
) -> Option<DiagnosticsHierarchy<Key>>
where
Key: AsRef<str>,
{
let starts_empty = root_node.children.is_empty() && root_node.properties.is_empty();
if filter_hierarchy_helper(&mut root_node, &[hierarchy_matcher]) {
if !starts_empty && root_node.children.is_empty() && root_node.properties.is_empty() {
return None;
}
return Some(root_node);
}
None
}
fn filter_hierarchy_helper<Key>(
node: &mut DiagnosticsHierarchy<Key>,
hierarchy_matchers: &[&HierarchyMatcher],
) -> bool
where
Key: AsRef<str>,
{
let child_matchers = eval_matchers_on_node_name(&node.name, hierarchy_matchers);
if child_matchers.is_empty() {
node.children.clear();
node.properties.clear();
return false;
}
if child_matchers.iter().any(|m| m.subtree) {
return true;
}
node.children.retain_mut(|child| filter_hierarchy_helper(child, &child_matchers));
node.properties.retain_mut(|prop| eval_matchers_on_property(prop.name(), &child_matchers));
!(node.children.is_empty() && node.properties.is_empty())
}
fn eval_matchers_on_node_name<'a>(
node_name: &'a str,
matchers: &'a [&'a HierarchyMatcher],
) -> Vec<&'a HierarchyMatcher> {
let mut result = vec![];
for matcher in matchers {
for (node_pattern, tree_matcher) in matcher.nodes.iter() {
if selectors::match_string(&node_pattern.0, node_name) {
result.push(tree_matcher);
}
}
}
result
}
fn eval_matchers_on_property(property_name: &str, matchers: &[&HierarchyMatcher]) -> bool {
matchers.iter().any(|matcher| {
matcher
.properties
.iter()
.any(|property_pattern| selectors::match_string(&property_pattern.0, property_name))
})
}
#[derive(Clone)]
pub struct ExponentialHistogramParams<T: Clone> {
pub floor: T,
pub initial_step: T,
pub step_multiplier: T,
pub buckets: usize,
}
#[derive(Clone)]
pub struct LinearHistogramParams<T: Clone> {
pub floor: T,
pub step_size: T,
pub buckets: usize,
}
pub trait DiagnosticsHierarchyGetter<K: Clone> {
fn get_diagnostics_hierarchy(&self) -> Cow<'_, DiagnosticsHierarchy<K>>;
}
impl<K: Clone> DiagnosticsHierarchyGetter<K> for DiagnosticsHierarchy<K> {
fn get_diagnostics_hierarchy(&self) -> Cow<'_, DiagnosticsHierarchy<K>> {
Cow::Borrowed(self)
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::testing::CondensableOnDemand;
use assert_matches::assert_matches;
use selectors::VerboseError;
use std::sync::Arc;
fn validate_hierarchy_iteration(
mut results_vec: Vec<(Vec<String>, Option<Property>)>,
test_hierarchy: DiagnosticsHierarchy,
) {
let expected_num_entries = results_vec.len();
let mut num_entries = 0;
for (key, val) in test_hierarchy.property_iter() {
num_entries += 1;
let (expected_key, expected_property) = results_vec.pop().unwrap();
assert_eq!(key.to_vec().join("/"), expected_key.to_vec().join("/"));
assert_eq!(val, expected_property.as_ref());
}
assert_eq!(num_entries, expected_num_entries);
}
#[fuchsia::test]
fn test_diagnostics_hierarchy_iteration() {
let double_array_data = vec![-1.2, 2.3, 3.4, 4.5, -5.6];
let chars = ['a', 'b', 'c', 'd', 'e', 'f', 'g'];
let string_data = chars.iter().cycle().take(6000).collect::<String>();
let bytes_data = (0u8..=9u8).cycle().take(5000).collect::<Vec<u8>>();
let test_hierarchy = DiagnosticsHierarchy::new(
"root".to_string(),
vec![
Property::Int("int-root".to_string(), 3),
Property::DoubleArray(
"property-double-array".to_string(),
ArrayContent::Values(double_array_data.clone()),
),
],
vec![DiagnosticsHierarchy::new(
"child-1".to_string(),
vec![
Property::Uint("property-uint".to_string(), 10),
Property::Double("property-double".to_string(), -3.4),
Property::String("property-string".to_string(), string_data.clone()),
Property::IntArray(
"property-int-array".to_string(),
ArrayContent::new(vec![1, 2, 1, 1, 1, 1, 1], ArrayFormat::LinearHistogram)
.unwrap(),
),
],
vec![DiagnosticsHierarchy::new(
"child-1-1".to_string(),
vec![
Property::Int("property-int".to_string(), -9),
Property::Bytes("property-bytes".to_string(), bytes_data.clone()),
Property::UintArray(
"property-uint-array".to_string(),
ArrayContent::new(
vec![1, 1, 2, 0, 1, 1, 2, 0, 0],
ArrayFormat::ExponentialHistogram,
)
.unwrap(),
),
],
vec![],
)],
)],
);
let results_vec = vec![
(
vec!["root".to_string(), "child-1".to_string(), "child-1-1".to_string()],
Some(Property::UintArray(
"property-uint-array".to_string(),
ArrayContent::new(
vec![1, 1, 2, 0, 1, 1, 2, 0, 0],
ArrayFormat::ExponentialHistogram,
)
.unwrap(),
)),
),
(
vec!["root".to_string(), "child-1".to_string(), "child-1-1".to_string()],
Some(Property::Bytes("property-bytes".to_string(), bytes_data)),
),
(
vec!["root".to_string(), "child-1".to_string(), "child-1-1".to_string()],
Some(Property::Int("property-int".to_string(), -9)),
),
(
vec!["root".to_string(), "child-1".to_string()],
Some(Property::IntArray(
"property-int-array".to_string(),
ArrayContent::new(vec![1, 2, 1, 1, 1, 1, 1], ArrayFormat::LinearHistogram)
.unwrap(),
)),
),
(
vec!["root".to_string(), "child-1".to_string()],
Some(Property::String("property-string".to_string(), string_data)),
),
(
vec!["root".to_string(), "child-1".to_string()],
Some(Property::Double("property-double".to_string(), -3.4)),
),
(
vec!["root".to_string(), "child-1".to_string()],
Some(Property::Uint("property-uint".to_string(), 10)),
),
(
vec!["root".to_string()],
Some(Property::DoubleArray(
"property-double-array".to_string(),
ArrayContent::Values(double_array_data),
)),
),
(vec!["root".to_string()], Some(Property::Int("int-root".to_string(), 3))),
];
validate_hierarchy_iteration(results_vec, test_hierarchy);
}
#[fuchsia::test]
fn test_getters() {
let a_prop = Property::Int("a".to_string(), 1);
let b_prop = Property::Uint("b".to_string(), 2);
let child2 = DiagnosticsHierarchy::new("child2".to_string(), vec![], vec![]);
let child = DiagnosticsHierarchy::new(
"child".to_string(),
vec![b_prop.clone()],
vec![child2.clone()],
);
let mut hierarchy = DiagnosticsHierarchy::new(
"root".to_string(),
vec![a_prop.clone()],
vec![child.clone()],
);
assert_matches!(hierarchy.get_child("child"), Some(node) if *node == child);
assert_matches!(hierarchy.get_child_mut("child"), Some(node) if *node == child);
assert_matches!(hierarchy.get_child_by_path(&["child", "child2"]),
Some(node) if *node == child2);
assert_matches!(hierarchy.get_child_by_path_mut(&["child", "child2"]),
Some(node) if *node == child2);
assert_matches!(hierarchy.get_property("a"), Some(prop) if *prop == a_prop);
assert_matches!(hierarchy.get_property_by_path(&["child", "b"]),
Some(prop) if *prop == b_prop);
}
#[fuchsia::test]
fn test_edge_case_hierarchy_iteration() {
let root_only_with_one_property_hierarchy = DiagnosticsHierarchy::new(
"root".to_string(),
vec![Property::Int("property-int".to_string(), -9)],
vec![],
);
let results_vec =
vec![(vec!["root".to_string()], Some(Property::Int("property-int".to_string(), -9)))];
validate_hierarchy_iteration(results_vec, root_only_with_one_property_hierarchy);
let empty_hierarchy = DiagnosticsHierarchy::new("root".to_string(), vec![], vec![]);
let results_vec = vec![(vec!["root".to_string()], None)];
validate_hierarchy_iteration(results_vec, empty_hierarchy);
let empty_root_populated_child = DiagnosticsHierarchy::new(
"root",
vec![],
vec![DiagnosticsHierarchy::new(
"foo",
vec![Property::Int("11".to_string(), -4)],
vec![],
)],
);
let results_vec = vec![
(
vec!["root".to_string(), "foo".to_string()],
Some(Property::Int("11".to_string(), -4)),
),
(vec!["root".to_string()], None),
];
validate_hierarchy_iteration(results_vec, empty_root_populated_child);
let empty_root_empty_child = DiagnosticsHierarchy::new(
"root",
vec![],
vec![DiagnosticsHierarchy::new("foo", vec![], vec![])],
);
let results_vec = vec![
(vec!["root".to_string(), "foo".to_string()], None),
(vec!["root".to_string()], None),
];
validate_hierarchy_iteration(results_vec, empty_root_empty_child);
}
#[fuchsia::test]
fn array_value() {
let values = vec![1, 2, 5, 7, 9, 11, 13];
let array = ArrayContent::<u64>::new(values.clone(), ArrayFormat::Default);
assert_matches!(array, Ok(ArrayContent::Values(vals)) if vals == values);
}
#[fuchsia::test]
fn linear_histogram_array_value() {
let values = vec![1, 2, 5, 7, 9, 11, 13];
let array = ArrayContent::<i64>::new(values, ArrayFormat::LinearHistogram);
assert_matches!(array, Ok(ArrayContent::LinearHistogram(hist))
if hist == LinearHistogram {
floor: 1,
step: 2,
counts: vec![5, 7, 9, 11, 13],
indexes: None,
size: 5,
}
);
}
#[fuchsia::test]
fn exponential_histogram_array_value() {
let values = vec![1.0, 2.0, 5.0, 7.0, 9.0, 11.0, 15.0];
let array = ArrayContent::<f64>::new(values, ArrayFormat::ExponentialHistogram);
assert_matches!(array, Ok(ArrayContent::ExponentialHistogram(hist))
if hist == ExponentialHistogram {
floor: 1.0,
initial_step: 2.0,
step_multiplier: 5.0,
counts: vec![7.0, 9.0, 11.0, 15.0],
indexes: None,
size: 4,
}
);
}
#[fuchsia::test]
fn deserialize_linear_int_histogram() -> Result<(), serde_json::Error> {
let json = r#"{
"root": {
"histogram": {
"floor": -2,
"step": 3,
"counts": [4, 5, 6],
"size": 3
}
}
}"#;
let parsed: DiagnosticsHierarchy = serde_json::from_str(json)?;
let expected = DiagnosticsHierarchy::new(
"root".to_string(),
vec![Property::IntArray(
"histogram".to_string(),
ArrayContent::new(vec![-2, 3, 4, 5, 6], ArrayFormat::LinearHistogram).unwrap(),
)],
vec![],
);
assert_eq!(parsed, expected);
Ok(())
}
#[fuchsia::test]
fn deserialize_exponential_int_histogram() -> Result<(), serde_json::Error> {
let json = r#"{
"root": {
"histogram": {
"floor": 1,
"initial_step": 3,
"step_multiplier": 2,
"counts": [4, 5, 6],
"size": 3
}
}
}"#;
let parsed: DiagnosticsHierarchy = serde_json::from_str(json)?;
let expected = DiagnosticsHierarchy::new(
"root".to_string(),
vec![Property::IntArray(
"histogram".to_string(),
ArrayContent::new(vec![1, 3, 2, 4, 5, 6], ArrayFormat::ExponentialHistogram)
.unwrap(),
)],
vec![],
);
assert_eq!(parsed, expected);
Ok(())
}
#[fuchsia::test]
fn deserialize_linear_uint_histogram() -> Result<(), serde_json::Error> {
let json = r#"{
"root": {
"histogram": {
"floor": 2,
"step": 3,
"counts": [4, 9223372036854775808, 6],
"size": 3
}
}
}"#;
let parsed: DiagnosticsHierarchy = serde_json::from_str(json)?;
let expected = DiagnosticsHierarchy::new(
"root".to_string(),
vec![Property::UintArray(
"histogram".to_string(),
ArrayContent::new(
vec![2, 3, 4, 9_223_372_036_854_775_808, 6],
ArrayFormat::LinearHistogram,
)
.unwrap(),
)],
vec![],
);
assert_eq!(parsed, expected);
Ok(())
}
#[fuchsia::test]
fn deserialize_linear_double_histogram() -> Result<(), serde_json::Error> {
let json = r#"{
"root": {
"histogram": {
"floor": 2.0,
"step": 3.0,
"counts": [4.0, 5.0, 6.0],
"size": 3
}
}
}"#;
let parsed: DiagnosticsHierarchy = serde_json::from_str(json)?;
let expected = DiagnosticsHierarchy::new(
"root".to_string(),
vec![Property::DoubleArray(
"histogram".to_string(),
ArrayContent::new(vec![2.0, 3.0, 4.0, 5.0, 6.0], ArrayFormat::LinearHistogram)
.unwrap(),
)],
vec![],
);
assert_eq!(parsed, expected);
Ok(())
}
#[fuchsia::test]
fn deserialize_sparse_histogram() -> Result<(), serde_json::Error> {
let json = r#"{
"root": {
"histogram": {
"floor": 2,
"step": 3,
"counts": [4, 5, 6],
"indexes": [1, 2, 4],
"size": 8
}
}
}"#;
let parsed: DiagnosticsHierarchy = serde_json::from_str(json)?;
let mut histogram =
ArrayContent::new(vec![2, 3, 0, 4, 5, 0, 6, 0, 0, 0], ArrayFormat::LinearHistogram)
.unwrap();
histogram.condense_histogram();
let expected = DiagnosticsHierarchy::new(
"root".to_string(),
vec![Property::IntArray("histogram".to_string(), histogram)],
vec![],
);
assert_eq!(parsed, expected);
Ok(())
}
#[fuchsia::test]
fn reject_histogram_incompatible_values() -> Result<(), serde_json::Error> {
let json = r#"{
"root": {
"histogram": {
"floor": -2,
"step": 3,
"counts": [4, 9223372036854775808, 6],
"size": 3
}
}
}"#;
let parsed: DiagnosticsHierarchy = serde_json::from_str(json)?;
assert_eq!(parsed.children.len(), 1);
assert_eq!(&parsed.children[0].name, "histogram");
Ok(())
}
#[fuchsia::test]
fn reject_histogram_bad_sparse_list() -> Result<(), serde_json::Error> {
let json = r#"{
"root": {
"histogram": {
"floor": -2,
"step": 3,
"counts": [4, 5, 6],
"indexes": [0, 1, 2, 3],
"size": 8
}
}
}"#;
let parsed: DiagnosticsHierarchy = serde_json::from_str(json)?;
assert_eq!(parsed.children.len(), 1);
assert_eq!(&parsed.children[0].name, "histogram");
Ok(())
}
#[fuchsia::test]
fn reject_histogram_bad_index() -> Result<(), serde_json::Error> {
let json = r#"{
"root": {
"histogram": {
"floor": -2,
"step": 3,
"counts": [4, 5, 6],
"indexes": [0, 1, 4],
"size": 4
}
}
}"#;
let parsed: DiagnosticsHierarchy = serde_json::from_str(json)?;
assert_eq!(parsed.children.len(), 1);
assert_eq!(&parsed.children[0].name, "histogram");
Ok(())
}
#[fuchsia::test]
fn reject_histogram_wrong_field() -> Result<(), serde_json::Error> {
let json = r#"{
"root": {
"histogram": {
"floor": 2,
"step": 3,
"counts": [4, 5, 6],
"incorrect": [0, 1, 3],
"size": 4
}
}
}"#;
let parsed: DiagnosticsHierarchy = serde_json::from_str(json)?;
assert_eq!(parsed.children.len(), 1);
assert_eq!(&parsed.children[0].name, "histogram");
Ok(())
}
#[fuchsia::test]
fn exponential_histogram() {
let values = vec![0, 2, 4, 0, 1, 2, 3, 4, 5];
let array = ArrayContent::new(values, ArrayFormat::ExponentialHistogram);
assert_matches!(array, Ok(ArrayContent::ExponentialHistogram(hist))
if hist == ExponentialHistogram {
floor: 0,
initial_step: 2,
step_multiplier: 4,
counts: vec![0, 1, 2, 3, 4, 5],
indexes: None,
size: 6,
}
);
}
#[fuchsia::test]
fn add_to_hierarchy() {
let mut hierarchy = DiagnosticsHierarchy::new_root();
let prop_1 = Property::String("x".to_string(), "foo".to_string());
let path_1 = vec!["root", "one"];
let prop_2 = Property::Uint("c".to_string(), 3);
let path_2 = vec!["root", "two"];
let prop_2_prime = Property::Int("z".to_string(), -4);
hierarchy.add_property_at_path(&path_1, prop_1.clone());
hierarchy.add_property_at_path(&path_2.clone(), prop_2.clone());
hierarchy.add_property_at_path(&path_2, prop_2_prime.clone());
assert_eq!(
hierarchy,
DiagnosticsHierarchy {
name: "root".to_string(),
children: vec![
DiagnosticsHierarchy {
name: "one".to_string(),
properties: vec![prop_1],
children: vec![],
missing: vec![],
},
DiagnosticsHierarchy {
name: "two".to_string(),
properties: vec![prop_2, prop_2_prime],
children: vec![],
missing: vec![],
}
],
properties: vec![],
missing: vec![],
}
);
}
#[fuchsia::test]
fn string_lists() {
let mut hierarchy = DiagnosticsHierarchy::new_root();
let prop_1 =
Property::StringList("x".to_string(), vec!["foo".to_string(), "bar".to_string()]);
let path_1 = vec!["root", "one"];
hierarchy.add_property_at_path(&path_1, prop_1.clone());
assert_eq!(
hierarchy,
DiagnosticsHierarchy {
name: "root".to_string(),
children: vec![DiagnosticsHierarchy {
name: "one".to_string(),
properties: vec![prop_1],
children: vec![],
missing: vec![],
},],
properties: vec![],
missing: vec![],
}
);
}
#[fuchsia::test]
#[cfg_attr(feature = "variant_asan", ignore)]
#[cfg_attr(feature = "variant_hwasan", ignore)]
#[should_panic]
fn no_empty_paths_allowed() {
let mut hierarchy = DiagnosticsHierarchy::<String>::new_root();
let path_1: Vec<&String> = vec![];
hierarchy.get_or_add_node(&path_1);
}
#[fuchsia::test]
#[should_panic]
fn path_must_start_at_self() {
let mut hierarchy = DiagnosticsHierarchy::<String>::new_root();
let path_1 = vec!["not_root", "a"];
hierarchy.get_or_add_node(&path_1);
}
#[fuchsia::test]
fn sort_hierarchy() {
let mut hierarchy = DiagnosticsHierarchy::new(
"root",
vec![
Property::String("x".to_string(), "foo".to_string()),
Property::Uint("c".to_string(), 3),
Property::Int("z".to_string(), -4),
],
vec![
DiagnosticsHierarchy::new(
"foo",
vec![
Property::Int("11".to_string(), -4),
Property::Bytes("123".to_string(), "foo".bytes().collect()),
Property::Double("0".to_string(), 8.1),
],
vec![],
),
DiagnosticsHierarchy::new("bar", vec![], vec![]),
],
);
hierarchy.sort();
let sorted_hierarchy = DiagnosticsHierarchy::new(
"root",
vec![
Property::Uint("c".to_string(), 3),
Property::String("x".to_string(), "foo".to_string()),
Property::Int("z".to_string(), -4),
],
vec![
DiagnosticsHierarchy::new("bar", vec![], vec![]),
DiagnosticsHierarchy::new(
"foo",
vec![
Property::Double("0".to_string(), 8.1),
Property::Int("11".to_string(), -4),
Property::Bytes("123".to_string(), "foo".bytes().collect()),
],
vec![],
),
],
);
assert_eq!(sorted_hierarchy, hierarchy);
}
fn parse_selectors_and_filter_hierarchy(
hierarchy: DiagnosticsHierarchy,
test_selectors: Vec<&str>,
) -> Option<DiagnosticsHierarchy> {
let parsed_test_selectors = test_selectors
.into_iter()
.map(|selector_string| {
Arc::new(
selectors::parse_selector::<VerboseError>(selector_string)
.expect("All test selectors are valid and parsable."),
)
})
.collect::<Vec<Arc<Selector>>>();
let hierarchy_matcher: HierarchyMatcher = parsed_test_selectors.try_into().unwrap();
filter_hierarchy(hierarchy, &hierarchy_matcher).map(|mut hierarchy| {
hierarchy.sort();
hierarchy
})
}
fn get_test_hierarchy() -> DiagnosticsHierarchy {
DiagnosticsHierarchy::new(
"root",
vec![
Property::String("x".to_string(), "foo".to_string()),
Property::Uint("c".to_string(), 3),
Property::Int("z".to_string(), -4),
],
vec![
DiagnosticsHierarchy::new(
"foo",
vec![
Property::Int("11".to_string(), -4),
Property::Bytes("123".to_string(), "foo".bytes().collect()),
Property::Double("0".to_string(), 8.1),
],
vec![DiagnosticsHierarchy::new(
"zed",
vec![Property::Int("13".to_string(), -4)],
vec![],
)],
),
DiagnosticsHierarchy::new(
"bar",
vec![Property::Int("12".to_string(), -4)],
vec![DiagnosticsHierarchy::new(
"zed",
vec![Property::Int("13/:".to_string(), -4)],
vec![],
)],
),
],
)
}
#[fuchsia::test]
fn test_filter_hierarchy() {
let test_selectors = vec!["*:root/foo:11", "*:root:z", r#"*:root/bar/zed:13\/\:"#];
assert_eq!(
parse_selectors_and_filter_hierarchy(get_test_hierarchy(), test_selectors),
Some(DiagnosticsHierarchy::new(
"root",
vec![Property::Int("z".to_string(), -4),],
vec![
DiagnosticsHierarchy::new(
"bar",
vec![],
vec![DiagnosticsHierarchy::new(
"zed",
vec![Property::Int("13/:".to_string(), -4)],
vec![],
)],
),
DiagnosticsHierarchy::new(
"foo",
vec![Property::Int("11".to_string(), -4),],
vec![],
)
],
))
);
let test_selectors = vec!["*:root"];
let mut sorted_expected = get_test_hierarchy();
sorted_expected.sort();
assert_eq!(
parse_selectors_and_filter_hierarchy(get_test_hierarchy(), test_selectors),
Some(sorted_expected)
);
}
#[fuchsia::test]
fn test_filter_does_not_include_empty_node() {
let test_selectors = vec!["*:root/foo:blorg"];
assert_eq!(
parse_selectors_and_filter_hierarchy(get_test_hierarchy(), test_selectors),
None,
);
}
#[fuchsia::test]
fn test_filter_empty_hierarchy() {
let test_selectors = vec!["*:root"];
assert_eq!(
parse_selectors_and_filter_hierarchy(
DiagnosticsHierarchy::new("root", vec![], vec![]),
test_selectors
),
Some(DiagnosticsHierarchy::new("root", vec![], vec![])),
);
}
#[fuchsia::test]
fn test_full_filtering() {
let test_selectors = vec!["*:non-existent-root"];
assert_eq!(
parse_selectors_and_filter_hierarchy(get_test_hierarchy(), test_selectors),
None,
);
let test_selectors = vec!["*:root/i-dont-exist:foo"];
assert_eq!(
parse_selectors_and_filter_hierarchy(get_test_hierarchy(), test_selectors),
None,
);
let test_selectors = vec!["*:root:i-dont-exist"];
assert_eq!(
parse_selectors_and_filter_hierarchy(get_test_hierarchy(), test_selectors),
None,
);
}
#[fuchsia::test]
fn test_subtree_selection_includes_empty_nodes() {
let test_selectors = vec!["*:root"];
let mut empty_hierarchy = DiagnosticsHierarchy::new(
"root",
vec![],
vec![
DiagnosticsHierarchy::new(
"foo",
vec![],
vec![DiagnosticsHierarchy::new("zed", vec![], vec![])],
),
DiagnosticsHierarchy::new(
"bar",
vec![],
vec![DiagnosticsHierarchy::new("zed", vec![], vec![])],
),
],
);
empty_hierarchy.sort();
assert_eq!(
parse_selectors_and_filter_hierarchy(empty_hierarchy.clone(), test_selectors),
Some(empty_hierarchy)
);
}
#[fuchsia::test]
fn test_empty_tree_filtering() {
let mut empty_hierarchy = DiagnosticsHierarchy::new("root", vec![], vec![]);
empty_hierarchy.sort();
let subtree_selector = vec!["*:root"];
assert_eq!(
parse_selectors_and_filter_hierarchy(empty_hierarchy.clone(), subtree_selector),
Some(empty_hierarchy.clone())
);
let fake_property_selector = vec!["*:root:blorp"];
assert_eq!(
parse_selectors_and_filter_hierarchy(empty_hierarchy.clone(), fake_property_selector),
None,
);
}
#[fuchsia::test]
fn test_select_from_hierarchy() {
let int_11 = Property::Int("11".to_string(), -4);
let double_0 = Property::Double("0".to_string(), 8.1);
let bytes_123 = Property::Bytes("123".to_string(), "foo".bytes().collect());
let int_13 = Property::Int("13".to_string(), -4);
let test_cases = vec![
("*:root/foo:11", vec![&int_11]),
("*:root/foo:*", vec![&double_0, &int_11, &bytes_123]),
("*:root/foo:nonexistant", vec![]),
("*:root/foo", vec![&double_0, &int_11, &bytes_123, &int_13]),
];
for (test_selector, expected_vector) in test_cases {
let hierarchy = get_test_hierarchy();
let parsed_selector = selectors::parse_selector::<VerboseError>(test_selector)
.expect("All test selectors are valid and parsable.");
let mut property_entry_vec = select_from_hierarchy(&hierarchy, &parsed_selector)
.expect("Selecting from hierarchy should succeed.");
property_entry_vec.sort_by(|p1, p2| {
let p1_string = p1.name().to_string();
let p2_string = p2.name().to_string();
p1_string.cmp(&p2_string)
});
assert_eq!(property_entry_vec, expected_vector);
}
}
#[fuchsia::test]
fn sort_numerical_value() {
let mut diagnostics_hierarchy = DiagnosticsHierarchy::new(
"root",
vec![
Property::Double("2".to_string(), 2.3),
Property::Int("0".to_string(), -4),
Property::Uint("10".to_string(), 3),
Property::String("1".to_string(), "test".to_string()),
],
vec![
DiagnosticsHierarchy::new("123", vec![], vec![]),
DiagnosticsHierarchy::new("34", vec![], vec![]),
DiagnosticsHierarchy::new("4", vec![], vec![]),
DiagnosticsHierarchy::new("023", vec![], vec![]),
DiagnosticsHierarchy::new("12", vec![], vec![]),
DiagnosticsHierarchy::new("1", vec![], vec![]),
],
);
diagnostics_hierarchy.sort();
assert_eq!(
diagnostics_hierarchy,
DiagnosticsHierarchy::new(
"root",
vec![
Property::Int("0".to_string(), -4),
Property::String("1".to_string(), "test".to_string()),
Property::Double("2".to_string(), 2.3),
Property::Uint("10".to_string(), 3),
],
vec![
DiagnosticsHierarchy::new("1", vec![], vec![]),
DiagnosticsHierarchy::new("4", vec![], vec![]),
DiagnosticsHierarchy::new("12", vec![], vec![]),
DiagnosticsHierarchy::new("023", vec![], vec![]),
DiagnosticsHierarchy::new("34", vec![], vec![]),
DiagnosticsHierarchy::new("123", vec![], vec![]),
]
)
);
}
#[fuchsia::test]
fn filter_hierarchy_doesnt_return_partial_matches() {
let hierarchy = DiagnosticsHierarchy::new(
"root",
vec![],
vec![DiagnosticsHierarchy::new("session_started_at", vec![], vec![])],
);
let test_selectors = vec!["*:root/session_started_at/0"];
assert_eq!(parse_selectors_and_filter_hierarchy(hierarchy, test_selectors), None);
}
#[fuchsia::test]
fn test_filter_tree() {
let test_selectors = vec!["root/foo:11", "root:z", r#"root/bar/zed:13\/\:"#];
let parsed_test_selectors = test_selectors
.into_iter()
.map(|s| {
selectors::parse_tree_selector::<VerboseError>(s)
.expect("All test selectors are valid and parsable.")
})
.collect::<Vec<_>>();
let result =
filter_tree(get_test_hierarchy(), &parsed_test_selectors).map(|mut hierarchy| {
hierarchy.sort();
hierarchy
});
assert_eq!(
result,
Some(DiagnosticsHierarchy::new(
"root",
vec![Property::Int("z".to_string(), -4),],
vec![
DiagnosticsHierarchy::new(
"bar",
vec![],
vec![DiagnosticsHierarchy::new(
"zed",
vec![Property::Int("13/:".to_string(), -4)],
vec![],
)],
),
DiagnosticsHierarchy::new(
"foo",
vec![Property::Int("11".to_string(), -4),],
vec![],
)
],
))
);
}
#[fuchsia::test]
fn test_matcher_from_iterator() {
let matcher = HierarchyMatcher::new(
["*:root/foo:11", "*:root:z", r#"*:root/bar/zed:13\/\:"#].into_iter().map(|s| {
selectors::parse_selector::<VerboseError>(s)
.expect("All test selectors are valid and parsable.")
}),
)
.expect("create matcher from iterator of selectors");
let result = filter_hierarchy(get_test_hierarchy(), &matcher).map(|mut hierarchy| {
hierarchy.sort();
hierarchy
});
assert_eq!(
result,
Some(DiagnosticsHierarchy::new(
"root",
vec![Property::Int("z".to_string(), -4),],
vec![
DiagnosticsHierarchy::new(
"bar",
vec![],
vec![DiagnosticsHierarchy::new(
"zed",
vec![Property::Int("13/:".to_string(), -4)],
vec![],
)],
),
DiagnosticsHierarchy::new(
"foo",
vec![Property::Int("11".to_string(), -4),],
vec![],
)
],
))
);
}
}