@Test
public void testProcessDeleteAndSchedule() throws Exception {
// Submit process with invalid property so that coord submit fails and bundle goes to failed
// state
Map<String, String> overlay = getUniqueOverlay();
String tmpFileName = overlayParametersOverTemplate(PROCESS_TEMPLATE, overlay);
Process process =
(Process) EntityType.PROCESS.getUnmarshaller().unmarshal(new File(tmpFileName));
Property prop = new Property();
prop.setName("newProp");
prop.setValue("${formatTim()}");
process.getProperties().getProperties().add(prop);
File tmpFile = getTempFile();
EntityType.PROCESS.getMarshaller().marshal(process, tmpFile);
scheduleProcess(tmpFile.getAbsolutePath(), overlay);
waitForBundleStart(Status.FAILED);
// Delete and re-submit the process with correct workflow
ClientResponse clientRepsonse =
this.service
.path("api/entities/delete/process/" + processName)
.header("Remote-User", REMOTE_USER)
.accept(MediaType.TEXT_XML)
.delete(ClientResponse.class);
assertSuccessful(clientRepsonse);
process.getWorkflow().setPath("/falcon/test/workflow");
tmpFile = getTempFile();
EntityType.PROCESS.getMarshaller().marshal(process, tmpFile);
clientRepsonse =
this.service
.path("api/entities/submitAndSchedule/process")
.header("Remote-User", REMOTE_USER)
.accept(MediaType.TEXT_XML)
.type(MediaType.TEXT_XML)
.post(ClientResponse.class, getServletInputStream(tmpFile.getAbsolutePath()));
assertSuccessful(clientRepsonse);
// Assert that new schedule creates new bundle
List<BundleJob> bundles = getBundles();
Assert.assertEquals(bundles.size(), 2);
}
private static org.apache.falcon.entity.v0.process.Process bindAttributesInTemplate(
final String processTemplate,
final Properties extensionProperties,
final String extensionName,
final String wfPath,
final String wfLibPath)
throws FalconException {
if (StringUtils.isBlank(processTemplate) || extensionProperties == null) {
throw new FalconException("Process template or properties cannot be null");
}
org.apache.falcon.entity.v0.process.Process process;
try {
Unmarshaller unmarshaller = EntityType.PROCESS.getUnmarshaller();
// Validation can be skipped for unmarshalling as we want to bind template with the
// properties.
// Vaildation is handled as part of marshalling
unmarshaller.setSchema(null);
unmarshaller.setEventHandler(
new ValidationEventHandler() {
public boolean handleEvent(ValidationEvent validationEvent) {
return true;
}
});
process =
(org.apache.falcon.entity.v0.process.Process)
unmarshaller.unmarshal(new StringReader(processTemplate));
} catch (Exception e) {
throw new FalconException(e);
}
/* For optional properties user might directly set them in the process xml and might not set it in properties
file. Before doing the submission validation is done to confirm process xml doesn't have
EXTENSION_VAR_PATTERN
*/
String processName = extensionProperties.getProperty(ExtensionProperties.JOB_NAME.getName());
if (StringUtils.isNotEmpty(processName)) {
process.setName(processName);
}
// DR process template has only one cluster
bindClusterProperties(process.getClusters().getClusters().get(0), extensionProperties);
// bind scheduling properties
String processFrequency =
extensionProperties.getProperty(ExtensionProperties.FREQUENCY.getName());
if (StringUtils.isNotEmpty(processFrequency)) {
process.setFrequency(Frequency.fromString(processFrequency));
}
String zone = extensionProperties.getProperty(ExtensionProperties.TIMEZONE.getName());
if (StringUtils.isNotBlank(zone)) {
process.setTimezone(TimeZone.getTimeZone(zone));
} else {
process.setTimezone(TimeZone.getTimeZone("UTC"));
}
bindWorkflowProperties(process.getWorkflow(), extensionName, wfPath, wfLibPath);
bindRetryProperties(process.getRetry(), extensionProperties);
bindNotificationProperties(process.getNotification(), extensionProperties);
bindACLProperties(process.getACL(), extensionProperties);
bindTagsProperties(process, extensionProperties);
bindCustomProperties(process.getProperties(), extensionProperties);
return process;
}
