@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());
}
}
