Billy Chan 1054d6451c
Use file!() macro to avoid manually defining the name of migration (#736)
* Add `DeriveMigrationName`

* CLI migrate generate without replacing migration name

* Update migrations

* Fixup
2022-06-29 00:42:59 +08:00

37 lines
945 B
Rust

pub fn get_file_stem(path: &str) -> &str {
std::path::Path::new(path)
.file_stem()
.map(|f| f.to_str().unwrap())
.unwrap()
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_get_file_stem() {
let pair = vec![
(
"m20220101_000001_create_table.rs",
"m20220101_000001_create_table",
),
(
"src/m20220101_000001_create_table.rs",
"m20220101_000001_create_table",
),
(
"migration/src/m20220101_000001_create_table.rs",
"m20220101_000001_create_table",
),
(
"/migration/src/m20220101_000001_create_table.tmp.rs",
"m20220101_000001_create_table.tmp",
),
];
for (path, expect) in pair {
assert_eq!(get_file_stem(path), expect);
}
}
}