77 lines
2.7 KiB
Java
77 lines
2.7 KiB
Java
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;
|
|
}
|
|
}
|