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.SessionManager;
  25 import freenet.clients.http.ToadletContext;
  26 import freenet.support.api.HTTPRequest;
  27
  28 import org.junit.Test;
  29
  30 /**
  31  * Unit test for {@link DefaultLoaders}.
  32  */
  33 public class DefaultLoadersTest {
  34
  35         private final Loaders loaders = new DefaultLoaders();
  36         private final StringWriter stringWriter = new StringWriter();
  37         private final TemplateContext templateContext = new TemplateContext();
  38
  39         @Test
  40         public void templateCanBeLoadedFromTheClasspath() {
  41                 Template template = loaders.loadTemplate("/net/pterodactylus/sone/main/template.txt");
  42                 template.render(templateContext, stringWriter);
  43                 assertThat(stringWriter.toString(), is("Template. bar\n"));
  44         }
  45
  46         @Test
  47         public void staticPageIsServedFromClasspath() throws IOException, URISyntaxException {
  48                 Page<FreenetRequest> staticPage = loaders.loadStaticPage("text/", "/net/pterodactylus/sone/main/", "text/plain");
  49                 URI uri = new URI("http://some.host/text/template.txt");
  50                 Method method = Method.GET;
  51                 HTTPRequest httpRequest = mock(HTTPRequest.class);
  52                 ToadletContext toadletContext = mock(ToadletContext.class);
  53                 SessionManager sessionManager = mock(SessionManager.class);
  54                 FreenetRequest request = new FreenetRequest(uri, method, httpRequest, toadletContext, sessionManager);
  55                 OutputStream outputStream = new ByteArrayOutputStream();
  56                 Response response = new Response(outputStream);
  57                 staticPage.handleRequest(request, response);
  58                 assertThat(response.getContentType(), startsWith("text/plain"));
  59                 assertThat(response.getStatusCode(), is(200));
  60         }
  61
  62         @Test
  63         public void templateIsLocatedInClasspath() {
  64                 TemplateProvider templateProvider = loaders.getTemplateProvider();
  65                 Template template = templateProvider.getTemplate(templateContext, "about.html");
  66                 assertThat(template, notNullValue());
  67         }
  68
  69 }