CipherStreamAPI/src/test/java/com/mattrixwv/cipherstream/controller/polysubstitution/PolybiusSquareControllerIntegrationTest.java

package com.mattrixwv.cipherstream.controller.polysubstitution;


import static org.hamcrest.Matchers.*;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.*;

import java.util.List;

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Tag;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.webmvc.test.autoconfigure.WebMvcTest;
import org.springframework.http.MediaType;
import org.springframework.test.util.ReflectionTestUtils;
import org.springframework.test.web.servlet.MockMvc;

import com.mattrixwv.cipherstream.controller.CipherStreamControllerIntegrationTestBase;
import com.mattrixwv.cipherstream.utils.CipherInfoUtil;
import com.mattrixwv.cipherstream.utils.CipherParameterUtil;

import tools.jackson.databind.node.ObjectNode;


@Tag("integration-test")
@WebMvcTest(controllers = PolybiusSquareController.class)
public class PolybiusSquareControllerIntegrationTest extends CipherStreamControllerIntegrationTestBase{
	@Autowired
	private MockMvc mockMvc;
	@Autowired
	private PolybiusSquareController polybiusSquareController;
	//Loggers
	//TODO: Fix logger testing
	//@Mock(name = "com.mattrixwv.cipherstream.controller.polysubstitution.PolybiusSquareController")
	//protected Logger polybiusLogger;
	//Fields
	private static final ObjectNode blankNode = mapper.createObjectNode();
	private static final String url = "/polybius";
	private static final String decodedString = "Message to^encode";
	private static final String encodedString = "41124545233212 5115^124225152212";
	private static final String keyword = "keyword";
	private static final String polybiusName = "polybiusName";
	private static final String polybiusDescription = "polybiusDescription";
	private static final List<String> polybiusExplanation = List.of("polybiusExplanation1", "polybiusExplanation2", "polybiusExplanation3");
	private static final List<String> polybiusFacts = List.of("polybiusFact1", "polybiusFact2", "polybiusFact3");


	@BeforeEach
	public void setup(){
		decodedNode = mapper.createObjectNode();
		decodedNode.put(CipherParameterUtil.PRESERVE_CAPITALS, true);
		decodedNode.put(CipherParameterUtil.PRESERVE_WHITESPACE, true);
		decodedNode.put(CipherParameterUtil.PRESERVE_SYMBOLS, true);
		decodedNode.put(CipherParameterUtil.KEYWORD, keyword);
		decodedNode.put(CipherParameterUtil.INPUT_STRING, decodedString);
		decodedNode.put(CipherParameterUtil.OUTPUT_STRING, encodedString);

		encodedNode = mapper.createObjectNode();
		encodedNode.put(CipherParameterUtil.PRESERVE_CAPITALS, true);
		encodedNode.put(CipherParameterUtil.PRESERVE_WHITESPACE, true);
		encodedNode.put(CipherParameterUtil.PRESERVE_SYMBOLS, true);
		encodedNode.put(CipherParameterUtil.KEYWORD, keyword);
		encodedNode.put(CipherParameterUtil.INPUT_STRING, encodedString);
		encodedNode.put(CipherParameterUtil.OUTPUT_STRING, decodedString.toUpperCase());

		ReflectionTestUtils.setField(polybiusSquareController, "polybiusName", polybiusName);
		ReflectionTestUtils.setField(polybiusSquareController, "polybiusDescription", polybiusDescription);
		ReflectionTestUtils.setField(polybiusSquareController, "polybiusExplanation", polybiusExplanation);
		ReflectionTestUtils.setField(polybiusSquareController, "polybiusFacts", polybiusFacts);
	}


	@Test
	public void testGetCipherInfo() throws Exception{
		mockMvc.perform(get(url)
				.header("X-Request-Id", requestId)
				.header("X-Forwarded-For", ipAddress))
			.andExpect(status().isOk())
			.andExpect(content().contentTypeCompatibleWith(MediaType.APPLICATION_JSON))
			.andExpect(jsonPath(CipherInfoUtil.CIPHER_NAME).value(polybiusName))
			.andExpect(jsonPath(CipherInfoUtil.CIPHER_DESCRIPTION).value(polybiusDescription))
			.andExpect(jsonPath(CipherInfoUtil.CIPHER_EXPLANATION).isArray())
			.andExpect(jsonPath(CipherInfoUtil.CIPHER_EXPLANATION, hasSize(polybiusExplanation.size())))
			.andExpect(jsonPath(CipherInfoUtil.CIPHER_EXPLANATION, hasItems(polybiusExplanation.get(0), polybiusExplanation.get(1), polybiusExplanation.get(2))))
			.andExpect(jsonPath(CipherInfoUtil.CIPHER_FACTS).isArray())
			.andExpect(jsonPath(CipherInfoUtil.CIPHER_FACTS, hasSize(polybiusFacts.size())))
			.andExpect(jsonPath(CipherInfoUtil.CIPHER_FACTS, hasItems(polybiusFacts.get(0), polybiusFacts.get(1), polybiusFacts.get(2))));

		//Filter
		//TODO: Fix logger testing
		/*
		super.verifyFilter(url);
		//Controller
		verify(polybiusLogger, times(1)).info("Getting info for {}", polybiusName);
		verifyNoMoreInteractions(polybiusLogger);
		verify(mdc, times(1)).put(CipherStreamLoggingAspect.CIPHER_NAME_LOGGING, polybiusName);
		*/
	}

