use std::fmt::Display;
use syn::Ident;
pub trait IdentExt {
fn append<S>(&self, suffix: S) -> Ident
where
S: Display;
fn prepend<S>(&self, prefix: S) -> Ident
where
S: Display;
}
impl IdentExt for Ident {
fn append<S>(&self, suffix: S) -> Ident
where
S: Display,
{
let appended = format!("{}{}", self, suffix);
Ident::new(&appended, self.span())
}
fn prepend<S>(&self, suffix: S) -> Ident
where
S: Display,
{
let prepended = format!("{}{}", suffix, self);
Ident::new(&prepended, self.span())
}
}
#[cfg(test)]
mod tests {
use proc_macro2::Span;
use syn::Ident;
use super::IdentExt;
#[test]
fn append_str_returns_appended_ident() {
let one = Ident::new("One", Span::call_site());
assert_eq!(Ident::new("OneTwo", Span::call_site()), one.append("Two"));
}
#[test]
fn append_ident_returns_appended_ident() {
let one = Ident::new("One", Span::call_site());
let two = Ident::new("Two", Span::call_site());
assert_eq!(Ident::new("OneTwo", Span::call_site()), one.append(two));
}
#[test]
fn append_ident_ref_returns_appended_ident() {
let one = Ident::new("One", Span::call_site());
let two = Ident::new("Two", Span::call_site());
assert_eq!(Ident::new("OneTwo", Span::call_site()), one.append(&two));
}
#[test]
fn prepend_str_returns_prepended_ident() {
let one = Ident::new("One", Span::call_site());
assert_eq!(Ident::new("TwoOne", Span::call_site()), one.prepend("Two"));
}
#[test]
fn prepend_ident_returns_prepended_ident() {
let one = Ident::new("One", Span::call_site());
let two = Ident::new("Two", Span::call_site());
assert_eq!(Ident::new("TwoOne", Span::call_site()), one.prepend(two));
}
#[test]
fn prepend_ident_ref_returns_prepended_ident() {
let one = Ident::new("One", Span::call_site());
let two = Ident::new("Two", Span::call_site());
assert_eq!(Ident::new("TwoOne", Span::call_site()), one.prepend(&two));
}
}