61 lines
1.6 KiB
Java
61 lines
1.6 KiB
Java
package com.mattrixwv.cipherstream.controller;
|
|
|
|
|
|
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.JsonNode;
|
|
import com.fasterxml.jackson.databind.ObjectMapper;
|
|
import com.fasterxml.jackson.databind.node.ObjectNode;
|
|
import com.mattrixwv.cipherstream.exception.InvalidCipherParameterException;
|
|
|
|
|
|
@Tag("unit-test")
|
|
@ExtendWith(MockitoExtension.class)
|
|
public class ExceptionRestControllerTest{
|
|
private static final ObjectMapper mapper = new ObjectMapper();
|
|
@InjectMocks
|
|
private ExceptionRestController controller;
|
|
//Fields
|
|
private static final String errorMessage = "Error message";
|
|
|
|
|
|
@Test
|
|
public void testInvalidCipherParameterHandler(){
|
|
//Setup variables
|
|
InvalidCipherParameterException error = new InvalidCipherParameterException(errorMessage);
|
|
ObjectNode expectedJson = mapper.createObjectNode();
|
|
expectedJson.put("message", errorMessage);
|
|
|
|
|
|
//Run the function
|
|
ObjectNode returnJson = controller.invalidCipherParameterHandler(error);
|
|
|
|
|
|
//Verify the data
|
|
assertEquals(expectedJson, returnJson);
|
|
}
|
|
|
|
|
|
@Test
|
|
public void testGenericExceptionHandler(){
|
|
//Setup variables
|
|
Exception error = new Exception(errorMessage);
|
|
ObjectNode expectedJson = mapper.createObjectNode();
|
|
expectedJson.put("message", errorMessage);
|
|
|
|
|
|
//Run the function
|
|
JsonNode returnJson = controller.genericExceptionHandler(error);
|
|
|
|
|
|
//Verify the data
|
|
assertEquals(expectedJson, returnJson);
|
|
}
|
|
}
|