Ything-RepoMirrors/poem
1
0
Fork 0
mirror of https://github.com/poem-web/poem.git synced 2026-01-24 20:08:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(poem-openapi): support externally tagged unions with primitive types

Browse Source
This commit is contained in:
Luca Iachini
2025-08-22 08:48:45 +02:00
committed by GitHub
parent c1a08f9c84
commit 1c2474ecee
2 changed files with 87 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
poem-openapi-derive/src/union.rs
View File

@@ -168,11 +168,6 @@ pub(crate) fn generate(args: DeriveInput) -> GeneratorResult<TokenStream> {
if args.externally_tagged {
create_schemas.push(quote! {
{
fn __check_is_object_type<T: #crate_name::types::IsObjectType>() {}
__check_is_object_type::<#object_ty>();
}
let schema = #crate_name::registry::MetaSchema {
description: #description,
all_of: ::std::vec![

87
poem-openapi/tests/union.rs
View File

@@ -844,3 +844,90 @@ fn with_externally_tagged_mapping() {
}))
);
}
#[test]
fn with_externally_tagged_primitives() {
#[derive(Union, Debug, PartialEq)]
#[oai(externally_tagged)]
enum MyObj {
A(String),
B(bool),
}
let schema = get_meta::<MyObj>();
assert_eq!(
schema,
MetaSchema {
rust_typename: Some("union::with_externally_tagged_primitives::MyObj"),
ty: "object",
any_of: vec![
MetaSchemaRef::Reference("MyObj_A".to_string()),
MetaSchemaRef::Reference("MyObj_B".to_string()),
],
..MetaSchema::ANY
}
);
let schema_myobj_a = get_meta_by_name::<MyObj>("MyObj_A");
assert_eq!(
schema_myobj_a,
MetaSchema {
all_of: vec![MetaSchemaRef::Inline(Box::new(MetaSchema {
required: vec!["A"],
properties: vec![(
"A",
MetaSchemaRef::Inline(Box::new(MetaSchema::new("string")))
)],
..MetaSchema::new("object")
})),],
..MetaSchema::ANY
}
);
let schema_myobj_b = get_meta_by_name::<MyObj>("MyObj_B");
assert_eq!(
schema_myobj_b,
MetaSchema {
all_of: vec![MetaSchemaRef::Inline(Box::new(MetaSchema {
required: vec!["B"],
properties: vec![(
"B",
MetaSchemaRef::Inline(Box::new(MetaSchema::new("boolean")))
)],
..MetaSchema::new("object")
})),],
..MetaSchema::ANY
}
);
assert_eq!(
MyObj::parse_from_json(Some(json!({
"A":"hello",
})))
.unwrap(),
MyObj::A("hello".to_string())
);
assert_eq!(
MyObj::B(true).to_json(),
Some(json!({
"B": true,
}))
);
assert_eq!(
MyObj::parse_from_json(Some(json!({
"A": "hello",
})))
.unwrap(),
MyObj::A("hello".to_string())
);
assert_eq!(
MyObj::B(true).to_json(),
Some(json!({
"B": true,
}))
);
}