implemented WebResourceServlet, which loads resources from the UberWebResourceLoader

scm-webapp/src/test/java/sonia/scm/WebResourceServletTest.java

@@ -0,0 +1,115 @@
+package sonia.scm;
+
+import com.google.common.base.Charsets;
+import com.google.common.io.Files;
+import org.junit.Before;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.TemporaryFolder;
+import org.junit.runner.RunWith;
+import org.mockito.Mock;
+import org.mockito.junit.MockitoJUnitRunner;
+import sonia.scm.plugin.PluginLoader;
+import sonia.scm.plugin.UberWebResourceLoader;
+
+import javax.servlet.ServletOutputStream;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.io.ByteArrayOutputStream;
+import java.io.File;
+import java.io.IOException;
+
+import static org.junit.Assert.*;
+import static org.mockito.Mockito.*;
+
+@RunWith(MockitoJUnitRunner.class)
+public class WebResourceServletTest {
+
+  @Rule
+  public TemporaryFolder temporaryFolder = new TemporaryFolder();
+
+  @Mock
+  private HttpServletRequest request;
+
+  @Mock
+  private HttpServletResponse response;
+
+  @Mock
+  private PluginLoader pluginLoader;
+
+  @Mock
+  private UberWebResourceLoader webResourceLoader;
+
+  private WebResourceServlet servlet;
+
+  @Before
+  public void setUpMocks() {
+    when(pluginLoader.getUberWebResourceLoader()).thenReturn(webResourceLoader);
+    when(request.getContextPath()).thenReturn("/scm");
+    servlet = new WebResourceServlet(pluginLoader);
+  }
+
+  @Test
+  public void testPattern() {
+    assertTrue("/some/resource".matches(WebResourceServlet.PATTERN));
+    assertTrue("/favicon.ico".matches(WebResourceServlet.PATTERN));
+    assertTrue("/other.html".matches(WebResourceServlet.PATTERN));
+    assertFalse("/api/v2/repositories".matches(WebResourceServlet.PATTERN));
+
+    // exclude old style ui template servlets
+    assertFalse("/".matches(WebResourceServlet.PATTERN));
+    assertFalse("/index.html".matches(WebResourceServlet.PATTERN));
+    assertFalse("/error.html".matches(WebResourceServlet.PATTERN));
+    assertFalse("/plugins/resources/js/sonia/scm/hg.config-wizard.js".matches(WebResourceServlet.PATTERN));
+  }
+
+  @Test
+  public void testDoGetWithNonExistingResource() {
+    when(request.getRequestURI()).thenReturn("/scm/awesome.jpg");
+    servlet.doGet(request, response);
+    verify(response).setStatus(HttpServletResponse.SC_NOT_FOUND);
+  }
+
+
+  @Test
+  public void testDoGet() throws IOException {
+    when(request.getRequestURI()).thenReturn("/scm/README.txt");
+    TestingOutputServletOutputStream output = new TestingOutputServletOutputStream();
+    when(response.getOutputStream()).thenReturn(output);
+
+    File file = temporaryFolder.newFile();
+    Files.write("hello".getBytes(Charsets.UTF_8), file);
+
+    when(webResourceLoader.getResource("/README.txt")).thenReturn(file.toURI().toURL());
+    servlet.doGet(request, response);
+
+    assertEquals("hello", output.buffer.toString());
+  }
+
+  @Test
+  public void testDoGetWithError() throws IOException {
+    when(request.getRequestURI()).thenReturn("/scm/README.txt");
+    ServletOutputStream output = mock(ServletOutputStream.class);
+    when(response.getOutputStream()).thenReturn(output);
+
+    File file = temporaryFolder.newFile();
+    assertTrue(file.delete());
+
+    when(webResourceLoader.getResource("/README.txt")).thenReturn(file.toURI().toURL());
+    servlet.doGet(request, response);
+
+    verify(output, never()).write(anyInt());
+    verify(response).setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
+  }
+
+  private static class TestingOutputServletOutputStream extends ServletOutputStream {
+
+    private ByteArrayOutputStream buffer = new ByteArrayOutputStream();
+
+    @Override
+    public void write(int b) {
+      buffer.write(b);
+    }
+  }
+
+}