1 package net.pterodactylus.sone.main;
3 import static org.hamcrest.MatcherAssert.assertThat;
4 import static org.hamcrest.Matchers.is;
5 import static org.hamcrest.Matchers.startsWith;
6 import static org.mockito.Mockito.mock;
8 import java.io.ByteArrayOutputStream;
9 import java.io.IOException;
10 import java.io.OutputStream;
11 import java.io.StringWriter;
13 import java.net.URISyntaxException;
15 import net.pterodactylus.sone.web.page.FreenetRequest;
16 import net.pterodactylus.util.template.Template;
17 import net.pterodactylus.util.template.TemplateContext;
18 import net.pterodactylus.util.web.Method;
19 import net.pterodactylus.util.web.Page;
20 import net.pterodactylus.util.web.Response;
22 import freenet.clients.http.ToadletContext;
23 import freenet.support.api.HTTPRequest;
25 import org.junit.Test;
28 * Unit test for {@link DefaultLoaders}.
30 * @author <a href="mailto:bombe@pterodactylus.net">David ‘Bombe’ Roden</a>
32 public class DefaultLoadersTest {
34 private final Loaders loaders = new DefaultLoaders();
35 private final StringWriter stringWriter = new StringWriter();
36 private final TemplateContext templateContext = new TemplateContext();
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"));
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));