/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.oozie.command.wf;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import org.apache.hadoop.conf.Configuration;
import org.apache.oozie.local.LocalOozie;
import org.apache.oozie.client.OozieClient;
import org.apache.oozie.command.CommandException;
import org.apache.oozie.test.XDataTestCase;
import org.apache.oozie.util.XConfiguration;
import org.apache.oozie.service.XLogService;
public class TestSubmitXCommand extends XDataTestCase {
@Override
protected void setUp() throws Exception {
super.setUp();
setSystemProperty(XLogService.LOG4J_FILE, "oozie-log4j.properties");
LocalOozie.start();
}
@Override
protected void tearDown() throws Exception {
LocalOozie.stop();
super.tearDown();
}
public void testSubmitReservedVars() throws Exception {
Configuration conf = new XConfiguration();
String appPath = getTestCaseDir();
String appXml = "<workflow-app xmlns='uri:oozie:workflow:0.1' name='map-reduce-wf'> " + "<start to='end' /> "
+ "<end name='end' /> " + "</workflow-app>";
writeToFile(appXml, appPath + "/workflow.xml");
conf.set(OozieClient.APP_PATH, "file://" + appPath + "/workflow.xml");
conf.set(OozieClient.USER_NAME, getTestUser());
conf.set("GB", "5");
SubmitXCommand sc = new SubmitXCommand(conf, "UNIT_TESTING");
try {
sc.call();
fail("WF job submission should fail with reserved variable definitions.");
}
catch (CommandException ce) {
}
}
public void testAppPathIsDir() throws Exception {
Configuration conf = new XConfiguration();
String appPath = getTestCaseDir();
String appXml = "<workflow-app xmlns='uri:oozie:workflow:0.1' name='map-reduce-wf'> " + "<start to='end' /> "
+ "<end name='end' /> " + "</workflow-app>";
writeToFile(appXml, appPath + "/workflow.xml");
conf.set(OozieClient.APP_PATH, "file://" + appPath);
conf.set(OozieClient.USER_NAME, getTestUser());
SubmitXCommand sc = new SubmitXCommand(conf, "UNIT_TESTING");
try {
sc.call();
}
catch (CommandException ce) {
fail("Should succeed");
}
}
public void testAppPathIsFile1() throws Exception {
Configuration conf = new XConfiguration();
String appPath = getTestCaseDir();
String appXml = "<workflow-app xmlns='uri:oozie:workflow:0.1' name='map-reduce-wf'> " + "<start to='end' /> "
+ "<end name='end' /> " + "</workflow-app>";
writeToFile(appXml, appPath + "/workflow.xml");
conf.set(OozieClient.APP_PATH, "file://" + appPath + "/workflow.xml");
conf.set(OozieClient.USER_NAME, getTestUser());
SubmitXCommand sc = new SubmitXCommand(conf, "UNIT_TESTING");
try {
sc.call();
}
catch (CommandException ce) {
fail("Should succeed");
}
}
public void testAppPathIsFile2() throws Exception {
Configuration conf = new XConfiguration();
String appPath = getTestCaseDir();
String appXml = "<workflow-app xmlns='uri:oozie:workflow:0.1' name='map-reduce-wf'> " + "<start to='end' /> "
+ "<end name='end' /> " + "</workflow-app>";
writeToFile(appXml, appPath + "/test.xml");
conf.set(OozieClient.APP_PATH, "file://" + appPath + "/test.xml");
conf.set(OozieClient.USER_NAME, getTestUser());
SubmitXCommand sc = new SubmitXCommand(conf, "UNIT_TESTING");
try {
sc.call();
}
catch (CommandException ce) {
fail("Should succeed");
}
}
public void testAppPathIsFileNegative() throws Exception {
Configuration conf = new XConfiguration();
String appPath = getTestCaseDir();
String appXml = "<workflow-app xmlns='uri:oozie:workflow:0.1' name='map-reduce-wf'> " + "<start to='end' /> "
+ "<end name='end' /> " + "</workflow-app>";
writeToFile(appXml, appPath + "/test.xml");
conf.set(OozieClient.APP_PATH, "file://" + appPath + "/does_not_exist.xml");
conf.set(OozieClient.USER_NAME, getTestUser());
SubmitXCommand sc = new SubmitXCommand(conf, "UNIT_TESTING");
try {
sc.call();
fail("should fail");
}
catch (CommandException ce) {
}
}
private void writeToFile(String appXml, String appPath) throws IOException {
// TODO Auto-generated method stub
File wf = new File(appPath);
PrintWriter out = null;
try {
out = new PrintWriter(new FileWriter(wf));
out.println(appXml);
}
catch (IOException iex) {
throw iex;
}
finally {
if (out != null) {
out.close();
}
}
}
}