1pub mod config;
2pub mod file_format;
3pub mod name_case;
4
5pub use config::{SeaOrmConfig, VespertideConfig, default_migration_filename_pattern};
6pub use file_format::FileFormat;
7pub use name_case::NameCase;
8
9#[cfg(test)]
10mod tests {
11 use std::path::{Path, PathBuf};
12
13 use super::*;
14
15 #[test]
16 fn default_values_are_snake_and_standard_paths() {
17 let cfg = VespertideConfig::default();
18 assert_eq!(cfg.models_dir, PathBuf::from("models"));
19 assert_eq!(cfg.migrations_dir, PathBuf::from("migrations"));
20 assert!(cfg.table_case().is_snake());
21 assert!(cfg.column_case().is_snake());
22 }
23
24 #[test]
25 fn overrides_work_via_struct_update() {
26 let cfg = VespertideConfig {
27 models_dir: PathBuf::from("custom_models"),
28 migrations_dir: PathBuf::from("custom_migrations"),
29 table_naming_case: NameCase::Camel,
30 column_naming_case: NameCase::Pascal,
31 ..Default::default()
32 };
33
34 assert_eq!(cfg.models_dir(), Path::new("custom_models"));
35 assert_eq!(cfg.migrations_dir(), Path::new("custom_migrations"));
36 assert!(cfg.table_case().is_camel());
37 assert!(cfg.column_case().is_pascal());
38 }
39
40 #[test]
41 fn seaorm_config_default_has_vespera_schema() {
42 let cfg = SeaOrmConfig::default();
43 assert_eq!(cfg.extra_enum_derives(), &["vespera::Schema".to_string()]);
44 assert!(cfg.extra_model_derives().is_empty());
45 }
46
47 #[test]
48 fn seaorm_config_accessors() {
49 let cfg = SeaOrmConfig {
50 extra_enum_derives: vec!["A".to_string(), "B".to_string()],
51 extra_model_derives: vec!["C".to_string()],
52 };
53 assert_eq!(cfg.extra_enum_derives(), &["A", "B"]);
54 assert_eq!(cfg.extra_model_derives(), &["C"]);
55 }
56
57 #[test]
58 fn vespertide_config_seaorm_accessor() {
59 let cfg = VespertideConfig::default();
60 let seaorm = cfg.seaorm();
61 assert_eq!(
62 seaorm.extra_enum_derives(),
63 &["vespera::Schema".to_string()]
64 );
65 }
66
67 #[test]
68 fn seaorm_config_deserialize_with_defaults() {
69 let json = r#"{}"#;
70 let cfg: SeaOrmConfig = serde_json::from_str(json).unwrap();
71 assert_eq!(cfg.extra_enum_derives(), &["vespera::Schema".to_string()]);
72 assert!(cfg.extra_model_derives().is_empty());
73 }
74
75 #[test]
76 fn seaorm_config_deserialize_with_custom_values() {
77 let json = r#"{"extraEnumDerives": ["Custom"], "extraModelDerives": ["Model"]}"#;
78 let cfg: SeaOrmConfig = serde_json::from_str(json).unwrap();
79 assert_eq!(cfg.extra_enum_derives(), &["Custom"]);
80 assert_eq!(cfg.extra_model_derives(), &["Model"]);
81 }
82
83 #[test]
84 fn vespertide_config_deserialize_with_seaorm() {
85 let json = r#"{
86 "modelsDir": "models",
87 "migrationsDir": "migrations",
88 "tableNamingCase": "snake",
89 "columnNamingCase": "snake",
90 "seaorm": {
91 "extraEnumDerives": ["MyDerive"]
92 }
93 }"#;
94 let cfg: VespertideConfig = serde_json::from_str(json).unwrap();
95 assert_eq!(cfg.seaorm().extra_enum_derives(), &["MyDerive"]);
96 }
97}