diff --git a/src/gen/mod.rs b/src/gen/mod.rs index 84565f4a3..68ca4d1b0 100644 --- a/src/gen/mod.rs +++ b/src/gen/mod.rs @@ -103,7 +103,7 @@ pub fn generate(component: Component) -> Result<()> { println!("{}", scaffold::generate::(&rrgen, &name, &fields)?); } Component::Controller { name } => { - let vars = json!({ "name": name }); + let vars = json!({ "name": name, "pkg_name": H::app_name()}); rrgen.generate(CONTROLLER_T, &vars)?; rrgen.generate(CONTROLLER_TEST_T, &vars)?; } diff --git a/src/gen/scaffold.rs b/src/gen/scaffold.rs index e2de1d559..fbe982ba8 100644 --- a/src/gen/scaffold.rs +++ b/src/gen/scaffold.rs @@ -49,7 +49,7 @@ pub fn generate( } } - let vars = json!({"name": name, "columns": columns}); + let vars = json!({"name": name, "columns": columns, "pkg_name": H::app_name()}); let res1 = rrgen.generate(CONTROLLER_SCAFFOLD_T, &vars)?; let res2 = rrgen.generate(CONTROLLER_TEST_T, &vars)?; let messages = collect_messages(vec![res1, res2]); diff --git a/src/gen/templates/request_test.t b/src/gen/templates/request_test.t index 197696641..51a045a9f 100644 --- a/src/gen/templates/request_test.t +++ b/src/gen/templates/request_test.t @@ -8,7 +8,7 @@ injections: append: true content: "pub mod {{ file_name }};" --- -use blo::app::App; +use {{pkg_name}}::app::App; use migration::Migrator; use loco_rs::testing; use serial_test::serial;