package org.oransc.policyagent.dmaap;
+import static ch.qos.logback.classic.Level.WARN;
import static org.assertj.core.api.Assertions.assertThat;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.jupiter.api.Assertions.assertTrue;
-import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.verifyNoMoreInteractions;
-import ch.qos.logback.classic.Level;
import ch.qos.logback.classic.spi.ILoggingEvent;
import ch.qos.logback.core.read.ListAppender;
import com.google.gson.JsonObject;
import java.io.IOException;
+import java.nio.charset.Charset;
import java.util.Optional;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.ArgumentCaptor;
-import org.onap.dmaap.mr.client.MRBatchingPublisher;
-import org.onap.dmaap.mr.client.response.MRPublisherResponse;
import org.oransc.policyagent.clients.AsyncRestClient;
import org.oransc.policyagent.dmaap.DmaapRequestMessage.Operation;
import org.oransc.policyagent.repository.ImmutablePolicyType;
import org.oransc.policyagent.utils.LoggingUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
+import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.reactive.function.client.WebClientResponseException;
private static final Logger logger = LoggerFactory.getLogger(DmaapMessageHandlerTest.class);
private static final String URL = "url";
- private final MRBatchingPublisher dmaapClient = mock(MRBatchingPublisher.class);
+ private final AsyncRestClient dmaapClient = mock(AsyncRestClient.class);
private final AsyncRestClient agentClient = mock(AsyncRestClient.class);
private DmaapMessageHandler testedObject;
private static Gson gson = new GsonBuilder() //
return Mono.just(entity);
}
+ private Mono<ResponseEntity<String>> notOkResponse() {
+ ResponseEntity<String> entity = new ResponseEntity<>("NOK", HttpStatus.BAD_GATEWAY);
+ return Mono.just(entity);
+ }
+
@Test
void testMessageParsing() {
String message = dmaapInputMessage(Operation.DELETE);
@Test
void unparseableMessage_thenWarning() {
- final ListAppender<ILoggingEvent> logAppender = LoggingUtils.getLogListAppender(DmaapMessageHandler.class);
+ final ListAppender<ILoggingEvent> logAppender =
+ LoggingUtils.getLogListAppender(DmaapMessageHandler.class, WARN);
- testedObject.handleDmaapMsg("bad message");
+ String msg = "bad message";
+ testedObject.handleDmaapMsg(msg);
- assertThat(logAppender.list.get(0).getLevel()).isEqualTo(Level.WARN);
- assertThat(logAppender.list.toString().contains("handleDmaapMsg failure ")).isTrue();
+ assertThat(logAppender.list.get(0).getFormattedMessage()).startsWith(
+ "handleDmaapMsg failure org.oransc.policyagent.exceptions.ServiceException: Received unparsable "
+ + "message from DMAAP: \"" + msg + "\", reason: ");
}
@Test
void successfulDelete() throws IOException {
doReturn(okResponse()).when(agentClient).deleteForEntity(anyString());
- doReturn(1).when(dmaapClient).send(anyString());
- doReturn(new MRPublisherResponse()).when(dmaapClient).sendBatchWithResponse();
+ doReturn(Mono.just("OK")).when(dmaapClient).post(anyString(), anyString());
String message = dmaapInputMessage(Operation.DELETE);
verify(agentClient).deleteForEntity(URL);
verifyNoMoreInteractions(agentClient);
- verify(dmaapClient).send(anyString());
- verify(dmaapClient).sendBatchWithResponse();
+ verify(dmaapClient).post(anyString(), anyString());
+
verifyNoMoreInteractions(dmaapClient);
}
@Test
void successfulGet() throws IOException {
doReturn(okResponse()).when(agentClient).getForEntity(anyString());
- doReturn(1).when(dmaapClient).send(anyString());
- doReturn(new MRPublisherResponse()).when(dmaapClient).sendBatchWithResponse();
+ doReturn(Mono.just("OK")).when(dmaapClient).post(anyString(), anyString());
StepVerifier //
.create(testedObject.createTask(dmaapInputMessage(Operation.GET))) //
verify(agentClient).getForEntity(URL);
verifyNoMoreInteractions(agentClient);
- verify(dmaapClient).send(anyString());
- verify(dmaapClient).sendBatchWithResponse();
+ verify(dmaapClient).post(anyString(), anyString());
verifyNoMoreInteractions(dmaapClient);
}
+ @Test
+ void exceptionFromAgentWhenGet_thenPostError() throws IOException {
+ String errorBody = "Unavailable";
+ WebClientResponseException webClientResponseException = new WebClientResponseException(
+ HttpStatus.SERVICE_UNAVAILABLE.value(), "", (HttpHeaders) null, errorBody.getBytes(), (Charset) null);
+ doReturn(Mono.error(webClientResponseException)).when(agentClient).getForEntity(anyString());
+ doReturn(Mono.just("OK")).when(dmaapClient).post(anyString(), anyString());
+
+ StepVerifier //
+ .create(testedObject.createTask(dmaapInputMessage(Operation.GET))) //
+ .expectSubscription() //
+ .verifyComplete(); //
+
+ ArgumentCaptor<String> captor = ArgumentCaptor.forClass(String.class);
+ verify(dmaapClient).post(anyString(), captor.capture());
+ String actualMessage = captor.getValue();
+ assertThat(actualMessage).contains(HttpStatus.SERVICE_UNAVAILABLE.toString()) //
+ .contains(errorBody);
+ }
+
@Test
void successfulPut() throws IOException {
doReturn(okResponse()).when(agentClient).putForEntity(anyString(), anyString());
- doReturn(1).when(dmaapClient).send(anyString());
- doReturn(new MRPublisherResponse()).when(dmaapClient).sendBatchWithResponse();
+ doReturn(Mono.just("OK")).when(dmaapClient).post(anyString(), anyString());
StepVerifier //
.create(testedObject.createTask(dmaapInputMessage(Operation.PUT))) //
verify(agentClient).putForEntity(URL, payloadAsString());
verifyNoMoreInteractions(agentClient);
- verify(dmaapClient).send(anyString());
- verify(dmaapClient).sendBatchWithResponse();
+ verify(dmaapClient).post(anyString(), anyString());
verifyNoMoreInteractions(dmaapClient);
}
@Test
void successfulPost() throws IOException {
doReturn(okResponse()).when(agentClient).postForEntity(anyString(), anyString());
- doReturn(1).when(dmaapClient).send(anyString());
- doReturn(new MRPublisherResponse()).when(dmaapClient).sendBatchWithResponse();
+ doReturn(Mono.just("OK")).when(dmaapClient).post(anyString(), anyString());
StepVerifier //
.create(testedObject.createTask(dmaapInputMessage(Operation.POST))) //
verify(agentClient).postForEntity(URL, payloadAsString());
verifyNoMoreInteractions(agentClient);
- verify(dmaapClient).send(anyString());
- verify(dmaapClient).sendBatchWithResponse();
+ verify(dmaapClient).post(anyString(), anyString());
verifyNoMoreInteractions(dmaapClient);
}
@Test
void exceptionWhenCallingPolicyAgent_thenNotFoundResponse() throws IOException {
- WebClientResponseException except = new WebClientResponseException(400, "Refused", null, null, null, null);
- doReturn(Mono.error(except)).when(agentClient).putForEntity(anyString(), any());
- doReturn(1).when(dmaapClient).send(anyString());
- doReturn(new MRPublisherResponse()).when(dmaapClient).sendBatchWithResponse();
- StepVerifier //
- .create(testedObject.createTask(dmaapInputMessage(Operation.PUT))) //
- .expectSubscription() //
- .verifyComplete(); //
+ doReturn(notOkResponse()).when(agentClient).putForEntity(anyString(), anyString());
+ doReturn(Mono.just("OK")).when(dmaapClient).post(anyString(), anyString());
+
+ testedObject.createTask(dmaapInputMessage(Operation.PUT)).block();
verify(agentClient).putForEntity(anyString(), anyString());
verifyNoMoreInteractions(agentClient);
ArgumentCaptor<String> captor = ArgumentCaptor.forClass(String.class);
- verify(dmaapClient).send(captor.capture());
+ verify(dmaapClient).post(anyString(), captor.capture());
String actualMessage = captor.getValue();
- assertThat(actualMessage.contains(HttpStatus.BAD_REQUEST.toString())).isTrue();
+ assertThat(actualMessage).as("Message \"%s\" sent to DMaaP contains %s", actualMessage, HttpStatus.BAD_GATEWAY)
+ .contains(HttpStatus.BAD_GATEWAY.toString());
- verify(dmaapClient).sendBatchWithResponse();
verifyNoMoreInteractions(dmaapClient);
}
testedObject.handleDmaapMsg(message);
ArgumentCaptor<String> captor = ArgumentCaptor.forClass(String.class);
- verify(dmaapClient).send(captor.capture());
+ verify(dmaapClient).post(anyString(), captor.capture());
String actualMessage = captor.getValue();
- assertThat(actualMessage
- .contains(HttpStatus.BAD_REQUEST + "\",\"message\":\"Not implemented operation: " + badOperation)).isTrue();
-
- verify(dmaapClient).sendBatchWithResponse();
- verifyNoMoreInteractions(dmaapClient);
+ assertThat(actualMessage).contains("Not implemented operation") //
+ .contains("BAD_REQUEST");
}
@Test
String message = dmaapInputMessage(Operation.PUT).toString();
message = message.replace(",\"payload\":{\"name\":\"name\",\"schema\":\"schema\"}", "");
- final ListAppender<ILoggingEvent> logAppender = LoggingUtils.getLogListAppender(DmaapMessageHandler.class);
+ final ListAppender<ILoggingEvent> logAppender =
+ LoggingUtils.getLogListAppender(DmaapMessageHandler.class, WARN);
testedObject.handleDmaapMsg(message);
- assertThat(logAppender.list.get(0).getLevel()).isEqualTo(Level.WARN);
- assertThat(logAppender.list.toString().contains("Expected payload in message from DMAAP: ")).isTrue();
+ assertThat(logAppender.list.get(0).getFormattedMessage())
+ .startsWith("Expected payload in message from DMAAP: ");
}
}