From b164d711994b45c0bf29800fa9abe6f2767a0cd7 Mon Sep 17 00:00:00 2001
From: Andreas Penski <18-andreas.penski@users.noreply.projekte.kosit.org>
Date: Mon, 11 May 2020 09:07:40 +0200
Subject: [PATCH] test for gui handler

---
 .../kosit/validationtool/daemon/BaseHandler.java |  7 ++++---
 .../validationtool/daemon/GuiHandlerIT.java      | 16 ++++++++++++++++
 2 files changed, 20 insertions(+), 3 deletions(-)
 create mode 100644 src/test/java/de/kosit/validationtool/daemon/GuiHandlerIT.java

diff --git a/src/main/java/de/kosit/validationtool/daemon/BaseHandler.java b/src/main/java/de/kosit/validationtool/daemon/BaseHandler.java
index 4a7fe54..9aaae1c 100644
--- a/src/main/java/de/kosit/validationtool/daemon/BaseHandler.java
+++ b/src/main/java/de/kosit/validationtool/daemon/BaseHandler.java
@@ -29,10 +29,11 @@ abstract class BaseHandler implements HttpHandler {
         os.close();
     }
 
-    protected static void error(final HttpExchange httpExchange, final int statusCode, final String message) throws IOException {
+    protected static void error(final HttpExchange exchange, final int statusCode, final String message) throws IOException {
         final byte[] bytes = message.getBytes();
-        httpExchange.sendResponseHeaders(statusCode, bytes.length);
-        final OutputStream os = httpExchange.getResponseBody();
+        exchange.getResponseHeaders().add("Content-Type", "text/plain");
+        exchange.sendResponseHeaders(statusCode, bytes.length);
+        final OutputStream os = exchange.getResponseBody();
         os.write(bytes);
         os.close();
     }
diff --git a/src/test/java/de/kosit/validationtool/daemon/GuiHandlerIT.java b/src/test/java/de/kosit/validationtool/daemon/GuiHandlerIT.java
new file mode 100644
index 0000000..bd62201
--- /dev/null
+++ b/src/test/java/de/kosit/validationtool/daemon/GuiHandlerIT.java
@@ -0,0 +1,16 @@
+package de.kosit.validationtool.daemon;
+
+import io.restassured.http.ContentType;
+import org.junit.Test;
+
+import static io.restassured.RestAssured.given;
+
+public class GuiHandlerIT extends BaseIT {
+
+    @Test
+    public void checkGui() {
+        given().when().get("/").then().statusCode(200).and().contentType(ContentType.HTML);
+        given().when().get("/README.md").then().statusCode(200).and().contentType("text/markdown");
+        given().when().get("/unknown.md").then().statusCode(404).and().contentType(ContentType.TEXT);
+    }
+}