hoot.services.controllers.ingest.RasterToTilesResourceTest.java Source code

Introduction

Here is the source code for hoot.services.controllers.ingest.RasterToTilesResourceTest.java

Source

/*
 * This file is part of Hootenanny.
 *
 * Hootenanny is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 * --------------------------------------------------------------------
 *
 * The following copyright notices are generated automatically. If you
 * have a new notice to add, please use the format:
 * " * @copyright Copyright ..."
 * This will properly maintain the copyright information. DigitalGlobe
 * copyrights will be updated automatically.
 *
 * @copyright Copyright (C) 2015, 2016 DigitalGlobe (http://www.digitalglobe.com/)
 */
package hoot.services.controllers.ingest;

import static hoot.services.HootProperties.*;

import java.io.File;
import java.lang.reflect.Method;

import org.apache.commons.io.FileUtils;
import org.json.simple.JSONArray;
import org.json.simple.JSONObject;
import org.json.simple.parser.JSONParser;
import org.junit.AfterClass;
import org.junit.Assert;
import org.junit.Test;
import org.junit.experimental.categories.Category;

import hoot.services.UnitTest;
import hoot.services.utils.HootCustomPropertiesSetter;

public class RasterToTilesResourceTest {
    private static final File homeFolder;
    private static final String tileServerPath;
    private static final String original_HOME_FOLDER;
    private static final String original_TILE_SERVER_PATH;

    static {
        try {
            original_HOME_FOLDER = HOME_FOLDER;
            homeFolder = new File(FileUtils.getTempDirectory(), "RasterToTilesResourceTest");
            FileUtils.forceMkdir(homeFolder);
            Assert.assertTrue(homeFolder.exists());
            HootCustomPropertiesSetter.setProperty("HOME_FOLDER", homeFolder.getAbsolutePath());

            //tileServerPath=$(homeFolder)/ingest/processed

            original_TILE_SERVER_PATH = TILE_SERVER_PATH;
            File processedFolder = new File(homeFolder, "ingest/processed");
            FileUtils.forceMkdir(processedFolder);
            Assert.assertTrue(processedFolder.exists());
            tileServerPath = processedFolder.getAbsolutePath();
            Assert.assertNotNull(tileServerPath);
            Assert.assertTrue(!tileServerPath.isEmpty());
            HootCustomPropertiesSetter.setProperty("TILE_SERVER_PATH", processedFolder.getAbsolutePath());
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    @AfterClass
    public static void oneTimeTearDown() throws Exception {
        FileUtils.deleteDirectory(homeFolder);
        HootCustomPropertiesSetter.setProperty("HOME_FOLDER", original_HOME_FOLDER);
        HootCustomPropertiesSetter.setProperty("TILE_SERVER_PATH", original_TILE_SERVER_PATH);
    }

    @Test
    @Category(UnitTest.class)
    public void TestIngestOSMResource() throws Exception {
        String processScriptName = RASTER_TO_TILES;
        Assert.assertNotNull(processScriptName);
        Assert.assertTrue(!processScriptName.isEmpty());

        RasterToTilesService rts = new RasterToTilesService();

        JSONObject oExpected = new JSONObject();
        oExpected.put("caller", "RasterToTilesService");
        oExpected.put("exec", processScriptName);

        JSONArray params = new JSONArray();
        JSONObject param = new JSONObject();
        param.put("RASTER_OUTPUT_DIR", tileServerPath);
        params.add(param);

        param = new JSONObject();
        param.put("INPUT", "test");
        params.add(param);

        param = new JSONObject();
        param.put("ZOOM_LIST", "0-1 2-3");
        params.add(param);

        param = new JSONObject();
        param.put("RASTER_SIZE", "500");
        params.add(param);

        param = new JSONObject();
        param.put("MAP_ID", "1");
        params.add(param);

        oExpected.put("params", params);

        oExpected.put("exectype", "make");
        oExpected.put("erroraswarning", "true");

        Method createCommandMethod = RasterToTilesService.class.getDeclaredMethod("createCommand", String.class,
                String.class, int.class, long.class);

        createCommandMethod.setAccessible(true);

        String actual = (String) createCommandMethod.invoke(rts, "test", "0-1 2-3", 500, 1);

        JSONParser parser = new JSONParser();
        JSONObject actualObj = (JSONObject) parser.parse(actual);

        Assert.assertEquals(oExpected, actualObj);
    }

    @Test
    @Category(UnitTest.class)
    public void TestGetZoomInfo() throws Exception {
        RasterToTilesService rts = new RasterToTilesService();
        Method getZoomInfoMethod = RasterToTilesService.class.getDeclaredMethod("getZoomInfo", double.class);
        getZoomInfoMethod.setAccessible(true);

        JSONObject oActual = (JSONObject) getZoomInfoMethod.invoke(rts, 0.025);

        Assert.assertEquals("0-1 2-3 4-5 6-7 8-9 10-11 12-13 14-15 16-17", oActual.get("zoomlist").toString());
        Assert.assertEquals(500, oActual.get("rastersize"));
    }
}