use crate::metrics::Metrics;
use crate::puppet::DiffType;
use anyhow::{bail, format_err, Error};
use base64::engine::general_purpose::STANDARD as BASE64_STANDARD;
use base64::engine::Engine as _;
use diagnostics_hierarchy::{
ArrayContent, DiagnosticsHierarchy, ExponentialHistogram, LinearHistogram,
Property as iProperty,
};
use fidl_diagnostics_validate::{self as validate, Value};
use inspect_format::{ArrayFormat, BlockIndex, LinkNodeDisposition};
use num_traits::Zero;
use std::collections::{HashMap, HashSet};
use std::fmt;
mod scanner;
pub use scanner::Scanner;
mod fetch;
pub use fetch::LazyNode;
#[cfg(test)]
use num_traits::ToPrimitive;
const ROOT_NAME: &str = "root";
#[derive(Debug, Clone)]
pub struct Data {
nodes: HashMap<BlockIndex, Node>,
properties: HashMap<BlockIndex, Property>,
tombstone_nodes: HashSet<BlockIndex>,
tombstone_properties: HashSet<BlockIndex>,
}
#[derive(Debug, Clone)]
pub struct Node {
name: String,
parent: BlockIndex,
children: HashSet<BlockIndex>,
properties: HashSet<BlockIndex>,
}
#[derive(Debug, Clone)]
pub struct Property {
name: String,
#[allow(unused)]
id: BlockIndex,
parent: BlockIndex,
payload: Payload,
}
#[derive(Debug, Clone)]
enum Payload {
String(#[allow(unused)] String),
Bytes(Vec<u8>),
Int(i64),
Uint(u64),
Double(f64),
Bool(#[allow(unused)] bool),
IntArray(Vec<i64>, ArrayFormat),
UintArray(Vec<u64>, ArrayFormat),
DoubleArray(Vec<f64>, ArrayFormat),
StringArray(Vec<String>),
Link { disposition: LinkNodeDisposition, parsed_data: Data },
GenericNumber(#[allow(unused)] String),
GenericArray(#[allow(unused)] Vec<String>),
GenericLinearHistogram(#[allow(unused)] Vec<String>),
GenericExponentialHistogram(#[allow(unused)] Vec<String>),
}
fn to_string<T: std::fmt::Display>(v: T) -> String {
format!("{}", v)
}
fn stringify_list<T: std::fmt::Display>(values: Vec<T>) -> Vec<String> {
values.into_iter().map(|x| to_string(x)).collect()
}
fn stringify_linear_histogram<T: Clone + std::fmt::Display + Zero>(
histogram: LinearHistogram<T>,
) -> Vec<String> {
let mut values = vec![T::zero(); histogram.size + 2];
values[0] = histogram.floor;
values[1] = histogram.step;
if let Some(indexes) = histogram.indexes {
for (index, count) in indexes.iter().zip(histogram.counts) {
values[index + 2] = count.clone();
}
} else {
for (index, count) in histogram.counts.iter().enumerate() {
values[index + 2] = count.clone();
}
}
stringify_list(values)
}
fn stringify_exponential_histogram<T: Clone + std::fmt::Display + Zero>(
histogram: ExponentialHistogram<T>,
) -> Vec<String> {
let mut values = vec![T::zero(); histogram.size + 3];
values[0] = histogram.floor;
values[1] = histogram.initial_step;
values[2] = histogram.step_multiplier;
if let Some(indexes) = histogram.indexes {
for (index, count) in indexes.iter().zip(histogram.counts) {
values[index + 3] = count.clone();
}
} else {
for (index, count) in histogram.counts.iter().enumerate() {
values[index + 3] = count.clone();
}
}
stringify_list(values)
}
fn handle_array(values: Vec<String>, format: ArrayFormat) -> Payload {
match format {
ArrayFormat::LinearHistogram => Payload::GenericLinearHistogram(values),
ArrayFormat::ExponentialHistogram => Payload::GenericExponentialHistogram(values),
_ => Payload::GenericArray(values),
}
}
trait ToGeneric {
fn to_generic(self) -> Payload;
}
impl ToGeneric for Payload {
fn to_generic(self) -> Self {
match self {
Payload::IntArray(val, format) => handle_array(stringify_list(val), format),
Payload::UintArray(val, format) => handle_array(stringify_list(val), format),
Payload::DoubleArray(val, format) => handle_array(stringify_list(val), format),
Payload::StringArray(val) => handle_array(val, ArrayFormat::Default),
Payload::Bytes(val) => Payload::String(format!("b64:{}", BASE64_STANDARD.encode(&val))),
Payload::Int(val) => Payload::GenericNumber(to_string(val)),
Payload::Uint(val) => Payload::GenericNumber(to_string(val)),
Payload::Double(val) => Payload::GenericNumber(to_string(val)),
Payload::Link { disposition, parsed_data } => {
Payload::Link { disposition, parsed_data: parsed_data.clone_generic() }
}
val => val,
}
}
}
impl<T: std::fmt::Display + Clone + Zero> ToGeneric for ArrayContent<T> {
fn to_generic(self) -> Payload {
match self {
ArrayContent::Values(values) => Payload::GenericArray(stringify_list(values)),
ArrayContent::LinearHistogram(histogram) => {
Payload::GenericLinearHistogram(stringify_linear_histogram(histogram))
}
ArrayContent::ExponentialHistogram(histogram) => {
Payload::GenericExponentialHistogram(stringify_exponential_histogram(histogram))
}
}
}
}
struct FormattedEntries {
nodes: Vec<String>,
properties: Vec<String>,
}
impl Property {
fn to_string(&self, prefix: &str) -> String {
format!("{}{}: {:?}", prefix, self.name, &self.payload)
}
fn format_entries(&self, prefix: &str) -> FormattedEntries {
match &self.payload {
Payload::Link { disposition, parsed_data } => match disposition {
LinkNodeDisposition::Child => FormattedEntries {
nodes: vec![format!(
"{}{} ->\n{}",
prefix,
self.name,
parsed_data.nodes[&0.into()].to_string(prefix, parsed_data, true)
)],
properties: vec![],
},
LinkNodeDisposition::Inline => {
let root = &parsed_data.nodes[&BlockIndex::ROOT];
let mut nodes = root
.children
.iter()
.map(|v| {
parsed_data.nodes.get(v).map_or("Missing child".into(), |n| {
n.to_string(prefix, parsed_data, false)
})
})
.collect::<Vec<_>>();
let mut properties = vec![];
for FormattedEntries { nodes: mut n, properties: mut p } in
root.properties.iter().map(|v| {
parsed_data.properties.get(v).map_or(
FormattedEntries {
nodes: vec![],
properties: vec!["Missing property".into()],
},
|p| p.format_entries(prefix),
)
})
{
nodes.append(&mut n);
properties.append(&mut p);
}
FormattedEntries { nodes, properties }
}
},
_ => FormattedEntries { nodes: vec![], properties: vec![self.to_string(prefix)] },
}
}
}
impl Node {
fn to_string(&self, prefix: &str, tree: &Data, hide_root: bool) -> String {
let sub_prefix = format!("{}> ", prefix);
let mut nodes = vec![];
for node_id in self.children.iter() {
nodes.push(
tree.nodes
.get(node_id)
.map_or("Missing child".into(), |n| n.to_string(&sub_prefix, tree, hide_root)),
);
}
let mut properties = vec![];
for property_id in self.properties.iter() {
let FormattedEntries { nodes: mut n, properties: mut p } =
tree.properties.get(property_id).map_or(
FormattedEntries {
nodes: vec![],
properties: vec!["Missing property".to_string()],
},
|p| p.format_entries(&sub_prefix),
);
properties.append(&mut p);
nodes.append(&mut n);
}
nodes.sort_unstable();
properties.sort_unstable();
let mut output_lines = vec![];
if self.name != ROOT_NAME || !hide_root {
output_lines.push(format!("{}{} ->", prefix, self.name));
}
output_lines.append(&mut properties);
output_lines.append(&mut nodes);
output_lines.join("\n")
}
}
struct Op {
int: fn(i64, i64) -> i64,
uint: fn(u64, u64) -> u64,
double: fn(f64, f64) -> f64,
string: fn(String, String) -> String,
name: &'static str,
}
const ADD: Op = Op {
int: |a, b| a + b,
uint: |a, b| a + b,
double: |a, b| a + b,
string: |_, _| unreachable!(),
name: "add",
};
const SUBTRACT: Op = Op {
int: |a, b| a - b,
uint: |a, b| a - b,
double: |a, b| a - b,
string: |_, _| unreachable!(),
name: "subtract",
};
const SET: Op =
Op { int: |_a, b| b, uint: |_a, b| b, double: |_a, b| b, string: |_a, b| b, name: "set" };
macro_rules! insert_linear_fn {
($name:ident, $type:ident) => {
fn $name(numbers: &mut [$type], value: $type, count: u64) -> Result<(), Error> {
let buckets: $type = (numbers.len() as i32 - 4).try_into().unwrap();
let floor = numbers[0];
let step_size = numbers[1];
let index: usize = if value < floor {
2
} else if value >= floor + buckets * step_size {
numbers.len() - 1
} else {
(((value - floor) / step_size) as $type + 3 as $type) as i32 as usize
};
numbers[index] += count as $type;
Ok(())
}
};
}
insert_linear_fn! {insert_linear_i, i64}
insert_linear_fn! {insert_linear_u, u64}
insert_linear_fn! {insert_linear_d, f64}
macro_rules! insert_exponential_fn {
($name:ident, $type:ident, $fudge_factor:expr) => {
fn $name(numbers: &mut [$type], value: $type, count: u64) -> Result<(), Error> {
let buckets = numbers.len() - 5;
let floor = numbers[0];
let initial_step = numbers[1];
let step_multiplier = numbers[2];
let index = if value < floor {
3
} else if value < floor + initial_step {
4
} else if value
>= floor + initial_step * (step_multiplier as f64).powi(buckets as i32 - 1) as $type
{
numbers.len() - 1
} else {
((((value as f64 - floor as f64) / initial_step as f64) as f64).log2()
/ (step_multiplier as f64 + $fudge_factor).log2())
.trunc() as usize
+ 5
};
numbers[index] += count as $type;
Ok(())
}
};
}
insert_exponential_fn! {insert_exponential_i, i64, 0.0000000000000000000001}
insert_exponential_fn! {insert_exponential_u, u64, 0.0000000000000000000001}
insert_exponential_fn! {insert_exponential_d, f64, 0.0}
impl Data {
pub fn apply(&mut self, action: &validate::Action) -> Result<(), Error> {
match action {
validate::Action::CreateNode(validate::CreateNode { parent, id, name }) => {
self.create_node(parent.into(), id.into(), name)
}
validate::Action::DeleteNode(validate::DeleteNode { id }) => {
self.delete_node(id.into())
}
validate::Action::CreateNumericProperty(validate::CreateNumericProperty {
parent,
id,
name,
value,
}) => self.add_property(
parent.into(),
id.into(),
name,
match value {
validate::Value::IntT(value) => Payload::Int(*value),
validate::Value::UintT(value) => Payload::Uint(*value),
validate::Value::DoubleT(value) => Payload::Double(*value),
unknown => return Err(format_err!("Unknown number type {:?}", unknown)),
},
),
validate::Action::CreateBytesProperty(validate::CreateBytesProperty {
parent,
id,
name,
value,
}) => self.add_property(parent.into(), id.into(), name, Payload::Bytes(value.clone())),
validate::Action::CreateStringProperty(validate::CreateStringProperty {
parent,
id,
name,
value,
}) => self.add_property(
parent.into(),
id.into(),
name,
Payload::String(value.to_string()),
),
validate::Action::CreateBoolProperty(validate::CreateBoolProperty {
parent,
id,
name,
value,
}) => self.add_property(parent.into(), id.into(), name, Payload::Bool(*value)),
validate::Action::DeleteProperty(validate::DeleteProperty { id }) => {
self.delete_property(id.into())
}
validate::Action::SetNumber(validate::SetNumber { id, value }) => {
self.modify_number(id.into(), value, SET)
}
validate::Action::AddNumber(validate::AddNumber { id, value }) => {
self.modify_number(id.into(), value, ADD)
}
validate::Action::SubtractNumber(validate::SubtractNumber { id, value }) => {
self.modify_number(id.into(), value, SUBTRACT)
}
validate::Action::SetBytes(validate::SetBytes { id, value }) => {
self.set_bytes(id.into(), value)
}
validate::Action::SetString(validate::SetString { id, value }) => {
self.set_string(id.into(), value)
}
validate::Action::SetBool(validate::SetBool { id, value }) => {
self.set_bool(id.into(), *value)
}
validate::Action::CreateArrayProperty(validate::CreateArrayProperty {
parent,
id,
name,
slots,
value_type,
}) => self.add_property(
parent.into(),
id.into(),
name,
match value_type {
validate::ValueType::Int => {
Payload::IntArray(vec![0; *slots as usize], ArrayFormat::Default)
}
validate::ValueType::Uint => {
Payload::UintArray(vec![0; *slots as usize], ArrayFormat::Default)
}
validate::ValueType::Double => {
Payload::DoubleArray(vec![0.0; *slots as usize], ArrayFormat::Default)
}
validate::ValueType::String => {
Payload::StringArray(vec![String::new(); *slots as usize])
}
unknown => return Err(format_err!("unknown value type {unknown:?}")),
},
),
validate::Action::ArrayAdd(validate::ArrayAdd { id, index, value }) => {
self.modify_array(id.into(), *index, value, ADD)
}
validate::Action::ArraySubtract(validate::ArraySubtract { id, index, value }) => {
self.modify_array(id.into(), *index, value, SUBTRACT)
}
validate::Action::ArraySet(validate::ArraySet { id, index, value }) => {
self.modify_array(id.into(), *index, value, SET)
}
validate::Action::CreateLinearHistogram(validate::CreateLinearHistogram {
parent,
id,
name,
floor,
step_size,
buckets,
}) => self.add_property(
parent.into(),
id.into(),
name,
match (floor, step_size) {
(validate::Value::IntT(floor), validate::Value::IntT(step_size)) => {
let mut data = vec![0; *buckets as usize + 4];
data[0] = *floor;
data[1] = *step_size;
Payload::IntArray(data, ArrayFormat::LinearHistogram)
}
(validate::Value::UintT(floor), validate::Value::UintT(step_size)) => {
let mut data = vec![0; *buckets as usize + 4];
data[0] = *floor;
data[1] = *step_size;
Payload::UintArray(data, ArrayFormat::LinearHistogram)
}
(validate::Value::DoubleT(floor), validate::Value::DoubleT(step_size)) => {
let mut data = vec![0.0; *buckets as usize + 4];
data[0] = *floor;
data[1] = *step_size;
Payload::DoubleArray(data, ArrayFormat::LinearHistogram)
}
unexpected => {
return Err(format_err!(
"Bad types in CreateLinearHistogram: {:?}",
unexpected
))
}
},
),
validate::Action::CreateExponentialHistogram(
validate::CreateExponentialHistogram {
parent,
id,
name,
floor,
initial_step,
step_multiplier,
buckets,
},
) => self.add_property(
parent.into(),
id.into(),
name,
match (floor, initial_step, step_multiplier) {
(
validate::Value::IntT(floor),
validate::Value::IntT(initial_step),
validate::Value::IntT(step_multiplier),
) => {
let mut data = vec![0i64; *buckets as usize + 5];
data[0] = *floor;
data[1] = *initial_step;
data[2] = *step_multiplier;
Payload::IntArray(data, ArrayFormat::ExponentialHistogram)
}
(
validate::Value::UintT(floor),
validate::Value::UintT(initial_step),
validate::Value::UintT(step_multiplier),
) => {
let mut data = vec![0u64; *buckets as usize + 5];
data[0] = *floor;
data[1] = *initial_step;
data[2] = *step_multiplier;
Payload::UintArray(data, ArrayFormat::ExponentialHistogram)
}
(
validate::Value::DoubleT(floor),
validate::Value::DoubleT(initial_step),
validate::Value::DoubleT(step_multiplier),
) => {
let mut data = vec![0.0f64; *buckets as usize + 5];
data[0] = *floor;
data[1] = *initial_step;
data[2] = *step_multiplier;
Payload::DoubleArray(data, ArrayFormat::ExponentialHistogram)
}
unexpected => {
return Err(format_err!(
"Bad types in CreateExponentialHistogram: {:?}",
unexpected
))
}
},
),
validate::Action::Insert(validate::Insert { id, value }) => {
if let Some(mut property) = self.properties.get_mut(&id.into()) {
match (&mut property, value) {
(
Property {
payload: Payload::IntArray(numbers, ArrayFormat::LinearHistogram),
..
},
Value::IntT(value),
) => insert_linear_i(numbers, *value, 1),
(
Property {
payload:
Payload::IntArray(numbers, ArrayFormat::ExponentialHistogram),
..
},
Value::IntT(value),
) => insert_exponential_i(numbers, *value, 1),
(
Property {
payload: Payload::UintArray(numbers, ArrayFormat::LinearHistogram),
..
},
Value::UintT(value),
) => insert_linear_u(numbers, *value, 1),
(
Property {
payload:
Payload::UintArray(numbers, ArrayFormat::ExponentialHistogram),
..
},
Value::UintT(value),
) => insert_exponential_u(numbers, *value, 1),
(
Property {
payload: Payload::DoubleArray(numbers, ArrayFormat::LinearHistogram),
..
},
Value::DoubleT(value),
) => insert_linear_d(numbers, *value, 1),
(
Property {
payload:
Payload::DoubleArray(numbers, ArrayFormat::ExponentialHistogram),
..
},
Value::DoubleT(value),
) => insert_exponential_d(numbers, *value, 1),
unexpected => {
Err(format_err!("Type mismatch {:?} trying to insert", unexpected))
}
}
} else {
Err(format_err!("Tried to insert number on nonexistent property {}", id))
}
}
validate::Action::InsertMultiple(validate::InsertMultiple { id, value, count }) => {
if let Some(mut property) = self.properties.get_mut(&id.into()) {
match (&mut property, value) {
(
Property {
payload: Payload::IntArray(numbers, ArrayFormat::LinearHistogram),
..
},
Value::IntT(value),
) => insert_linear_i(numbers, *value, *count),
(
Property {
payload:
Payload::IntArray(numbers, ArrayFormat::ExponentialHistogram),
..
},
Value::IntT(value),
) => insert_exponential_i(numbers, *value, *count),
(
Property {
payload: Payload::UintArray(numbers, ArrayFormat::LinearHistogram),
..
},
Value::UintT(value),
) => insert_linear_u(numbers, *value, *count),
(
Property {
payload:
Payload::UintArray(numbers, ArrayFormat::ExponentialHistogram),
..
},
Value::UintT(value),
) => insert_exponential_u(numbers, *value, *count),
(
Property {
payload: Payload::DoubleArray(numbers, ArrayFormat::LinearHistogram),
..
},
Value::DoubleT(value),
) => insert_linear_d(numbers, *value, *count),
(
Property {
payload:
Payload::DoubleArray(numbers, ArrayFormat::ExponentialHistogram),
..
},
Value::DoubleT(value),
) => insert_exponential_d(numbers, *value, *count),
unexpected => Err(format_err!(
"Type mismatch {:?} trying to insert multiple",
unexpected
)),
}
} else {
Err(format_err!(
"Tried to insert_multiple number on nonexistent property {}",
id
))
}
}
_ => Err(format_err!("Unknown action {:?}", action)),
}
}
pub fn apply_lazy(&mut self, lazy_action: &validate::LazyAction) -> Result<(), Error> {
match lazy_action {
validate::LazyAction::CreateLazyNode(validate::CreateLazyNode {
parent,
id,
name,
disposition,
actions,
}) => self.add_lazy_node(parent.into(), id.into(), name, disposition, actions),
validate::LazyAction::DeleteLazyNode(validate::DeleteLazyNode { id }) => {
self.delete_property(id.into())
}
_ => Err(format_err!("Unknown lazy action {:?}", lazy_action)),
}
}
fn create_node(&mut self, parent: BlockIndex, id: BlockIndex, name: &str) -> Result<(), Error> {
let node = Node {
name: name.to_owned(),
parent,
children: HashSet::new(),
properties: HashSet::new(),
};
if self.tombstone_nodes.contains(&id) {
return Err(format_err!("Tried to create implicitly deleted node {}", id));
}
if self.nodes.insert(id, node).is_some() {
return Err(format_err!("Create called when node already existed at {}", id));
}
if let Some(parent_node) = self.nodes.get_mut(&parent) {
parent_node.children.insert(id);
} else {
return Err(format_err!("Parent {} of created node {} doesn't exist", parent, id));
}
Ok(())
}
fn delete_node(&mut self, id: BlockIndex) -> Result<(), Error> {
if id == BlockIndex::ROOT {
return Err(format_err!("Do not try to delete node 0"));
}
if self.tombstone_nodes.remove(&id) {
return Ok(());
}
if let Some(node) = self.nodes.remove(&id) {
for child in node.children.clone().iter() {
self.make_tombstone_node(*child)?;
}
for property in node.properties.clone().iter() {
self.make_tombstone_property(*property)?;
}
if let Some(parent) = self.nodes.get_mut(&node.parent) {
if !parent.children.remove(&id) {
bail!(
"Internal error! Parent {} didn't know about this child {}",
node.parent,
id
);
}
}
} else {
return Err(format_err!("Delete of nonexistent node {}", id));
}
Ok(())
}
fn make_tombstone_node(&mut self, id: BlockIndex) -> Result<(), Error> {
if id == BlockIndex::ROOT {
return Err(format_err!("Internal error! Do not try to delete node 0."));
}
if let Some(node) = self.nodes.remove(&id) {
for child in node.children.clone().iter() {
self.make_tombstone_node(*child)?;
}
for property in node.properties.clone().iter() {
self.make_tombstone_property(*property)?;
}
} else {
return Err(format_err!("Internal error! Tried to tombstone nonexistent node {}", id));
}
self.tombstone_nodes.insert(id);
Ok(())
}
fn make_tombstone_property(&mut self, id: BlockIndex) -> Result<(), Error> {
if self.properties.remove(&id).is_none() {
return Err(format_err!(
"Internal error! Tried to tombstone nonexistent property {}",
id
));
}
self.tombstone_properties.insert(id);
Ok(())
}
fn add_property(
&mut self,
parent: BlockIndex,
id: BlockIndex,
name: &str,
payload: Payload,
) -> Result<(), Error> {
if let Some(node) = self.nodes.get_mut(&parent) {
node.properties.insert(id);
} else {
return Err(format_err!("Parent {} of property {} not found", parent, id));
}
if self.tombstone_properties.contains(&id) {
return Err(format_err!("Tried to create implicitly deleted property {}", id));
}
let property = Property { parent, id, name: name.into(), payload };
if self.properties.insert(id, property).is_some() {
return Err(format_err!("Property insert called on existing id {}", id));
}
Ok(())
}
fn delete_property(&mut self, id: BlockIndex) -> Result<(), Error> {
if self.tombstone_properties.remove(&id) {
return Ok(());
}
if let Some(property) = self.properties.remove(&id) {
if let Some(node) = self.nodes.get_mut(&property.parent) {
if !node.properties.remove(&id) {
bail!(
"Internal error! Property {}'s parent {} didn't have it as child",
id,
property.parent
);
}
} else {
bail!(
"Internal error! Property {}'s parent {} doesn't exist on delete",
id,
property.parent
);
}
} else {
return Err(format_err!("Delete of nonexistent property {}", id));
}
Ok(())
}
fn modify_number(
&mut self,
id: BlockIndex,
value: &validate::Value,
op: Op,
) -> Result<(), Error> {
if let Some(property) = self.properties.get_mut(&id) {
match (&property, value) {
(Property { payload: Payload::Int(old), .. }, Value::IntT(value)) => {
property.payload = Payload::Int((op.int)(*old, *value));
}
(Property { payload: Payload::Uint(old), .. }, Value::UintT(value)) => {
property.payload = Payload::Uint((op.uint)(*old, *value));
}
(Property { payload: Payload::Double(old), .. }, Value::DoubleT(value)) => {
property.payload = Payload::Double((op.double)(*old, *value));
}
unexpected => {
return Err(format_err!("Bad types {:?} trying to set number", unexpected))
}
}
} else {
return Err(format_err!("Tried to {} number on nonexistent property {}", op.name, id));
}
Ok(())
}
fn modify_array(
&mut self,
id: BlockIndex,
index64: u64,
value: &validate::Value,
op: Op,
) -> Result<(), Error> {
if let Some(mut property) = self.properties.get_mut(&id) {
let index = index64 as usize;
let array_len = match &property {
Property { payload: Payload::IntArray(numbers, ArrayFormat::Default), .. } => {
numbers.len()
}
Property { payload: Payload::UintArray(numbers, ArrayFormat::Default), .. } => {
numbers.len()
}
Property {
payload: Payload::DoubleArray(numbers, ArrayFormat::Default), ..
} => numbers.len(),
Property { payload: Payload::StringArray(values), .. } => values.len(),
unexpected => {
return Err(format_err!(
"Bad types {:?} trying to set number of elements",
unexpected
))
}
};
if index >= array_len {
return Ok(());
}
match (&mut property, value) {
(Property { payload: Payload::IntArray(numbers, _), .. }, Value::IntT(value)) => {
numbers[index] = (op.int)(numbers[index], *value);
}
(Property { payload: Payload::UintArray(numbers, _), .. }, Value::UintT(value)) => {
numbers[index] = (op.uint)(numbers[index], *value);
}
(
Property { payload: Payload::DoubleArray(numbers, _), .. },
Value::DoubleT(value),
) => {
numbers[index] = (op.double)(numbers[index], *value);
}
(Property { payload: Payload::StringArray(values), .. }, Value::StringT(value)) => {
values[index] = (op.string)(values[index].clone(), value.clone());
}
unexpected => {
return Err(format_err!("Type mismatch {:?} trying to set value", unexpected))
}
}
} else {
return Err(format_err!("Tried to {} number on nonexistent property {}", op.name, id));
}
Ok(())
}
fn set_string(&mut self, id: BlockIndex, value: &String) -> Result<(), Error> {
if let Some(property) = self.properties.get_mut(&id) {
match &property {
Property { payload: Payload::String(_), .. } => {
property.payload = Payload::String(value.to_owned())
}
unexpected => {
return Err(format_err!("Bad type {:?} trying to set string", unexpected))
}
}
} else {
return Err(format_err!("Tried to set string on nonexistent property {}", id));
}
Ok(())
}
fn set_bytes(&mut self, id: BlockIndex, value: &Vec<u8>) -> Result<(), Error> {
if let Some(property) = self.properties.get_mut(&id) {
match &property {
Property { payload: Payload::Bytes(_), .. } => {
property.payload = Payload::Bytes(value.to_owned())
}
unexpected => {
return Err(format_err!("Bad type {:?} trying to set bytes", unexpected))
}
}
} else {
return Err(format_err!("Tried to set bytes on nonexistent property {}", id));
}
Ok(())
}
fn set_bool(&mut self, id: BlockIndex, value: bool) -> Result<(), Error> {
if let Some(property) = self.properties.get_mut(&id) {
match &property {
Property { payload: Payload::Bool(_), .. } => {
property.payload = Payload::Bool(value)
}
unexpected => {
return Err(format_err!("Bad type {:?} trying to set bool", unexpected))
}
}
} else {
return Err(format_err!("Tried to set bool on nonexistent property {}", id));
}
Ok(())
}
fn add_lazy_node(
&mut self,
parent: BlockIndex,
id: BlockIndex,
name: &str,
disposition: &validate::LinkDisposition,
actions: &Vec<validate::Action>,
) -> Result<(), Error> {
let mut parsed_data = Data::new();
parsed_data.apply_multiple(actions)?;
self.add_property(
parent,
id,
name,
Payload::Link {
disposition: match disposition {
validate::LinkDisposition::Child => LinkNodeDisposition::Child,
validate::LinkDisposition::Inline => LinkNodeDisposition::Inline,
},
parsed_data,
},
)?;
Ok(())
}
fn clone_generic(&self) -> Self {
let mut clone = self.clone();
let mut to_remove = vec![];
let mut names = HashSet::new();
clone.properties = clone
.properties
.into_iter()
.filter_map(|(id, mut v)| {
if !names.insert((v.parent, v.name.clone())) {
to_remove.push((v.parent, id));
None
} else {
v.payload = v.payload.to_generic();
Some((id, v))
}
})
.collect::<HashMap<BlockIndex, Property>>();
for (parent, id) in to_remove {
if clone.nodes.contains_key(&parent) {
clone.nodes.get_mut(&parent).unwrap().properties.remove(&id);
}
}
clone
}
pub fn compare_to_json(&self, other: &Data, diff_type: DiffType) -> Result<(), Error> {
self.clone_generic().compare(other, diff_type)
}
pub fn compare(&self, other: &Data, diff_type: DiffType) -> Result<(), Error> {
let self_string = self.to_string();
let other_string = other.to_string();
let difference::Changeset { diffs, distance, .. } =
difference::Changeset::new(&self_string, &other_string, "\n");
if distance == 0 {
Ok(())
} else {
let diff_lines = diffs
.into_iter()
.flat_map(|diff| {
let (prefix, val) = match diff {
difference::Difference::Same(val) => (" same", val),
difference::Difference::Add(val) => ("other", val),
difference::Difference::Rem(val) => ("local", val),
};
val.split("\n")
.map(|line| format!("{}: {:?}", prefix, line))
.collect::<Vec<_>>()
})
.collect::<Vec<_>>();
match diff_type {
DiffType::Full => Err(format_err!(
"Trees differ:\n-- LOCAL --\n{}\n-- OTHER --\n{}",
self_string,
other_string
)),
DiffType::Diff => {
Err(format_err!("Trees differ:\n-- DIFF --\n{}", diff_lines.join("\n")))
}
DiffType::Both => Err(format_err!(
"Trees differ:\n-- LOCAL --\n{}\n-- OTHER --\n{}\n-- DIFF --\n{}",
self_string,
other_string,
diff_lines.join("\n")
)),
}
}
}
pub fn new() -> Data {
let mut ret = Data {
nodes: HashMap::new(),
properties: HashMap::new(),
tombstone_nodes: HashSet::new(),
tombstone_properties: HashSet::new(),
};
ret.nodes.insert(
BlockIndex::ROOT,
Node {
name: ROOT_NAME.into(),
parent: BlockIndex::ROOT,
children: HashSet::new(),
properties: HashSet::new(),
},
);
ret
}
fn build(nodes: HashMap<BlockIndex, Node>, properties: HashMap<BlockIndex, Property>) -> Data {
Data {
nodes,
properties,
tombstone_nodes: HashSet::new(),
tombstone_properties: HashSet::new(),
}
}
fn apply_multiple(&mut self, actions: &Vec<validate::Action>) -> Result<(), Error> {
for action in actions {
self.apply(action)?;
}
Ok(())
}
pub fn remove_tree(&mut self, name: &str) {
if self.is_empty() {
return;
}
let mut target_node_id = None;
let root = &self.nodes[&BlockIndex::ROOT];
for child_id in &root.children {
if let Some(node) = self.nodes.get(child_id) {
if node.name == name {
target_node_id = Some(*child_id);
break;
}
}
}
if let Some(id) = target_node_id {
self.delete_node(id).unwrap();
}
}
pub fn is_empty(&self) -> bool {
if !self.nodes.contains_key(&BlockIndex::ROOT) {
return true;
}
if self.properties.len() == 0 {
return true;
}
let root = &self.nodes[&BlockIndex::ROOT];
root.children.is_subset(&self.tombstone_nodes) && root.properties.len() == 0
}
}
impl fmt::Display for Data {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let s = if let Some(node) = self.nodes.get(&BlockIndex::ROOT) {
node.to_string("", self, false)
} else {
"No root node; internal error\n".to_owned()
};
write!(f, "{s}",)
}
}
impl From<DiagnosticsHierarchy> for Data {
fn from(hierarchy: DiagnosticsHierarchy) -> Self {
let mut nodes = HashMap::new();
let mut properties = HashMap::new();
nodes.insert(
0.into(),
Node {
name: hierarchy.name.clone(),
parent: 0.into(),
children: HashSet::new(),
properties: HashSet::new(),
},
);
let mut queue = vec![(0.into(), &hierarchy)];
let mut next_id = BlockIndex::new(1);
while let Some((id, value)) = queue.pop() {
for node in value.children.iter() {
let child_id = next_id;
next_id += 1;
nodes.insert(
child_id,
Node {
name: node.name.clone(),
parent: id,
children: HashSet::new(),
properties: HashSet::new(),
},
);
nodes.get_mut(&id).expect("parent must exist").children.insert(child_id);
queue.push((child_id, node));
}
for property in value.properties.iter() {
let prop_id = next_id;
next_id += 1;
let (name, payload) = match property.clone() {
iProperty::String(n, v) => (n, Payload::String(v).to_generic()),
iProperty::Bytes(n, v) => (n, Payload::Bytes(v).to_generic()),
iProperty::Int(n, v) => (n, Payload::Int(v).to_generic()),
iProperty::Uint(n, v) => (n, Payload::Uint(v).to_generic()),
iProperty::Double(n, v) => (n, Payload::Double(v).to_generic()),
iProperty::Bool(n, v) => (n, Payload::Bool(v).to_generic()),
iProperty::IntArray(n, content) => (n, content.to_generic()),
iProperty::UintArray(n, content) => (n, content.to_generic()),
iProperty::DoubleArray(n, content) => (n, content.to_generic()),
iProperty::StringList(n, content) => {
(n, Payload::StringArray(content).to_generic())
}
};
properties.insert(prop_id, Property { name, id: prop_id, parent: id, payload });
nodes.get_mut(&id).expect("parent must exist").properties.insert(prop_id);
}
}
Data {
nodes,
properties,
tombstone_nodes: HashSet::new(),
tombstone_properties: HashSet::new(),
}
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::*;
use fidl_diagnostics_validate::{ValueType, ROOT_ID};
use fuchsia_inspect::reader::ArrayContent as iArrayContent;
use inspect_format::BlockType;
use num_derive::{FromPrimitive, ToPrimitive};
#[fuchsia::test]
fn test_basic_data_strings() -> Result<(), Error> {
let mut info = Data::new();
assert_eq!(info.to_string(), "root ->");
info.apply(&create_node!(parent: ROOT_ID, id: 1, name: "foo"))?;
assert_eq!(info.to_string(), "root ->\n> foo ->");
info.apply(&delete_node!( id: 1 ))?;
assert_eq!(info.to_string(), "root ->");
Ok(())
}
const EXPECTED_HIERARCHY: &str = r#"root ->
> double: GenericNumber("2.5")
> int: GenericNumber("-5")
> string: String("value")
> uint: GenericNumber("10")
> child ->
> > bytes: String("b64:AQI=")
> > grandchild ->
> > > double_a: GenericArray(["0.5", "1"])
> > > double_eh: GenericExponentialHistogram(["0.5", "0.5", "2", "1", "2", "3"])
> > > double_lh: GenericLinearHistogram(["0.5", "0.5", "1", "2", "3"])
> > > int_a: GenericArray(["-1", "-2"])
> > > int_eh: GenericExponentialHistogram(["-1", "1", "2", "1", "2", "3"])
> > > int_lh: GenericLinearHistogram(["-1", "1", "1", "2", "3"])
> > > uint_a: GenericArray(["1", "2"])
> > > uint_eh: GenericExponentialHistogram(["1", "1", "2", "1", "2", "3"])
> > > uint_lh: GenericLinearHistogram(["1", "1", "1", "2", "3"])"#;
#[derive(Debug, PartialEq, Eq, FromPrimitive, ToPrimitive)]
enum ArrayType {
Int = 4,
Uint = 5,
Double = 6,
}
#[fuchsia::test]
fn test_parse_hierarchy() -> Result<(), Error> {
let hierarchy = DiagnosticsHierarchy {
name: "root".to_string(),
properties: vec![
iProperty::String("string".to_string(), "value".to_string()),
iProperty::Uint("uint".to_string(), 10u64),
iProperty::Int("int".to_string(), -5i64),
iProperty::Double("double".to_string(), 2.5f64),
],
children: vec![DiagnosticsHierarchy {
name: "child".to_string(),
properties: vec![iProperty::Bytes("bytes".to_string(), vec![1u8, 2u8])],
children: vec![DiagnosticsHierarchy {
name: "grandchild".to_string(),
properties: vec![
iProperty::UintArray(
"uint_a".to_string(),
iArrayContent::Values(vec![1, 2]),
),
iProperty::IntArray(
"int_a".to_string(),
iArrayContent::Values(vec![-1i64, -2i64]),
),
iProperty::DoubleArray(
"double_a".to_string(),
iArrayContent::Values(vec![0.5, 1.0]),
),
iProperty::UintArray(
"uint_lh".to_string(),
iArrayContent::new(vec![1, 1, 1, 2, 3], ArrayFormat::LinearHistogram)
.unwrap(),
),
iProperty::IntArray(
"int_lh".to_string(),
iArrayContent::new(
vec![-1i64, 1, 1, 2, 3],
ArrayFormat::LinearHistogram,
)
.unwrap(),
),
iProperty::DoubleArray(
"double_lh".to_string(),
iArrayContent::new(
vec![0.5, 0.5, 1.0, 2.0, 3.0],
ArrayFormat::LinearHistogram,
)
.unwrap(),
),
iProperty::UintArray(
"uint_eh".to_string(),
iArrayContent::new(
vec![1, 1, 2, 1, 2, 3],
ArrayFormat::ExponentialHistogram,
)
.unwrap(),
),
iProperty::IntArray(
"int_eh".to_string(),
iArrayContent::new(
vec![-1i64, 1, 2, 1, 2, 3],
ArrayFormat::ExponentialHistogram,
)
.unwrap(),
),
iProperty::DoubleArray(
"double_eh".to_string(),
iArrayContent::new(
vec![0.5, 0.5, 2.0, 1.0, 2.0, 3.0],
ArrayFormat::ExponentialHistogram,
)
.unwrap(),
),
],
children: vec![],
missing: vec![],
}],
missing: vec![],
}],
missing: vec![],
};
let data: Data = hierarchy.into();
assert_eq!(EXPECTED_HIERARCHY, data.to_string());
Ok(())
}
#[fuchsia::test]
fn test_creation_deletion() -> Result<(), Error> {
let mut info = Data::new();
assert!(!info.to_string().contains("child ->"));
info.apply(&create_node!(parent: ROOT_ID, id: 1, name: "child"))?;
assert!(info.to_string().contains("child ->"));
info.apply(&create_node!(parent: 1, id: 2, name: "grandchild"))?;
assert!(
info.to_string().contains("grandchild ->") && info.to_string().contains("child ->")
);
info.apply(
&create_numeric_property!(parent: ROOT_ID, id: 3, name: "int-42", value: Value::IntT(-42)),
)?;
assert!(info.to_string().contains("int-42: Int(-42)")); info.apply(&create_string_property!(parent: 1, id: 4, name: "stringfoo", value: "foo"))?;
assert_eq!(
info.to_string(),
"root ->\n> int-42: Int(-42)\n> child ->\
\n> > stringfoo: String(\"foo\")\n> > grandchild ->"
);
info.apply(&create_numeric_property!(parent: ROOT_ID, id: 5, name: "uint", value: Value::UintT(1024)))?;
assert!(info.to_string().contains("uint: Uint(1024)"));
info.apply(&create_numeric_property!(parent: ROOT_ID, id: 6, name: "frac", value: Value::DoubleT(0.5)))?;
assert!(info.to_string().contains("frac: Double(0.5)"));
info.apply(
&create_bytes_property!(parent: ROOT_ID, id: 7, name: "bytes", value: vec!(1u8, 2u8)),
)?;
assert!(info.to_string().contains("bytes: Bytes([1, 2])"));
info.apply(&create_array_property!(parent: ROOT_ID, id: 8, name: "i_ntarr", slots: 1, type: ValueType::Int))?;
assert!(info.to_string().contains("i_ntarr: IntArray([0], Default)"));
info.apply(&create_array_property!(parent: ROOT_ID, id: 9, name: "u_intarr", slots: 2, type: ValueType::Uint))?;
assert!(info.to_string().contains("u_intarr: UintArray([0, 0], Default)"));
info.apply(&create_array_property!(parent: ROOT_ID, id: 10, name: "dblarr", slots: 3, type: ValueType::Double))?;
assert!(info.to_string().contains("dblarr: DoubleArray([0.0, 0.0, 0.0], Default)"));
info.apply(&create_linear_histogram!(parent: ROOT_ID, id: 11, name: "ILhist", floor: 12,
step_size: 3, buckets: 2, type: IntT))?;
assert!(info
.to_string()
.contains("ILhist: IntArray([12, 3, 0, 0, 0, 0], LinearHistogram)"));
info.apply(&create_linear_histogram!(parent: ROOT_ID, id: 12, name: "ULhist", floor: 34,
step_size: 5, buckets: 2, type: UintT))?;
assert!(info
.to_string()
.contains("ULhist: UintArray([34, 5, 0, 0, 0, 0], LinearHistogram)"));
info.apply(
&create_linear_histogram!(parent: ROOT_ID, id: 13, name: "DLhist", floor: 56.0,
step_size: 7.0, buckets: 2, type: DoubleT),
)?;
assert!(info
.to_string()
.contains("DLhist: DoubleArray([56.0, 7.0, 0.0, 0.0, 0.0, 0.0], LinearHistogram)"));
info.apply(&create_exponential_histogram!(parent: ROOT_ID, id: 14, name: "IEhist",
floor: 12, initial_step: 3, step_multiplier: 5, buckets: 2, type: IntT))?;
assert!(info
.to_string()
.contains("IEhist: IntArray([12, 3, 5, 0, 0, 0, 0], ExponentialHistogram)"));
info.apply(&create_exponential_histogram!(parent: ROOT_ID, id: 15, name: "UEhist",
floor: 34, initial_step: 9, step_multiplier: 6, buckets: 2, type: UintT))?;
assert!(info
.to_string()
.contains("UEhist: UintArray([34, 9, 6, 0, 0, 0, 0], ExponentialHistogram)"));
info.apply(&create_exponential_histogram!(parent: ROOT_ID, id: 16, name: "DEhist",
floor: 56.0, initial_step: 27.0, step_multiplier: 7.0, buckets: 2, type: DoubleT))?;
assert!(info.to_string().contains(
"DEhist: DoubleArray([56.0, 27.0, 7.0, 0.0, 0.0, 0.0, 0.0], ExponentialHistogram)"
));
info.apply(&create_bool_property!(parent: ROOT_ID, id: 17, name: "bool", value: true))?;
assert!(info.to_string().contains("bool: Bool(true)"));
info.apply(&delete_property!(id: 3))?;
assert!(!info.to_string().contains("int-42") && info.to_string().contains("stringfoo"));
info.apply(&delete_property!(id: 4))?;
assert!(!info.to_string().contains("stringfoo"));
info.apply(&delete_property!(id: 5))?;
assert!(!info.to_string().contains("uint"));
info.apply(&delete_property!(id: 6))?;
assert!(!info.to_string().contains("frac"));
info.apply(&delete_property!(id: 7))?;
assert!(!info.to_string().contains("bytes"));
info.apply(&delete_property!(id: 8))?;
assert!(!info.to_string().contains("i_ntarr"));
info.apply(&delete_property!(id: 9))?;
assert!(!info.to_string().contains("u_intarr"));
info.apply(&delete_property!(id: 10))?;
assert!(!info.to_string().contains("dblarr"));
info.apply(&delete_property!(id: 11))?;
assert!(!info.to_string().contains("ILhist"));
info.apply(&delete_property!(id: 12))?;
assert!(!info.to_string().contains("ULhist"));
info.apply(&delete_property!(id: 13))?;
assert!(!info.to_string().contains("DLhist"));
info.apply(&delete_property!(id: 14))?;
assert!(!info.to_string().contains("IEhist"));
info.apply(&delete_property!(id: 15))?;
assert!(!info.to_string().contains("UEhist"));
info.apply(&delete_property!(id: 16))?;
assert!(!info.to_string().contains("DEhist"));
info.apply(&delete_property!(id: 17))?;
assert!(!info.to_string().contains("bool"));
info.apply(&delete_node!(id:2))?;
assert!(!info.to_string().contains("grandchild") && info.to_string().contains("child"));
info.apply(&delete_node!( id: 1 ))?;
assert_eq!(info.to_string(), "root ->");
Ok(())
}
#[fuchsia::test]
fn test_basic_int_ops() -> Result<(), Error> {
let mut info = Data::new();
info.apply(&create_numeric_property!(parent: ROOT_ID, id: 3, name: "value",
value: Value::IntT(-42)))?;
assert!(info.apply(&add_number!(id: 3, value: Value::IntT(3))).is_ok());
assert!(info.to_string().contains("value: Int(-39)"));
assert!(info.apply(&add_number!(id: 3, value: Value::UintT(3))).is_err());
assert!(info.apply(&add_number!(id: 3, value: Value::DoubleT(3.0))).is_err());
assert!(info.to_string().contains("value: Int(-39)"));
assert!(info.apply(&subtract_number!(id: 3, value: Value::IntT(5))).is_ok());
assert!(info.to_string().contains("value: Int(-44)"));
assert!(info.apply(&subtract_number!(id: 3, value: Value::UintT(5))).is_err());
assert!(info.apply(&subtract_number!(id: 3, value: Value::DoubleT(5.0))).is_err());
assert!(info.to_string().contains("value: Int(-44)"));
assert!(info.apply(&set_number!(id: 3, value: Value::IntT(22))).is_ok());
assert!(info.to_string().contains("value: Int(22)"));
assert!(info.apply(&set_number!(id: 3, value: Value::UintT(23))).is_err());
assert!(info.apply(&set_number!(id: 3, value: Value::DoubleT(24.0))).is_err());
assert!(info.to_string().contains("value: Int(22)"));
Ok(())
}
#[fuchsia::test]
fn test_array_string_ops() -> Result<(), Error> {
let mut info = Data::new();
info.apply(&create_array_property!(parent: ROOT_ID, id: 3, name: "value", slots: 3,
type: ValueType::String))?;
assert!(info.apply(&array_add!(id: 3, index: 1, value: Value::IntT(3))).is_err());
assert!(info.to_string().contains(r#"value: StringArray(["", "", ""])"#));
assert!(info.apply(&array_add!(id: 3, index: 1, value: Value::UintT(3))).is_err());
assert!(info.apply(&array_add!(id: 3, index: 1, value: Value::DoubleT(3.0))).is_err());
assert!(info.to_string().contains(r#"value: StringArray(["", "", ""]"#));
assert!(info.apply(&array_subtract!(id: 3, index: 2, value: Value::IntT(5))).is_err());
assert!(info.apply(&array_subtract!(id: 3, index: 2, value: Value::UintT(5))).is_err());
assert!(info
.apply(&array_subtract!(id: 3, index: 2,
value: Value::DoubleT(5.0)))
.is_err());
assert!(info.to_string().contains(r#"value: StringArray(["", "", ""])"#));
assert!(info
.apply(&array_set!(id: 3, index: 1, value: Value::StringT("data".into())))
.is_ok());
assert!(info.to_string().contains(r#"value: StringArray(["", "data", ""])"#));
assert!(info.apply(&array_set!(id: 3, index: 1, value: Value::UintT(23))).is_err());
assert!(info.apply(&array_set!(id: 3, index: 1, value: Value::DoubleT(24.0))).is_err());
assert!(info.to_string().contains(r#"value: StringArray(["", "data", ""])"#));
let long: String = (0..3000).map(|_| ".").collect();
let expected = format!(r#"value: StringArray(["{}", "data", ""])"#, long.clone());
assert!(info.apply(&array_set!(id: 3, index: 0, value: Value::StringT(long))).is_ok());
assert!(info.to_string().contains(&expected));
Ok(())
}
#[fuchsia::test]
fn test_array_int_ops() -> Result<(), Error> {
let mut info = Data::new();
info.apply(&create_array_property!(parent: ROOT_ID, id: 3, name: "value", slots: 3,
type: ValueType::Int))?;
assert!(info.apply(&array_add!(id: 3, index: 1, value: Value::IntT(3))).is_ok());
assert!(info.to_string().contains("value: IntArray([0, 3, 0], Default)"));
assert!(info.apply(&array_add!(id: 3, index: 1, value: Value::UintT(3))).is_err());
assert!(info.apply(&array_add!(id: 3, index: 1, value: Value::DoubleT(3.0))).is_err());
assert!(info.to_string().contains("value: IntArray([0, 3, 0], Default)"));
assert!(info.apply(&array_subtract!(id: 3, index: 2, value: Value::IntT(5))).is_ok());
assert!(info.to_string().contains("value: IntArray([0, 3, -5], Default)"));
assert!(info.apply(&array_subtract!(id: 3, index: 2, value: Value::UintT(5))).is_err());
assert!(info
.apply(&array_subtract!(id: 3, index: 2,
value: Value::DoubleT(5.0)))
.is_err());
assert!(info.to_string().contains("value: IntArray([0, 3, -5], Default)"));
assert!(info.apply(&array_set!(id: 3, index: 1, value: Value::IntT(22))).is_ok());
assert!(info.to_string().contains("value: IntArray([0, 22, -5], Default)"));
assert!(info.apply(&array_set!(id: 3, index: 1, value: Value::UintT(23))).is_err());
assert!(info.apply(&array_set!(id: 3, index: 1, value: Value::DoubleT(24.0))).is_err());
assert!(info.to_string().contains("value: IntArray([0, 22, -5], Default)"));
Ok(())
}
#[fuchsia::test]
fn test_linear_int_ops() -> Result<(), Error> {
let mut info = Data::new();
info.apply(&create_linear_histogram!(parent: ROOT_ID, id: 3, name: "value",
floor: 4, step_size: 2, buckets: 2, type: IntT))?;
assert!(info.to_string().contains("value: IntArray([4, 2, 0, 0, 0, 0], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(4))).is_ok());
assert!(info.to_string().contains("value: IntArray([4, 2, 0, 1, 0, 0], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(5))).is_ok());
assert!(info.to_string().contains("value: IntArray([4, 2, 0, 2, 0, 0], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(6))).is_ok());
assert!(info.to_string().contains("value: IntArray([4, 2, 0, 2, 1, 0], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(8))).is_ok());
assert!(info.to_string().contains("value: IntArray([4, 2, 0, 2, 1, 1], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(i64::MAX))).is_ok());
assert!(info.to_string().contains("value: IntArray([4, 2, 0, 2, 1, 2], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(0))).is_ok());
assert!(info.to_string().contains("value: IntArray([4, 2, 1, 2, 1, 2], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(i64::MIN))).is_ok());
assert!(info.to_string().contains("value: IntArray([4, 2, 2, 2, 1, 2], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::UintT(0))).is_err());
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(0.0))).is_err());
assert!(info.apply(&array_set!(id: 3, index: 1, value: Value::IntT(222))).is_err());
assert!(info.to_string().contains("value: IntArray([4, 2, 2, 2, 1, 2], LinearHistogram)"));
assert!(info.apply(&insert_multiple!(id: 3, value: Value::IntT(7), count: 4)).is_ok());
assert!(info.to_string().contains("value: IntArray([4, 2, 2, 2, 5, 2], LinearHistogram)"));
Ok(())
}
#[fuchsia::test]
fn test_exponential_int_ops() -> Result<(), Error> {
let mut info = Data::new();
info.apply(&create_exponential_histogram!(parent: ROOT_ID, id: 3, name: "value",
floor: 5, initial_step: 2,
step_multiplier: 4, buckets: 2, type: IntT))?;
assert!(info
.to_string()
.contains("value: IntArray([5, 2, 4, 0, 0, 0, 0], ExponentialHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(5))).is_ok());
assert!(info
.to_string()
.contains("value: IntArray([5, 2, 4, 0, 1, 0, 0], ExponentialHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(6))).is_ok());
assert!(info
.to_string()
.contains("value: IntArray([5, 2, 4, 0, 2, 0, 0], ExponentialHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(7))).is_ok());
assert!(info
.to_string()
.contains("value: IntArray([5, 2, 4, 0, 2, 1, 0], ExponentialHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(13))).is_ok());
assert!(info
.to_string()
.contains("value: IntArray([5, 2, 4, 0, 2, 1, 1], ExponentialHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(i64::MAX))).is_ok());
assert!(info
.to_string()
.contains("value: IntArray([5, 2, 4, 0, 2, 1, 2], ExponentialHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(0))).is_ok());
assert!(info
.to_string()
.contains("value: IntArray([5, 2, 4, 1, 2, 1, 2], ExponentialHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(i64::MIN))).is_ok());
assert!(info
.to_string()
.contains("value: IntArray([5, 2, 4, 2, 2, 1, 2], ExponentialHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::UintT(0))).is_err());
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(0.0))).is_err());
assert!(info.apply(&array_set!(id: 3, index: 1, value: Value::IntT(222))).is_err());
assert!(info
.to_string()
.contains("value: IntArray([5, 2, 4, 2, 2, 1, 2], ExponentialHistogram)"));
assert!(info.apply(&insert_multiple!(id: 3, value: Value::IntT(12), count: 4)).is_ok());
assert!(info
.to_string()
.contains("value: IntArray([5, 2, 4, 2, 2, 5, 2], ExponentialHistogram)"));
Ok(())
}
#[fuchsia::test]
fn test_array_out_of_bounds_nop() -> Result<(), Error> {
let mut info = Data::new();
info.apply(&create_array_property!(parent: ROOT_ID, id: 3, name: "value", slots: 3,
type: ValueType::Int))?;
assert!(info.apply(&array_add!(id: 3, index: 1, value: Value::IntT(3))).is_ok());
assert!(info.to_string().contains("value: IntArray([0, 3, 0], Default)"));
assert!(info.apply(&array_add!(id: 3, index: 3, value: Value::IntT(3))).is_ok());
assert!(info.apply(&array_add!(id: 3, index: 6, value: Value::IntT(3))).is_ok());
assert!(info.apply(&array_add!(id: 3, index: 12345, value: Value::IntT(3))).is_ok());
assert!(info.to_string().contains("value: IntArray([0, 3, 0], Default)"));
Ok(())
}
#[fuchsia::test]
fn test_basic_uint_ops() -> Result<(), Error> {
let mut info = Data::new();
info.apply(&create_numeric_property!(parent: ROOT_ID, id: 3, name: "value",
value: Value::UintT(42)))?;
assert!(info.apply(&add_number!(id: 3, value: Value::UintT(3))).is_ok());
assert!(info.to_string().contains("value: Uint(45)"));
assert!(info.apply(&add_number!(id: 3, value: Value::IntT(3))).is_err());
assert!(info.apply(&add_number!(id: 3, value: Value::DoubleT(3.0))).is_err());
assert!(info.to_string().contains("value: Uint(45)"));
assert!(info.apply(&subtract_number!(id: 3, value: Value::UintT(5))).is_ok());
assert!(info.to_string().contains("value: Uint(40)"));
assert!(info.apply(&subtract_number!(id: 3, value: Value::IntT(5))).is_err());
assert!(info.apply(&subtract_number!(id: 3, value: Value::DoubleT(5.0))).is_err());
assert!(info.to_string().contains("value: Uint(40)"));
assert!(info.apply(&set_number!(id: 3, value: Value::UintT(22))).is_ok());
assert!(info.to_string().contains("value: Uint(22)"));
assert!(info.apply(&set_number!(id: 3, value: Value::IntT(23))).is_err());
assert!(info.apply(&set_number!(id: 3, value: Value::DoubleT(24.0))).is_err());
assert!(info.to_string().contains("value: Uint(22)"));
Ok(())
}
#[fuchsia::test]
fn test_array_uint_ops() -> Result<(), Error> {
let mut info = Data::new();
info.apply(&create_array_property!(parent: ROOT_ID, id: 3, name: "value", slots: 3,
type: ValueType::Uint))?;
assert!(info.apply(&array_add!(id: 3, index: 1, value: Value::UintT(3))).is_ok());
assert!(info.to_string().contains("value: UintArray([0, 3, 0], Default)"));
assert!(info.apply(&array_add!(id: 3, index: 1, value: Value::IntT(3))).is_err());
assert!(info.apply(&array_add!(id: 3, index: 1, value: Value::DoubleT(3.0))).is_err());
assert!(info.to_string().contains("value: UintArray([0, 3, 0], Default)"));
assert!(info.apply(&array_set!(id: 3, index: 1, value: Value::UintT(22))).is_ok());
assert!(info.to_string().contains("value: UintArray([0, 22, 0], Default)"));
assert!(info.apply(&array_set!(id: 3, index: 1, value: Value::IntT(23))).is_err());
assert!(info.apply(&array_set!(id: 3, index: 1, value: Value::DoubleT(24.0))).is_err());
assert!(info.to_string().contains("value: UintArray([0, 22, 0], Default)"));
assert!(info.apply(&array_subtract!(id: 3, index: 1, value: Value::UintT(5))).is_ok());
assert!(info.to_string().contains("value: UintArray([0, 17, 0], Default)"));
assert!(info.apply(&array_subtract!(id: 3, index: 1, value: Value::IntT(5))).is_err());
assert!(info.apply(&array_subtract!(id: 3, index: 1, value: Value::DoubleT(5.0))).is_err());
assert!(info.to_string().contains("value: UintArray([0, 17, 0], Default)"));
Ok(())
}
#[fuchsia::test]
fn test_linear_uint_ops() -> Result<(), Error> {
let mut info = Data::new();
info.apply(&create_linear_histogram!(parent: ROOT_ID, id: 3, name: "value",
floor: 4, step_size: 2, buckets: 2, type: UintT))?;
assert!(info.to_string().contains("value: UintArray([4, 2, 0, 0, 0, 0], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::UintT(4))).is_ok());
assert!(info.to_string().contains("value: UintArray([4, 2, 0, 1, 0, 0], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::UintT(5))).is_ok());
assert!(info.to_string().contains("value: UintArray([4, 2, 0, 2, 0, 0], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::UintT(6))).is_ok());
assert!(info.to_string().contains("value: UintArray([4, 2, 0, 2, 1, 0], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::UintT(8))).is_ok());
assert!(info.to_string().contains("value: UintArray([4, 2, 0, 2, 1, 1], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::UintT(u64::MAX))).is_ok());
assert!(info.to_string().contains("value: UintArray([4, 2, 0, 2, 1, 2], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::UintT(0))).is_ok());
assert!(info.to_string().contains("value: UintArray([4, 2, 1, 2, 1, 2], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(0))).is_err());
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(0.0))).is_err());
assert!(info.apply(&array_set!(id: 3, index: 1, value: Value::UintT(222))).is_err());
assert!(info.to_string().contains("value: UintArray([4, 2, 1, 2, 1, 2], LinearHistogram)"));
assert!(info.apply(&insert_multiple!(id: 3, value: Value::UintT(7), count: 4)).is_ok());
assert!(info.to_string().contains("value: UintArray([4, 2, 1, 2, 5, 2], LinearHistogram)"));
Ok(())
}
#[fuchsia::test]
fn test_exponential_uint_ops() -> Result<(), Error> {
let mut info = Data::new();
info.apply(&create_exponential_histogram!(parent: ROOT_ID, id: 3, name: "value",
floor: 5, initial_step: 2,
step_multiplier: 4, buckets: 2, type: UintT))?;
assert!(info
.to_string()
.contains("value: UintArray([5, 2, 4, 0, 0, 0, 0], ExponentialHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::UintT(5))).is_ok());
assert!(info
.to_string()
.contains("value: UintArray([5, 2, 4, 0, 1, 0, 0], ExponentialHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::UintT(6))).is_ok());
assert!(info
.to_string()
.contains("value: UintArray([5, 2, 4, 0, 2, 0, 0], ExponentialHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::UintT(7))).is_ok());
assert!(info
.to_string()
.contains("value: UintArray([5, 2, 4, 0, 2, 1, 0], ExponentialHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::UintT(13))).is_ok());
assert!(info
.to_string()
.contains("value: UintArray([5, 2, 4, 0, 2, 1, 1], ExponentialHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::UintT(u64::MAX))).is_ok());
assert!(info
.to_string()
.contains("value: UintArray([5, 2, 4, 0, 2, 1, 2], ExponentialHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::UintT(0))).is_ok());
assert!(info
.to_string()
.contains("value: UintArray([5, 2, 4, 1, 2, 1, 2], ExponentialHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(0))).is_err());
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(0.0))).is_err());
assert!(info.apply(&array_set!(id: 3, index: 1, value: Value::UintT(222))).is_err());
assert!(info
.to_string()
.contains("value: UintArray([5, 2, 4, 1, 2, 1, 2], ExponentialHistogram)"));
assert!(info.apply(&insert_multiple!(id: 3, value: Value::UintT(12), count: 4)).is_ok());
assert!(info
.to_string()
.contains("value: UintArray([5, 2, 4, 1, 2, 5, 2], ExponentialHistogram)"));
Ok(())
}
#[fuchsia::test]
fn test_basic_double_ops() -> Result<(), Error> {
let mut info = Data::new();
info.apply(&create_numeric_property!(parent: ROOT_ID, id: 3, name: "value",
value: Value::DoubleT(42.0)))?;
assert!(info.apply(&add_number!(id: 3, value: Value::DoubleT(3.0))).is_ok());
assert!(info.to_string().contains("value: Double(45.0)"));
assert!(info.apply(&add_number!(id: 3, value: Value::IntT(3))).is_err());
assert!(info.apply(&add_number!(id: 3, value: Value::UintT(3))).is_err());
assert!(info.to_string().contains("value: Double(45.0)"));
assert!(info.apply(&subtract_number!(id: 3, value: Value::DoubleT(5.0))).is_ok());
assert!(info.to_string().contains("value: Double(40.0)"));
assert!(info.apply(&subtract_number!(id: 3, value: Value::UintT(5))).is_err());
assert!(info.apply(&subtract_number!(id: 3, value: Value::UintT(5))).is_err());
assert!(info.to_string().contains("value: Double(40.0)"));
assert!(info.apply(&set_number!(id: 3, value: Value::DoubleT(22.0))).is_ok());
assert!(info.to_string().contains("value: Double(22.0)"));
assert!(info.apply(&set_number!(id: 3, value: Value::UintT(23))).is_err());
assert!(info.apply(&set_number!(id: 3, value: Value::UintT(24))).is_err());
assert!(info.to_string().contains("value: Double(22.0)"));
Ok(())
}
#[fuchsia::test]
fn test_array_double_ops() -> Result<(), Error> {
let mut info = Data::new();
info.apply(&create_array_property!(parent: ROOT_ID, id: 3, name: "value", slots: 3,
type: ValueType::Double))?;
assert!(info.apply(&array_add!(id: 3, index: 1, value: Value::DoubleT(3.0))).is_ok());
assert!(info.to_string().contains("value: DoubleArray([0.0, 3.0, 0.0], Default)"));
assert!(info.apply(&array_add!(id: 3, index: 1, value: Value::IntT(3))).is_err());
assert!(info.apply(&array_add!(id: 3, index: 1, value: Value::UintT(3))).is_err());
assert!(info.to_string().contains("value: DoubleArray([0.0, 3.0, 0.0], Default)"));
assert!(info.apply(&array_subtract!(id: 3, index: 2, value: Value::DoubleT(5.0))).is_ok());
assert!(info.to_string().contains("value: DoubleArray([0.0, 3.0, -5.0], Default)"));
assert!(info.apply(&array_subtract!(id: 3, index: 2, value: Value::IntT(5))).is_err());
assert!(info.apply(&array_subtract!(id: 3, index: 2, value: Value::UintT(5))).is_err());
assert!(info.to_string().contains("value: DoubleArray([0.0, 3.0, -5.0], Default)"));
assert!(info.apply(&array_set!(id: 3, index: 1, value: Value::DoubleT(22.0))).is_ok());
assert!(info.to_string().contains("value: DoubleArray([0.0, 22.0, -5.0], Default)"));
assert!(info.apply(&array_set!(id: 3, index: 1, value: Value::IntT(23))).is_err());
assert!(info.apply(&array_set!(id: 3, index: 1, value: Value::IntT(24))).is_err());
assert!(info.to_string().contains("value: DoubleArray([0.0, 22.0, -5.0], Default)"));
Ok(())
}
#[fuchsia::test]
fn test_linear_double_ops() -> Result<(), Error> {
let mut info = Data::new();
info.apply(&create_linear_histogram!(parent: ROOT_ID, id: 3, name: "value",
floor: 4.0, step_size: 0.5, buckets: 2, type: DoubleT))?;
assert!(info
.to_string()
.contains("value: DoubleArray([4.0, 0.5, 0.0, 0.0, 0.0, 0.0], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(4.0))).is_ok());
assert!(info
.to_string()
.contains("value: DoubleArray([4.0, 0.5, 0.0, 1.0, 0.0, 0.0], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(4.25))).is_ok());
assert!(info
.to_string()
.contains("value: DoubleArray([4.0, 0.5, 0.0, 2.0, 0.0, 0.0], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(4.75))).is_ok());
assert!(info
.to_string()
.contains("value: DoubleArray([4.0, 0.5, 0.0, 2.0, 1.0, 0.0], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(5.1))).is_ok());
assert!(info
.to_string()
.contains("value: DoubleArray([4.0, 0.5, 0.0, 2.0, 1.0, 1.0], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(f64::MAX))).is_ok());
assert!(info
.to_string()
.contains("value: DoubleArray([4.0, 0.5, 0.0, 2.0, 1.0, 2.0], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(f64::MIN_POSITIVE))).is_ok());
assert!(info
.to_string()
.contains("value: DoubleArray([4.0, 0.5, 1.0, 2.0, 1.0, 2.0], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(f64::MIN))).is_ok());
assert!(info
.to_string()
.contains("value: DoubleArray([4.0, 0.5, 2.0, 2.0, 1.0, 2.0], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(0.0))).is_ok());
assert!(info
.to_string()
.contains("value: DoubleArray([4.0, 0.5, 3.0, 2.0, 1.0, 2.0], LinearHistogram)"));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(0))).is_err());
assert!(info.apply(&insert!(id: 3, value: Value::UintT(0))).is_err());
assert!(info.apply(&array_set!(id: 3, index: 1, value: Value::DoubleT(222.0))).is_err());
assert!(info
.to_string()
.contains("value: DoubleArray([4.0, 0.5, 3.0, 2.0, 1.0, 2.0], LinearHistogram)"));
assert!(info.apply(&insert_multiple!(id: 3, value: Value::DoubleT(4.5), count: 4)).is_ok());
assert!(info
.to_string()
.contains("value: DoubleArray([4.0, 0.5, 3.0, 2.0, 5.0, 2.0], LinearHistogram)"));
Ok(())
}
#[fuchsia::test]
fn test_exponential_double_ops() -> Result<(), Error> {
let mut info = Data::new();
info.apply(&create_exponential_histogram!(parent: ROOT_ID, id: 3, name: "value",
floor: 5.0, initial_step: 2.0,
step_multiplier: 4.0, buckets: 3, type: DoubleT))?;
assert!(info.to_string().contains(
"value: DoubleArray([5.0, 2.0, 4.0, 0.0, 0.0, 0.0, 0.0, 0.0], ExponentialHistogram)"
));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(5.0))).is_ok());
assert!(info.to_string().contains(
"value: DoubleArray([5.0, 2.0, 4.0, 0.0, 1.0, 0.0, 0.0, 0.0], ExponentialHistogram)"
));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(6.9))).is_ok());
assert!(info.to_string().contains(
"value: DoubleArray([5.0, 2.0, 4.0, 0.0, 2.0, 0.0, 0.0, 0.0], ExponentialHistogram)"
));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(7.1))).is_ok());
assert!(info.to_string().contains(
"value: DoubleArray([5.0, 2.0, 4.0, 0.0, 2.0, 1.0, 0.0, 0.0], ExponentialHistogram)"
));
assert!(info
.apply(&insert_multiple!(id: 3, value: Value::DoubleT(12.9), count: 4))
.is_ok());
assert!(info.to_string().contains(
"value: DoubleArray([5.0, 2.0, 4.0, 0.0, 2.0, 5.0, 0.0, 0.0], ExponentialHistogram)"
));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(13.1))).is_ok());
assert!(info.to_string().contains(
"value: DoubleArray([5.0, 2.0, 4.0, 0.0, 2.0, 5.0, 1.0, 0.0], ExponentialHistogram)"
));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(36.9))).is_ok());
assert!(info.to_string().contains(
"value: DoubleArray([5.0, 2.0, 4.0, 0.0, 2.0, 5.0, 2.0, 0.0], ExponentialHistogram)"
));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(37.1))).is_ok());
assert!(info.to_string().contains(
"value: DoubleArray([5.0, 2.0, 4.0, 0.0, 2.0, 5.0, 2.0, 1.0], ExponentialHistogram)"
));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(f64::MAX))).is_ok());
assert!(info.to_string().contains(
"value: DoubleArray([5.0, 2.0, 4.0, 0.0, 2.0, 5.0, 2.0, 2.0], ExponentialHistogram)"
));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(f64::MIN_POSITIVE))).is_ok());
assert!(info.to_string().contains(
"value: DoubleArray([5.0, 2.0, 4.0, 1.0, 2.0, 5.0, 2.0, 2.0], ExponentialHistogram)"
));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(f64::MIN))).is_ok());
assert!(info.to_string().contains(
"value: DoubleArray([5.0, 2.0, 4.0, 2.0, 2.0, 5.0, 2.0, 2.0], ExponentialHistogram)"
));
assert!(info.apply(&insert!(id: 3, value: Value::DoubleT(0.0))).is_ok());
assert!(info.to_string().contains(
"value: DoubleArray([5.0, 2.0, 4.0, 3.0, 2.0, 5.0, 2.0, 2.0], ExponentialHistogram)"
));
assert!(info.apply(&insert!(id: 3, value: Value::IntT(0))).is_err());
assert!(info.apply(&insert!(id: 3, value: Value::UintT(0))).is_err());
assert!(info.apply(&array_set!(id: 3, index: 1, value: Value::DoubleT(222.0))).is_err());
assert!(info.to_string().contains(
"value: DoubleArray([5.0, 2.0, 4.0, 3.0, 2.0, 5.0, 2.0, 2.0], ExponentialHistogram)"
));
Ok(())
}
#[fuchsia::test]
fn test_basic_vector_ops() -> Result<(), Error> {
let mut info = Data::new();
info.apply(&create_string_property!(parent: ROOT_ID, id: 3, name: "value",
value: "foo"))?;
assert!(info.to_string().contains("value: String(\"foo\")"));
assert!(info.apply(&set_string!(id: 3, value: "bar")).is_ok());
assert!(info.to_string().contains("value: String(\"bar\")"));
assert!(info.apply(&set_bytes!(id: 3, value: vec!(3u8))).is_err());
assert!(info.to_string().contains("value: String(\"bar\")"));
info.apply(&create_bytes_property!(parent: ROOT_ID, id: 4, name: "bvalue",
value: vec!(1u8, 2u8)))?;
assert!(info.to_string().contains("bvalue: Bytes([1, 2])"));
assert!(info.apply(&set_bytes!(id: 4, value: vec!(3u8, 4u8))).is_ok());
assert!(info.to_string().contains("bvalue: Bytes([3, 4])"));
assert!(info.apply(&set_string!(id: 4, value: "baz")).is_err());
assert!(info.to_string().contains("bvalue: Bytes([3, 4])"));
Ok(())
}
#[fuchsia::test]
fn test_basic_lazy_node_ops() -> Result<(), Error> {
let mut info = Data::new();
info.apply_lazy(&create_lazy_node!(parent: ROOT_ID, id: 1, name: "child", disposition: validate::LinkDisposition::Child, actions: vec![create_bytes_property!(parent: ROOT_ID, id: 1, name: "child_bytes",value: vec!(3u8, 4u8))]))?;
info.apply_lazy(&create_lazy_node!(parent: ROOT_ID, id: 2, name: "inline", disposition: validate::LinkDisposition::Inline, actions: vec![create_bytes_property!(parent: ROOT_ID, id: 1, name: "inline_bytes",value: vec!(3u8, 4u8))]))?;
assert_eq!(
info.to_string(),
"root ->\n> inline_bytes: Bytes([3, 4])\n> child ->\n> > child_bytes: Bytes([3, 4])"
);
info.apply_lazy(&delete_lazy_node!(id: 1))?;
assert_eq!(info.to_string(), "root ->\n> inline_bytes: Bytes([3, 4])");
Ok(())
}
#[fuchsia::test]
fn test_illegal_node_actions() -> Result<(), Error> {
let mut info = Data::new();
assert!(info.apply(&create_node!(parent: 42, id: 1, name: "child")).is_err());
info = Data::new();
info.apply(&create_node!(parent: ROOT_ID, id: 1, name: "child"))?;
assert!(info.apply(&create_node!(parent: ROOT_ID, id: 1, name: "another_child")).is_err());
info = Data::new();
assert!(info.apply(&delete_node!(id: ROOT_ID)).is_err());
info = Data::new();
assert!(info.apply(&delete_node!(id: 333)).is_err());
Ok(())
}
#[fuchsia::test]
fn test_illegal_property_actions() -> Result<(), Error> {
let mut info = Data::new();
assert!(info
.apply(
&create_numeric_property!(parent: 42, id: 1, name: "answer", value: Value::IntT(42))
)
.is_err());
info = Data::new();
info.apply(&create_numeric_property!(parent: ROOT_ID, id: 1, name: "answer",
value: Value::IntT(42)))?;
assert!(info
.apply(&create_numeric_property!(parent: ROOT_ID, id: 1, name: "another_answer",
value: Value::IntT(7)))
.is_err());
info = Data::new();
assert!(info.apply(&delete_property!(id: 1)).is_err());
info = Data::new();
info.apply(&create_numeric_property!(parent: ROOT_ID, id: 3, name: "value",
value: Value::IntT(42)))?;
info.apply(&create_array_property!(parent: ROOT_ID, id: 4, name: "array", slots: 2,
type: ValueType::Int))?;
info.apply(&create_linear_histogram!(parent: ROOT_ID, id: 5, name: "lin",
floor: 5, step_size: 2,
buckets: 2, type: IntT))?;
info.apply(&create_exponential_histogram!(parent: ROOT_ID, id: 6, name: "exp",
floor: 5, initial_step: 2,
step_multiplier: 2, buckets: 2, type: IntT))?;
assert!(info.apply(&set_number!(id: 3, value: Value::IntT(5))).is_ok());
assert!(info.apply(&array_set!(id: 4, index: 0, value: Value::IntT(5))).is_ok());
assert!(info.apply(&insert!(id: 5, value: Value::IntT(2))).is_ok());
assert!(info.apply(&insert!(id: 6, value: Value::IntT(2))).is_ok());
assert!(info.apply(&insert_multiple!(id: 5, value: Value::IntT(2), count: 3)).is_ok());
assert!(info.apply(&insert_multiple!(id: 6, value: Value::IntT(2), count: 3)).is_ok());
assert!(info.apply(&set_number!(id: 4, value: Value::IntT(5))).is_err());
assert!(info.apply(&set_number!(id: 5, value: Value::IntT(5))).is_err());
assert!(info.apply(&set_number!(id: 6, value: Value::IntT(5))).is_err());
assert!(info.apply(&array_set!(id: 3, index: 0, value: Value::IntT(5))).is_err());
assert!(info.apply(&array_set!(id: 5, index: 0, value: Value::IntT(5))).is_err());
assert!(info.apply(&array_set!(id: 6, index: 0, value: Value::IntT(5))).is_err());
assert!(info.apply(&insert!(id: 3, value: Value::IntT(2))).is_err());
assert!(info.apply(&insert!(id: 4, value: Value::IntT(2))).is_err());
assert!(info.apply(&insert_multiple!(id: 3, value: Value::IntT(2), count: 3)).is_err());
assert!(info.apply(&insert_multiple!(id: 4, value: Value::IntT(2), count: 3)).is_err());
Ok(())
}
#[fuchsia::test]
fn test_enum_values() {
assert_eq!(BlockType::IntValue.to_isize().unwrap(), ArrayType::Int.to_isize().unwrap());
assert_eq!(BlockType::UintValue.to_isize().unwrap(), ArrayType::Uint.to_isize().unwrap());
assert_eq!(
BlockType::DoubleValue.to_isize().unwrap(),
ArrayType::Double.to_isize().unwrap()
);
}
#[fuchsia::test]
fn test_create_node_checks() {
let mut data = Data::new();
assert!(data.apply(&create_node!(parent: 0, id: 1, name: "first")).is_ok());
assert!(data.apply(&create_node!(parent: 0, id: 2, name: "second")).is_ok());
assert!(data.apply(&create_node!(parent: 1, id: 3, name: "child")).is_ok());
assert!(data.apply(&create_node!(parent: 0, id: 2, name: "double")).is_err());
let mut data = Data::new();
assert!(data.apply(&create_node!(parent: 1, id: 2, name: "orphan")).is_err());
}
#[fuchsia::test]
fn test_delete_node_checks() {
let mut data = Data::new();
assert!(data.apply(&delete_node!(id: 0)).is_err());
let mut data = Data::new();
data.apply(&create_node!(parent: 0, id: 1, name: "first")).ok();
assert!(data.apply(&delete_node!(id: 1)).is_ok());
assert!(data.apply(&delete_node!(id: 1)).is_err());
}
#[fuchsia::test]
fn test_node_tombstoning() {
let mut data = Data::new();
assert!(data.apply(&create_node!(parent: 0, id: 1, name: "first")).is_ok());
assert!(data
.apply(&create_numeric_property!(parent: 1, id: 2,
name: "answer", value: Value::IntT(42)))
.is_ok());
assert!(data.apply(&delete_node!(id: 1)).is_ok());
assert!(data.apply(&delete_property!(id: 2)).is_ok());
assert!(data.apply(&delete_property!(id: 2)).is_err());
let mut data = Data::new();
assert!(data.apply(&create_node!(parent: 0, id: 1, name: "first")).is_ok());
assert!(data
.apply(&create_numeric_property!(parent: 1, id: 2,
name: "answer", value: Value::IntT(42)))
.is_ok());
assert!(data.apply(&delete_node!(id: 1)).is_ok());
assert!(data.apply(&delete_property!(id: 2)).is_ok());
assert!(data
.apply(&create_numeric_property!(parent: 0, id: 2,
name: "root_answer", value: Value::IntT(42)))
.is_ok());
let mut data = Data::new();
assert!(data.apply(&create_node!(parent: 0, id: 1, name: "first")).is_ok());
assert!(data
.apply(&create_numeric_property!(parent: 1, id: 2,
name: "answer", value: Value::IntT(42)))
.is_ok());
assert!(data.apply(&delete_node!(id: 1)).is_ok());
assert!(data
.apply(&create_numeric_property!(parent: 0, id: 2,
name: "root_answer", value: Value::IntT(42)))
.is_err());
}
#[fuchsia::test]
fn test_property_tombstoning() {
let mut data = Data::new();
assert!(data.apply(&create_node!(parent: 0, id: 1, name: "first")).is_ok());
assert!(data.apply(&create_node!(parent: 1, id: 2, name: "second")).is_ok());
assert!(data.apply(&delete_node!(id: 1)).is_ok());
assert!(data.apply(&delete_node!(id: 2)).is_ok());
assert!(data.apply(&delete_node!(id: 2)).is_err());
let mut data = Data::new();
assert!(data.apply(&create_node!(parent: 0, id: 1, name: "first")).is_ok());
assert!(data.apply(&create_node!(parent: 1, id: 2, name: "second")).is_ok());
assert!(data.apply(&delete_node!(id: 1)).is_ok());
assert!(data.apply(&delete_node!(id: 2)).is_ok());
assert!(data.apply(&create_node!(parent: 0, id: 2, name: "new_root_second")).is_ok());
let mut data = Data::new();
assert!(data.apply(&create_node!(parent: 0, id: 1, name: "first")).is_ok());
assert!(data.apply(&create_node!(parent: 1, id: 2, name: "second")).is_ok());
assert!(data.apply(&delete_node!(id: 1)).is_ok());
assert!(data.apply(&create_node!(parent: 0, id: 2, name: "new_root_second")).is_err());
}
const DIFF_STRING: &str = r#"-- DIFF --
same: "root ->"
same: "> node ->"
local: "> > prop1: String(\"foo\")"
other: "> > prop1: String(\"bar\")""#;
const FULL_STRING: &str = r#"-- LOCAL --
root ->
> node ->
> > prop1: String("foo")
-- OTHER --
root ->
> node ->
> > prop1: String("bar")"#;
#[fuchsia::test]
fn diff_modes_work() -> Result<(), Error> {
let mut local = Data::new();
let mut remote = Data::new();
local.apply(&create_node!(parent: 0, id: 1, name: "node"))?;
local.apply(&create_string_property!(parent: 1, id: 2, name: "prop1", value: "foo"))?;
remote.apply(&create_node!(parent: 0, id: 1, name: "node"))?;
remote.apply(&create_string_property!(parent: 1, id: 2, name: "prop1", value: "bar"))?;
match local.compare(&remote, DiffType::Diff) {
Err(error) => {
let error_string = format!("{:?}", error);
assert_eq!("Trees differ:\n".to_string() + DIFF_STRING, error_string);
}
_ => return Err(format_err!("Didn't get failure")),
}
match local.compare(&remote, DiffType::Full) {
Err(error) => {
let error_string = format!("{:?}", error);
assert_eq!("Trees differ:\n".to_string() + FULL_STRING, error_string);
}
_ => return Err(format_err!("Didn't get failure")),
}
match local.compare(&remote, DiffType::Both) {
Err(error) => {
let error_string = format!("{:?}", error);
assert_eq!(["Trees differ:", FULL_STRING, DIFF_STRING].join("\n"), error_string);
}
_ => return Err(format_err!("Didn't get failure")),
}
Ok(())
}
}