PythonCatalogModTest.java

package org.geoserver.python;

import java.sql.SQLException;

import org.geoserver.catalog.Catalog;
import org.geoserver.catalog.DataStoreInfo;
import org.geoserver.catalog.FeatureTypeInfo;
import org.geoserver.catalog.NamespaceInfo;
import org.geoserver.catalog.impl.CatalogImpl;
import org.geoserver.platform.GeoServerExtensions;
import org.geotools.data.DataStore;
import org.geotools.feature.simple.SimpleFeatureTypeBuilder;
import org.h2.tools.DeleteDbFiles;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.opengis.feature.simple.SimpleFeatureType;
import org.opengis.feature.type.FeatureType;
import org.springframework.context.ApplicationContext;

import static org.easymock.EasyMock.createNiceMock;
import static org.easymock.EasyMock.expect;
import static org.easymock.EasyMock.replay;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;

public class PythonCatalogModTest extends PythonTestSupport {

    Catalog cat;
    
    @Before
    public void setUp() throws Exception {
        DeleteDbFiles.execute("target", "bar", false);
        
        cat = new CatalogImpl();
        
        ApplicationContext context = createNiceMock(ApplicationContext.class);
        expect(context.getBean("catalog")).andReturn(cat).anyTimes();
        replay(context);
        
        new GeoServerExtensions().setApplicationContext(context);
    }
    
    @After
    public void tearDown() throws Exception {
        new GeoServerExtensions().setApplicationContext(null);
        DeleteDbFiles.execute("target", "bar", false);
    }
    
    @Test
    public void testAddWorkspace() {
        assertEquals(0, cat.getWorkspaces().size());
        
        pi.exec("from geoserver.catalog import Workspace");
        pi.exec("ws = Workspace('foo')");
        pi.exec("ws.save()");
        
        assertEquals(1, cat.getWorkspaces().size());
        assertNotNull(cat.getWorkspaceByName("foo"));
    }
    
    @Test
    public void testSaveWorkspace() {
        testAddWorkspace();
        
        assertNotNull(cat.getWorkspaceByName("foo"));
        
        pi.exec("from geoserver.catalog import Workspace");
        pi.exec("ws = Workspace('foo')");
        pi.exec("ws.name = 'bar'");
        pi.exec("ws.save()");
  
        assertNull(cat.getWorkspaceByName("foo"));
        assertNotNull(cat.getWorkspaceByName("bar"));
    }
   
    @Test
    public void testAddStore() throws Exception {
        testAddWorkspace();
        
        assertNull(cat.getDataStoreByName("foo", "bar"));
      
        pi.exec("from geoserver.catalog import Store");
        pi.exec("st = Store('bar')");
        pi.exec("st.connectionParameters.putAll({'database':'target/bar', 'dbtype': 'h2'})");
        pi.exec("st.save()");
        
        DataStoreInfo ds = cat.getDataStoreByName("foo", "bar");
        assertNotNull(ds);
        assertNotNull(ds.getDataStore(null));
    }
    
    @Test
    public void testAddStoreNonDefaultWorkspace() throws Exception {
        testAddWorkspace();
        
        assertNull(cat.getDataStoreByName("acme", "bar"));
        
        pi.exec("from geoserver.catalog import Workspace, Store");
        pi.exec("ws = Workspace('acme')");
        pi.exec("ws.save()");
        
        pi.exec("st = Store('bar', ws)");
        pi.exec("st.connectionParameters.putAll({'database':'target/bar', 'dbtype': 'h2'})");
        pi.exec("st.save()");
        
        assertNotNull(cat.getDataStoreByName("acme", "bar"));
    }
    
    @Test
    public void testSaveStore() throws Exception {
        testAddStore();
        
        assertNull(cat.getDataStoreByName("foo", "bar").getDescription());
        
        pi.exec("from geoserver.catalog import Store");
        pi.exec("st = Store('bar', 'foo')");
        pi.exec("st.description = 'foobar'");
        pi.exec("st.save()");
        
        assertEquals("foobar", cat.getDataStoreByName("foo", "bar").getDescription());
    }
    
    @Test
    public void testAddFeatureType() throws Exception {
        testAddStore();
        
        NamespaceInfo ns = cat.getFactory().createNamespace();
        ns.setPrefix("foo");
        ns.setURI("http://foo.org");
        cat.add(ns);
        
        DataStoreInfo ds = cat.getDataStoreByName("foo", "bar");
        
        SimpleFeatureTypeBuilder tb = new SimpleFeatureTypeBuilder();
        tb.setName("baz");
        tb.add("x", Integer.class);
        tb.add("y", String.class);
        ((DataStore)ds.getDataStore(null)).createSchema(tb.buildFeatureType());
        
        assertNull(cat.getFeatureTypeByDataStore(ds, "baz"));
        
        pi.exec("from geoserver.catalog import Catalog");
        pi.exec("from geoserver.catalog import Layer");
        pi.exec("cat = Catalog()");
        pi.exec("st = cat['foo']['bar']");
        pi.exec("l = Layer('baz', st)");
        pi.exec("l.save()");
        
        FeatureTypeInfo ft = cat.getFeatureTypeByDataStore(ds, "baz"); 
        
        assertNotNull(ft);
        assertNotNull(ft.getFeatureType());
        assertNotNull(ft.getFeatureSource(null, null));
    }
    
    @Test
    public void testSaveFeatureType() throws Exception {
        testAddStore();
        
        
    }
}