1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
use crate::{Driver, TypeGenerator};
use alloc::{boxed::Box, string::String, vec::Vec};

impl<T: TypeGenerator> TypeGenerator for Box<T> {
    fn generate<D: Driver>(driver: &mut D) -> Option<Self> {
        Some(Box::new(driver.gen()?))
    }

    fn mutate<D: Driver>(&mut self, driver: &mut D) -> Option<()> {
        self.as_mut().mutate(driver)
    }
}

impl<T: TypeGenerator> TypeGenerator for Box<[T]> {
    fn generate<D: Driver>(driver: &mut D) -> Option<Self> {
        Some(driver.gen::<Vec<T>>()?.into_boxed_slice())
    }

    // TODO mutate
}

impl TypeGenerator for Box<str> {
    fn generate<D: Driver>(driver: &mut D) -> Option<Self> {
        Some(driver.gen::<String>()?.into_boxed_str())
    }

    // TODO mutate
}