Add WorldImageCommands unit tests

Author: jericks

src/test/java/org/geoserver/shell/BaseTest.java

@@ -8,7 +8,11 @@
 
 import java.io.File;
 import java.io.IOException;
-import java.net.URL;
+import java.util.ArrayList;
+import java.util.Enumeration;
+import java.util.List;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipFile;
 
 public class BaseTest {
 
@@ -33,4 +37,16 @@ protected String getResourceString(String path) throws IOException {
         return Resources.toString(Resources.getResource(path), Charsets.UTF_8);
     }
 
+    protected List<String> getFileNamesFromZip(File zipFile) throws IOException {
+        List<String> names = new ArrayList<String>();
+        ZipFile zip = new ZipFile(zipFile);
+        Enumeration<? extends ZipEntry> zipEntries = zip.entries();
+        while (zipEntries.hasMoreElements()) {
+            ZipEntry zipEntry = zipEntries.nextElement();
+            String name = zipEntry.getName();
+            names.add(name);
+        }
+        return names;
+    }
+
 }

src/test/java/org/geoserver/shell/WorldImageCommandsTest.java

@@ -0,0 +1,56 @@
+package org.geoserver.shell;
+
+import org.glassfish.grizzly.http.Method;
+import org.glassfish.grizzly.http.util.HttpStatus;
+import org.junit.Test;
+
+import java.io.File;
+import java.util.List;
+import java.util.Map;
+
+import static com.xebialabs.restito.builder.stub.StubHttp.whenHttp;
+import static com.xebialabs.restito.builder.verify.VerifyHttp.verifyHttp;
+import static com.xebialabs.restito.semantics.Action.status;
+import static com.xebialabs.restito.semantics.Action.stringContent;
+import static com.xebialabs.restito.semantics.Condition.*;
+import static junit.framework.Assert.assertEquals;
+import static junit.framework.Assert.assertTrue;
+
+public class WorldImageCommandsTest extends BaseTest {
+
+    @Test
+    public void zipWorldImae() throws Exception {
+        WorldImageCommands commands = new WorldImageCommands();
+        File file = getResourceFile("raster.jpg");
+        File zipFile = File.createTempFile("raster", ".zip");
+        boolean result = commands.zip(file, zipFile);
+        assertTrue(result);
+        List<String> names = getFileNamesFromZip(zipFile);
+        assertEquals(3, names.size());
+        assertTrue(names.contains("raster.jpg"));
+        assertTrue(names.contains("raster.jgw"));
+        assertTrue(names.contains("raster.prj"));
+    }
+
+    @Test
+    public void uploadWorldImage() throws Exception {
+        String workspace = "nurc";
+        String coverageStore = "terrain";
+        File file = getResourceFile("coverageStore.xml");
+        String coverage = "myterrain";
+        String url = "/geoserver/rest/workspaces/nurc/coveragestores/terrain/file.worldimage";
+        whenHttp(server).match(put(url)).then(stringContent("true"), status(HttpStatus.OK_200));
+        Geoserver geoserver = new Geoserver("http://00.0.0.0:8888/geoserver", "admin", "geoserver");
+        WorldImageCommands commands = new WorldImageCommands();
+        commands.setGeoserver(geoserver);
+        boolean result = commands.publish(workspace, coverageStore, file, coverage);
+        assertTrue(result);
+        String body = server.getCalls().get(0).getPostBody();
+        String contentType = server.getCalls().get(0).getContentType();
+        assertEquals("application/zip", contentType);
+        Map<String, String[]> params = server.getCalls().get(0).getParameters();
+        assertEquals("first", params.get("configure")[0]);
+        assertEquals(coverage, params.get("coverageName")[0]);
+        verifyHttp(server).once(method(Method.PUT), uri(url));
+    }
+}

src/test/resources/raster.jgw

@@ -0,0 +1,6 @@
+0.4
+0.0
+0.0
+-0.4
+-179.8
+89.8

src/test/resources/raster.jpg

@@ -0,0 +1,9 @@
+GEOGCS["WGS 84", 
+  DATUM["World Geodetic System 1984", 
+    SPHEROID["WGS 84", 6378137.0, 298.257223563, AUTHORITY["EPSG","7030"]], 
+    AUTHORITY["EPSG","6326"]], 
+  PRIMEM["Greenwich", 0.0, AUTHORITY["EPSG","8901"]], 
+  UNIT["degree", 0.017453292519943295], 
+  AXIS["Geodetic longitude", EAST], 
+  AXIS["Geodetic latitude", NORTH], 
+  AUTHORITY["EPSG","4326"]]