core/src/test/java/org/apache/oozie/executor/jpa/TestSLARegistrationQueryExecutor.java

/**
 * 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 javax.persistence.EntityManager;
import javax.persistence.Query;
import org.apache.oozie.client.event.SLAEvent.SLAStatus;
import org.apache.oozie.executor.jpa.SLARegistrationQueryExecutor.SLARegQuery;
import org.apache.oozie.service.JPAService;
import org.apache.oozie.service.Services;
import org.apache.oozie.sla.SLARegistrationBean;
import org.apache.oozie.test.XDataTestCase;

public class TestSLARegistrationQueryExecutor extends XDataTestCase {
    Services services;
    JPAService jpaService;

    @Override
    protected void setUp() throws Exception {
        super.setUp();
        services = new Services();
        services.init();
        jpaService = Services.get().get(JPAService.class);
    }

    @Override
    protected void tearDown() throws Exception {
        services.destroy();
        super.tearDown();
    }

    public void testGetUpdateQuery() throws Exception {
        EntityManager em = jpaService.getEntityManager();
        SLARegistrationBean bean = addRecordToSLARegistrationTable("test-application", SLAStatus.MET);

        // UPDATE_SLA_REG_ALL
        Query query = SLARegistrationQueryExecutor.getInstance().getUpdateQuery(SLARegQuery.UPDATE_SLA_REG_ALL, bean,
                em);

        assertEquals(query.getParameterValue("jobId"), bean.getId());
        assertEquals(query.getParameterValue("nominalTime"), bean.getNominalTimestamp());
        assertEquals(query.getParameterValue("expectedStartTime"), bean.getExpectedStartTimestamp());
        assertEquals(query.getParameterValue("expectedEndTime"), bean.getExpectedEndTimestamp());
        assertEquals(query.getParameterValue("expectedDuration"), bean.getExpectedDuration());
        assertEquals(query.getParameterValue("slaConfig"), bean.getSlaConfig());
        assertEquals(query.getParameterValue("notificationMsg"), bean.getNotificationMsg());
        assertEquals(query.getParameterValue("upstreamApps"), bean.getUpstreamApps());
        assertEquals(query.getParameterValue("appType"), bean.getAppType().toString());
        assertEquals(query.getParameterValue("appName"), bean.getAppName());
        assertEquals(query.getParameterValue("user"), bean.getUser());
        assertEquals(query.getParameterValue("parentId"), bean.getParentId());
        assertEquals(query.getParameterValue("jobData"), bean.getJobData());

        em.close();
    }

    public void testGetSelectQuery() throws Exception {
        EntityManager em = jpaService.getEntityManager();
        SLARegistrationBean bean = addRecordToSLARegistrationTable("test-application", SLAStatus.MET);
        // GET_SLA_REG_ALL
        Query query = SLARegistrationQueryExecutor.getInstance().getSelectQuery(SLARegQuery.GET_SLA_REG_ALL, em,
                bean.getId());
        assertEquals(query.getParameterValue("id"), bean.getId());

        // GET_WORKFLOW_SUSPEND
        query = SLARegistrationQueryExecutor.getInstance().getSelectQuery(SLARegQuery.GET_SLA_REG_ON_RESTART, em,
                bean.getId());
        assertEquals(query.getParameterValue("id"), bean.getId());
    }

    public void testGet() throws Exception {

        SLARegistrationBean bean = addRecordToSLARegistrationTable("test-application", SLAStatus.MET);
        //GET_SLA_REG_ON_RESTART
        SLARegistrationBean retBean = SLARegistrationQueryExecutor.getInstance().get(
                SLARegQuery.GET_SLA_REG_ON_RESTART, bean.getId());
        assertEquals(bean.getJobData(), retBean.getJobData());
        assertEquals(bean.getSlaConfig(), retBean.getSlaConfig());
        assertEquals(bean.getUpstreamApps(), retBean.getUpstreamApps());
        assertEquals(bean.getNotificationMsg(), retBean.getNotificationMsg());
        assertNull(retBean.getAppName());
        assertNull(retBean.getExpectedEnd());
        assertNull(retBean.getExpectedStart());
        assertNull(retBean.getCreatedTime());
        assertNull(retBean.getNominalTime());
        assertNull(retBean.getUser());
        assertNull(retBean.getParentId());
        //GET_SLA_REG_ALL
        retBean = SLARegistrationQueryExecutor.getInstance().get(SLARegQuery.GET_SLA_REG_ALL, bean.getId());
        assertEquals(bean.getId(), retBean.getId());
        assertEquals(bean.getAppName(), retBean.getAppName());
        assertEquals(bean.getAppType(), retBean.getAppType());
        assertEquals(bean.getExpectedDuration(), retBean.getExpectedDuration());
        assertEquals(bean.getExpectedStart().getTime(), retBean.getExpectedStart().getTime());
        assertEquals(bean.getExpectedEnd().getTime(), retBean.getExpectedEnd().getTime());
        assertEquals(bean.getCreatedTime().getTime(), retBean.getCreatedTime().getTime());
        assertEquals(bean.getNominalTime().getTime(), retBean.getNominalTime().getTime());
        assertEquals(bean.getNotificationMsg(), retBean.getNotificationMsg());
        assertEquals(bean.getJobData(), retBean.getJobData());
        assertEquals(bean.getParentId(), retBean.getParentId());
        assertEquals(bean.getSlaConfig(), retBean.getSlaConfig());
        assertEquals(bean.getUpstreamApps(), retBean.getUpstreamApps());
        assertEquals(bean.getUser(), retBean.getUser());
}

    public void testExecuteUpdate() throws Exception {
        // TODO
    }

    public void testGetList() throws Exception {
        // TODO
    }

    public void testInsert() throws Exception {
        // TODO
    }
}