[policy/drools-pdp.git] /

/*-
 * ============LICENSE_START=======================================================
 * feature-session-persistence
 * ================================================================================
 * Copyright (C) 2017-2018, 2020 AT&T Intellectual Property. All rights reserved.
 * ================================================================================
 * Licensed 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.
 * ============LICENSE_END=========================================================
 */

package org.onap.policy.drools.persistence;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.doThrow;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import java.util.HashMap;
import java.util.Map;
import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import javax.persistence.EntityTransaction;
import javax.persistence.Persistence;
import org.junit.After;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;

public class JpaDroolsSessionConnectorTest {

    private EntityManagerFactory emf;
    private JpaDroolsSessionConnector conn;

    @BeforeClass
    public static void setUpBeforeClass() {
        System.setProperty("com.arjuna.ats.arjuna.objectstore.objectStoreDir", "target/tm");
        System.setProperty("ObjectStoreEnvironmentBean.objectStoreDir", "target/tm");
    }

    /**
     * Setup.
     *
     * @throws Exception exception
     */
    @Before
    public void setUp() throws Exception {
        Map<String, Object> propMap = new HashMap<>();

        propMap.put("javax.persistence.jdbc.driver", "org.h2.Driver");
        propMap.put("javax.persistence.jdbc.url", "jdbc:h2:mem:JpaDroolsSessionConnectorTest");

        emf = Persistence.createEntityManagerFactory("junitDroolsSessionEntityPU", propMap);

        conn = new JpaDroolsSessionConnector(emf);
    }

    @After
    public void tearDown() {
        // this will cause the memory db to be dropped
        emf.close();
    }

    @Test
    public void testGet() {
        /*
         * Load up the DB with some data.
         */

        addSession("nameA", 10);
        addSession("nameY", 20);

        /*
         * Now test the functionality.
         */

        // not found
        assertNull(conn.get("unknown"));

        assertEquals("{name=nameA, id=10}", conn.get("nameA").toString());

        assertEquals("{name=nameY, id=20}", conn.get("nameY").toString());
    }

    @Test(expected = IllegalArgumentException.class)
    public void testGet_NewEx() {
        EntityManagerFactory emf = mock(EntityManagerFactory.class);
        EntityManager em = mock(EntityManager.class);

        when(emf.createEntityManager()).thenReturn(em);
        when(em.find(any(), any())).thenThrow(new IllegalArgumentException("expected exception"));

        conn = new JpaDroolsSessionConnector(emf);
        conn.get("xyz");
    }

    @Test(expected = IllegalArgumentException.class)
    public void testGet_FindEx() {
        EntityManagerFactory emf = mock(EntityManagerFactory.class);
        EntityManager em = mock(EntityManager.class);
        EntityTransaction tr = mock(EntityTransaction.class);

        when(emf.createEntityManager()).thenReturn(em);
        when(em.getTransaction()).thenReturn(tr);
        when(em.find(any(), any())).thenThrow(new IllegalArgumentException("expected exception"));

        new JpaDroolsSessionConnector(emf).get("xyz");
    }

    @Test(expected = IllegalArgumentException.class)
    public void testGet_FindEx_CloseEx() {
        EntityManagerFactory emf = mock(EntityManagerFactory.class);
        EntityManager em = mock(EntityManager.class);
        EntityTransaction tr = mock(EntityTransaction.class);

        when(emf.createEntityManager()).thenReturn(em);
        when(em.getTransaction()).thenReturn(tr);
        when(em.find(any(), any())).thenThrow(new IllegalArgumentException("expected exception"));
        doThrow(new IllegalArgumentException("expected exception #2")).when(em).close();

        new JpaDroolsSessionConnector(emf).get("xyz");
    }

    @Test
    public void testReplace_Existing() {
        addSession("nameA", 10);

        DroolsSessionEntity sess = new DroolsSessionEntity("nameA", 30);

        conn.replace(sess);

        // id should be changed
        assertEquals(sess.toString(), conn.get("nameA").toString());
    }

    @Test
    public void testReplace_New() {
        DroolsSessionEntity sess = new DroolsSessionEntity("nameA", 30);

        conn.replace(sess);

        assertEquals(sess.toString(), conn.get("nameA").toString());
    }

    @Test
    public void testAdd() {
        DroolsSessionEntity sess = new DroolsSessionEntity("nameA", 30);

        conn.replace(sess);

        assertEquals(sess.toString(), conn.get("nameA").toString());
    }

    @Test
    public void testUpdate() {
        addSession("nameA", 10);

        DroolsSessionEntity sess = new DroolsSessionEntity("nameA", 30);

        conn.replace(sess);

        // id should be changed
        assertEquals("{name=nameA, id=30}", conn.get("nameA").toString());
    }

    /**
     * Adds a session to the DB.
     *
     * @param sessnm session name
     * @param sessid session id
     */
    private void addSession(String sessnm, int sessid) {
        EntityManager em = emf.createEntityManager();

        try (EntityMgrTrans trans = new EntityMgrTrans(em)) {
            DroolsSessionEntity ent = new DroolsSessionEntity();

            ent.setSessionName(sessnm);
            ent.setSessionId(sessid);

            em.persist(ent);

            trans.commit();
        }
    }
}