public MockCatalogBuilder workspace(String name, String uri) {
String wsId = newId();
String nsId = newId();
final WorkspaceInfo ws = createNiceMock(WorkspaceInfo.class);
workspaces.add(ws);
expect(ws.getId()).andReturn(wsId).anyTimes();
expect(ws.getName()).andReturn(name).anyTimes();
expect(ws.getMetadata()).andReturn(new MetadataMap()).anyTimes();
expect(catalog.getWorkspace(wsId)).andReturn(ws).anyTimes();
expect(catalog.getWorkspaceByName(name)).andReturn(ws).anyTimes();
final NamespaceInfo ns = createNiceMock(NamespaceInfo.class);
namespaces.add(ns);
expect(ns.getId()).andReturn(nsId).anyTimes();
expect(ns.getName()).andReturn(name).anyTimes();
expect(ns.getPrefix()).andReturn(name).anyTimes();
expect(ns.getMetadata()).andReturn(new MetadataMap()).anyTimes();
expect(catalog.getNamespace(nsId)).andReturn(ns).anyTimes();
expect(catalog.getNamespaceByPrefix(name)).andReturn(ns).anyTimes();
expect(catalog.getNamespaceByURI(uri)).andReturn(ns).anyTimes();
ws.accept((CatalogVisitor) anyObject());
expectLastCall()
.andAnswer(
new VisitAnswer() {
@Override
protected void doVisit(CatalogVisitor visitor) {
visitor.visit(ws);
}
})
.anyTimes();
ns.accept((CatalogVisitor) anyObject());
expectLastCall()
.andAnswer(
new VisitAnswer() {
@Override
protected void doVisit(CatalogVisitor visitor) {
visitor.visit(ns);
}
})
.anyTimes();
callback.onWorkspace(name, ws, this);
replay(ws, ns);
return this;
}
public void visit(WorkspaceInfo workspace) {
// remove owned stores
for (StoreInfo s : catalog.getStoresByWorkspace(workspace, StoreInfo.class)) {
s.accept(this);
}
// remove any linked namespaces
NamespaceInfo ns = catalog.getNamespaceByPrefix(workspace.getName());
if (ns != null) {
ns.accept(this);
}
catalog.remove(workspace);
}
/** Checks that the namespace/workspace listener keeps on working after a catalog reload */
@Test
public void testNamespaceWorkspaceListenerAttached() throws Exception {
Catalog catalog = getCatalog();
NamespaceInfo ns = catalog.getNamespaceByPrefix(MockData.CITE_PREFIX);
String newName = "XYWZ1234";
ns.setPrefix(newName);
catalog.save(ns);
assertNotNull(catalog.getWorkspaceByName(newName));
assertNotNull(catalog.getNamespaceByPrefix(newName));
// force a reload
int listenersBefore = catalog.getListeners().size();
getGeoServer().reload();
int listenersAfter = catalog.getListeners().size();
assertEquals(listenersBefore, listenersAfter);
// check the NamespaceWorkspaceListener is still attached and working
ns = catalog.getNamespaceByPrefix(newName);
ns.setPrefix(MockData.CITE_PREFIX);
catalog.save(ns);
assertNotNull(catalog.getWorkspaceByName(MockData.CITE_PREFIX));
// make sure we only have one resource pool listener and one catalog persister
int countCleaner = 0;
int countPersister = 0;
for (CatalogListener listener : catalog.getListeners()) {
if (listener instanceof ResourcePool.CacheClearingListener) {
countCleaner++;
} else if (listener instanceof GeoServerPersister) {
countPersister++;
}
}
assertEquals(1, countCleaner);
assertEquals(1, countPersister);
}
@Test
public void modificationProxySerializeTest() throws Exception {
Catalog catalog = getCatalog();
// workspace
WorkspaceInfo ws = catalog.getWorkspaceByName(MockData.CITE_PREFIX);
WorkspaceInfo ws2 = serialize(ws);
assertSame(ModificationProxy.unwrap(ws), ModificationProxy.unwrap(ws2));
// namespace
NamespaceInfo ns = catalog.getNamespaceByPrefix(MockData.CITE_PREFIX);
NamespaceInfo ns2 = serialize(ns);
assertSame(ModificationProxy.unwrap(ns), ModificationProxy.unwrap(ns2));
// data store and related objects
DataStoreInfo ds = catalog.getDataStoreByName(MockData.CITE_PREFIX);
DataStoreInfo ds2 = serialize(ds);
assertSame(ModificationProxy.unwrap(ds), ModificationProxy.unwrap(ds2));
assertSame(
ModificationProxy.unwrap(ds.getWorkspace()), ModificationProxy.unwrap(ds2.getWorkspace()));
// coverage store and related objects
CoverageStoreInfo cs = catalog.getCoverageStoreByName(MockData.TASMANIA_DEM.getLocalPart());
CoverageStoreInfo cs2 = serialize(cs);
assertSame(ModificationProxy.unwrap(cs), ModificationProxy.unwrap(cs2));
assertSame(
ModificationProxy.unwrap(cs.getWorkspace()), ModificationProxy.unwrap(cs2.getWorkspace()));
// feature type and related objects
FeatureTypeInfo ft = catalog.getFeatureTypeByName(getLayerId(MockData.BRIDGES));
FeatureTypeInfo ft2 = serialize(ft);
assertSame(ModificationProxy.unwrap(ft), ModificationProxy.unwrap(ft2));
assertSame(ModificationProxy.unwrap(ft.getStore()), ModificationProxy.unwrap(ft2.getStore()));
// coverage and related objects
CoverageInfo ci = catalog.getCoverageByName(getLayerId(MockData.TASMANIA_DEM));
CoverageInfo ci2 = serialize(ci);
assertSame(ModificationProxy.unwrap(ci), ModificationProxy.unwrap(ci2));
assertSame(ModificationProxy.unwrap(ci.getStore()), ModificationProxy.unwrap(ci.getStore()));
// style
StyleInfo streamsStyle = catalog.getStyleByName("Streams");
StyleInfo si2 = serialize(streamsStyle);
assertSame(ModificationProxy.unwrap(streamsStyle), ModificationProxy.unwrap(si2));
// layer and related objects
LayerInfo li = catalog.getLayerByName(getLayerId(MockData.BRIDGES));
// ... let's add an extra style
li.getStyles().add(streamsStyle);
catalog.save(li);
LayerInfo li2 = serialize(li);
assertSame(ModificationProxy.unwrap(li), ModificationProxy.unwrap(li2));
assertSame(
ModificationProxy.unwrap(li.getResource()), ModificationProxy.unwrap(li2.getResource()));
assertSame(
ModificationProxy.unwrap(li.getDefaultStyle()),
ModificationProxy.unwrap(li2.getDefaultStyle()));
assertSame(
ModificationProxy.unwrap(li.getStyles().iterator().next()),
ModificationProxy.unwrap(li2.getStyles().iterator().next()));
// try a group layer
CatalogBuilder cb = new CatalogBuilder(catalog);
LayerGroupInfo lg = catalog.getFactory().createLayerGroup();
lg.getLayers().add(catalog.getLayerByName(getLayerId(MockData.ROAD_SEGMENTS)));
lg.getLayers().add(catalog.getLayerByName(getLayerId(MockData.PONDS)));
cb.calculateLayerGroupBounds(lg);
lg.setName("test-lg");
catalog.add(lg);
// ... make sure we get a proxy
lg = catalog.getLayerGroupByName("test-lg");
if (lg instanceof SecuredLayerGroupInfo) {
lg = ((SecuredLayerGroupInfo) lg).unwrap(LayerGroupInfo.class);
}
LayerGroupInfo lg2 = serialize(lg);
assertSame(ModificationProxy.unwrap(lg), ModificationProxy.unwrap(lg2));
assertSame(
ModificationProxy.unwrap(lg.getLayers().get(0)),
ModificationProxy.unwrap(lg2.getLayers().get(0)));
assertSame(
ModificationProxy.unwrap(lg.getLayers().get(1)),
ModificationProxy.unwrap(lg2.getLayers().get(1)));
// now check a half modified proxy
LayerInfo lim = catalog.getLayerByName(getLayerId(MockData.BRIDGES));
// ... let's add an extra style
lim.setDefaultStyle(streamsStyle);
lim.getStyles().add(streamsStyle);
// clone and check
LayerInfo lim2 = serialize(lim);
assertSame(
ModificationProxy.unwrap(lim.getDefaultStyle()),
ModificationProxy.unwrap(lim2.getDefaultStyle()));
assertSame(
ModificationProxy.unwrap(lim.getStyles().iterator().next()),
ModificationProxy.unwrap(lim2.getStyles().iterator().next()));
// mess a bit with the metadata map too
String key = "workspaceKey";
lim.getMetadata().put(key, ws);
LayerInfo lim3 = serialize(lim);
assertSame(ModificationProxy.unwrap(lim), ModificationProxy.unwrap(lim3));
assertSame(
ModificationProxy.unwrap(lim.getMetadata().get(key)),
ModificationProxy.unwrap(lim3.getMetadata().get(key)));
}
