Merge pull request #596 from jannispinter/593-unittest

Add unit tests requested in issue #593

Author: ic0ns

TLS-Core/src/test/java/de/rub/nds/tlsattacker/core/https/HttpsRequestHandlerTest.java

@@ -0,0 +1,59 @@
+/**
+ * TLS-Attacker - A Modular Penetration Testing Framework for TLS
+ *
+ * Copyright 2014-2017 Ruhr University Bochum / Hackmanit GmbH
+ *
+ * Licensed under Apache License 2.0
+ * http://www.apache.org/licenses/LICENSE-2.0
+ */
+package de.rub.nds.tlsattacker.core.https;
+
+import de.rub.nds.tlsattacker.core.constants.ProtocolVersion;
+import de.rub.nds.tlsattacker.core.state.TlsContext;
+import org.junit.Before;
+import org.junit.Test;
+
+import java.nio.charset.Charset;
+
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
+
+public class HttpsRequestHandlerTest {
+
+    private TlsContext context;
+    private HttpsRequestMessage message;
+    private HttpsRequestHandler handler;
+
+    @Before
+    public void setUp() {
+        context = new TlsContext();
+
+        String rawMessage = "GET /index.html HTTP/1.1\r\nUser-Agent: Test\r\nHost: www.rub.de\r\n\r\n";
+        HttpsRequestParser parser = new HttpsRequestParser(0, rawMessage.getBytes(Charset.forName("UTF-8")),
+                ProtocolVersion.TLS12);
+        message = parser.parse();
+
+        handler = new HttpsRequestHandler(context);
+    }
+
+    @Test
+    public void testGetParser() {
+        assertTrue(handler.getParser(new byte[1], 0) instanceof HttpsRequestParser);
+    }
+
+    @Test
+    public void testGetPreparator() {
+        assertTrue(handler.getPreparator(new HttpsRequestMessage()) instanceof HttpsRequestPreparator);
+    }
+
+    @Test
+    public void testGetSerializer() {
+        assertTrue(handler.getSerializer(new HttpsRequestMessage()) instanceof HttpsRequestSerializer);
+    }
+
+    @Test
+    public void testAdjustTLSContext() {
+        handler.adjustTLSContext(message);
+        assertEquals(context.getHttpContext().getLastRequestPath(), message.getRequestPath().getValue());
+    }
+}

TLS-Core/src/test/java/de/rub/nds/tlsattacker/core/https/HttpsRequestPreparatorTest.java

@@ -0,0 +1,48 @@
+/**
+ * TLS-Attacker - A Modular Penetration Testing Framework for TLS
+ *
+ * Copyright 2014-2017 Ruhr University Bochum / Hackmanit GmbH
+ *
+ * Licensed under Apache License 2.0
+ * http://www.apache.org/licenses/LICENSE-2.0
+ */
+package de.rub.nds.tlsattacker.core.https;
+
+import de.rub.nds.tlsattacker.core.constants.ProtocolVersion;
+import de.rub.nds.tlsattacker.core.state.TlsContext;
+import org.junit.Before;
+import org.junit.Test;
+
+import java.nio.charset.Charset;
+
+import static org.junit.Assert.*;
+
+public class HttpsRequestPreparatorTest {
+
+    private TlsContext context;
+    private HttpsRequestMessage message;
+    private HttpsRequestPreparator preparator;
+
+    @Before
+    public void setUp() {
+        context = new TlsContext();
+
+        String rawMessage = "GET /index.html HTTP/1.1\r\nUser-Agent: Test\r\nHost: www.rub.de\r\n\r\n";
+        HttpsRequestParser parser = new HttpsRequestParser(0, rawMessage.getBytes(Charset.forName("UTF-8")),
+                ProtocolVersion.TLS12);
+        message = parser.parse();
+
+        preparator = new HttpsRequestPreparator(context.getChooser(), message);
+    }
+
+    @Test
+    public void testPrepareProtocolMessageContents() {
+        preparator.prepareProtocolMessageContents();
+
+        assertEquals("/", message.getRequestPath().getOriginalValue());
+        assertEquals("HTTP/1.1", message.getRequestProtocol().getOriginalValue());
+        assertEquals("GET", message.getRequestType().getOriginalValue());
+        assertEquals(2, message.getHeader().size());
+
+    }
+}