Fix extracting OpenAPI entity name

shopstic · Nov 25, 2024 · cfc690f · cfc690f
1 parent 9683f1a
commit cfc690f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/chopsticks-openapi/src/main/scala/dev/chopsticks/openapi/common/OpenApiConverterUtils.scala b/chopsticks-openapi/src/main/scala/dev/chopsticks/openapi/common/OpenApiConverterUtils.scala
@@ -15,6 +15,8 @@ object OpenApiConverterUtils {
         getEntityName(Some(r.id), parsed)
       case r: Schema.Lazy[_] =>
         getEntityName(r.schema)
+      case r: zio.schema.Schema.Transform[_, _, _] =>
+        getEntityName(r.schema)
       case other =>
         OpenApiParsedAnnotations.extractAnnotations(other.annotations).entityName
     }
@@ -33,6 +35,8 @@ object OpenApiConverterUtils {
         getEntityName(Some(r.id), parsed)
       case r: Schema.Lazy[_] =>
         getEntityName(r.schema, parsed)
+      case r: zio.schema.Schema.Transform[_, _, _] =>
+        getEntityName(r.schema)
       case _ =>
         parsed.entityName
     }