/*
 *    GeoTools - The Open Source Java GIS Toolkit
 *    http://geotools.org
 *
 *    (C) 2008, Open Source Geospatial Foundation (OSGeo)
 *
 *    This library is free software; you can redistribute it and/or
 *    modify it under the terms of the GNU Lesser General Public
 *    License as published by the Free Software Foundation;
 *    version 2.1 of the License.
 *
 *    This library 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
 *    Lesser General Public License for more details.
 */
package org.geotools.data.ws;

import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;

import java.io.File;
import java.io.IOException;
import java.io.Serializable;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.HashMap;
import java.util.Map;

import org.geotools.data.ws.XmlDataStore;
import org.geotools.data.ws.WSDataStoreFactory;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

public class WSDataStoreFactoryTest {

    private static final String BASE_DIRECTORY = "./src/test/resources/test-data/";

    private WSDataStoreFactory dsf;

    private Map<String, Serializable> params;

    @Before
    public void setUp() throws Exception {
        dsf = new WSDataStoreFactory();
        params = new HashMap<String, Serializable>();
    }

    @After
    public void tearDown() throws Exception {
        dsf = null;
        params = null;
    }

    @Test
    public void testCanProcess() {
        // URL not set
        assertFalse(dsf.canProcess(params));

        params.put(WSDataStoreFactory.GET_CONNECTION_URL.key,
                "http://someserver.example.org/wfs?request=GetCapabilities");
        assertFalse(dsf.canProcess(params));

        params.put(WSDataStoreFactory.TEMPLATE_NAME.key, "request.ftl");
        assertFalse(dsf.canProcess(params));
        
        try {
            params.put(WSDataStoreFactory.TEMPLATE_DIRECTORY.key, new URL(
                    "file:./src/test/resources/test-data"));
        } catch (MalformedURLException e) {
            fail(e.getMessage());
        }
        assertFalse(dsf.canProcess(params));

        try {
            params.put(WSDataStoreFactory.CAPABILITIES_FILE_LOCATION.key, new URL(
                    "file:./src/test/resources/test-data/ws_capabilities.xml"));
        } catch (MalformedURLException e) {
            fail(e.getMessage());
        }
        assertTrue(dsf.canProcess(params));

        params.put(WSDataStoreFactory.TIMEOUT.key, "30000");
        assertTrue(dsf.canProcess(params));       
    }

    @Test
    public void testCreateDataStoreWS() throws IOException {
        String capabilitiesFile;
        capabilitiesFile = "ws_capabilities_equals_removed.xml";
        testCreateDataStore_WS(capabilitiesFile);
    }

    private void testCreateDataStore_WS(final String capabilitiesFile) throws IOException {
         final WSDataStoreFactory dsf = new WSDataStoreFactory(); 
        Map<String, Serializable> params = new HashMap<String, Serializable>();
         
            
        File file = new File(BASE_DIRECTORY + capabilitiesFile);
        if (!file.exists()) {
            throw new IllegalArgumentException(capabilitiesFile + " not found");
        }    
        URL url = file.toURL();
       
        params.put(WSDataStoreFactory.GET_CONNECTION_URL.key, url);
        params.put(WSDataStoreFactory.TEMPLATE_DIRECTORY.key, new URL("file:" + BASE_DIRECTORY));
        params.put(WSDataStoreFactory.TEMPLATE_NAME.key, "request.ftl");
        params.put(WSDataStoreFactory.CAPABILITIES_FILE_LOCATION.key, url);

        XmlDataStore dataStore = dsf.createDataStore(params);
        assertTrue(dataStore instanceof WS_DataStore);
    }
}