@@ -12,6 +12,7 @@ import io.openapiprocessor.core.parser.OpenApiParser
1212import io.openapiprocessor.core.writer.SourceFormatter
1313import io.openapiprocessor.core.writer.java.*
1414import io.openapiprocessor.spring.Versions
15+ import io.openapiprocessor.spring.processor.SpringAnnotations.Companion.EXCHANGE
1516import io.openapiprocessor.spring.writer.java.*
1617import io.openapiprocessor.spring.writer.java.MappingAnnotationFactory
1718import io.openapiprocessor.spring.writer.java.ParameterAnnotationWriter
@@ -39,18 +40,22 @@ class SpringProcessor : OpenApiProcessorTest {
3940 openapi.printWarnings()
4041 }
4142
42- val framework = SpringFramework ( )
43+ val kind = SpringAnnotations .valueOf(processorOptions[ " annotations " ]?.toString() )
4344
44- val annotations = when (processorOptions[ " annotations " ]?.toString() ) {
45- " service-client " -> SpringFrameworkExchange ()
45+ val annotations = when (kind ) {
46+ EXCHANGE -> SpringFrameworkExchange ()
4647 else -> SpringFrameworkAnnotations ()
4748 }
49+ val annotationFactory = when (kind) {
50+ EXCHANGE -> ExchangeAnnotationFactory (annotations)
51+ else -> MappingAnnotationFactory (annotations)
52+ }
4853
4954 val options = convertOptions(processorOptions)
5055 val identifier = JavaIdentifier (IdentifierOptions (
5156 options.identifierWordBreakFromDigitToLetter,
5257 options.identifierPrefixInvalidEnumStart))
53- val cv = ApiConverter (options, identifier, framework )
58+ val cv = ApiConverter (options, identifier, SpringFramework () )
5459 val api = cv.convert(openapi)
5560
5661 val writerFactory = SpringWriterFactory (options)
@@ -73,7 +78,7 @@ class SpringProcessor : OpenApiProcessorTest {
7378 options,
7479 identifier,
7580 StatusAnnotationWriter (annotations),
76- MappingAnnotationFactory (annotations) ,
81+ annotationFactory ,
7782 ParameterAnnotationWriter (annotations),
7883 beanValidations,
7984 javaDocWriter
0 commit comments