public boolean typeExists(String type) {
List<ConsumerType> types = consumerTypeResource.list();
boolean found = false;
for (ConsumerType aType : types) {
found = type.equals(aType.getLabel());
if (found) {
break;
}
}
return found;
}
private void exportConsumerTypes(File baseDir) throws IOException {
File typeDir = new File(baseDir.getCanonicalPath(), "consumer_types");
typeDir.mkdir();
for (ConsumerType type : consumerTypeCurator.listAll()) {
File file = new File(typeDir.getCanonicalPath(), type.getLabel() + ".json");
FileWriter writer = new FileWriter(file);
consumerType.export(mapper, writer, type);
writer.close();
}
}
@Test
public void testCRUD() {
String type = "JarJar-" + System.currentTimeMillis();
ConsumerType consumerType = new ConsumerType();
consumerType.setLabel(type);
assertEquals("the label getter works", type, consumerType.getLabel());
// Ensure it is not there
assertTrue("The type already exists", !this.typeExists(type));
// Create it
consumerType = consumerTypeResource.create(consumerType);
// Make sure it is there
assertTrue("The type was not found", this.typeExists(type));
// Find it by ID
ConsumerType consumerType2 = consumerTypeResource.getConsumerType(consumerType.getId());
assertNotNull("The type was not found by ID", consumerType2);
// Update it
String type2 = "JarJar2-" + System.currentTimeMillis();
consumerType.setLabel(type2);
consumerTypeResource.update(consumerType);
assertTrue("The update was not found", this.typeExists(type2));
// Delete It
consumerTypeResource.deleteConsumerType(consumerType.getId());
// Make sure it is not there
assertTrue("The type was found after a delete", !this.typeExists(type));
}