/* (c) 2014 Open Source Geospatial Foundation - all rights reserved
* (c) 2001 - 2013 OpenPlans
* This code is licensed under the GPL 2.0 license, available at the root
* application directory.
*/
package org.geoserver.script.wfs;
import static org.easymock.EasyMock.expect;
import static org.easymock.classextension.EasyMock.createNiceMock;
import static org.easymock.classextension.EasyMock.replay;
import static org.geoserver.data.test.MockData.PRIMITIVEGEOFEATURE;
import java.util.Iterator;
import javax.xml.namespace.QName;
import junit.framework.TestCase;
import net.opengis.wfs.TransactionType;
import net.opengis.wfs.WfsFactory;
import org.geoserver.script.ScriptManager;
import org.geoserver.script.wfs.TransactionDetail.Entry;
import org.geoserver.wfs.TransactionEvent;
import org.geoserver.wfs.TransactionEventType;
import org.geoserver.wfs.request.TransactionRequest;
import org.geotools.data.simple.SimpleFeatureCollection;
import com.google.common.collect.Multimap;
import com.google.common.io.Files;
public class ScriptTransctionPluginTest extends TestCase {
ScriptManager scriptMgr;
@Override
protected void setUp() throws Exception {
super.setUp();
scriptMgr = createScriptMgr();
}
ScriptManager createScriptMgr() throws Exception {
ScriptManager mgr = createNiceMock(ScriptManager.class);
expect(mgr.getWfsTxRoot()).andReturn(Files.createTempDir()).anyTimes();
replay(mgr);
return mgr;
}
public void testTransactionDetails() throws Exception {
SimpleFeatureCollection inserted = createNiceMock(SimpleFeatureCollection.class);
SimpleFeatureCollection updated = createNiceMock(SimpleFeatureCollection.class);
SimpleFeatureCollection deleted = createNiceMock(SimpleFeatureCollection.class);
replay(inserted, updated, deleted);
TransactionType t = WfsFactory.eINSTANCE.createTransactionType();
TransactionEvent e1 = new TransactionEvent(TransactionEventType.PRE_INSERT,
TransactionRequest.adapt(t), PRIMITIVEGEOFEATURE, inserted);
TransactionEvent e2 = new TransactionEvent(TransactionEventType.PRE_UPDATE,
TransactionRequest.adapt(t), PRIMITIVEGEOFEATURE, updated);
TransactionEvent e3 = new TransactionEvent(TransactionEventType.PRE_DELETE,
TransactionRequest.adapt(t), PRIMITIVEGEOFEATURE, deleted);
ScriptTransactionPlugin plugin = new ScriptTransactionPlugin(scriptMgr);
plugin.dataStoreChange(e1);
plugin.dataStoreChange(e2);
plugin.dataStoreChange(e3);
TransactionDetail detail =
(TransactionDetail) t.getExtendedProperties().get(TransactionDetail.class);
assertNotNull(detail);
Multimap<QName, Entry> entries = detail.getEntries();
assertTrue(entries.containsKey(PRIMITIVEGEOFEATURE));
Iterator<Entry> it = entries.get(PRIMITIVEGEOFEATURE).iterator();
assertTrue(it.hasNext());
Entry e = it.next();
assertEquals(TransactionEventType.PRE_INSERT, e.type);
assertEquals(inserted, e.features);
assertTrue(it.hasNext());
e = it.next();
assertEquals(TransactionEventType.PRE_UPDATE, e.type);
assertEquals(updated, e.features);
assertTrue(it.hasNext());
e = it.next();
assertEquals(TransactionEventType.PRE_DELETE, e.type);
assertEquals(deleted, e.features);
assertFalse(it.hasNext());
}
}