use super::*;
#[derive(Clone, Debug, Default)]
pub struct LinkedList<M> {
mutator: M,
}
pub fn linked_list<M>(mutator: M) -> LinkedList<M> {
LinkedList { mutator }
}
impl<M, T> Mutate<alloc::collections::LinkedList<T>> for LinkedList<M>
where
M: Generate<T> + Mutate<T>,
{
#[inline]
fn mutation_count(
&self,
value: &alloc::collections::LinkedList<T>,
shrink: bool,
) -> core::option::Option<u32> {
let mut count = 0u32;
count += !shrink as u32;
count += !value.is_empty() as u32;
for x in value.iter() {
count += self.mutator.mutation_count(x, shrink)?;
}
Some(count)
}
#[inline]
fn mutate(
&mut self,
c: &mut Candidates,
value: &mut alloc::collections::LinkedList<T>,
) -> Result<()> {
if !c.shrink() {
c.mutation(|ctx| {
let elem = self.mutator.generate(ctx)?;
if ctx.rng().gen_bool() {
value.push_front(elem);
} else {
value.push_back(elem);
}
Ok(())
})?;
}
if !value.is_empty() {
c.mutation(|ctx| {
if ctx.rng().gen_bool() {
value.pop_front();
} else {
value.pop_back();
}
Ok(())
})?;
}
for x in value.iter_mut() {
self.mutator.mutate(c, x)?;
}
Ok(())
}
}
impl<M, T> Generate<alloc::collections::LinkedList<T>> for LinkedList<M>
where
M: Generate<T> + Mutate<T>,
{
#[inline]
fn generate(&mut self, ctx: &mut Context) -> Result<alloc::collections::LinkedList<T>> {
self.generate_via_mutate(ctx, 1)
}
}
impl<T> DefaultMutate for alloc::collections::LinkedList<T>
where
T: DefaultMutate,
T::DefaultMutate: Generate<T>,
{
type DefaultMutate = LinkedList<T::DefaultMutate>;
}