#27, test: Fixed broken tests

ebowman · Sep 21, 2016 · 4b4dbbd · 4b4dbbd
1 parent e27007a
commit 4b4dbbd
Show file tree

Hide file tree

Showing 20 changed files with 59 additions and 59 deletions.
diff --git a/play-scala-generator/src/main/scala/de/zalando/apifirst/generators/DataGeneratorsStep.scala b/play-scala-generator/src/main/scala/de/zalando/apifirst/generators/DataGeneratorsStep.scala
@@ -99,10 +99,10 @@ trait DataGeneratorsStep extends EnrichmentStep[Type] {
   }
 
   private def containerTypeName(c: Container, t: DenotationTable, ref: Reference): String = {
-    val className = typeNameDenotation(t, c.tpe.name)
+    lazy val className = typeNameDenotation(t, c.tpe.name)
     c match {
-      case ArrResult(tpe, _) => s": Gen[List[$className]]"
-      case c @ CatchAll(tpe, _) => s": Gen[Map[String, $className]]"
+      case ArrResult(_, _) => s": Gen[List[$className]]"
+      case CatchAll(_, _) => s": Gen[Map[String, $className]]"
       case _ => ""
     }
   }

diff --git a/...-generator/src/test/resources/expected_results/test_data/additional_properties_yaml.scala b/...-generator/src/test/resources/expected_results/test_data/additional_properties_yaml.scala
@@ -16,8 +16,8 @@ object Generators extends JsValueGenerators {
 
 
 
-    def KeyedArraysAdditionalPropertiesGenerator = _genMap[String,KeyedArraysAdditionalPropertiesCatchAll](arbitrary[String], KeyedArraysAdditionalPropertiesCatchAllGenerator)
-    def KeyedArraysAdditionalPropertiesCatchAllGenerator = Gen.containerOf[List,BigInt](arbitrary[BigInt])
+    def KeyedArraysAdditionalPropertiesGenerator: Gen[Map[String, KeyedArraysAdditionalPropertiesCatchAll]] = _genMap[String,KeyedArraysAdditionalPropertiesCatchAll](arbitrary[String], KeyedArraysAdditionalPropertiesCatchAllGenerator)
+    def KeyedArraysAdditionalPropertiesCatchAllGenerator: Gen[List[BigInt]] = Gen.containerOf[List,BigInt](arbitrary[BigInt])
 
 
     def createKeyedArraysGenerator = _generate(KeyedArraysGenerator)

diff --git a/...ala-generator/src/test/resources/expected_results/test_data/basic_polymorphism_yaml.scala b/...ala-generator/src/test/resources/expected_results/test_data/basic_polymorphism_yaml.scala
@@ -17,7 +17,7 @@ object Generators extends JsValueGenerators {
 
 
 
-    def ZooTiersOptGenerator = Gen.containerOf[List,Pet](PetGenerator)
+    def ZooTiersOptGenerator: Gen[List[Pet]] = Gen.containerOf[List,Pet](PetGenerator)
     def ZooTiersGenerator = Gen.option(ZooTiersOptGenerator)
     def CatHuntingSkillGenerator = { import CatHuntingSkill._ ; Gen.oneOf(Seq(Clueless, Lazy, Adventurous, Aggressive)) }
     def PutDummyGenerator = Gen.option(PetGenerator)

diff --git a/...-generator/src/test/resources/expected_results/test_data/cross_spec_references_yaml.scala b/...-generator/src/test/resources/expected_results/test_data/cross_spec_references_yaml.scala
@@ -36,12 +36,12 @@ object Generators extends JsValueGenerators {
     def ModelSchemaRootLinksGenerator = Gen.option(ModelSchemaRootLinksOptGenerator)
     def PetTagsGenerator = Gen.option(PetTagsOptGenerator)
     def ModelSchemaSilhouetteIdGenerator = { import ModelSchemaSilhouetteId._ ; Gen.oneOf(Seq(Kitchen, Bikini_top, Toys, Nightwear_combination, Bra, One_piece_underwear, Ball, Cleansing, Skincare, Jewellery, Headgear, Bustier, Beach_trouser, Bedroom, Lounge, Nail, Undershirt, Combination_clothing, Gloves, Fragrance, Other_equipment, Fitness, Bathroom, One_piece_nightwear, Sleeping_bag, Coat, Case, Sandals, Ankle_boots, Stocking, Shirt, Backpack, Face_cosmetic, Travel_equipment, Hair, Sneaker, Beauty_equipment, Bikini_combination, Backless_slipper, Beach_accessoires, Scarf, First_shoe, Voucher, Wallet, Peeling, Glasses, Boards, Sun, Shave, Low_shoe, Underwear_combination, Nightdress, Suit_accessoires, Watch, Headphones, Skates, Boots, Jacket, Etui, Night_shirt, Other_accessoires, Vest, Bag, System, Racket, Trouser, Lip_cosmetic, Keychain, Belt, Ballerina_shoe, One_piece_suit, Night_trouser, Skirt, Tights, Beach_shirt, Dress, Bicycle, Protector, Eye_cosmetic, Bathrobe, Bicycle_equipment, Pullover, One_piece_beachwear, Underpant, Living, Cardigan, Corsage, Shoe_accessoires, Umbrella, Pumps, Tent, T_shirt_top, Ski)) }
-    def PetPhotoUrlsGenerator = Gen.containerOf[List,String](arbitrary[String])
+    def PetPhotoUrlsGenerator: Gen[List[String]] = Gen.containerOf[List,String](arbitrary[String])
     def ModelSchemaLengthRegisterGenerator = Gen.option(arbitrary[String])
-    def ModelSchemaAgeGroupsGenerator = Gen.containerOf[List,ModelSchemaAgeGroupsArrResult](ModelSchemaAgeGroupsArrResultGenerator)
+    def ModelSchemaAgeGroupsGenerator: Gen[List[ModelSchemaAgeGroupsArrResult]] = Gen.containerOf[List,ModelSchemaAgeGroupsArrResult](ModelSchemaAgeGroupsArrResultGenerator)
     def PetCategoryGenerator = Gen.option(PetCategoryOptGenerator)
     def ModelSchemaAgeGroupsArrResultGenerator = { import ModelSchemaAgeGroupsArrResult._ ; Gen.oneOf(Seq(Baby, Kid, Teen, Adult)) }
-    def PetTagsOptGenerator = Gen.containerOf[List,PetCategoryOpt](PetCategoryOptGenerator)
+    def PetTagsOptGenerator: Gen[List[PetCategoryOpt]] = Gen.containerOf[List,PetCategoryOpt](PetCategoryOptGenerator)
     def ModelSchemaRootMetaGenerator = Gen.option(ModelSchemaRootMetaOptGenerator)
 
 

diff --git a/play-scala-generator/src/test/resources/expected_results/test_data/error_in_array_yaml.scala b/play-scala-generator/src/test/resources/expected_results/test_data/error_in_array_yaml.scala
@@ -27,18 +27,18 @@ object Generators extends JsValueGenerators {
 
 
 
-    def ModelSchemaSpecialDescriptionsOptGenerator = Gen.containerOf[List,String](arbitrary[String])
+    def ModelSchemaSpecialDescriptionsOptGenerator: Gen[List[String]] = Gen.containerOf[List,String](arbitrary[String])
     def MetaCopyrightGenerator = Gen.option(arbitrary[String])
     def ModelSchemaKeywordsGenerator = Gen.option(arbitrary[String])
     def ModelSchemaSpecialDescriptionsGenerator = Gen.option(ModelSchemaSpecialDescriptionsOptGenerator)
-    def ErrorsErrorsOptGenerator = Gen.containerOf[List,Error](ErrorGenerator)
+    def ErrorsErrorsOptGenerator: Gen[List[Error]] = Gen.containerOf[List,Error](ErrorGenerator)
     def ModelSchemaRootDataGenerator = Gen.option(ModelSchemaGenerator)
     def ErrorSourceGenerator = Gen.option(ErrorSourceNameClashGenerator)
     def ModelSchemaRootLinksGenerator = Gen.option(LinksGenerator)
     def ModelSchemaSilhouetteIdGenerator = { import ModelSchemaSilhouetteId._ ; Gen.oneOf(Seq(Kitchen, Bikini_top, Toys, Nightwear_combination, Bra, One_piece_underwear, Ball, Cleansing, Skincare, Jewellery, Headgear, Bustier, Beach_trouser, Bedroom, Lounge, Nail, Undershirt, Combination_clothing, Gloves, Fragrance, Other_equipment, Fitness, Bathroom, One_piece_nightwear, Sleeping_bag, Coat, Case, Sandals, Ankle_boots, Stocking, Shirt, Backpack, Face_cosmetic, Travel_equipment, Hair, Sneaker, Beauty_equipment, Bikini_combination, Backless_slipper, Beach_accessoires, Scarf, First_shoe, Voucher, Wallet, Peeling, Glasses, Boards, Sun, Shave, Low_shoe, Underwear_combination, Nightdress, Suit_accessoires, Watch, Headphones, Skates, Boots, Jacket, Etui, Night_shirt, Other_accessoires, Vest, Bag, System, Racket, Trouser, Lip_cosmetic, Keychain, Belt, Ballerina_shoe, One_piece_suit, Night_trouser, Skirt, Tights, Beach_shirt, Dress, Bicycle, Protector, Eye_cosmetic, Bathrobe, Bicycle_equipment, Pullover, One_piece_beachwear, Underpant, Living, Cardigan, Corsage, Shoe_accessoires, Umbrella, Pumps, Tent, T_shirt_top, Ski)) }
     def ModelSchemaLengthRegisterGenerator = Gen.option(arbitrary[String])
     def ErrorsErrorsGenerator = Gen.option(ErrorsErrorsOptGenerator)
-    def ModelSchemaAgeGroupsGenerator = Gen.containerOf[List,ModelSchemaAgeGroupsArrResult](ModelSchemaAgeGroupsArrResultGenerator)
+    def ModelSchemaAgeGroupsGenerator: Gen[List[ModelSchemaAgeGroupsArrResult]] = Gen.containerOf[List,ModelSchemaAgeGroupsArrResult](ModelSchemaAgeGroupsArrResultGenerator)
     def ModelSchemaAgeGroupsArrResultGenerator = { import ModelSchemaAgeGroupsArrResult._ ; Gen.oneOf(Seq(Baby, Kid, Teen, Adult)) }
     def ModelSchemaRootMetaGenerator = Gen.option(MetaGenerator)
 

diff --git a/...-generator/src/test/resources/expected_results/test_data/expanded_polymorphism_yaml.scala b/...-generator/src/test/resources/expected_results/test_data/expanded_polymorphism_yaml.scala
@@ -25,7 +25,7 @@ object Generators extends JsValueGenerators {
     def PetsGetLimitGenerator = Gen.option(arbitrary[Int])
     def PetsGetTagsOptGenerator = _genList(arbitrary[String], "csv")
     def NewPetTagGenerator = Gen.option(arbitrary[String])
-    def PetsGetResponses200Generator = Gen.containerOf[List,Pet](PetGenerator)
+    def PetsGetResponses200Generator: Gen[List[Pet]] = Gen.containerOf[List,Pet](PetGenerator)
     def PetsGetTagsGenerator = Gen.option(PetsGetTagsOptGenerator)
 
 

diff --git a/...cala-generator/src/test/resources/expected_results/test_data/full_petstore_api_yaml.scala b/...cala-generator/src/test/resources/expected_results/test_data/full_petstore_api_yaml.scala
@@ -38,9 +38,9 @@ object Generators extends JsValueGenerators {
     def NullGenerator = arbitrary[Null]
     def OrderStatusGenerator = Gen.option(arbitrary[String])
     def PetsFindByStatusGetStatusOptGenerator = _genList(arbitrary[String], "multi")
-    def UsersCreateWithListPostBodyOptGenerator = Gen.containerOf[List,User](UserGenerator)
+    def UsersCreateWithListPostBodyOptGenerator: Gen[List[User]] = Gen.containerOf[List,User](UserGenerator)
     def OrderPetIdGenerator = Gen.option(arbitrary[Long])
-    def PetsFindByStatusGetResponses200Generator = Gen.containerOf[List,Pet](PetGenerator)
+    def PetsFindByStatusGetResponses200Generator: Gen[List[Pet]] = Gen.containerOf[List,Pet](PetGenerator)
     def PetsPostBodyGenerator = Gen.option(PetGenerator)
     def OrderShipDateGenerator = Gen.option(arbitrary[ZonedDateTime])
     def UsersUsernamePutBodyGenerator = Gen.option(UserGenerator)
@@ -49,11 +49,11 @@ object Generators extends JsValueGenerators {
     def PetTagsGenerator = Gen.option(PetTagsOptGenerator)
     def LongGenerator = arbitrary[Long]
     def OrderQuantityGenerator = Gen.option(arbitrary[Int])
-    def PetPhotoUrlsGenerator = Gen.containerOf[List,String](arbitrary[String])
+    def PetPhotoUrlsGenerator: Gen[List[String]] = Gen.containerOf[List,String](arbitrary[String])
     def UsersCreateWithListPostBodyGenerator = Gen.option(UsersCreateWithListPostBodyOptGenerator)
     def PetsFindByStatusGetStatusGenerator = Gen.option(PetsFindByStatusGetStatusOptGenerator)
     def PetCategoryGenerator = Gen.option(TagGenerator)
-    def PetTagsOptGenerator = Gen.containerOf[List,Tag](TagGenerator)
+    def PetTagsOptGenerator: Gen[List[Tag]] = Gen.containerOf[List,Tag](TagGenerator)
 
 
     def createUserGenerator = _generate(UserGenerator)

diff --git a/play-scala-generator/src/test/resources/expected_results/test_data/hackweek_yaml.scala b/play-scala-generator/src/test/resources/expected_results/test_data/hackweek_yaml.scala
@@ -27,18 +27,18 @@ object Generators extends JsValueGenerators {
 
 
 
-    def ModelSchemaSpecialDescriptionsOptGenerator = Gen.containerOf[List,String](arbitrary[String])
+    def ModelSchemaSpecialDescriptionsOptGenerator: Gen[List[String]] = Gen.containerOf[List,String](arbitrary[String])
     def MetaCopyrightGenerator = Gen.option(arbitrary[String])
     def ModelSchemaKeywordsGenerator = Gen.option(arbitrary[String])
     def ModelSchemaSpecialDescriptionsGenerator = Gen.option(ModelSchemaSpecialDescriptionsOptGenerator)
-    def ErrorsErrorsOptGenerator = Gen.containerOf[List,Error](ErrorGenerator)
+    def ErrorsErrorsOptGenerator: Gen[List[Error]] = Gen.containerOf[List,Error](ErrorGenerator)
     def ModelSchemaRootDataGenerator = Gen.option(ModelSchemaGenerator)
     def ErrorSourceGenerator = Gen.option(ErrorSourceNameClashGenerator)
     def ModelSchemaRootLinksGenerator = Gen.option(LinksGenerator)
     def ModelSchemaSilhouetteIdGenerator = { import ModelSchemaSilhouetteId._ ; Gen.oneOf(Seq(Kitchen, Bikini_top, Toys, Nightwear_combination, Bra, One_piece_underwear, Ball, Cleansing, Skincare, Jewellery, Headgear, Bustier, Beach_trouser, Bedroom, Lounge, Nail, Undershirt, Combination_clothing, Gloves, Fragrance, Other_equipment, Fitness, Bathroom, One_piece_nightwear, Sleeping_bag, Coat, Case, Sandals, Ankle_boots, Stocking, Shirt, Backpack, Face_cosmetic, Travel_equipment, Hair, Sneaker, Beauty_equipment, Bikini_combination, Backless_slipper, Beach_accessoires, Scarf, First_shoe, Voucher, Wallet, Peeling, Glasses, Boards, Sun, Shave, Low_shoe, Underwear_combination, Nightdress, Suit_accessoires, Watch, Headphones, Skates, Boots, Jacket, Etui, Night_shirt, Other_accessoires, Vest, Bag, System, Racket, Trouser, Lip_cosmetic, Keychain, Belt, Ballerina_shoe, One_piece_suit, Night_trouser, Skirt, Tights, Beach_shirt, Dress, Bicycle, Protector, Eye_cosmetic, Bathrobe, Bicycle_equipment, Pullover, One_piece_beachwear, Underpant, Living, Cardigan, Corsage, Shoe_accessoires, Umbrella, Pumps, Tent, T_shirt_top, Ski)) }
     def ModelSchemaLengthRegisterGenerator = Gen.option(arbitrary[String])
     def ErrorsErrorsGenerator = Gen.option(ErrorsErrorsOptGenerator)
-    def ModelSchemaAgeGroupsGenerator = Gen.containerOf[List,ModelSchemaAgeGroupsArrResult](ModelSchemaAgeGroupsArrResultGenerator)
+    def ModelSchemaAgeGroupsGenerator: Gen[List[ModelSchemaAgeGroupsArrResult]] = Gen.containerOf[List,ModelSchemaAgeGroupsArrResult](ModelSchemaAgeGroupsArrResultGenerator)
     def ModelSchemaAgeGroupsArrResultGenerator = { import ModelSchemaAgeGroupsArrResult._ ; Gen.oneOf(Seq(Baby, Kid, Teen, Adult)) }
     def ModelSchemaRootMetaGenerator = Gen.option(MetaGenerator)
 

diff --git a/...la-generator/src/test/resources/expected_results/test_data/heroku_petstore_api_yaml.scala b/...la-generator/src/test/resources/expected_results/test_data/heroku_petstore_api_yaml.scala
@@ -26,7 +26,7 @@ object Generators extends JsValueGenerators {
     def NullGenerator = arbitrary[Null]
     def BigIntGenerator = arbitrary[BigInt]
     def PutPetGenerator = Gen.option(PetGenerator)
-    def GetResponses200Generator = Gen.containerOf[List,Pet](PetGenerator)
+    def GetResponses200Generator: Gen[List[Pet]] = Gen.containerOf[List,Pet](PetGenerator)
 
 
     def createPetGenerator = _generate(PetGenerator)

diff --git a/play-scala-generator/src/test/resources/expected_results/test_data/instagram_api_yaml.scala b/play-scala-generator/src/test/resources/expected_results/test_data/instagram_api_yaml.scala
@@ -59,21 +59,21 @@ object Generators extends JsValueGenerators {
     def MediaIdGenerator = Gen.option(arbitrary[BigInt])
     def MediaVideosLow_resolutionGenerator = Gen.option(ImageGenerator)
     def UsersUser_idRelationshipPostActionGenerator = Gen.option(UsersUser_idRelationshipPostActionOptGenerator)
-    def MediaTagsOptGenerator = Gen.containerOf[List,Tag](TagGenerator)
+    def MediaTagsOptGenerator: Gen[List[Tag]] = Gen.containerOf[List,Tag](TagGenerator)
     def MediaImagesGenerator = Gen.option(MediaImagesOptGenerator)
     def MediaLikesGenerator = Gen.option(MediaLikesOptGenerator)
     def BigIntNameClashGenerator = arbitrary[BigInt]
-    def MediaUsers_in_photoOptGenerator = Gen.containerOf[List,MiniProfile](MiniProfileGenerator)
-    def MediaMedia_idLikesGetResponses200DataOptGenerator = Gen.containerOf[List,Like](LikeGenerator)
+    def MediaUsers_in_photoOptGenerator: Gen[List[MiniProfile]] = Gen.containerOf[List,MiniProfile](MiniProfileGenerator)
+    def MediaMedia_idLikesGetResponses200DataOptGenerator: Gen[List[Like]] = Gen.containerOf[List,Like](LikeGenerator)
     def LocationsSearchGetResponses200DataGenerator = Gen.option(LocationsSearchGetResponses200DataOptGenerator)
     def MediaComments_escGenerator = Gen.option(MediaComments_OptGenerator)
-    def MediaSearchGetResponses200DataOptGenerator = Gen.containerOf[List,MediaSearchGetResponses200DataOptArrResult](MediaSearchGetResponses200DataOptArrResultGenerator)
+    def MediaSearchGetResponses200DataOptGenerator: Gen[List[MediaSearchGetResponses200DataOptArrResult]] = Gen.containerOf[List,MediaSearchGetResponses200DataOptArrResult](MediaSearchGetResponses200DataOptArrResultGenerator)
     def CommentFromGenerator = Gen.option(MiniProfileGenerator)
-    def LocationsSearchGetResponses200DataOptGenerator = Gen.containerOf[List,Location](LocationGenerator)
+    def LocationsSearchGetResponses200DataOptGenerator: Gen[List[Location]] = Gen.containerOf[List,Location](LocationGenerator)
     def MediaSearchGetResponses200DataGenerator = Gen.option(MediaSearchGetResponses200DataOptGenerator)
-    def UsersSelfFeedGetResponses200DataOptGenerator = Gen.containerOf[List,Media](MediaGenerator)
+    def UsersSelfFeedGetResponses200DataOptGenerator: Gen[List[Media]] = Gen.containerOf[List,Media](MediaGenerator)
     def UsersUser_idRelationshipPostActionOptGenerator = { import UsersUser_idRelationshipPostActionOpt._ ; Gen.oneOf(Seq(Follow, Approve, Unfollow, Block, Unblock)) }
-    def MediaComments_DataOptGenerator = Gen.containerOf[List,Comment](CommentGenerator)
+    def MediaComments_DataOptGenerator: Gen[List[Comment]] = Gen.containerOf[List,Comment](CommentGenerator)
     def UsersUser_idGetResponses200DataGenerator = Gen.option(UserGenerator)
     def MediaVideosGenerator = Gen.option(MediaVideosOptGenerator)
     def MediaLocationGenerator = Gen.option(LocationGenerator)

diff --git a/play-scala-generator/src/test/resources/expected_results/test_data/nakadi_yaml.scala b/play-scala-generator/src/test/resources/expected_results/test_data/nakadi_yaml.scala
@@ -31,16 +31,16 @@ object Generators extends JsValueGenerators {
     def TopicsTopicEventsGetStream_timeoutGenerator = Gen.option(arbitrary[Int])
     def IntGenerator = arbitrary[Int]
     def EventEvent_typeGenerator = Gen.option(arbitrary[String])
-    def SimpleStreamEventEventsOptGenerator = Gen.containerOf[List,Event](EventGenerator)
+    def SimpleStreamEventEventsOptGenerator: Gen[List[Event]] = Gen.containerOf[List,Event](EventGenerator)
     def EventMetaDataParent_idGenerator = Gen.option(arbitrary[UUID])
     def EventMetadataGenerator = Gen.option(EventMetaDataNameClashGenerator)
     def NullGenerator = arbitrary[Null]
-    def EventMetaDataScopesOptGenerator = Gen.containerOf[List,String](arbitrary[String])
-    def TopicsTopicPartitionsGetResponses200Generator = Gen.containerOf[List,TopicPartition](TopicPartitionGenerator)
+    def EventMetaDataScopesOptGenerator: Gen[List[String]] = Gen.containerOf[List,String](arbitrary[String])
+    def TopicsTopicPartitionsGetResponses200Generator: Gen[List[TopicPartition]] = Gen.containerOf[List,TopicPartition](TopicPartitionGenerator)
     def TopicsTopicEventsBatchPostEventGenerator = Gen.option(EventGenerator)
     def SimpleStreamEventEventsGenerator = Gen.option(SimpleStreamEventEventsOptGenerator)
     def EventMetaDataScopesGenerator = Gen.option(EventMetaDataScopesOptGenerator)
-    def TopicsGetResponses200Generator = Gen.containerOf[List,Topic](TopicGenerator)
+    def TopicsGetResponses200Generator: Gen[List[Topic]] = Gen.containerOf[List,Topic](TopicGenerator)
 
 
     def createEventMetaDataNameClashGenerator = _generate(EventMetaDataNameClashGenerator)

diff --git a/play-scala-generator/src/test/resources/expected_results/test_data/nested_arrays_yaml.scala b/play-scala-generator/src/test/resources/expected_results/test_data/nested_arrays_yaml.scala
@@ -21,14 +21,14 @@ object Generators extends JsValueGenerators {
 
 
 
-    def ExampleNestedArraysOptArrResultArrResultGenerator = Gen.containerOf[List,ExampleNestedArraysOptArrResultArrResultArrResult](ExampleNestedArraysOptArrResultArrResultArrResultGenerator)
-    def ExampleNestedArraysOptGenerator = Gen.containerOf[List,ExampleNestedArraysOptArrResult](ExampleNestedArraysOptArrResultGenerator)
-    def ExampleMessagesOptGenerator = Gen.containerOf[List,ExampleMessagesOptArrResult](ExampleMessagesOptArrResultGenerator)
+    def ExampleNestedArraysOptArrResultArrResultGenerator: Gen[List[ExampleNestedArraysOptArrResultArrResultArrResult]] = Gen.containerOf[List,ExampleNestedArraysOptArrResultArrResultArrResult](ExampleNestedArraysOptArrResultArrResultArrResultGenerator)
+    def ExampleNestedArraysOptGenerator: Gen[List[ExampleNestedArraysOptArrResult]] = Gen.containerOf[List,ExampleNestedArraysOptArrResult](ExampleNestedArraysOptArrResultGenerator)
+    def ExampleMessagesOptGenerator: Gen[List[ExampleMessagesOptArrResult]] = Gen.containerOf[List,ExampleMessagesOptArrResult](ExampleMessagesOptArrResultGenerator)
     def ExampleMessagesGenerator = Gen.option(ExampleMessagesOptGenerator)
-    def ExampleNestedArraysOptArrResultArrResultArrResultGenerator = Gen.containerOf[List,String](arbitrary[String])
-    def ExampleNestedArraysOptArrResultGenerator = Gen.containerOf[List,ExampleNestedArraysOptArrResultArrResult](ExampleNestedArraysOptArrResultArrResultGenerator)
+    def ExampleNestedArraysOptArrResultArrResultArrResultGenerator: Gen[List[String]] = Gen.containerOf[List,String](arbitrary[String])
+    def ExampleNestedArraysOptArrResultGenerator: Gen[List[ExampleNestedArraysOptArrResultArrResult]] = Gen.containerOf[List,ExampleNestedArraysOptArrResultArrResult](ExampleNestedArraysOptArrResultArrResultGenerator)
     def ExampleNestedArraysGenerator = Gen.option(ExampleNestedArraysOptGenerator)
-    def ExampleMessagesOptArrResultGenerator = Gen.containerOf[List,Activity](ActivityGenerator)
+    def ExampleMessagesOptArrResultGenerator: Gen[List[Activity]] = Gen.containerOf[List,Activity](ActivityGenerator)
     def ActivityActionsGenerator = Gen.option(arbitrary[String])
 
 

diff --git a/play-scala-generator/src/test/resources/expected_results/test_data/options_yaml.scala b/play-scala-generator/src/test/resources/expected_results/test_data/options_yaml.scala
@@ -14,7 +14,7 @@ object Generators extends JsValueGenerators {
 
 
 
-    def BasicRequiredGenerator = Gen.containerOf[List,String](arbitrary[String])
+    def BasicRequiredGenerator: Gen[List[String]] = Gen.containerOf[List,String](arbitrary[String])
     def BasicOptionalGenerator = Gen.option(BasicRequiredGenerator)
 
 

diff --git a/play-scala-generator/src/test/resources/expected_results/test_data/security_api_yaml.scala b/play-scala-generator/src/test/resources/expected_results/test_data/security_api_yaml.scala
@@ -17,7 +17,7 @@ object Generators extends JsValueGenerators {
 
 
     def PetsIdGetIdGenerator = _genList(arbitrary[String], "csv")
-    def PetsIdGetResponses200Generator = Gen.containerOf[List,Pet](PetGenerator)
+    def PetsIdGetResponses200Generator: Gen[List[Pet]] = Gen.containerOf[List,Pet](PetGenerator)
     def PetTagGenerator = Gen.option(arbitrary[String])