dendi-wang/thingsboard
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix test

Browse Source
This commit is contained in:
desoliture 2022-03-16 22:42:46 +02:00
parent f3a29aadf2
commit a72534f461
1 changed files with 66 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

81
rule-engine/rule-engine-components/src/test/java/org/thingsboard/rule/engine/rest/TbHttpClientTest.java
View File

@ -19,24 +19,32 @@ package org.thingsboard.rule.engine.rest;
import io.netty.channel.EventLoopGroup; import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup; import io.netty.channel.nio.NioEventLoopGroup;
import org.junit.After; import org.junit.After;
import org.junit.Assert;
import org.junit.Before; import org.junit.Before;
import org.junit.Test; import org.junit.Test;
import org.mockito.ArgumentCaptor;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpMethod;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.scheduling.annotation.AsyncResult;
import org.springframework.web.client.AsyncRestTemplate;
import org.springframework.web.util.UriComponentsBuilder;
import org.thingsboard.rule.engine.api.TbContext; import org.thingsboard.rule.engine.api.TbContext;
import org.thingsboard.rule.engine.credentials.AnonymousCredentials;
import org.thingsboard.rule.engine.credentials.ClientCredentials;
import org.thingsboard.server.common.data.id.DeviceId; import org.thingsboard.server.common.data.id.DeviceId;
import org.thingsboard.server.common.data.id.EntityId; import org.thingsboard.server.common.data.id.EntityId;
import org.thingsboard.server.common.msg.TbMsg; import org.thingsboard.server.common.msg.TbMsg;
import org.thingsboard.server.common.msg.TbMsgMetaData; import org.thingsboard.server.common.msg.TbMsgMetaData;
import java.util.Map; import java.net.URI;
import static org.hamcrest.MatcherAssert.assertThat; import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.instanceOf; import static org.hamcrest.Matchers.instanceOf;
import static org.hamcrest.Matchers.is; import static org.hamcrest.Matchers.is;
import static org.mockito.ArgumentMatchers.any; import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.BDDMockito.willCallRealMethod; import static org.mockito.BDDMockito.willCallRealMethod;
import static org.mockito.Mockito.doCallRealMethod;
import static org.mockito.Mockito.mock; import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times; import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify; import static org.mockito.Mockito.verify;
@ -47,6 +55,9 @@ public class TbHttpClientTest {
EventLoopGroup eventLoop; EventLoopGroup eventLoop;
TbHttpClient client; TbHttpClient client;
private final String ENDPOINT_URL = "http://localhost/api?data=[{\\\"test\\\":\\\"test\\\"}]";
private final String GET_METHOD = "GET";
@Before @Before
public void setUp() throws Exception { public void setUp() throws Exception {
client = mock(TbHttpClient.class); client = mock(TbHttpClient.class);
@ -73,21 +84,61 @@ public class TbHttpClientTest {
} }
@Test @Test
public void testProcessMessageWithJsonInUrlVariable() throws Exception{ public void testProcessMessageWithJsonInUrlVariable() throws Exception {
var config = mock(TbRestApiCallNodeConfiguration.class); var config = new TbRestApiCallNodeConfiguration()
when(config.getCredentials()).thenReturn(new AnonymousCredentials()); .defaultConfiguration();
when(config.getRequestMethod()).thenReturn("GET"); config.setRequestMethod(GET_METHOD);
when(config.getRestEndpointUrlPattern()) config.setRestEndpointUrlPattern(ENDPOINT_URL);
.thenReturn("http://localhost/api?data=[{\\\"test\\\":\\\"test\\\"}]"); config.setUseSimpleClientHttpFactory(true);
var httpClient = new TbHttpClient(config, eventLoop); var asyncRestTemplate = mock(AsyncRestTemplate.class);
var ctx = mock(TbContext.class); var uriCaptor = ArgumentCaptor.forClass(URI.class);
var msg = TbMsg.newMsg(
"Main", "GET", new DeviceId(EntityId.NULL_UUID), var responseEntity = new ResponseEntity<>(
TbMsgMetaData.EMPTY, "" "{}",
new HttpHeaders(),
HttpStatus.OK
); );
when(asyncRestTemplate.exchange(
uriCaptor.capture(),
any(),
any(),
eq(String.class)
)).thenReturn(new AsyncResult<>(responseEntity));
var httpClient = new TbHttpClient(config, eventLoop);
httpClient.setHttpClient(asyncRestTemplate);
var msg = TbMsg.newMsg(
"Main", "GET", new DeviceId(EntityId.NULL_UUID),
TbMsgMetaData.EMPTY, "{}"
);
var successMsg = TbMsg.newMsg(
"SUCCESS", msg.getOriginator(),
msg.getMetaData(), msg.getData()
);
var ctx = mock(TbContext.class);
when(ctx.transformMsg(
eq(msg), eq(msg.getType()),
eq(msg.getOriginator()),
eq(msg.getMetaData()),
eq(msg.getData())
)).thenReturn(successMsg);
httpClient.processMessage(ctx, msg); httpClient.processMessage(ctx, msg);
verify(ctx, times(1)).transformMsg(
eq(msg), eq(msg.getType()),
eq(msg.getOriginator()),
eq(msg.getMetaData()),
eq(msg.getData())
);
verify(ctx, times(1))
.tellSuccess(eq(successMsg));
URI uri = UriComponentsBuilder.fromUriString(ENDPOINT_URL).build().encode().toUri();
Assert.assertEquals("URI encoding was not performed!!", uri, uriCaptor.getValue());
} }
} }