@Test public void shouldGetVdbModel() throws Exception { loadVdbs(); // get Properties settings = _uriBuilder.createSettings(SettingNames.VDB_NAME, TestUtilities.PORTFOLIO_VDB_NAME); _uriBuilder.addSetting(settings, SettingNames.MODEL_NAME, "PersonalValuations"); URI uri = _uriBuilder.buildVdbModelUri(LinkType.SELF, settings); this.response = request(uri).get(); final String entity = this.response.readEntity(String.class); assertThat(entity, is(notNullValue())); // System.out.println("Response from uri " + uri + ":\n" + entity); RestVdbModel model = KomodoJsonMarshaller.unmarshall(entity, RestVdbModel.class); assertNotNull(model); assertEquals("PersonalValuations", model.getId()); assertEquals(KomodoType.MODEL, model.getkType()); assertEquals(true, model.isVisible()); assertEquals(Type.PHYSICAL, model.getModelType()); Collection<RestLink> links = model.getLinks(); assertEquals(4, links.size()); }
@Test public void shouldGetVdbModels() throws Exception { loadVdbs(); // get URI uri = _uriBuilder.generateVdbChildGroupUri(TestUtilities.PORTFOLIO_VDB_NAME, LinkType.MODELS); this.response = request(uri).get(); final String entity = this.response.readEntity(String.class); assertThat(entity, is(notNullValue())); // System.out.println("Response:\n" + entity); RestVdbModel[] models = KomodoJsonMarshaller.unmarshallArray(entity, RestVdbModel[].class); assertNotNull(models); assertEquals(5, models.length); for (RestVdbModel model : models) { assertEquals(KomodoType.MODEL, model.getkType()); assertEquals(true, model.isVisible()); if ("MarketData".equals(model.getId())) { assertEquals(Type.PHYSICAL, model.getModelType()); } else if ("Accounts".equals(model.getId())) { assertEquals(Type.PHYSICAL, model.getModelType()); } else if ("PersonalValuations".equals(model.getId())) { assertEquals(Type.PHYSICAL, model.getModelType()); } else if ("Stocks".equals(model.getId())) { assertEquals(Type.VIRTUAL, model.getModelType()); } else if ("StocksMatModel".equals(model.getId())) { assertEquals(Type.VIRTUAL, model.getModelType()); } else fail("Model has invalid id"); Collection<RestLink> links = model.getLinks(); assertEquals(4, links.size()); } }