Commit 666d92e9 by Shwetha GS

fixed add trait check

parent ec046d19
...@@ -332,7 +332,7 @@ public class DefaultMetadataService implements MetadataService { ...@@ -332,7 +332,7 @@ public class DefaultMetadataService implements MetadataService {
Preconditions.checkArgument(typeSystem.isRegistered(traitName), Preconditions.checkArgument(typeSystem.isRegistered(traitName),
"trait=%s should be defined in type system before it can be added", traitName); "trait=%s should be defined in type system before it can be added", traitName);
// ensure trait is not already defined // ensure trait is not already defined
Preconditions.checkArgument(getTraitNames(guid).contains(traitName), Preconditions.checkArgument(!getTraitNames(guid).contains(traitName),
"trait=%s is already defined for entity=%s", traitName, guid); "trait=%s is already defined for entity=%s", traitName, guid);
repository.addTrait(guid, traitInstance); repository.addTrait(guid, traitInstance);
......
...@@ -64,6 +64,7 @@ public class EntityJerseyResourceIT extends BaseResourceIT { ...@@ -64,6 +64,7 @@ public class EntityJerseyResourceIT extends BaseResourceIT {
private Referenceable tableInstance; private Referenceable tableInstance;
private Id tableId; private Id tableId;
private String traitName;
@BeforeClass @BeforeClass
public void setUp() throws Exception { public void setUp() throws Exception {
...@@ -120,8 +121,8 @@ public class EntityJerseyResourceIT extends BaseResourceIT { ...@@ -120,8 +121,8 @@ public class EntityJerseyResourceIT extends BaseResourceIT {
tableId = createInstance(tableInstance); tableId = createInstance(tableInstance);
Assert.fail("Was expecting an exception here "); Assert.fail("Was expecting an exception here ");
} catch (MetadataServiceException e) { } catch (MetadataServiceException e) {
Assert.assertTrue(e.getMessage() Assert.assertTrue(
.contains("\"error\":\"Cannot convert value '2014-07-11' to datatype date\"")); e.getMessage().contains("\"error\":\"Cannot convert value '2014-07-11' to datatype date\""));
} }
} }
...@@ -315,8 +316,7 @@ public class EntityJerseyResourceIT extends BaseResourceIT { ...@@ -315,8 +316,7 @@ public class EntityJerseyResourceIT extends BaseResourceIT {
private void addNewType() throws Exception { private void addNewType() throws Exception {
HierarchicalTypeDefinition<ClassType> testTypeDefinition = HierarchicalTypeDefinition<ClassType> testTypeDefinition =
TypesUtil.createClassTypeDef("test", TypesUtil.createClassTypeDef("test", ImmutableList.<String>of(),
ImmutableList.<String>of(),
TypesUtil.createRequiredAttrDef("name", DataTypes.STRING_TYPE), TypesUtil.createRequiredAttrDef("name", DataTypes.STRING_TYPE),
TypesUtil.createRequiredAttrDef("description", DataTypes.STRING_TYPE)); TypesUtil.createRequiredAttrDef("description", DataTypes.STRING_TYPE));
...@@ -349,7 +349,7 @@ public class EntityJerseyResourceIT extends BaseResourceIT { ...@@ -349,7 +349,7 @@ public class EntityJerseyResourceIT extends BaseResourceIT {
@Test(dependsOnMethods = "testGetTraitNames") @Test(dependsOnMethods = "testGetTraitNames")
public void testAddTrait() throws Exception { public void testAddTrait() throws Exception {
final String traitName = "PII_Trait"; traitName = "PII_Trait" + randomString();
HierarchicalTypeDefinition<TraitType> piiTrait = HierarchicalTypeDefinition<TraitType> piiTrait =
TypesUtil.createTraitTypeDef(traitName, ImmutableList.<String>of()); TypesUtil.createTraitTypeDef(traitName, ImmutableList.<String>of());
String traitDefinitionAsJSON = TypesSerialization$.MODULE$.toJson(piiTrait, true); String traitDefinitionAsJSON = TypesSerialization$.MODULE$.toJson(piiTrait, true);
...@@ -469,7 +469,6 @@ public class EntityJerseyResourceIT extends BaseResourceIT { ...@@ -469,7 +469,6 @@ public class EntityJerseyResourceIT extends BaseResourceIT {
@Test(dependsOnMethods = "testAddTrait") @Test(dependsOnMethods = "testAddTrait")
public void testDeleteTrait() throws Exception { public void testDeleteTrait() throws Exception {
final String traitName = "PII_Trait";
final String guid = tableId._getId(); final String guid = tableId._getId();
ClientResponse clientResponse = service ClientResponse clientResponse = service
...@@ -510,7 +509,8 @@ public class EntityJerseyResourceIT extends BaseResourceIT { ...@@ -510,7 +509,8 @@ public class EntityJerseyResourceIT extends BaseResourceIT {
JSONObject response = new JSONObject(responseAsString); JSONObject response = new JSONObject(responseAsString);
Assert.assertNotNull(response.get(MetadataServiceClient.ERROR)); Assert.assertNotNull(response.get(MetadataServiceClient.ERROR));
Assert.assertEquals(response.getString(MetadataServiceClient.ERROR), "trait=" + traitName + " should be defined in type system before it can be deleted"); Assert.assertEquals(response.getString(MetadataServiceClient.ERROR),
"trait=" + traitName + " should be defined in type system before it can be deleted");
Assert.assertNotNull(response.get(MetadataServiceClient.STACKTRACE)); Assert.assertNotNull(response.get(MetadataServiceClient.STACKTRACE));
} }
...@@ -518,6 +518,10 @@ public class EntityJerseyResourceIT extends BaseResourceIT { ...@@ -518,6 +518,10 @@ public class EntityJerseyResourceIT extends BaseResourceIT {
return RandomStringUtils.random(10); return RandomStringUtils.random(10);
} }
private String randomString() {
return RandomStringUtils.randomAlphanumeric(10);
}
@Test @Test
public void testUTF8() throws Exception { public void testUTF8() throws Exception {
String classType = random(); String classType = random();
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment