WorkspaceInfo ws1 = geoServer.getCatalog().getFactory().createWorkspace();
ws1.setName("TEST-WORKSPACE-1");
geoServer.getCatalog().add(ws1);
// Make a service for that workspace
ServiceInfo ws1wms = new org.geoserver.wms.WMSInfoImpl();
ws1wms.setWorkspace(ws1);
ws1wms.setName("WMS1");
ws1wms.setTitle("WMS for WS1");
geoServer.add(ws1wms);
// Make a second for that workspace
ServiceInfo ws1wfs = new org.geoserver.wfs.WFSInfoImpl();
ws1wfs.setWorkspace(ws1);
ws1wfs.setName("WFS1");
ws1wfs.setTitle("WFS for WS1");
geoServer.add(ws1wfs);
// Make a global service
ServiceInfo gwms = new org.geoserver.wms.WMSInfoImpl();
gwms.setName("WMSG");
gwms.setTitle("Global WMS");
geoServer.add(gwms);
// Make a second global service
ServiceInfo gwfs = new org.geoserver.wfs.WFSInfoImpl();
gwfs.setName("WFSG");
gwfs.setTitle("Global WFS");
geoServer.add(gwfs);
// Make a workspace
WorkspaceInfo ws2 = geoServer.getCatalog().getFactory().createWorkspace();
ws2.setName("TEST-WORKSPACE-2");
geoServer.getCatalog().add(ws2);
// Make a service for that workspace
ServiceInfo ws2wms = new org.geoserver.wms.WMSInfoImpl();
ws2wms.setWorkspace(ws2);
ws2wms.setName("WMS2");
ws2wms.setTitle("WMS for WS2");
geoServer.add(ws2wms);
// Make a second for that workspace
ServiceInfo ws2wfs = new org.geoserver.wfs.WFSInfoImpl();
ws2wfs.setWorkspace(ws2);
ws2wfs.setName("WFS2");
ws2wfs.setTitle("WFS for WS2");
geoServer.add(ws2wfs);
// Check that we get the services we expect to
assertThat(geoServer.getService(org.geoserver.wms.WMSInfo.class), equalTo(gwms));
assertThat(geoServer.getService(org.geoserver.wfs.WFSInfo.class), equalTo(gwfs));