/* (c) 2016 Open Source Geospatial Foundation - all rights reserved
 * This code is licensed under the GPL 2.0 license, available at the root
 * application directory.
 */
package org.geoserver.backuprestore.rest;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;

import java.util.logging.Level;

import org.geoserver.backuprestore.BackupRestoreTestSupport;
import org.geoserver.platform.resource.Resource;
import org.junit.Test;
import org.springframework.mock.web.MockHttpServletResponse;
import org.springframework.util.Assert;

import net.sf.json.JSONObject;

/**
 * 
 * @author Alessio Fabiani, GeoSolutions
 *
 */
public class RESTRestoreTest extends BackupRestoreTestSupport {
    @Test
    public void testNewRestore() throws Exception {
        Resource archiveFile = file("geoserver-alfa2-backup.zip");
        
        if (archiveFile == null) {
            LOGGER.log(Level.WARNING, "Could not find source archive file.");
        } else {
            String json = 
                    "{\"restore\": {" + 
                    "   \"archiveFile\": \""+archiveFile.path()+"\", " +  
                    "   \"options\": { \"option\": [\"BK_DRY_RUN=true\", \"BK_BEST_EFFORT=true\"] }" +
                    "  }" + 
                    "}";
            
            JSONObject restore = postNewRestore(json);
            
            Assert.notNull(restore);

            JSONObject execution = readExecutionStatus(restore.getJSONObject("execution").getLong("id"));

            assertTrue("STARTED".equals(execution.getString("status")) || 
                    "STARTING".equals(execution.getString("status")));

            while ("STARTED".equals(execution.getString("status")) || 
                    "STARTING".equals(execution.getString("status"))) {
                execution = readExecutionStatus(execution.getLong("id"));

                Thread.sleep(100);
            }

            assertTrue("COMPLETED".equals(execution.getString("status")));   
        }
    }

    JSONObject postNewRestore(String body) throws Exception {
        MockHttpServletResponse resp = postAsServletResponse("/rest/br/restore", body, "application/json");

        assertEquals(201, resp.getStatus());
        assertEquals("application/json", resp.getContentType());

        JSONObject json = (JSONObject) json(resp);
        JSONObject execution = json.getJSONObject("restore");

        assertNotNull(execution);

        return execution;
    }

    JSONObject readExecutionStatus(long executionId) throws Exception {
        JSONObject json = (JSONObject) getAsJSON("/rest/br/restore/" + executionId + ".json");

        JSONObject restore = json.getJSONObject("restore");

        assertNotNull(restore);
        
        JSONObject execution = restore.getJSONObject("execution");

        assertNotNull(execution);

        return execution;
    }
}