@Test
public void testValidateWithValidator_fail() throws Exception {
Properties properties = new Properties();
properties.put("p1", "v1");
Configuration ambariConfig = new Configuration(properties);
ViewConfig config = ViewConfigTest.getConfig(xml_invalid_instance);
ViewEntity viewEntity =
ViewRegistryTest.getViewEntity(config, ambariConfig, getClass().getClassLoader(), "");
ViewInstanceEntity viewInstanceEntity =
ViewRegistryTest.getViewInstanceEntity(viewEntity, config.getInstances().get(0));
ViewEntityTest.TestValidator validator = new ViewEntityTest.TestValidator();
validator.result = new ValidationResultImpl(false, "detail");
viewEntity.setValidator(validator);
try {
viewInstanceEntity.validate(viewEntity, Validator.ValidationContext.PRE_CREATE);
Assert.fail("Expected an IllegalStateException");
} catch (ValidationException e) {
// expected
}
}
@Test
public void testGetValidationResult() throws Exception {
Properties properties = new Properties();
properties.put("p1", "v1");
Configuration ambariConfig = new Configuration(properties);
ViewConfig config = ViewConfigTest.getConfig(xml_valid_instance);
ViewEntity viewEntity =
ViewRegistryTest.getViewEntity(config, ambariConfig, getClass().getClassLoader(), "");
ViewInstanceEntity viewInstanceEntity =
ViewRegistryTest.getViewInstanceEntity(viewEntity, config.getInstances().get(0));
ViewEntityTest.TestValidator validator = new ViewEntityTest.TestValidator();
validator.result = new ValidationResultImpl(true, "detail");
viewEntity.setValidator(validator);
InstanceValidationResultImpl result =
viewInstanceEntity.getValidationResult(viewEntity, Validator.ValidationContext.PRE_CREATE);
Map<String, ValidationResult> propertyResults = result.getPropertyResults();
junit.framework.Assert.assertEquals(2, propertyResults.size());
junit.framework.Assert.assertTrue(propertyResults.containsKey("p1"));
junit.framework.Assert.assertTrue(propertyResults.containsKey("p2"));
junit.framework.Assert.assertTrue(propertyResults.get("p1").isValid());
junit.framework.Assert.assertTrue(propertyResults.get("p2").isValid());
}
@Test
public void testContextPath() throws Exception {
ViewInstanceEntity viewInstanceDefinition = getViewInstanceEntity();
Assert.assertEquals(
AmbariAuthorizationFilter.VIEWS_CONTEXT_PATH_PREFIX + "MY_VIEW/1.0.0/INSTANCE1",
viewInstanceDefinition.getContextPath());
}
@Test
public void testGetId() throws Exception {
ViewInstanceEntity viewInstanceDefinition = getViewInstanceEntity();
Assert.assertNull(viewInstanceDefinition.getViewInstanceId());
viewInstanceDefinition.setViewInstanceId(99L);
Assert.assertEquals(99L, (long) viewInstanceDefinition.getViewInstanceId());
}
@Test
public void testGetConfiguration() throws Exception {
InstanceConfig instanceConfig = InstanceConfigTest.getInstanceConfigs().get(0);
ViewEntity viewDefinition = ViewEntityTest.getViewEntity();
ViewInstanceEntity viewInstanceDefinition =
new ViewInstanceEntity(viewDefinition, instanceConfig);
Assert.assertEquals(instanceConfig, viewInstanceDefinition.getConfiguration());
}
@Test
public void testAddGetResourceProvider() throws Exception {
ViewInstanceEntity viewInstanceDefinition = getViewInstanceEntity();
ResourceProvider provider = createNiceMock(ResourceProvider.class);
Resource.Type type = new Resource.Type("MY_VIEW{1.0.0}/myType");
viewInstanceDefinition.addResourceProvider(type, provider);
Assert.assertEquals(provider, viewInstanceDefinition.getResourceProvider(type));
Assert.assertEquals(provider, viewInstanceDefinition.getResourceProvider("myType"));
}
@Test
public void testAddGetService() throws Exception {
ViewInstanceEntity viewInstanceDefinition = getViewInstanceEntity();
Object service = new Object();
viewInstanceDefinition.addService("resources", service);
Object service2 = new Object();
viewInstanceDefinition.addService("subresources", service2);
Assert.assertEquals(service, viewInstanceDefinition.getService("resources"));
Assert.assertEquals(service2, viewInstanceDefinition.getService("subresources"));
}
@Test
public void testAddGetProperty() throws Exception {
ViewInstanceEntity viewInstanceDefinition = getViewInstanceEntity();
viewInstanceDefinition.putProperty("p1", "v1");
viewInstanceDefinition.putProperty("p2", "v2");
viewInstanceDefinition.putProperty("p3", "v3");
Map<String, String> properties = viewInstanceDefinition.getPropertyMap();
Assert.assertEquals(3, properties.size());
Assert.assertEquals("v1", properties.get("p1"));
Assert.assertEquals("v2", properties.get("p2"));
Assert.assertEquals("v3", properties.get("p3"));
}
@Test
public void testValidate() throws Exception {
Properties properties = new Properties();
properties.put("p1", "v1");
Configuration ambariConfig = new Configuration(properties);
ViewConfig config = ViewConfigTest.getConfig(xml_valid_instance);
ViewEntity viewEntity =
ViewRegistryTest.getViewEntity(config, ambariConfig, getClass().getClassLoader(), "");
ViewInstanceEntity viewInstanceEntity =
ViewRegistryTest.getViewInstanceEntity(viewEntity, config.getInstances().get(0));
viewInstanceEntity.validate(viewEntity, Validator.ValidationContext.PRE_CREATE);
}
@Test
public void testGetDescription() throws Exception {
InstanceConfig instanceConfig =
InstanceConfigTest.getInstanceConfigs(xml_with_instance_label).get(0);
ViewEntity viewDefinition = ViewEntityTest.getViewEntity();
ViewInstanceEntity viewInstanceDefinition =
new ViewInstanceEntity(viewDefinition, instanceConfig);
Assert.assertEquals(
"This is an instance description.", viewInstanceDefinition.getDescription());
instanceConfig = InstanceConfigTest.getInstanceConfigs(xml_without_instance_label).get(0);
viewDefinition = ViewEntityTest.getViewEntity();
viewInstanceDefinition = new ViewInstanceEntity(viewDefinition, instanceConfig);
Assert.assertNull(viewInstanceDefinition.getDescription());
}
public static ViewInstanceEntity getViewInstanceEntity() throws Exception {
InstanceConfig instanceConfig = InstanceConfigTest.getInstanceConfigs().get(0);
ViewEntity viewDefinition = ViewEntityTest.getViewEntity();
ViewInstanceEntity viewInstanceEntity = new ViewInstanceEntity(viewDefinition, instanceConfig);
ResourceTypeEntity resourceTypeEntity = new ResourceTypeEntity();
resourceTypeEntity.setId(10);
resourceTypeEntity.setName(viewDefinition.getName());
viewDefinition.setResourceType(resourceTypeEntity);
ResourceEntity resourceEntity = new ResourceEntity();
resourceEntity.setId(20L);
resourceEntity.setResourceType(resourceTypeEntity);
viewInstanceEntity.setResource(resourceEntity);
return viewInstanceEntity;
}
@Test
public void testGetIcon64() throws Exception {
InstanceConfig instanceConfig =
InstanceConfigTest.getInstanceConfigs(xml_with_instance_label).get(0);
ViewEntity viewDefinition = ViewEntityTest.getViewEntity();
ViewInstanceEntity viewInstanceDefinition =
new ViewInstanceEntity(viewDefinition, instanceConfig);
Assert.assertEquals(
"/this/is/the/icon/url/instance_1_icon64.png", viewInstanceDefinition.getIcon64());
viewInstanceDefinition.setIcon64("/a/different/icon.png");
Assert.assertEquals("/a/different/icon.png", viewInstanceDefinition.getIcon64());
instanceConfig = InstanceConfigTest.getInstanceConfigs(xml_with_instance_label).get(1);
viewDefinition = ViewEntityTest.getViewEntity();
viewInstanceDefinition = new ViewInstanceEntity(viewDefinition, instanceConfig);
Assert.assertEquals("/this/is/the/icon/url/icon64.png", viewInstanceDefinition.getIcon64());
}
@Test
public void testGetLabel() throws Exception {
// with an instance label
InstanceConfig instanceConfig =
InstanceConfigTest.getInstanceConfigs(xml_with_instance_label).get(0);
ViewEntity viewDefinition = ViewEntityTest.getViewEntity();
ViewInstanceEntity viewInstanceDefinition =
new ViewInstanceEntity(viewDefinition, instanceConfig);
Assert.assertEquals("My Instance 1!", viewInstanceDefinition.getLabel());
// without an instance label
instanceConfig = InstanceConfigTest.getInstanceConfigs(xml_without_instance_label).get(0);
viewDefinition = ViewEntityTest.getViewEntity();
viewInstanceDefinition = new ViewInstanceEntity(viewDefinition, instanceConfig);
// should default to view label
Assert.assertEquals("My View!", viewInstanceDefinition.getLabel());
}
@Test
public void testIsVisible() throws Exception {
InstanceConfig instanceConfig =
InstanceConfigTest.getInstanceConfigs(xml_with_instance_label).get(0);
ViewEntity viewDefinition = ViewEntityTest.getViewEntity();
ViewInstanceEntity viewInstanceDefinition =
new ViewInstanceEntity(viewDefinition, instanceConfig);
Assert.assertTrue(viewInstanceDefinition.isVisible());
instanceConfig = InstanceConfigTest.getInstanceConfigs(xml_with_instance_label).get(1);
viewDefinition = ViewEntityTest.getViewEntity();
viewInstanceDefinition = new ViewInstanceEntity(viewDefinition, instanceConfig);
Assert.assertFalse(viewInstanceDefinition.isVisible());
instanceConfig = InstanceConfigTest.getInstanceConfigs(xml_without_instance_label).get(0);
viewDefinition = ViewEntityTest.getViewEntity();
viewInstanceDefinition = new ViewInstanceEntity(viewDefinition, instanceConfig);
Assert.assertTrue(viewInstanceDefinition.isVisible());
}
public static ViewInstanceEntity getViewInstanceEntity(SecurityHelper securityHelper)
throws Exception {
ViewInstanceEntity viewInstanceEntity = getViewInstanceEntity();
viewInstanceEntity.setSecurityHelper(securityHelper);
return viewInstanceEntity;
}
@Test
public void testGetName() throws Exception {
ViewInstanceEntity viewInstanceDefinition = getViewInstanceEntity();
Assert.assertEquals("INSTANCE1", viewInstanceDefinition.getName());
}
@Test
public void testInstanceData() throws Exception {
TestSecurityHelper securityHelper = new TestSecurityHelper("user1");
ViewInstanceEntity viewInstanceDefinition = getViewInstanceEntity(securityHelper);
viewInstanceDefinition.putInstanceData("key1", "foo");
ViewInstanceDataEntity dataEntity = viewInstanceDefinition.getInstanceData("key1");
Assert.assertNotNull(dataEntity);
Assert.assertEquals("foo", dataEntity.getValue());
Assert.assertEquals("user1", dataEntity.getUser());
viewInstanceDefinition.putInstanceData("key2", "bar");
viewInstanceDefinition.putInstanceData("key3", "baz");
viewInstanceDefinition.putInstanceData("key4", "monkey");
viewInstanceDefinition.putInstanceData("key5", "runner");
Map<String, String> dataMap = viewInstanceDefinition.getInstanceDataMap();
Assert.assertEquals(5, dataMap.size());
Assert.assertEquals("foo", dataMap.get("key1"));
Assert.assertEquals("bar", dataMap.get("key2"));
Assert.assertEquals("baz", dataMap.get("key3"));
Assert.assertEquals("monkey", dataMap.get("key4"));
Assert.assertEquals("runner", dataMap.get("key5"));
viewInstanceDefinition.removeInstanceData("key3");
dataMap = viewInstanceDefinition.getInstanceDataMap();
Assert.assertEquals(4, dataMap.size());
Assert.assertFalse(dataMap.containsKey("key3"));
securityHelper.setUser("user2");
dataMap = viewInstanceDefinition.getInstanceDataMap();
Assert.assertTrue(dataMap.isEmpty());
viewInstanceDefinition.putInstanceData("key1", "aaa");
viewInstanceDefinition.putInstanceData("key2", "bbb");
viewInstanceDefinition.putInstanceData("key3", "ccc");
dataMap = viewInstanceDefinition.getInstanceDataMap();
Assert.assertEquals(3, dataMap.size());
Assert.assertEquals("aaa", dataMap.get("key1"));
Assert.assertEquals("bbb", dataMap.get("key2"));
Assert.assertEquals("ccc", dataMap.get("key3"));
securityHelper.setUser("user1");
dataMap = viewInstanceDefinition.getInstanceDataMap();
Assert.assertEquals(4, dataMap.size());
Assert.assertEquals("foo", dataMap.get("key1"));
Assert.assertEquals("bar", dataMap.get("key2"));
Assert.assertNull(dataMap.get("key3"));
Assert.assertEquals("monkey", dataMap.get("key4"));
Assert.assertEquals("runner", dataMap.get("key5"));
}
