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

check test

Browse Source
This commit is contained in:
Yuriy Lytvynchuk 2022-09-26 09:17:06 +03:00
parent 7b1651b860
commit bd0a70b017
1 changed files with 4 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
rule-engine/rule-engine-components/src/test/java/org/thingsboard/rule/engine/metadata/TbGetTenantAttributeNodeTest.java
View File

@ -15,7 +15,6 @@
*/ */
package org.thingsboard.rule.engine.metadata; package org.thingsboard.rule.engine.metadata;
import com.google.common.util.concurrent.Futures;
import org.junit.Before; import org.junit.Before;
import org.junit.Test; import org.junit.Test;
import org.junit.runner.RunWith; import org.junit.runner.RunWith;
@ -31,8 +30,6 @@ import org.thingsboard.server.common.data.id.UserId;
import java.util.UUID; import java.util.UUID;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.when; import static org.mockito.Mockito.when;
@RunWith(MockitoJUnitRunner.class) @RunWith(MockitoJUnitRunner.class)
@ -45,6 +42,7 @@ public class TbGetTenantAttributeNodeTest extends AbstractAttributeNodeTest {
@Before @Before
public void initDataForTests() throws TbNodeException { public void initDataForTests() throws TbNodeException {
init(new TbGetTenantAttributeNode()); init(new TbGetTenantAttributeNode());
user.setTenantId(tenantId); user.setTenantId(tenantId);
user.setId(new UserId(UUID.randomUUID())); user.setId(new UserId(UUID.randomUUID()));
@ -53,6 +51,8 @@ public class TbGetTenantAttributeNodeTest extends AbstractAttributeNodeTest {
device.setTenantId(tenantId); device.setTenantId(tenantId);
device.setId(new DeviceId(UUID.randomUUID())); device.setId(new DeviceId(UUID.randomUUID()));
when(ctx.getTenantId()).thenReturn(tenantId);
} }
@Override @Override
@ -67,20 +67,17 @@ public class TbGetTenantAttributeNodeTest extends AbstractAttributeNodeTest {
@Test @Test
public void errorThrownIfCannotLoadAttributes() { public void errorThrownIfCannotLoadAttributes() {
mockFindUser(user);
errorThrownIfCannotLoadAttributes(user); errorThrownIfCannotLoadAttributes(user);
} }
@Test @Test
public void errorThrownIfCannotLoadAttributesAsync() { public void errorThrownIfCannotLoadAttributesAsync() {
mockFindUser(user);
errorThrownIfCannotLoadAttributesAsync(user); errorThrownIfCannotLoadAttributesAsync(user);
} }
@Test @Test
public void failedChainUsedIfCustomerCannotBeFound() { public void failedChainUsedIfCustomerCannotBeFound() {
when(ctx.getUserService()).thenReturn(userService); when(ctx.getTenantId()).thenReturn(null);
when(userService.findUserByIdAsync(any(), eq(user.getId()))).thenReturn(Futures.immediateFuture(null));
failedChainUsedIfCustomerCannotBeFound(user); failedChainUsedIfCustomerCannotBeFound(user);
} }
@ -91,25 +88,21 @@ public class TbGetTenantAttributeNodeTest extends AbstractAttributeNodeTest {
@Test @Test
public void usersCustomerAttributesFetched() { public void usersCustomerAttributesFetched() {
mockFindUser(user);
usersCustomerAttributesFetched(user); usersCustomerAttributesFetched(user);
} }
@Test @Test
public void assetsCustomerAttributesFetched() { public void assetsCustomerAttributesFetched() {
mockFindAsset(asset);
assetsCustomerAttributesFetched(asset); assetsCustomerAttributesFetched(asset);
} }
@Test @Test
public void deviceCustomerAttributesFetched() { public void deviceCustomerAttributesFetched() {
mockFindDevice(device);
deviceCustomerAttributesFetched(device); deviceCustomerAttributesFetched(device);
} }
@Test @Test
public void deviceCustomerTelemetryFetched() throws TbNodeException { public void deviceCustomerTelemetryFetched() throws TbNodeException {
mockFindDevice(device);
deviceCustomerTelemetryFetched(device); deviceCustomerTelemetryFetched(device);
} }
} }