Merge pull request #583 from TanerBoecek/add-unittest

Add unittest

Author: ic0ns

TLS-Core/src/test/java/de/rub/nds/tlsattacker/core/protocol/handler/HelloRetryRequestHandlerTest.java

@@ -0,0 +1,71 @@
+/**
+ * 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.protocol.handler;
+
+import de.rub.nds.tlsattacker.core.constants.CipherSuite;
+import de.rub.nds.tlsattacker.core.constants.ProtocolVersion;
+import de.rub.nds.tlsattacker.core.constants.ExtensionType;
+import de.rub.nds.tlsattacker.core.protocol.message.HelloRetryRequestMessage;
+import de.rub.nds.tlsattacker.core.protocol.message.extension.ExtensionMessage;
+import de.rub.nds.tlsattacker.core.protocol.message.extension.EncryptThenMacExtensionMessage;
+import de.rub.nds.tlsattacker.core.protocol.parser.HelloRetryRequestParser;
+import de.rub.nds.tlsattacker.core.protocol.preparator.HelloRetryRequestPreparator;
+import de.rub.nds.tlsattacker.core.protocol.serializer.HelloRetryRequestSerializer;
+import de.rub.nds.tlsattacker.core.state.TlsContext;
+import org.junit.Before;
+import org.junit.Test;
+
+import static org.junit.Assert.*;
+
+public class HelloRetryRequestHandlerTest {
+
+    private TlsContext context;
+    private HelloRetryRequestHandler handler;
+
+    @Before
+    public void setUp() {
+        context = new TlsContext();
+        handler = new HelloRetryRequestHandler(context);
+    }
+
+    @Test
+    public void testGetParser() {
+        assertTrue(handler.getParser(new byte[1], 0) instanceof HelloRetryRequestParser);
+    }
+
+    @Test
+    public void testGetPreparator() {
+        assertTrue(handler.getPreparator(new HelloRetryRequestMessage()) instanceof HelloRetryRequestPreparator);
+    }
+
+    @Test
+    public void testGetSerializer() {
+        assertTrue(handler.getSerializer(new HelloRetryRequestMessage()) instanceof HelloRetryRequestSerializer);
+    }
+
+    @Test
+    public void testAdjustTLSContext() {
+        HelloRetryRequestMessage message = new HelloRetryRequestMessage();
+        ExtensionMessage extensionMessage = new EncryptThenMacExtensionMessage();
+        ProtocolVersion protocolVersion = ProtocolVersion.SSL2;
+        CipherSuite cipherSuite = CipherSuite.TLS_DH_DSS_WITH_DES_CBC_SHA;
+        ExtensionType extensionType = ExtensionType.ENCRYPT_THEN_MAC;
+
+        message.setProtocolVersion(protocolVersion.getValue());
+        message.setSelectedCipherSuite(cipherSuite.getByteValue());
+        message.setExtensionBytes(extensionType.getValue());
+        message.addExtension(extensionMessage);
+
+        handler.adjustTLSContext(message);
+        assertSame(context.getSelectedProtocolVersion(), protocolVersion);
+        assertSame(context.getSelectedCipherSuite(), cipherSuite);
+        assertTrue(context.isExtensionProposed(extensionType));
+    }
+
+}

TLS-Core/src/test/java/de/rub/nds/tlsattacker/core/workflow/action/CopyBufferedRecordsActionTest.java

@@ -0,0 +1,67 @@
+/**
+ * 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.workflow.action;
+
+import de.rub.nds.modifiablevariable.bytearray.ModifiableByteArray;
+import de.rub.nds.tlsattacker.core.constants.ProtocolMessageType;
+import de.rub.nds.tlsattacker.core.record.AbstractRecord;
+import de.rub.nds.tlsattacker.core.record.Record;
+import de.rub.nds.tlsattacker.core.state.TlsContext;
+import org.junit.Before;
+import org.junit.Test;
+
+import java.util.LinkedList;
+
+import static org.junit.Assert.*;
+
+public class CopyBufferedRecordsActionTest {
+
+    private CopyBufferedRecordsAction action;
+
+    @Before
+    public void setUp() {
+        action = new CopyBufferedRecordsAction("src", "dst");
+    }
+
+    @Test
+    public void testCopyField() {
+        TlsContext src = new TlsContext();
+        TlsContext dst = new TlsContext();
+        ModifiableByteArray byteArray = new ModifiableByteArray();
+        AbstractRecord record = new Record();
+        LinkedList<AbstractRecord> recordBuffer = new LinkedList<>();
+
+        record.setProtocolMessageBytes(byteArray);
+        record.setContentMessageType(ProtocolMessageType.HANDSHAKE);
+        record.setMaxRecordLengthConfig(18);
+        record.setCleanProtocolMessageBytes(new byte[1]);
+        record.setCompleteRecordBytes(new byte[1]);
+        recordBuffer.add(record);
+        src.setRecordBuffer(recordBuffer);
+
+        action.copyField(src, dst);
+        assertSame(src.getRecordBuffer(), dst.getRecordBuffer());
+    }
+
+    @Test
+    public void testExecutedAsPlanned() {
+        action.setExecuted(true);
+        assertTrue(action.executedAsPlanned());
+        action.setExecuted(false);
+        assertFalse(action.executedAsPlanned());
+    }
+
+    @Test
+    public void testReset() {
+        action.setExecuted(true);
+        action.reset();
+        assertFalse(action.isExecuted());
+    }
+
+}