ODataEntityAccessorTest.java

/**
 * personium.io
 * Copyright 2014 FUJITSU LIMITED
 *
 * 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.
 */
package com.fujitsu.dc.core.model.impl.es.accessor;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.fail;

import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.json.simple.JSONObject;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;

import com.fujitsu.dc.common.es.EsClient;
import com.fujitsu.dc.common.es.EsIndex;
import com.fujitsu.dc.common.es.response.DcIndexResponse;
import com.fujitsu.dc.core.DcCoreConfig;
import com.fujitsu.dc.core.model.impl.es.ads.AdsException;
import com.fujitsu.dc.core.model.impl.es.ads.JdbcAds;
import com.fujitsu.dc.core.model.impl.es.doc.EntitySetDocHandler;
import com.fujitsu.dc.core.model.impl.es.doc.OEntityDocHandler;
import com.fujitsu.dc.test.categories.Unit;
import com.fujitsu.dc.test.jersey.DcRunner;
import com.fujitsu.dc.test.unit.core.UrlUtils;

/**
 * ODataEntityAccessorTestの単体テストケース.
 */
@RunWith(DcRunner.class)
@Category({Unit.class })
public class ODataEntityAccessorTest {

    private static final String INDEX_NAME = "index_for_test";
    private static final String TYPE_NAME = "TypeForTest";
    private static final String ROUTING_ID = "RoutingIdTest";

    private static EsClient esClient;

    /**
     * 各テスト実行前の初期化処理.
     * @throws Exception 異常が発生した場合の例外
     */
    @Before
    public void setUp() throws Exception {
        esClient = new EsClient(DcCoreConfig.getEsClusterName(), DcCoreConfig.getEsHosts());
    }

    /**
     * 各テスト実行後のクリーンアップ処理.
     * @throws Exception 異常が発生した場合の例外
     */
    @After
    public void tearDown() throws Exception {
        String esUnitPrefix = DcCoreConfig.getEsUnitPrefix();
        EsIndex index = esClient.idxUser(esUnitPrefix, INDEX_NAME);
        try {
            index.delete();
            JdbcAds ads = new JdbcAds();
            ads.deleteIndex(esUnitPrefix + "_" + INDEX_NAME);
        } catch (Exception ex) {
            System.out.println("");
        }
    }

    /**
     * 例外用Mock.
     * @author Administrator
     */
    class JdbcAdsMock extends JdbcAds {

        JdbcAdsMock() throws Exception {
            super();
        }

        @Override
        public void createEntity(String index, EntitySetDocHandler docHandler) throws AdsException {
            throw new AdsException("MockErrorCreare");
        }

        @Override
        public void updateEntity(String index, EntitySetDocHandler docHandler) throws AdsException {
            throw new AdsException("MockErrorUpdate");
        }

        @Override
        public void deleteEntity(String index, String id) throws AdsException {
            throw new AdsException("MockErrorDelete");
        }
    }

    /**
     * create処理が正常に終了する.
     */
    @Test
    public void create処理が正常に終了する() {
        // 事前準備
        String esUnitPrefix = DcCoreConfig.getEsUnitPrefix();
        EsIndex index = esClient.idxUser(esUnitPrefix, INDEX_NAME);
        assertNotNull(index);
        ODataEntityAccessor entityAccessor = new ODataEntityAccessor(index, TYPE_NAME, ROUTING_ID);
        OEntityDocHandler docHandler = createTestOEntityDocHandler();

        // データ登録実行
        DcIndexResponse response = entityAccessor.create(docHandler);

        // レスポンスのチェック
        assertNotNull(response);
        assertFalse(response.getId().equals(""));

        // マスタにデータが登録されていることを確認
        if (entityAccessor.getAds() != null) {
            JdbcAds ads = null;
            try {
                ads = new JdbcAds();
                assertEquals(1, ads.countEntity(esUnitPrefix + "_" + INDEX_NAME));
            } catch (Exception e) {
                fail(e.getMessage());
            }
        }

        // データを削除する
        entityAccessor.delete(docHandler);
    }

    /**
     * create処理にてAdsが例外を上げた場合でも正常に終了すること.
     */
    @Test
    public void create処理にてAdsが例外を上げた場合でも正常に終了すること() {
        // 事前準備
        String esUnitPrefix = DcCoreConfig.getEsUnitPrefix();
        EsIndex index = esClient.idxUser(esUnitPrefix, INDEX_NAME);
        assertNotNull(index);
        ODataEntityAccessor entityAccessor = new ODataEntityAccessor(index, TYPE_NAME, ROUTING_ID);
        OEntityDocHandler docHandler = createTestOEntityDocHandler();
        try {
            entityAccessor.setAds(new JdbcAdsMock());
        } catch (Exception e1) {
            throw new RuntimeException(e1);
        }

        // データ登録実行
        DcIndexResponse response = entityAccessor.create(docHandler);

        // レスポンスのチェック
        assertNotNull(response);
        assertFalse(response.getId().equals(""));
    }