	@Test
	public void testEncodePolybius() throws Exception{
		mockMvc.perform(post(url + "/encode")
				.header("X-Request-Id", requestId)
				.header("X-Forwarded-For", ipAddress)
				.contentType(MediaType.APPLICATION_JSON)
				.content(decodedNode.toString()))
			.andExpect(status().isOk())
			.andExpect(content().contentTypeCompatibleWith(MediaType.APPLICATION_JSON))
			.andExpect(jsonPath(CipherParameterUtil.OUTPUT_STRING).value(encodedString));

		//Filter
		//TODO: Fix logger testing
		/*
		super.verifyFilter(url + "/encode");
		//Controller
		verify(polybiusLogger, times(1)).info("Encoding {}", polybiusName);
		verifyNoMoreInteractions(polybiusLogger);
		verify(mdc, times(1)).put(CipherStreamLoggingAspect.CIPHER_NAME_LOGGING, polybiusName);
		//Cipher Aspect
		verifyAspectLogging(decodedNode);
		*/
	}

	@Test
	public void testEncodepolybius_error() throws Exception{
		mockMvc.perform(post(url + "/encode")
				.header("X-Request-Id", requestId)
				.header("X-Forwarded-For", ipAddress)
				.contentType(MediaType.APPLICATION_JSON)
				.content(blankNode.toString()))
			.andExpect(status().isBadRequest())
			.andExpect(content().contentTypeCompatibleWith(MediaType.APPLICATION_JSON))
			.andExpect(jsonPath("message").value(CipherParameterUtil.PRESERVE_WHITESPACE + CipherParameterUtil.PRESENT_MESSAGE));

		//Filter
		//TODO: Fix logger testing
		/*
		super.verifyFilter(url + "/encode");
		//Controller
		verify(polybiusLogger, times(1)).info("Encoding {}", polybiusName);
		verifyNoMoreInteractions(polybiusLogger);
		verify(mdc, times(1)).put(CipherStreamLoggingAspect.CIPHER_NAME_LOGGING, polybiusName);
		//Cipher Aspect
		verifyNoInteractions(aspectLogger);
		*/
	}

	@Test
	public void testDecodePolybius() throws Exception{
		mockMvc.perform(post(url + "/decode")
				.header("X-Request-Id", requestId)
				.header("X-Forwarded-For", ipAddress)
				.contentType(MediaType.APPLICATION_JSON)
				.content(encodedNode.toString()))
			.andExpect(status().isOk())
			.andExpect(content().contentTypeCompatibleWith(MediaType.APPLICATION_JSON))
			.andExpect(jsonPath(CipherParameterUtil.OUTPUT_STRING).value(decodedString.toUpperCase()));

		//Filter
		//TODO: Fix logger testing
		/*
		super.verifyFilter(url + "/decode");
		//Controller
		verify(polybiusLogger, times(1)).info("Decoding {}", polybiusName);
		verifyNoMoreInteractions(polybiusLogger);
		verify(mdc, times(1)).put(CipherStreamLoggingAspect.CIPHER_NAME_LOGGING, polybiusName);
		//Cipher Aspect
		verifyAspectLogging(encodedNode);
		*/
	}

	@Test
	public void testDecodepolybius_error() throws Exception{
		mockMvc.perform(post(url + "/decode")
				.header("X-Request-Id", requestId)
				.header("X-Forwarded-For", ipAddress)
				.contentType(MediaType.APPLICATION_JSON)
				.content(blankNode.toString()))
			.andExpect(status().isBadRequest())
			.andExpect(content().contentTypeCompatibleWith(MediaType.APPLICATION_JSON))
			.andExpect(jsonPath("message").value(CipherParameterUtil.PRESERVE_WHITESPACE + CipherParameterUtil.PRESENT_MESSAGE));

		//Filter
		//TODO: Fix logger testing
		/*
		super.verifyFilter(url + "/decode");
		//Controller
		verify(polybiusLogger, times(1)).info("Decoding {}", polybiusName);
		verifyNoMoreInteractions(polybiusLogger);
		verify(mdc, times(1)).put(CipherStreamLoggingAspect.CIPHER_NAME_LOGGING, polybiusName);
		//Cipher Aspect
		verifyNoInteractions(aspectLogger);
		*/
	}
}