add an attribute to inner doctest fn

This commit is contained in:
Arpad Borsos
2020-12-20 10:00:32 +01:00
parent 830ceaa419
commit f9fa3fe65f
2 changed files with 7 additions and 6 deletions

View File

@@ -551,17 +551,18 @@ crate fn make_test(
} else {
"_inner".into()
};
let inner_attr = if test_id.is_some() { "#[allow(non_snake_case)] " } else { "" };
let (main_pre, main_post) = if returns_result {
(
format!(
"fn main() {{ fn {}() -> Result<(), impl core::fmt::Debug> {{\n",
inner_fn_name
"fn main() {{ {}fn {}() -> Result<(), impl core::fmt::Debug> {{\n",
inner_attr, inner_fn_name
),
format!("\n}}; {}().unwrap() }}", inner_fn_name),
)
} else if test_id.is_some() {
(
format!("fn main() {{ fn {}() {{\n", inner_fn_name),
format!("fn main() {{ {}fn {}() {{\n", inner_attr, inner_fn_name),
format!("\n}}; {}() }}", inner_fn_name),
)
} else {