    /**
     * update処理が正常に終了する.
     */
    @Test
    public void update処理が正常に終了する() {
        // 事前準備
        String esUnitPrefix = DcCoreConfig.getEsUnitPrefix();
        EsIndex index = esClient.idxUser(esUnitPrefix, INDEX_NAME);
        assertNotNull(index);
        ODataEntityAccessor entityAccessor = new ODataEntityAccessor(index, TYPE_NAME, ROUTING_ID);
        OEntityDocHandler docHandler = createTestOEntityDocHandler();
        DcIndexResponse createResponse = entityAccessor.create(docHandler);
        assertNotNull(createResponse);
        assertFalse(createResponse.getId().equals(""));

        // データ更新実行
        Map<String, Object> staticFields = new HashMap<String, Object>();
        staticFields.put("test", "testdata");
        docHandler.setStaticFields(staticFields);

        DcIndexResponse updateResponse = entityAccessor.update(createResponse.getId(), docHandler);

        // レスポンスのチェック
        assertNotNull(updateResponse);
        assertEquals(createResponse.getId(), updateResponse.getId());

        // マスタにデータが更新されていることを確認
        if (entityAccessor.getAds() != null) {
            JdbcAds ads = null;
            try {
                ads = new JdbcAds();
                List<JSONObject> list = ads.getEntityList(esUnitPrefix + "_" + INDEX_NAME, 0, 1);
                assertEquals(1, list.size());
                assertEquals(JSONObject.toJSONString(staticFields), ((JSONObject) list.get(0).get("source")).get("s"));
            } catch (Exception e) {
                fail(e.getMessage());
            }
        }

        // データを削除する
        entityAccessor.delete(docHandler);
    }

    /**
     * update処理にてAdsが例外を上げた場合でも正常に終了すること.
     */
    @Test
    public void update処理にてAdsが例外を上げた場合でも正常に終了すること() {
        // 事前準備
        String esUnitPrefix = DcCoreConfig.getEsUnitPrefix();
        EsIndex index = esClient.idxUser(esUnitPrefix, INDEX_NAME);
        assertNotNull(index);
        ODataEntityAccessor entityAccessor = new ODataEntityAccessor(index, TYPE_NAME, ROUTING_ID);
        OEntityDocHandler docHandler = createTestOEntityDocHandler();
        DcIndexResponse createResponse = entityAccessor.create(docHandler);
        assertNotNull(createResponse);
        assertFalse(createResponse.getId().equals(""));
        try {
            entityAccessor.setAds(new JdbcAdsMock());
        } catch (Exception e1) {
            throw new RuntimeException(e1);
        }

        // データ更新実行
        Map<String, Object> staticFields = new HashMap<String, Object>();
        staticFields.put("test", "testdata");
        docHandler.setStaticFields(staticFields);

        DcIndexResponse updateResponse = entityAccessor.update(createResponse.getId(), docHandler);

        // レスポンスのチェック
        assertNotNull(updateResponse);
        assertEquals(createResponse.getId(), updateResponse.getId());
    }

    /**
     * delete処理が正常に終了する.
     */
    @Test
    public void delete処理が正常に終了する() {
        // 事前準備
        String esUnitPrefix = DcCoreConfig.getEsUnitPrefix();
        EsIndex index = esClient.idxUser(esUnitPrefix, INDEX_NAME);
        assertNotNull(index);
        ODataEntityAccessor entityAccessor = new ODataEntityAccessor(index, TYPE_NAME, ROUTING_ID);
        OEntityDocHandler docHandler = createTestOEntityDocHandler();
        DcIndexResponse response = entityAccessor.create(docHandler);
        assertNotNull(response);
        assertFalse(response.getId().equals(""));

        // データを削除する
        entityAccessor.delete(docHandler);

        // データが削除されていることを確認する
        if (entityAccessor.getAds() != null) {
            JdbcAds ads = null;
            try {
                ads = new JdbcAds();
                assertEquals(0, ads.countEntity(esUnitPrefix + "_" + INDEX_NAME));
            } catch (Exception e) {
                fail(e.getMessage());
            }
        }
    }

    /**
     * delete処理にてAdsが例外を上げた場合でも正常に終了すること.
     */
    @Test
    public void delete処理にてAdsが例外を上げた場合でも正常に終了すること() {
        // 事前準備
        String esUnitPrefix = DcCoreConfig.getEsUnitPrefix();
        EsIndex index = esClient.idxUser(esUnitPrefix, INDEX_NAME);
        assertNotNull(index);
        ODataEntityAccessor entityAccessor = new ODataEntityAccessor(index, TYPE_NAME, ROUTING_ID);
        OEntityDocHandler docHandler = createTestOEntityDocHandler();
        DcIndexResponse response = entityAccessor.create(docHandler);
        assertNotNull(response);
        assertFalse(response.getId().equals(""));
        try {
            entityAccessor.setAds(new JdbcAdsMock());
        } catch (Exception e1) {
            throw new RuntimeException(e1);
        }

        // データを削除する
        entityAccessor.delete(docHandler);
    }

    /**
     * OEntityDocHandlerを生成する.
     * @return
     */
    private OEntityDocHandler createTestOEntityDocHandler() {
        long dateTime = new Date().getTime();
        Map<String, Object> dynamicField = new HashMap<String, Object>();
        Map<String, Object> staticFields = new HashMap<String, Object>();
        Map<String, Object> hiddenFields = new HashMap<String, Object>();
        Map<String, Object> link = new HashMap<String, Object>();
        OEntityDocHandler docHandler = new OEntityDocHandler();
        docHandler.setType("testType");
        docHandler.setCellId("testCellId");
        docHandler.setBoxId("testBoxId");
        docHandler.setNodeId("testNodeId");
        docHandler.setPublished(dateTime);
        docHandler.setUpdated(dateTime);
        docHandler.setDynamicFields(dynamicField);
        docHandler.setStaticFields(staticFields);
        docHandler.setHiddenFields(hiddenFields);
        docHandler.setManyToOnelinkId(link);
        String url = UrlUtils.getBaseUrl() + "#" + INDEX_NAME;
        hiddenFields.put("Owner", url);
        docHandler.resolveUnitUserName(hiddenFields);
        return docHandler;
    }
}