/**
* 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.executor.jpa;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.apache.oozie.CoordinatorJobBean;
import org.apache.oozie.CoordinatorJobInfo;
import org.apache.oozie.client.CoordinatorJob;
import org.apache.oozie.client.OozieClient;
import org.apache.oozie.local.LocalOozie;
import org.apache.oozie.service.JPAService;
import org.apache.oozie.service.Services;
import org.apache.oozie.test.XDataTestCase;
public class TestCoordJobInfoGetJPAExecutor extends XDataTestCase {
Services services;
@Override
protected void setUp() throws Exception {
super.setUp();
services = new Services();
services.init();
cleanUpDBTables();
LocalOozie.start();
}
@Override
protected void tearDown() throws Exception {
LocalOozie.stop();
services.destroy();
super.tearDown();
}
public void testCoordJobGet() throws Exception {
CoordinatorJobBean coordinatorJob1 = addRecordToCoordJobTable(CoordinatorJob.Status.RUNNING, false, false);
addRecordToCoordJobTable(CoordinatorJob.Status.KILLED, false, false);
_testGetJobInfoForStatus();
_testGetJobInfoForGroup();
addRecordToCoordJobTable(CoordinatorJob.Status.KILLED, false, false);
_testGetJobInfoForAppName();
_testGetJobInfoForUser();
_testGetJobInfoForUserAndStatus();
_testGetJobInfoForFrequency();
_testGetJobInfoForId(coordinatorJob1.getId());
_testGetJobInfoForFrequencyAndUnit();
}
private void _testGetJobInfoForStatus() throws Exception {
JPAService jpaService = Services.get().get(JPAService.class);
assertNotNull(jpaService);
Map<String, List<String>> filter = new HashMap<String, List<String>>();
List<String> list = new ArrayList<String>();
list.add("RUNNING");
list.add("KILLED");
filter.put(OozieClient.FILTER_STATUS, list);
CoordJobInfoGetJPAExecutor coordInfoGetCmd = new CoordJobInfoGetJPAExecutor(filter, 1, 20);
CoordinatorJobInfo ret = jpaService.execute(coordInfoGetCmd);
assertNotNull(ret);
assertEquals(ret.getCoordJobs().size(), 2);
}
private void _testGetJobInfoForGroup() throws Exception {
JPAService jpaService = Services.get().get(JPAService.class);
assertNotNull(jpaService);
Map<String, List<String>> filter = new HashMap<String, List<String>>();
List<String> list = new ArrayList<String>();
list.add(getTestGroup());
filter.put(OozieClient.FILTER_GROUP, list);
CoordJobInfoGetJPAExecutor coordInfoGetCmd = new CoordJobInfoGetJPAExecutor(filter, 1, 20);
CoordinatorJobInfo ret = jpaService.execute(coordInfoGetCmd);
assertNotNull(ret);
assertEquals(ret.getCoordJobs().size(), 2);
}
private void _testGetJobInfoForAppName() throws Exception {
JPAService jpaService = Services.get().get(JPAService.class);
assertNotNull(jpaService);
Map<String, List<String>> filter = new HashMap<String, List<String>>();
List<String> list = new ArrayList<String>();
list.add("COORD-TEST");
filter.put(OozieClient.FILTER_NAME, list);
CoordJobInfoGetJPAExecutor coordInfoGetCmd = new CoordJobInfoGetJPAExecutor(filter, 1, 20);
CoordinatorJobInfo ret = jpaService.execute(coordInfoGetCmd);
assertNotNull(ret);
assertEquals(ret.getCoordJobs().size(), 3);
}
private void _testGetJobInfoForUser() throws Exception {
JPAService jpaService = Services.get().get(JPAService.class);
assertNotNull(jpaService);
Map<String, List<String>> filter = new HashMap<String, List<String>>();
List<String> list = new ArrayList<String>();
list.add(getTestUser());
filter.put(OozieClient.FILTER_USER, list);
CoordJobInfoGetJPAExecutor coordInfoGetCmd = new CoordJobInfoGetJPAExecutor(filter, 1, 20);
CoordinatorJobInfo ret = jpaService.execute(coordInfoGetCmd);
assertNotNull(ret);
assertEquals(ret.getCoordJobs().size(), 3);
}
private void _testGetJobInfoForUserAndStatus() throws Exception {
JPAService jpaService = Services.get().get(JPAService.class);
assertNotNull(jpaService);
Map<String, List<String>> filter = new HashMap<String, List<String>>();
List<String> list1 = new ArrayList<String>();
list1.add(getTestUser());
filter.put(OozieClient.FILTER_USER, list1);
List<String> list2 = new ArrayList<String>();
list2.add("KILLED");
filter.put(OozieClient.FILTER_STATUS, list2);
CoordJobInfoGetJPAExecutor coordInfoGetCmd = new CoordJobInfoGetJPAExecutor(filter, 1, 20);
CoordinatorJobInfo ret = jpaService.execute(coordInfoGetCmd);
assertNotNull(ret);
assertEquals(ret.getCoordJobs().size(), 2);
}
private void _testGetJobInfoForFrequency() throws Exception {
JPAService jpaService = Services.get().get(JPAService.class);
assertNotNull(jpaService);
Map<String, List<String>> filter = new HashMap<String, List<String>>();
List<String> frequencyList = new ArrayList<String>();
frequencyList.add("1");
filter.put(OozieClient.FILTER_FREQUENCY, frequencyList);
CoordJobInfoGetJPAExecutor coordInfoGetCmd = new CoordJobInfoGetJPAExecutor(filter, 1, 20);
CoordinatorJobInfo ret = jpaService.execute(coordInfoGetCmd);
assertNotNull(ret);
assertEquals(ret.getCoordJobs().size(), 3);
}
private void _testGetJobInfoForId(String jobId) throws Exception {
JPAService jpaService = Services.get().get(JPAService.class);
assertNotNull(jpaService);
Map<String, List<String>> filter = new HashMap<String, List<String>>();
List<String> jobIdList = new ArrayList<String>();
jobIdList.add(jobId);
filter.put(OozieClient.FILTER_ID, jobIdList);
CoordJobInfoGetJPAExecutor coordInfoGetCmd = new CoordJobInfoGetJPAExecutor(filter, 1, 20);
CoordinatorJobInfo ret = jpaService.execute(coordInfoGetCmd);
assertNotNull(ret);
assertEquals(ret.getCoordJobs().size(), 1);
}
/**
* Test to verify various combinations of frequency and time unit filters for jobs
*
* @throws Exception
*/
private void _testGetJobInfoForFrequencyAndUnit() throws Exception {
JPAService jpaService = Services.get().get(JPAService.class);
assertNotNull(jpaService);
// Test specifying frequency value as 1 minute
Map<String, List<String>> filter = new HashMap<String, List<String>>();
List<String> unitList = new ArrayList<String>();
List<String> frequencyList = new ArrayList<String>();
unitList.add("MINUTE");
filter.put(OozieClient.FILTER_UNIT, unitList);
frequencyList = new ArrayList<String>();
frequencyList.add("1");
filter.put(OozieClient.FILTER_FREQUENCY, frequencyList);
CoordJobInfoGetJPAExecutor coordInfoGetCmd = new CoordJobInfoGetJPAExecutor(filter, 1, 20);
CoordinatorJobInfo ret = jpaService.execute(coordInfoGetCmd);
assertNotNull(ret);
assertEquals(ret.getCoordJobs().size(), 0);
frequencyList.remove(0);
unitList.remove(0);
// Test specifying frequency value as 3 days
unitList.add("DAY");
filter.put(OozieClient.FILTER_UNIT, unitList);
frequencyList.add("3");
filter.put(OozieClient.FILTER_FREQUENCY, frequencyList);
coordInfoGetCmd = new CoordJobInfoGetJPAExecutor(filter, 1, 20);
ret = jpaService.execute(coordInfoGetCmd);
assertNotNull(ret);
assertEquals(ret.getCoordJobs().size(), 0);
frequencyList.remove(0);
unitList.remove(0);
// Test specifying frequency value as 1 day
unitList.add("DAY");
filter.put(OozieClient.FILTER_UNIT, unitList);
frequencyList.add("1");
filter.put(OozieClient.FILTER_FREQUENCY, frequencyList);
coordInfoGetCmd = new CoordJobInfoGetJPAExecutor(filter, 1, 20);
ret = jpaService.execute(coordInfoGetCmd);
assertNotNull(ret);
assertEquals(ret.getCoordJobs().size(), 3);
}
}