caisin/sea-orm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

cargo fmt

Browse Source
This commit is contained in:
Billy Chan 2022-07-26 12:20:40 +08:00
parent b566ea6e12
commit f6672137a2
No known key found for this signature in database
GPG Key ID: A2D690CAC7DF3CC7
2 changed files with 52 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

94
sea-orm-codegen/src/entity/column.rs
View File

@ -128,7 +128,9 @@ impl Column {
} }
ColumnType::Time(_) => quote! { ColumnType::Time.def() }, ColumnType::Time(_) => quote! { ColumnType::Time.def() },
ColumnType::Date => quote! { ColumnType::Date.def() }, ColumnType::Date => quote! { ColumnType::Date.def() },
ColumnType::Binary(BlobSize::Blob(_)) | ColumnType::VarBinary(_) => quote! { ColumnType::Binary.def() }, ColumnType::Binary(BlobSize::Blob(_)) | ColumnType::VarBinary(_) => {
quote! { ColumnType::Binary.def() }
}
ColumnType::Binary(BlobSize::Tiny) => quote! { ColumnType::TinyBinary.def() }, ColumnType::Binary(BlobSize::Tiny) => quote! { ColumnType::TinyBinary.def() },
ColumnType::Binary(BlobSize::Medium) => quote! { ColumnType::MediumBinary.def() }, ColumnType::Binary(BlobSize::Medium) => quote! { ColumnType::MediumBinary.def() },
ColumnType::Binary(BlobSize::Long) => quote! { ColumnType::LongBinary.def() }, ColumnType::Binary(BlobSize::Long) => quote! { ColumnType::LongBinary.def() },
@ -466,14 +468,20 @@ mod tests {
#[test] #[test]
fn test_get_info() { fn test_get_info() {
let column: Column = ColumnDef::new(Alias::new("id")).string().to_owned().into(); let column: Column = ColumnDef::new(Alias::new("id")).string().to_owned().into();
assert_eq!(column.get_info(&DateTimeCrate::Chrono).as_str(), "Column `id`: Option<String>"); assert_eq!(
column.get_info(&DateTimeCrate::Chrono).as_str(),
"Column `id`: Option<String>"
);
let column: Column = ColumnDef::new(Alias::new("id")) let column: Column = ColumnDef::new(Alias::new("id"))
.string() .string()
.not_null() .not_null()
.to_owned() .to_owned()
.into(); .into();
assert_eq!(column.get_info(&DateTimeCrate::Chrono).as_str(), "Column `id`: String, not_null"); assert_eq!(
column.get_info(&DateTimeCrate::Chrono).as_str(),
"Column `id`: String, not_null"
);
let column: Column = ColumnDef::new(Alias::new("id")) let column: Column = ColumnDef::new(Alias::new("id"))
.string() .string()
@ -499,100 +507,100 @@ mod tests {
); );
let column: Column = ColumnDef::new(Alias::new("date_field")) let column: Column = ColumnDef::new(Alias::new("date_field"))
.date() .date()
.not_null() .not_null()
.to_owned() .to_owned()
.into(); .into();
assert_eq!( assert_eq!(
column.get_info(&DateTimeCrate::Chrono).as_str(), column.get_info(&DateTimeCrate::Chrono).as_str(),
"Column `date_field`: Date, not_null" "Column `date_field`: Date, not_null"
); );
let column: Column = ColumnDef::new(Alias::new("date_field")) let column: Column = ColumnDef::new(Alias::new("date_field"))
.date() .date()
.not_null() .not_null()
.to_owned() .to_owned()
.into(); .into();
assert_eq!( assert_eq!(
column.get_info(&DateTimeCrate::Time).as_str(), column.get_info(&DateTimeCrate::Time).as_str(),
"Column `date_field`: TimeDate, not_null" "Column `date_field`: TimeDate, not_null"
); );
let column: Column = ColumnDef::new(Alias::new("time_field")) let column: Column = ColumnDef::new(Alias::new("time_field"))
.time() .time()
.not_null() .not_null()
.to_owned() .to_owned()
.into(); .into();
assert_eq!( assert_eq!(
column.get_info(&DateTimeCrate::Chrono).as_str(), column.get_info(&DateTimeCrate::Chrono).as_str(),
"Column `time_field`: Time, not_null" "Column `time_field`: Time, not_null"
); );
let column: Column = ColumnDef::new(Alias::new("time_field")) let column: Column = ColumnDef::new(Alias::new("time_field"))
.time() .time()
.not_null() .not_null()
.to_owned() .to_owned()
.into(); .into();
assert_eq!( assert_eq!(
column.get_info(&DateTimeCrate::Time).as_str(), column.get_info(&DateTimeCrate::Time).as_str(),
"Column `time_field`: TimeTime, not_null" "Column `time_field`: TimeTime, not_null"
); );
let column: Column = ColumnDef::new(Alias::new("date_time_field")) let column: Column = ColumnDef::new(Alias::new("date_time_field"))
.date_time() .date_time()
.not_null() .not_null()
.to_owned() .to_owned()
.into(); .into();
assert_eq!( assert_eq!(
column.get_info(&DateTimeCrate::Chrono).as_str(), column.get_info(&DateTimeCrate::Chrono).as_str(),
"Column `date_time_field`: DateTime, not_null" "Column `date_time_field`: DateTime, not_null"
); );
let column: Column = ColumnDef::new(Alias::new("date_time_field")) let column: Column = ColumnDef::new(Alias::new("date_time_field"))
.date_time() .date_time()
.not_null() .not_null()
.to_owned() .to_owned()
.into(); .into();
assert_eq!( assert_eq!(
column.get_info(&DateTimeCrate::Time).as_str(), column.get_info(&DateTimeCrate::Time).as_str(),
"Column `date_time_field`: TimeDateTime, not_null" "Column `date_time_field`: TimeDateTime, not_null"
); );
let column: Column = ColumnDef::new(Alias::new("timestamp_field")) let column: Column = ColumnDef::new(Alias::new("timestamp_field"))
.timestamp() .timestamp()
.not_null() .not_null()
.to_owned() .to_owned()
.into(); .into();
assert_eq!( assert_eq!(
column.get_info(&DateTimeCrate::Chrono).as_str(), column.get_info(&DateTimeCrate::Chrono).as_str(),
"Column `timestamp_field`: DateTimeUtc, not_null" "Column `timestamp_field`: DateTimeUtc, not_null"
); );
let column: Column = ColumnDef::new(Alias::new("timestamp_field")) let column: Column = ColumnDef::new(Alias::new("timestamp_field"))
.timestamp() .timestamp()
.not_null() .not_null()
.to_owned() .to_owned()
.into(); .into();
assert_eq!( assert_eq!(
column.get_info(&DateTimeCrate::Time).as_str(), column.get_info(&DateTimeCrate::Time).as_str(),
"Column `timestamp_field`: TimeDateTime, not_null" "Column `timestamp_field`: TimeDateTime, not_null"
); );
let column: Column = ColumnDef::new(Alias::new("timestamp_with_timezone_field")) let column: Column = ColumnDef::new(Alias::new("timestamp_with_timezone_field"))
.timestamp_with_time_zone() .timestamp_with_time_zone()
.not_null() .not_null()
.to_owned() .to_owned()
.into(); .into();
assert_eq!( assert_eq!(
column.get_info(&DateTimeCrate::Chrono).as_str(), column.get_info(&DateTimeCrate::Chrono).as_str(),
"Column `timestamp_with_timezone_field`: DateTimeWithTimeZone, not_null" "Column `timestamp_with_timezone_field`: DateTimeWithTimeZone, not_null"
); );
let column: Column = ColumnDef::new(Alias::new("timestamp_with_timezone_field")) let column: Column = ColumnDef::new(Alias::new("timestamp_with_timezone_field"))
.timestamp_with_time_zone() .timestamp_with_time_zone()
.not_null() .not_null()
.to_owned() .to_owned()
.into(); .into();
assert_eq!( assert_eq!(
column.get_info(&DateTimeCrate::Time).as_str(), column.get_info(&DateTimeCrate::Time).as_str(),
"Column `timestamp_with_timezone_field`: TimeDateTimeWithTimeZone, not_null" "Column `timestamp_with_timezone_field`: TimeDateTimeWithTimeZone, not_null"