From 905a7746e7243491d2fd7d31a5ca353950622abb Mon Sep 17 00:00:00 2001 From: Tony Tam Date: Mon, 19 Oct 2015 21:45:47 -0700 Subject: [PATCH] updated test --- .../swagger/codegen/InlineModelResolverTest.java | 16 ++++++---------- 1 file changed, 6 insertions(+), 10 deletions(-) diff --git a/modules/swagger-codegen/src/test/java/io/swagger/codegen/InlineModelResolverTest.java b/modules/swagger-codegen/src/test/java/io/swagger/codegen/InlineModelResolverTest.java index 474cfad86e64..d1c9fc63d9e0 100644 --- a/modules/swagger-codegen/src/test/java/io/swagger/codegen/InlineModelResolverTest.java +++ b/modules/swagger-codegen/src/test/java/io/swagger/codegen/InlineModelResolverTest.java @@ -171,20 +171,16 @@ public class InlineModelResolverTest { assertNotNull(response.getSchema()); Property responseProperty = response.getSchema(); - assertTrue(responseProperty instanceof RefProperty); - Model inline = swagger.getDefinitions().get("inline_response_200"); - assertNotNull(inline); - assertTrue(inline instanceof ArrayModel); + // no need to flatten more + assertTrue(responseProperty instanceof ArrayProperty); - ArrayModel am = (ArrayModel) inline; - assertTrue(am.getItems() instanceof RefProperty); + ArrayProperty ap = (ArrayProperty) responseProperty; + Property p = ap.getItems(); - Model inlineInner = swagger.getDefinitions().get("inline_response_200_inner"); - assertNotNull(inlineInner); - assertTrue(inlineInner instanceof ModelImpl); + assertNotNull(p); - ModelImpl innerModel = (ModelImpl) inlineInner; + ObjectProperty innerModel = (ObjectProperty) p; assertTrue(innerModel.getProperties().size() == 1); assertNotNull(innerModel.getProperties().get("name")); }