@@ -18,17 +18,17 @@ package za.co.absa.springdocopenapiscala
1818
1919import io .swagger .v3 .oas .models .Components
2020import io .swagger .v3 .oas .models .media .Schema
21-
2221import org .scalatest
2322import org .scalatest .flatspec .AnyFlatSpec
2423
2524import java .time .{Instant , LocalDate , LocalDateTime , LocalTime , ZonedDateTime }
2625import java .util .UUID
2726import scala .collection .JavaConverters ._
2827import scala .reflect .runtime .universe ._
29-
3028import za .co .absa .springdocopenapiscala .OpenAPIModelRegistration .ExtraTypesHandling
3129
30+ import java .sql .Timestamp
31+
3232class OpenAPIModelRegistrationSpec extends AnyFlatSpec {
3333
3434 private case class OnlySimpleTypes (
@@ -47,7 +47,10 @@ class OpenAPIModelRegistrationSpec extends AnyFlatSpec {
4747 o : Instant ,
4848 p : LocalDateTime ,
4949 r : LocalDate ,
50- s : LocalTime
50+ s : LocalTime ,
51+ t : BigDecimal ,
52+ w : BigInt ,
53+ z : Timestamp
5154 )
5255
5356 private case class SimpleTypesMaybeInOption (
@@ -194,6 +197,9 @@ class OpenAPIModelRegistrationSpec extends AnyFlatSpec {
194197 assertTypeAndFormatAreAsExpected(actualSchemas, " OnlySimpleTypes.p" , " string" , Some (" date-time" ))
195198 assertTypeAndFormatAreAsExpected(actualSchemas, " OnlySimpleTypes.r" , " string" , Some (" date" ))
196199 assertTypeAndFormatAreAsExpected(actualSchemas, " OnlySimpleTypes.s" , " string" , Some (" time" ))
200+ assertTypeAndFormatAreAsExpected(actualSchemas, " OnlySimpleTypes.t" , " number" )
201+ assertTypeAndFormatAreAsExpected(actualSchemas, " OnlySimpleTypes.w" , " integer" )
202+ assertTypeAndFormatAreAsExpected(actualSchemas, " OnlySimpleTypes.z" , " string" , Some (" date-time" ))
197203 }
198204
199205 it should " mark all non-Option fields of case class as required" in {
0 commit comments