diff options
Diffstat (limited to 'macros/src/injectable/implementation.rs')
| -rw-r--r-- | macros/src/injectable/implementation.rs | 835 |
1 files changed, 683 insertions, 152 deletions
diff --git a/macros/src/injectable/implementation.rs b/macros/src/injectable/implementation.rs index c907f32..9542a98 100644 --- a/macros/src/injectable/implementation.rs +++ b/macros/src/injectable/implementation.rs @@ -1,10 +1,11 @@ use std::error::Error; +use proc_macro2::Ident; use quote::{format_ident, quote, ToTokens}; use syn::parse::{Parse, ParseStream}; -use syn::{parse_str, ExprMethodCall, FnArg, Generics, ItemImpl, Type}; +use syn::{parse_str, ExprMethodCall, FnArg, Generics, ImplItemMethod, ItemImpl, Type}; -use crate::injectable::dependency::Dependency; +use crate::injectable::dependency::IDependency; use crate::util::item_impl::find_impl_method_by_name_mut; use crate::util::string::camelcase_to_snakecase; use crate::util::syn_path::syn_path_to_string; @@ -12,131 +13,144 @@ use crate::util::syn_path::syn_path_to_string; const DI_CONTAINER_VAR_NAME: &str = "di_container"; const DEPENDENCY_HISTORY_VAR_NAME: &str = "dependency_history"; -pub struct InjectableImpl +pub struct InjectableImpl<Dep: IDependency> { - pub dependencies: Vec<Dependency>, + pub dependencies: Vec<Dep>, pub self_type: Type, pub generics: Generics, pub original_impl: ItemImpl, } -impl Parse for InjectableImpl +impl<Dep: IDependency> Parse for InjectableImpl<Dep> { + #[cfg(not(tarpaulin_include))] fn parse(input: ParseStream) -> syn::Result<Self> { - let mut impl_parsed_input = input.parse::<ItemImpl>()?; + let input_fork = input.fork(); - let dependencies = Self::build_dependencies(&mut impl_parsed_input) - .map_err(|err| input.error(err))?; + let mut item_impl = input.parse::<ItemImpl>()?; + + let new_method = find_impl_method_by_name_mut(&mut item_impl, "new") + .map_or_else(|| Err(input_fork.error("Missing a 'new' method")), Ok)?; + + let dependencies = + Self::build_dependencies(new_method).map_err(|err| input.error(err))?; + + Self::remove_method_argument_attrs(new_method); Ok(Self { dependencies, - self_type: impl_parsed_input.self_ty.as_ref().clone(), - generics: impl_parsed_input.generics.clone(), - original_impl: impl_parsed_input, + self_type: item_impl.self_ty.as_ref().clone(), + generics: item_impl.generics.clone(), + original_impl: item_impl, }) } } -impl InjectableImpl +impl<Dep: IDependency> InjectableImpl<Dep> { + #[cfg(not(tarpaulin_include))] pub fn expand(&self, no_doc_hidden: bool, is_async: bool) -> proc_macro2::TokenStream { - let Self { - dependencies, - self_type, - generics, - original_impl, - } = self; - let di_container_var = format_ident!("{}", DI_CONTAINER_VAR_NAME); let dependency_history_var = format_ident!("{}", DEPENDENCY_HISTORY_VAR_NAME); let maybe_doc_hidden = if no_doc_hidden { quote! {} } else { - quote! { - #[doc(hidden)] - } + quote! { #[doc(hidden)] } }; let maybe_prevent_circular_deps = if cfg!(feature = "prevent-circular") { - quote! { - if #dependency_history_var.contains(&self_type_name) { - #dependency_history_var.push(self_type_name); - - let dependency_trace = - syrette::dependency_trace::create_dependency_trace( - #dependency_history_var.as_slice(), - self_type_name - ); - - return Err(InjectableError::DetectedCircular {dependency_trace }); - } - - #dependency_history_var.push(self_type_name); - } + Self::expand_prevent_circular_deps(&dependency_history_var) } else { quote! {} }; let injectable_impl = if is_async { - let async_get_dep_method_calls = - Self::create_get_dep_method_calls(dependencies, true); + self.expand_async_impl( + &maybe_doc_hidden, + &di_container_var, + &dependency_history_var, + &maybe_prevent_circular_deps, + &Self::create_get_dep_method_calls(&self.dependencies, true).unwrap(), + ) + } else { + self.expand_blocking_impl( + &maybe_doc_hidden, + &di_container_var, + &dependency_history_var, + &maybe_prevent_circular_deps, + &Self::create_get_dep_method_calls(&self.dependencies, false).unwrap(), + ) + }; - quote! { - #maybe_doc_hidden - impl #generics syrette::interfaces::async_injectable::AsyncInjectable for #self_type - { - fn resolve<'di_container, 'fut>( - #di_container_var: &'di_container std::sync::Arc< - syrette::AsyncDIContainer - >, - mut #dependency_history_var: Vec<&'static str>, - ) -> syrette::future::BoxFuture< - 'fut, - Result< - syrette::ptr::TransientPtr<Self>, - syrette::errors::injectable::InjectableError - > - > - where - Self: Sized + 'fut, - 'di_container: 'fut - { - Box::pin(async move { - use std::any::type_name; + let original_impl = &self.original_impl; - use syrette::errors::injectable::InjectableError; + quote! { + #original_impl - let self_type_name = type_name::<#self_type>(); + #injectable_impl + } + } - #maybe_prevent_circular_deps + #[cfg(not(tarpaulin_include))] + fn expand_prevent_circular_deps( + dependency_history_var: &Ident, + ) -> proc_macro2::TokenStream + { + quote! { + if #dependency_history_var.contains(&self_type_name) { + #dependency_history_var.push(self_type_name); - Ok(syrette::ptr::TransientPtr::new(Self::new( - #(#async_get_dep_method_calls),* - ))) - }) - } - } + let dependency_trace = + syrette::dependency_trace::create_dependency_trace( + #dependency_history_var.as_slice(), + self_type_name + ); + return Err(InjectableError::DetectedCircular {dependency_trace }); } - } else { - let get_dep_method_calls = - Self::create_get_dep_method_calls(dependencies, false); - quote! { - #maybe_doc_hidden - impl #generics syrette::interfaces::injectable::Injectable for #self_type - { - fn resolve( - #di_container_var: &std::rc::Rc<syrette::DIContainer>, - mut #dependency_history_var: Vec<&'static str>, - ) -> Result< + #dependency_history_var.push(self_type_name); + } + } + + #[cfg(not(tarpaulin_include))] + fn expand_async_impl( + &self, + maybe_doc_hidden: &proc_macro2::TokenStream, + di_container_var: &Ident, + dependency_history_var: &Ident, + maybe_prevent_circular_deps: &proc_macro2::TokenStream, + get_dep_method_calls: &Vec<proc_macro2::TokenStream>, + ) -> proc_macro2::TokenStream + { + let generics = &self.generics; + let self_type = &self.self_type; + + quote! { + #maybe_doc_hidden + impl #generics syrette::interfaces::async_injectable::AsyncInjectable for #self_type + { + fn resolve<'di_container, 'fut>( + #di_container_var: &'di_container std::sync::Arc< + syrette::AsyncDIContainer + >, + mut #dependency_history_var: Vec<&'static str>, + ) -> syrette::future::BoxFuture< + 'fut, + Result< syrette::ptr::TransientPtr<Self>, - syrette::errors::injectable::InjectableError> - { + syrette::errors::injectable::InjectableError + > + > + where + Self: Sized + 'fut, + 'di_container: 'fut + { + Box::pin(async move { use std::any::type_name; use syrette::errors::injectable::InjectableError; @@ -145,99 +159,147 @@ impl InjectableImpl #maybe_prevent_circular_deps - return Ok(syrette::ptr::TransientPtr::new(Self::new( + Ok(syrette::ptr::TransientPtr::new(Self::new( #(#get_dep_method_calls),* - ))); - } + ))) + }) } } - }; + } + } + + #[cfg(not(tarpaulin_include))] + fn expand_blocking_impl( + &self, + maybe_doc_hidden: &proc_macro2::TokenStream, + di_container_var: &Ident, + dependency_history_var: &Ident, + maybe_prevent_circular_deps: &proc_macro2::TokenStream, + get_dep_method_calls: &Vec<proc_macro2::TokenStream>, + ) -> proc_macro2::TokenStream + { + let generics = &self.generics; + let self_type = &self.self_type; quote! { - #original_impl + #maybe_doc_hidden + impl #generics syrette::interfaces::injectable::Injectable for #self_type + { + fn resolve( + #di_container_var: &std::rc::Rc<syrette::DIContainer>, + mut #dependency_history_var: Vec<&'static str>, + ) -> Result< + syrette::ptr::TransientPtr<Self>, + syrette::errors::injectable::InjectableError> + { + use std::any::type_name; - #injectable_impl + use syrette::errors::injectable::InjectableError; + + let self_type_name = type_name::<#self_type>(); + + #maybe_prevent_circular_deps + + return Ok(syrette::ptr::TransientPtr::new(Self::new( + #(#get_dep_method_calls),* + ))); + } + } } } fn create_get_dep_method_calls( - dependencies: &[Dependency], + dependencies: &[Dep], is_async: bool, - ) -> Vec<proc_macro2::TokenStream> + ) -> Result<Vec<proc_macro2::TokenStream>, Box<dyn Error>> { dependencies .iter() .filter_map(|dependency| { - let dep_interface_str = match &dependency.interface { - Type::TraitObject(interface_trait) => { - Some(interface_trait.to_token_stream().to_string()) - } - Type::Path(path_interface) => { - Some(syn_path_to_string(&path_interface.path)) - } - &_ => None, + match dependency.get_interface() { + Type::TraitObject(_) | Type::Path(_) => Some(()), + _ => None, }?; - let method_call = parse_str::<ExprMethodCall>( - format!( - "{}.get_bound::<{}>({}.clone(), {})", - DI_CONTAINER_VAR_NAME, - dep_interface_str, - DEPENDENCY_HISTORY_VAR_NAME, - dependency.name.as_ref().map_or_else( - || "None".to_string(), - |name| format!("Some(\"{}\")", name.value()) - ) - ) - .as_str(), - ) - .ok()?; - - Some((method_call, dependency)) - }) - .map(|(method_call, dep_type)| { - let ptr_name = dep_type.ptr.to_string(); - - let to_ptr = format_ident!( - "{}", - camelcase_to_snakecase(&ptr_name.replace("Ptr", "")) - ); - - let do_method_call = if is_async { - quote! { #method_call.await } - } else { - quote! { #method_call } - }; - - let resolve_failed_error = if is_async { - quote! { InjectableError::AsyncResolveFailed } - } else { - quote! { InjectableError::ResolveFailed } - }; - - quote! { - #do_method_call.map_err(|err| #resolve_failed_error { - reason: Box::new(err), - affected: self_type_name - })?.#to_ptr().unwrap() - } + Some(Self::create_single_get_dep_method_call( + dependency, is_async, + )) }) .collect() } - fn build_dependencies( - item_impl: &mut ItemImpl, - ) -> Result<Vec<Dependency>, Box<dyn Error>> + fn create_single_get_dep_method_call( + dependency: &Dep, + is_async: bool, + ) -> Result<proc_macro2::TokenStream, Box<dyn Error>> + { + let dep_interface_str = match &dependency.get_interface() { + Type::TraitObject(interface_trait) => { + Ok(interface_trait.to_token_stream().to_string()) + } + Type::Path(path_interface) => Ok(syn_path_to_string(&path_interface.path)), + &_ => Err("Invalid type. Expected trait type or path type"), + }?; + + let method_call = parse_str::<ExprMethodCall>( + format!( + "{}.get_bound::<{}>({}.clone(), {})", + DI_CONTAINER_VAR_NAME, + dep_interface_str, + DEPENDENCY_HISTORY_VAR_NAME, + dependency.get_name().as_ref().map_or_else( + || "None".to_string(), + |name| format!("Some(\"{}\")", name.value()) + ) + ) + .as_str(), + )?; + + let ptr_name = dependency.get_ptr().to_string(); + + let to_ptr = + format_ident!("{}", camelcase_to_snakecase(&ptr_name.replace("Ptr", ""))); + + let do_method_call = if is_async { + quote! { #method_call.await } + } else { + quote! { #method_call } + }; + + let resolve_failed_error = if is_async { + quote! { InjectableError::AsyncResolveFailed } + } else { + quote! { InjectableError::ResolveFailed } + }; + + Ok(quote! { + #do_method_call.map_err(|err| #resolve_failed_error { + reason: Box::new(err), + affected: self_type_name + })?.#to_ptr().unwrap() + }) + } + + fn build_dependencies(new_method: &ImplItemMethod) + -> Result<Vec<Dep>, Box<dyn Error>> { - let new_method_impl_item = find_impl_method_by_name_mut(item_impl, "new") - .map_or_else(|| Err("Missing a 'new' method"), Ok)?; + let new_method_args = &new_method.sig.inputs; - let new_method_args = &mut new_method_impl_item.sig.inputs; + let dependencies_result: Result<Vec<_>, _> = + new_method_args.iter().map(Dep::build).collect(); - let dependencies: Result<Vec<_>, _> = - new_method_args.iter().map(Dependency::build).collect(); + let deps = dependencies_result?; - for arg in new_method_args { + Ok(deps) + } + + // Removes argument attributes from a method, as they are not actually valid Rust. + // Not doing this would cause a compilation error. + fn remove_method_argument_attrs(method: &mut ImplItemMethod) + { + let method_args = &mut method.sig.inputs; + + for arg in method_args { let typed_arg = if let FnArg::Typed(typed_arg) = arg { typed_arg } else { @@ -265,7 +327,476 @@ impl InjectableImpl typed_arg.attrs.remove(attr_index); } } + } +} + +#[cfg(test)] +mod tests +{ + use std::sync::{Mutex, MutexGuard}; + + use once_cell::sync::Lazy; + use pretty_assertions::assert_eq; + use proc_macro2::{Span, TokenStream}; + use syn::token::{Brace, Bracket, Colon, Paren, Pound}; + use syn::{ + parse2, + AttrStyle, + Attribute, + Block, + ImplItemMethod, + LitStr, + Pat, + PatType, + Visibility, + }; + + use super::*; + use crate::injectable::dependency::MockIDependency; + use crate::injectable::named_attr_input::NamedAttrInput; + use crate::test_utils; + + static TEST_MUTEX: Lazy<Mutex<()>> = Lazy::new(|| Mutex::new(())); + + // When a test panics, it will poison the Mutex. Since we don't actually + // care about the state of the data we ignore that it is poisoned and grab + // the lock regardless. If you just do `let _lock = &TEST_MUTEX.lock().unwrap()`, one + // test panicking will cause all other tests that try and acquire a lock on + // that Mutex to also panic. + fn get_lock(m: &'static Mutex<()>) -> MutexGuard<'static, ()> + { + match m.lock() { + Ok(guard) => guard, + Err(poisoned) => poisoned.into_inner(), + } + } - dependencies + #[test] + fn can_build_dependencies() -> Result<(), Box<dyn Error>> + { + let method = ImplItemMethod { + attrs: vec![], + vis: Visibility::Inherited, + defaultness: None, + sig: test_utils::create_signature( + format_ident!("new"), + vec![ + ( + test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment( + format_ident!("TransientPtr"), + &[test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment( + format_ident!("Foo"), + &[], + ), + ]))], + ), + ])), + vec![], + ), + ( + test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment( + format_ident!("FactoryPtr"), + &[test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment( + format_ident!("BarFactory"), + &[], + ), + ]))], + ), + ])), + vec![], + ), + ], + test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment(format_ident!("Self"), &[]), + ])), + ), + block: Block { + brace_token: Brace::default(), + stmts: vec![], + }, + }; + + let _lock = get_lock(&TEST_MUTEX); + + let build_context = MockIDependency::build_context(); + + build_context + .expect() + .returning(|_| Ok(MockIDependency::new())); + + let dependencies = + InjectableImpl::<MockIDependency>::build_dependencies(&method)?; + + assert_eq!(dependencies.len(), 2); + + Ok(()) + } + + #[test] + fn can_build_named_dependencies() -> Result<(), Box<dyn Error>> + { + let method = ImplItemMethod { + attrs: vec![], + vis: Visibility::Inherited, + defaultness: None, + sig: test_utils::create_signature( + format_ident!("new"), + vec![ + ( + test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment( + format_ident!("TransientPtr"), + &[test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment( + format_ident!("Foo"), + &[], + ), + ]))], + ), + ])), + vec![], + ), + ( + test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment( + format_ident!("FactoryPtr"), + &[test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment( + format_ident!("BarFactory"), + &[], + ), + ]))], + ), + ])), + vec![Attribute { + pound_token: Pound::default(), + style: AttrStyle::Outer, + bracket_token: Bracket::default(), + path: test_utils::create_path(&[ + test_utils::create_path_segment( + format_ident!("named"), + &[], + ), + ]), + tokens: NamedAttrInput { + paren: Paren::default(), + name: LitStr::new("awesome", Span::call_site()), + } + .to_token_stream(), + }], + ), + ], + test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment(format_ident!("Self"), &[]), + ])), + ), + block: Block { + brace_token: Brace::default(), + stmts: vec![], + }, + }; + + let _lock = get_lock(&TEST_MUTEX); + + let build_context = MockIDependency::build_context(); + + build_context + .expect() + .returning(|_| Ok(MockIDependency::new())) + .times(2); + + let dependencies = + InjectableImpl::<MockIDependency>::build_dependencies(&method)?; + + assert_eq!(dependencies.len(), 2); + + Ok(()) + } + + #[test] + fn can_remove_method_argument_attrs() + { + let first_arg_type = test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment( + format_ident!("TransientPtr"), + &[test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment(format_ident!("Foo"), &[]), + ]))], + ), + ])); + + let second_arg_type = test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment( + format_ident!("FactoryPtr"), + &[test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment(format_ident!("BarFactory"), &[]), + ]))], + ), + ])); + + let mut method = ImplItemMethod { + attrs: vec![], + vis: Visibility::Inherited, + defaultness: None, + sig: test_utils::create_signature( + format_ident!("new"), + vec![ + ( + first_arg_type.clone(), + vec![Attribute { + pound_token: Pound::default(), + style: AttrStyle::Outer, + bracket_token: Bracket::default(), + path: test_utils::create_path(&[ + test_utils::create_path_segment( + format_ident!("named"), + &[], + ), + ]), + tokens: NamedAttrInput { + paren: Paren::default(), + name: LitStr::new("cool", Span::call_site()), + } + .to_token_stream(), + }], + ), + ( + second_arg_type.clone(), + vec![Attribute { + pound_token: Pound::default(), + style: AttrStyle::Outer, + bracket_token: Bracket::default(), + path: test_utils::create_path(&[ + test_utils::create_path_segment( + format_ident!("named"), + &[], + ), + ]), + tokens: NamedAttrInput { + paren: Paren::default(), + name: LitStr::new("awesome", Span::call_site()), + } + .to_token_stream(), + }], + ), + ], + test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment(format_ident!("Self"), &[]), + ])), + ), + block: Block { + brace_token: Brace::default(), + stmts: vec![], + }, + }; + + InjectableImpl::<MockIDependency>::remove_method_argument_attrs(&mut method); + + assert_eq!( + method.sig.inputs.first().unwrap().clone(), + FnArg::Typed(PatType { + attrs: vec![], + pat: Box::new(Pat::Verbatim(TokenStream::new())), + colon_token: Colon::default(), + ty: Box::new(first_arg_type), + }) + ); + + assert_eq!( + method.sig.inputs.last().unwrap().clone(), + FnArg::Typed(PatType { + attrs: vec![], + pat: Box::new(Pat::Verbatim(TokenStream::new())), + colon_token: Colon::default(), + ty: Box::new(second_arg_type), + }) + ); + } + + #[test] + fn can_create_single_get_dep_method_call() -> Result<(), Box<dyn Error>> + { + let mut mock_dependency = MockIDependency::new(); + + mock_dependency + .expect_get_interface() + .return_const(test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment(format_ident!("Foo"), &[]), + ]))); + + mock_dependency.expect_get_name().return_const(None); + + mock_dependency + .expect_get_ptr() + .return_const(format_ident!("TransientPtr")); + + let di_container_var_ident = format_ident!("{}", DI_CONTAINER_VAR_NAME); + let dep_history_var_ident = format_ident!("{}", DEPENDENCY_HISTORY_VAR_NAME); + + let output = + InjectableImpl::<MockIDependency>::create_single_get_dep_method_call( + &mock_dependency, + false, + )?; + + assert_eq!( + parse2::<ExprMethodCall>(output)?, + parse2::<ExprMethodCall>(quote! { + #di_container_var_ident + .get_bound::<Foo>(#dep_history_var_ident.clone(), None) + .map_err(|err| InjectableError::ResolveFailed { + reason: Box::new(err), + affected: self_type_name + })? + .transient() + .unwrap() + + })? + ); + + Ok(()) + } + + #[test] + fn can_create_single_get_dep_method_call_with_name() -> Result<(), Box<dyn Error>> + { + let mut mock_dependency = MockIDependency::new(); + + mock_dependency + .expect_get_interface() + .return_const(test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment(format_ident!("Foo"), &[]), + ]))); + + mock_dependency + .expect_get_name() + .return_const(Some(LitStr::new("special", Span::call_site()))); + + mock_dependency + .expect_get_ptr() + .return_const(format_ident!("TransientPtr")); + + let di_container_var_ident = format_ident!("{}", DI_CONTAINER_VAR_NAME); + let dep_history_var_ident = format_ident!("{}", DEPENDENCY_HISTORY_VAR_NAME); + + let output = + InjectableImpl::<MockIDependency>::create_single_get_dep_method_call( + &mock_dependency, + false, + )?; + + assert_eq!( + parse2::<ExprMethodCall>(output)?, + parse2::<ExprMethodCall>(quote! { + #di_container_var_ident + .get_bound::<Foo>(#dep_history_var_ident.clone(), Some("special")) + .map_err(|err| InjectableError::ResolveFailed { + reason: Box::new(err), + affected: self_type_name + })? + .transient() + .unwrap() + + })? + ); + + Ok(()) + } + + #[test] + fn can_create_single_get_dep_method_call_async() -> Result<(), Box<dyn Error>> + { + let mut mock_dependency = MockIDependency::new(); + + mock_dependency + .expect_get_interface() + .return_const(test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment(format_ident!("Foo"), &[]), + ]))); + + mock_dependency.expect_get_name().return_const(None); + + mock_dependency + .expect_get_ptr() + .return_const(format_ident!("TransientPtr")); + + let di_container_var_ident = format_ident!("{}", DI_CONTAINER_VAR_NAME); + let dep_history_var_ident = format_ident!("{}", DEPENDENCY_HISTORY_VAR_NAME); + + let output = + InjectableImpl::<MockIDependency>::create_single_get_dep_method_call( + &mock_dependency, + true, + )?; + + assert_eq!( + parse2::<ExprMethodCall>(output)?, + parse2::<ExprMethodCall>(quote! { + #di_container_var_ident + .get_bound::<Foo>(#dep_history_var_ident.clone(), None) + .await + .map_err(|err| InjectableError::AsyncResolveFailed { + reason: Box::new(err), + affected: self_type_name + })? + .transient() + .unwrap() + + })? + ); + + Ok(()) + } + + #[test] + fn can_create_single_get_dep_method_call_async_with_name( + ) -> Result<(), Box<dyn Error>> + { + let mut mock_dependency = MockIDependency::new(); + + mock_dependency + .expect_get_interface() + .return_const(test_utils::create_type(test_utils::create_path(&[ + test_utils::create_path_segment(format_ident!("Foo"), &[]), + ]))); + + mock_dependency + .expect_get_name() + .return_const(Some(LitStr::new("foobar", Span::call_site()))); + + mock_dependency + .expect_get_ptr() + .return_const(format_ident!("TransientPtr")); + + let di_container_var_ident = format_ident!("{}", DI_CONTAINER_VAR_NAME); + let dep_history_var_ident = format_ident!("{}", DEPENDENCY_HISTORY_VAR_NAME); + + let output = + InjectableImpl::<MockIDependency>::create_single_get_dep_method_call( + &mock_dependency, + true, + )?; + + assert_eq!( + parse2::<ExprMethodCall>(output)?, + parse2::<ExprMethodCall>(quote! { + #di_container_var_ident + .get_bound::<Foo>(#dep_history_var_ident.clone(), Some("foobar")) + .await + .map_err(|err| InjectableError::AsyncResolveFailed { + reason: Box::new(err), + affected: self_type_name + })? + .transient() + .unwrap() + + })? + ); + + Ok(()) } } |