zerokms_protocol/
testing.rs

1//! Implementations of `Dummy` for various types in the `zerokms-protocol` crate.
2use fake::{
3    faker::lorem::en::{Word, Words},
4    Dummy, Fake, Faker,
5};
6use rand::{Fill, Rng};
7use std::borrow::Cow;
8use uuid::Uuid;
9
10use crate::{
11    CreateClientRequest, CreateDatasetRequest, Dataset, GenerateKeyRequest, GenerateKeySpec,
12    GrantDatasetRequest,
13};
14
15/// Generate a random `CreateDatasetRequest` with a random description and the given name.
16impl Dummy<&'static str> for CreateDatasetRequest<'_> {
17    fn dummy_with_rng<R: Rng + ?Sized>(name: &&'static str, _: &mut R) -> Self {
18        Self {
19            name: Cow::Borrowed(*name),
20            description: Cow::Owned("description".to_string()),
21        }
22    }
23}
24
25impl Dummy<Faker> for Dataset {
26    fn dummy_with_rng<R: Rng + ?Sized>(_: &Faker, _: &mut R) -> Self {
27        Self {
28            id: Uuid::new_v4(),
29            name: Word().fake(),
30            description: Words(3..10).fake::<Vec<String>>().join(" "),
31            is_disabled: false,
32        }
33    }
34}
35
36/// Generate a random `CreateDatasetRequest` with a random name and description.
37impl Dummy<Faker> for CreateDatasetRequest<'_> {
38    fn dummy_with_rng<R: Rng + ?Sized>(_: &Faker, _: &mut R) -> Self {
39        let words: Vec<String> = Words(3..10).fake();
40
41        Self {
42            name: Cow::Owned(Word().fake()),
43            description: Cow::Owned(words.join(" ")),
44        }
45    }
46}
47
48/// Generate a random `CreateClientRequest` for the given `Dataset`.
49/// The name and description are random.
50impl Dummy<Dataset> for CreateClientRequest<'_> {
51    fn dummy_with_rng<R: Rng + ?Sized>(Dataset { id, .. }: &Dataset, _: &mut R) -> Self {
52        Self {
53            dataset_id: *id,
54            name: Cow::Owned(Word().fake()),
55            description: Cow::Owned(Word().fake()),
56        }
57    }
58}
59
60/// Generate a `GrantDatasetRequest` for the given `Dataset` and client ID (`Uuid`).
61impl Dummy<(&Dataset, Uuid)> for GrantDatasetRequest {
62    fn dummy_with_rng<R: Rng + ?Sized>((dataset, client_id): &(&Dataset, Uuid), _: &mut R) -> Self {
63        Self {
64            dataset_id: dataset.id,
65            client_id: *client_id,
66        }
67    }
68}
69
70impl Dummy<(&Dataset, Uuid)> for GenerateKeyRequest<'_> {
71    fn dummy_with_rng<R: Rng + ?Sized>((dataset, client_id): &(&Dataset, Uuid), _: &mut R) -> Self {
72        Self {
73            dataset_id: Some(dataset.id),
74            client_id: *client_id,
75            keys: Cow::Owned(fake::vec![GenerateKeySpec; 4]),
76        }
77    }
78}
79
80impl Dummy<Faker> for GenerateKeySpec<'_> {
81    fn dummy_with_rng<R: Rng + ?Sized>(_: &Faker, rng: &mut R) -> Self {
82        let mut buf: [u8; 16] = Default::default();
83        buf.try_fill(rng).unwrap();
84
85        Self {
86            iv: buf.into(),
87            descriptor: Cow::Owned(Word().fake()),
88            context: vec![],
89        }
90    }
91}
zerokms_protocol/testing.rs

zerokms_protocol/
testing.rs
