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

refactored tests

Browse Source
This commit is contained in:
IrynaMatveieva 2024-06-19 16:15:08 +03:00
parent da3b42754a
commit bef6cfd339
3 changed files with 27 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/action/TbSaveToCustomCassandraTableNode.java
View File

@ -103,18 +103,17 @@ public class TbSaveToCustomCassandraTableNode implements TbNode {
saveStmt = getSaveStmt();
}
void onTenantProfileUpdate(TenantProfile tenantProfile) {
private void onTenantProfileUpdate(TenantProfile tenantProfile) {
DefaultTenantProfileConfiguration configuration = (DefaultTenantProfileConfiguration) tenantProfile.getProfileData().getConfiguration();
long tenantProfileDefaultStorageTtl = TimeUnit.DAYS.toSeconds(configuration.getDefaultStorageTtlDays());
ttl = config.getDefaultTTL();
if (ttl == 0L) {
ttl = tenantProfileDefaultStorageTtl;
ttl = TimeUnit.DAYS.toSeconds(configuration.getDefaultStorageTtlDays());
}
}
@Override
public void onMsg(TbContext ctx, TbMsg msg) {
withCallback(save(msg, ctx, ttl), aVoid -> ctx.tellSuccess(msg), e -> ctx.tellFailure(msg, e), ctx.getDbCallbackExecutor());
withCallback(save(msg, ctx), aVoid -> ctx.tellSuccess(msg), e -> ctx.tellFailure(msg, e), ctx.getDbCallbackExecutor());
}
@Override
@ -187,7 +186,7 @@ public class TbSaveToCustomCassandraTableNode implements TbNode {
return query.toString();
}
private ListenableFuture<Void> save(TbMsg msg, TbContext ctx, long ttl) {
private ListenableFuture<Void> save(TbMsg msg, TbContext ctx) {
JsonElement data = JsonParser.parseString(msg.getData());
if (!data.isJsonObject()) {
throw new IllegalStateException("Invalid message structure, it is not a JSON Object: " + data);

2
rule-engine/rule-engine-components/src/main/java/org/thingsboard/rule/engine/telemetry/TbMsgTimeseriesNode.java
View File

@ -75,7 +75,7 @@ public class TbMsgTimeseriesNode implements TbNode {
onTenantProfileUpdate(ctx.getTenantProfile());
}
void onTenantProfileUpdate(TenantProfile tenantProfile) {
private void onTenantProfileUpdate(TenantProfile tenantProfile) {
DefaultTenantProfileConfiguration configuration = (DefaultTenantProfileConfiguration) tenantProfile.getProfileData().getConfiguration();
tenantProfileDefaultStorageTtl = TimeUnit.DAYS.toSeconds(configuration.getDefaultStorageTtlDays());
}

41
rule-engine/rule-engine-components/src/test/java/org/thingsboard/rule/engine/action/TbSaveToCustomCassandraTableNodeTest.java
View File

@ -57,6 +57,7 @@ import org.thingsboard.server.dao.nosql.CassandraStatementTask;
import org.thingsboard.server.dao.nosql.TbResultSet;
import org.thingsboard.server.dao.nosql.TbResultSetFuture;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.UUID;
import java.util.concurrent.TimeUnit;
@ -75,10 +76,10 @@ import static org.mockito.ArgumentMatchers.anyLong;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.BDDMockito.given;
import static org.mockito.BDDMockito.never;
import static org.mockito.BDDMockito.spy;
import static org.mockito.BDDMockito.then;
import static org.mockito.BDDMockito.willAnswer;
import static org.mockito.BDDMockito.never;
@ExtendWith(MockitoExtension.class)
public class TbSaveToCustomCassandraTableNodeTest extends AbstractRuleNodeUpgradeTest {
@ -136,7 +137,9 @@ public class TbSaveToCustomCassandraTableNodeTest extends AbstractRuleNodeUpgrad
var configuration = new TbNodeConfiguration(JacksonUtil.valueToTree(config));
assertThatThrownBy(() -> node.init(ctxMock, configuration))
.isInstanceOf(TbNodeException.class)
.hasMessage("Unable to connect to Cassandra database");
.hasMessage("Unable to connect to Cassandra database")
.extracting(e -> ((TbNodeException) e).isUnrecoverable())
.isEqualTo(false);
}
@Test
@ -238,15 +241,15 @@ public class TbSaveToCustomCassandraTableNodeTest extends AbstractRuleNodeUpgrad
return Stream.of(
Arguments.of(0, 0, "INSERT INTO cs_tb_readings(entityIdTableColumn) VALUES(?)",
(Consumer<BoundStatementBuilder>) builder -> {
then(builder).should(never()).setInt(anyInt(), eq(0));
then(builder).should(never()).setInt(anyInt(), anyInt());
}),
Arguments.of(0, 5, "INSERT INTO cs_tb_readings(entityIdTableColumn) VALUES(?) USING TTL ?",
(Consumer<BoundStatementBuilder>) builder -> {
then(builder).should().setInt(anyInt(), eq(432000));
then(builder).should().setInt(1, 432000);
}),
Arguments.of(20, 1, "INSERT INTO cs_tb_readings(entityIdTableColumn) VALUES(?) USING TTL ?",
(Consumer<BoundStatementBuilder>) builder -> {
then(builder).should().setInt(anyInt(), eq(20));
then(builder).should().setInt(1, 20);
})
);
}
@ -255,14 +258,14 @@ public class TbSaveToCustomCassandraTableNodeTest extends AbstractRuleNodeUpgrad
public void givenValidMsgStructure_whenOnMsg_thenSaveToCustomCassandraTable() throws TbNodeException {
config.setDefaultTTL(25L);
config.setTableName("readings");
config.setFieldsMapping(Map.of(
"$entityId", "entityIdTableColumn",
"doubleField", "doubleTableColumn",
"longField", "longTableColumn",
"booleanField", "booleanTableColumn",
"stringField", "stringTableColumn",
"jsonField", "jsonTableColumn"
));
Map<String, String> mappings = new LinkedHashMap<>();
mappings.put("$entityId", "entityIdTableColumn");
mappings.put("doubleField", "doubleTableColumn");
mappings.put("longField", "longTableColumn");
mappings.put("booleanField", "booleanTableColumn");
mappings.put("stringField", "stringTableColumn");
mappings.put("jsonField", "jsonTableColumn");
config.setFieldsMapping(mappings);
mockOnInit();
mockBoundStatementBuilder();
@ -350,12 +353,12 @@ public class TbSaveToCustomCassandraTableNodeTest extends AbstractRuleNodeUpgrad
}
private void verifySettingStatementBuilder() {
then(boundStatementBuilderMock).should().setUuid(anyInt(), eq(DEVICE_ID.getId()));
then(boundStatementBuilderMock).should().setDouble(anyInt(), eq(22.5));
then(boundStatementBuilderMock).should().setLong(anyInt(), eq(56L));
then(boundStatementBuilderMock).should().setBoolean(anyInt(), eq(true));
then(boundStatementBuilderMock).should().setString(anyInt(), eq("some string"));
then(boundStatementBuilderMock).should().setString(anyInt(), eq("{\"key\":\"value\"}"));
then(boundStatementBuilderMock).should().setUuid(0, DEVICE_ID.getId());
then(boundStatementBuilderMock).should().setDouble(1, 22.5);
then(boundStatementBuilderMock).should().setLong(2, 56L);
then(boundStatementBuilderMock).should().setBoolean(3, true);
then(boundStatementBuilderMock).should().setString(4, "some string");
then(boundStatementBuilderMock).should().setString(5, "{\"key\":\"value\"}");
then(boundStatementBuilderMock).should().setInt(anyInt(), eq(25));
}