src/test/java/net/pterodactylus/sone/main/DefaultLoadersTest.java

   1 package net.pterodactylus.sone.main;
   2
   3 import static org.hamcrest.MatcherAssert.assertThat;
   4 import static org.hamcrest.Matchers.is;
   5 import static org.hamcrest.Matchers.notNullValue;
   6 import static org.hamcrest.Matchers.startsWith;
   7 import static org.mockito.Mockito.mock;
   8
   9 import java.io.ByteArrayOutputStream;
  10 import java.io.IOException;
  11 import java.io.OutputStream;
  12 import java.io.StringWriter;
  13 import java.net.URI;
  14 import java.net.URISyntaxException;
  15
  16 import net.pterodactylus.sone.web.page.FreenetRequest;
  17 import net.pterodactylus.util.template.Template;
  18 import net.pterodactylus.util.template.TemplateContext;
  19 import net.pterodactylus.util.template.TemplateProvider;
  20 import net.pterodactylus.util.web.Method;
  21 import net.pterodactylus.util.web.Page;
  22 import net.pterodactylus.util.web.Response;
  23
  24 import freenet.clients.http.ToadletContext;
  25 import freenet.support.api.HTTPRequest;
  26
  27 import org.junit.Test;
  28
  29 /**
  30  * Unit test for {@link DefaultLoaders}.
  31  */
  32 public class DefaultLoadersTest {
  33
  34         private final Loaders loaders = new DefaultLoaders();
  35         private final StringWriter stringWriter = new StringWriter();
  36         private final TemplateContext templateContext = new TemplateContext();
  37
  38         @Test
  39         public void templateCanBeLoadedFromTheClasspath() {
  40                 Template template = loaders.loadTemplate("/net/pterodactylus/sone/main/template.txt");
  41                 template.render(templateContext, stringWriter);
  42                 assertThat(stringWriter.toString(), is("Template. bar\n"));
  43         }
  44
  45         @Test
  46         public void staticPageIsServedFromClasspath() throws IOException, URISyntaxException {
  47                 Page<FreenetRequest> staticPage = loaders.loadStaticPage("text/", "/net/pterodactylus/sone/main/", "text/plain");
  48                 URI uri = new URI("http://some.host/text/template.txt");
  49                 Method method = Method.GET;
  50                 HTTPRequest httpRequest = mock(HTTPRequest.class);
  51                 ToadletContext toadletContext = mock(ToadletContext.class);
  52                 FreenetRequest request = new FreenetRequest(uri, method, httpRequest, toadletContext);
  53                 OutputStream outputStream = new ByteArrayOutputStream();
  54                 Response response = new Response(outputStream);
  55                 staticPage.handleRequest(request, response);
  56                 assertThat(response.getContentType(), startsWith("text/plain"));
  57                 assertThat(response.getStatusCode(), is(200));
  58         }
  59
  60         @Test
  61         public void templateIsLocatedInClasspath() {
  62                 TemplateProvider templateProvider = loaders.getTemplateProvider();
  63                 Template template = templateProvider.getTemplate(templateContext, "about.html");
  64                 assertThat(template, notNullValue());
  65         }
  66
  67 }