use rust_writer::ast::{
finder::{Finder, ToFind},
implementors::{ItemToFile, ItemToImpl},
mutator::{Mutator, ToMutate},
};
use rust_writer_proc::{finder, mutator};
use syn::{parse_quote, visit_mut::VisitMut, ImplItem};
use test_builder::TestBuilder;
#[mutator(ItemToFile, ItemToImpl<'a>)]
#[finder(ItemToFile, ItemToImpl<'a>)]
#[impl_from]
struct SomeStruct;
#[test]
fn modified_unit_struct() {
TestBuilder::default().with_trait_and_impl_block_ast().execute(|mut builder| {
let item_to_file = ItemToFile {
item: parse_quote!(
use std::path::Path;
),
};
let item_to_impl: ItemToImpl = (
Some("SomeTrait"),
"SomeImplementor",
ImplItem::Fn(parse_quote! {
fn other_func(&self) -> bool{
false
}
}),
)
.into();
let some_struct: SomeStruct = (item_to_file, item_to_impl).into();
let ast = builder.get_mut_ast_file("trait_and_impl_block.rs").expect("This should exist");
let mut finder: SomeStructFinderWrapper = Finder::default().to_find(&some_struct).into();
assert!(!finder.find(ast, None));
let mut mutator: SomeStructMutatorWrapper =
Mutator::default().to_mutate(&some_struct).into();
assert!(mutator.mutate(ast, None).is_ok());
let mut finder: SomeStructFinderWrapper = Finder::default().to_find(&some_struct).into();
assert!(finder.find(ast, None));
});
}