logic_mesh/blocks/collections/
values.rs

1// Copyright (c) 2022-2023, Radu Racariu.
2
3use uuid::Uuid;
4
5use crate::base::{
6    block::{Block, BlockDesc, BlockProps, BlockState},
7    input::{input_reader::InputReader, Input, InputProps},
8    output::Output,
9};
10
11use libhaystack::val::{kind::HaystackKind, Value};
12
13use crate::{blocks::InputImpl, blocks::OutputImpl};
14
15/// Outputs the values of a dictionary.
16#[block]
17#[derive(BlockProps, Debug)]
18#[category = "collections"]
19pub struct Values {
20    #[input(name = "input", kind = "Dict")]
21    pub input: InputImpl,
22    #[output(kind = "List")]
23    pub out: OutputImpl,
24}
25
26impl Block for Values {
27    async fn execute(&mut self) {
28        self.read_inputs_until_ready().await;
29
30        match self.input.get_value() {
31            Some(Value::Dict(dict)) => self
32                .out
33                .set(dict.values().cloned().collect::<Vec<_>>().into()),
34            _ => self.out.set(Vec::default().into()),
35        }
36    }
37}
38
39#[cfg(test)]
40mod test {
41    use libhaystack::dict;
42    use libhaystack::val::Dict;
43
44    use crate::{
45        base::block::test_utils::write_block_inputs, base::block::Block,
46        blocks::collections::Values,
47    };
48
49    #[tokio::test]
50    async fn test_values_block() {
51        let mut block = Values::new();
52
53        write_block_inputs(&mut [(&mut block.input, (Dict::default()).into())]).await;
54        block.execute().await;
55        assert_eq!(block.out.value, vec![].into());
56
57        write_block_inputs(&mut [(&mut block.input, (dict! {"a" => 1.into()}).into())]).await;
58        block.execute().await;
59        assert_eq!(block.out.value, vec![1.into()].into());
60    }
61}
logic_mesh/blocks/collections/values.rs

logic_mesh/blocks/collections/
values.rs
