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

Refactor

Browse Source
This commit is contained in:
Chris Tsang 2024-05-29 01:00:14 +01:00
parent 461aa925f6
commit e3f64b1892
2 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
sea-orm-macros/src/derives/active_enum.rs
View File

@ -89,7 +89,7 @@ impl ActiveEnum {
_ => return Err(Error::InputNotEnum),
};
let mut is_string = false;
let mut is_string = rename_all_rule.is_some();
let mut is_int = false;
let mut variants = Vec::new();
@ -129,7 +129,7 @@ impl ActiveEnum {
.map_err(Error::Syn)?;
}
if (is_string || rename_rule.is_some() || rename_all_rule.is_some()) && is_int {
if is_string && is_int {
return Err(Error::TT(quote_spanned! {
ident_span => compile_error!("All enum variants should specify the same `*_value` macro attribute, either `string_value` or `num_value` but not both");
}));

12
sea-orm-macros/tests/derive_active_enum_test.rs
View File

@ -46,6 +46,16 @@ pub enum TestEnum2 {
HelloWorldTwo,
}
#[derive(Debug, EnumIter, DeriveActiveEnum, Eq, PartialEq)]
#[sea_orm(
rs_type = "String",
db_type = "String(StringLen::None)",
rename_all = "snake_case"
)]
pub enum TestEnum3 {
HelloWorld,
}
#[test]
fn derive_active_enum_value() {
assert_eq!(TestEnum::DefaultVariant.to_value(), "defaultVariant");
@ -124,4 +134,6 @@ fn derive_active_enum_from_value() {
fn derive_active_enum_value_2() {
assert_eq!(TestEnum2::HelloWorld.to_value(), "hello_world");
assert_eq!(TestEnum2::HelloWorldTwo.to_value(), "helloWorldTwo");
assert_eq!(TestEnum3::HelloWorld.to_value(), "hello_world");
}