Initial commit

src/test/java/com/mattrixwv/cipherstream/controller/combination/AdfgvxCipherControllerTest.java

@@ -0,0 +1,76 @@
+package com.mattrixwv.cipherstream.controller.combination;
+
+
+import static org.junit.jupiter.api.Assertions.*;
+
+import org.junit.jupiter.api.Tag;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.InjectMocks;
+import org.mockito.junit.jupiter.MockitoExtension;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.node.ObjectNode;
+import com.mattrixwv.cipherstream.exception.InvalidCipherParameterException;
+import com.mattrixwv.cipherstream.utils.CipherParameterUtil;
+
+
+@Tag("unit-test")
+@ExtendWith(MockitoExtension.class)
+public class AdfgvxCipherControllerTest{
+	@InjectMocks
+	private AdfgvxCipherController adfgvxCipherController;
+	private static final String KEYWORD = CipherParameterUtil.KEYWORD;
+	private static final String SQUARE_KEYWORD = CipherParameterUtil.SQUARE_KEYWORD;
+	private static final String INPUT_STRING = "Message to-encode";
+	private static final String OUTPUT_STRING = "AXgvdavfxgagfa afag-aaxdxfgdagda";
+	private ObjectMapper objectMapper = new ObjectMapper();
+
+
+	@Test
+	public void testEncodeAdfgvx(){
+		ObjectNode cipherParams = generateParams(KEYWORD, SQUARE_KEYWORD, INPUT_STRING);
+
+
+		ObjectNode returnedJson = adfgvxCipherController.encodeAdfgvx(cipherParams);
+
+
+		assertEquals(cipherParams, returnedJson);
+		assertEquals(OUTPUT_STRING, returnedJson.get(CipherParameterUtil.OUTPUT_STRING).asText());
+
+		//Verify invalid params are caught
+		final ObjectNode blankNode = objectMapper.createObjectNode();
+		assertThrows(InvalidCipherParameterException.class, () -> {
+			adfgvxCipherController.encodeAdfgvx(blankNode);
+		});
+	}
+
+	@Test
+	public void testDecodeAdfgvx(){
+		ObjectNode cipherParams = generateParams(KEYWORD, SQUARE_KEYWORD, OUTPUT_STRING);
+
+
+		ObjectNode returnedJson = adfgvxCipherController.decodeAdfgvx(cipherParams);
+
+
+		assertEquals(cipherParams, returnedJson);
+		assertEquals(INPUT_STRING, returnedJson.get(CipherParameterUtil.OUTPUT_STRING).asText());
+
+		//Verify invalid params are caught
+		final ObjectNode blankNode = objectMapper.createObjectNode();
+		assertThrows(InvalidCipherParameterException.class, () -> {
+			adfgvxCipherController.decodeAdfgvx(blankNode);
+		});
+	}
+
+	private ObjectNode generateParams(String keyword, String squareKeyword, String inputString){
+		ObjectNode cipherParams = objectMapper.createObjectNode();
+		cipherParams.put(CipherParameterUtil.PRESERVE_CAPITALS, true);
+		cipherParams.put(CipherParameterUtil.PRESERVE_WHITESPACE, true);
+		cipherParams.put(CipherParameterUtil.PRESERVE_SYMBOLS, true);
+		cipherParams.put(CipherParameterUtil.KEYWORD, keyword);
+		cipherParams.put(CipherParameterUtil.SQUARE_KEYWORD, squareKeyword);
+		cipherParams.put(CipherParameterUtil.INPUT_STRING, inputString);
+		return cipherParams;
+	}
+}