Andrew Shvayka
2 years ago
committed by
GitHub
GitHub
37 changed files with 1308 additions and 214 deletions
@ -0,0 +1,241 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf.ctx.state; |
||||
|
|
||||
|
import org.junit.jupiter.api.BeforeEach; |
||||
|
import org.junit.jupiter.api.Test; |
||||
|
import org.springframework.beans.factory.annotation.Autowired; |
||||
|
import org.springframework.boot.test.context.SpringBootTest; |
||||
|
import org.thingsboard.script.api.tbel.DefaultTbelInvokeService; |
||||
|
import org.thingsboard.script.api.tbel.TbelInvokeService; |
||||
|
import org.thingsboard.server.common.data.AttributeScope; |
||||
|
import org.thingsboard.server.common.data.cf.CalculatedField; |
||||
|
import org.thingsboard.server.common.data.cf.CalculatedFieldType; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.Argument; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.ArgumentType; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.CalculatedFieldConfiguration; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.Output; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.OutputType; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.ReferencedEntityKey; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.SimpleCalculatedFieldConfiguration; |
||||
|
import org.thingsboard.server.common.data.id.AssetId; |
||||
|
import org.thingsboard.server.common.data.id.DeviceId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.kv.BasicKvEntry; |
||||
|
import org.thingsboard.server.common.data.kv.LongDataEntry; |
||||
|
import org.thingsboard.server.service.cf.CalculatedFieldResult; |
||||
|
|
||||
|
import java.util.HashMap; |
||||
|
import java.util.Map; |
||||
|
import java.util.TreeMap; |
||||
|
import java.util.UUID; |
||||
|
import java.util.concurrent.ExecutionException; |
||||
|
|
||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
|
||||
|
@SpringBootTest(classes = DefaultTbelInvokeService.class) |
||||
|
public class ScriptCalculatedFieldStateTest { |
||||
|
|
||||
|
private final TenantId TENANT_ID = TenantId.fromUUID(UUID.fromString("5b18e321-3327-4290-b996-d72a65e90382")); |
||||
|
private final DeviceId DEVICE_ID = new DeviceId(UUID.fromString("5512071d-5abc-411d-a907-4cdb6539c2eb")); |
||||
|
private final AssetId ASSET_ID = new AssetId(UUID.fromString("5bc010ae-bcfd-46c8-98b9-8ee8c8955a76")); |
||||
|
|
||||
|
private final SingleValueArgumentEntry assetHumidityArgEntry = new SingleValueArgumentEntry(System.currentTimeMillis() - 10, new LongDataEntry("assetHumidity", 43L), 122L); |
||||
|
private final TsRollingArgumentEntry deviceTemperatureArgEntry = createRollingArgEntry(); |
||||
|
|
||||
|
private final long ts = System.currentTimeMillis(); |
||||
|
|
||||
|
private ScriptCalculatedFieldState state; |
||||
|
private CalculatedFieldCtx ctx; |
||||
|
|
||||
|
@Autowired |
||||
|
private TbelInvokeService tbelInvokeService; |
||||
|
|
||||
|
@BeforeEach |
||||
|
void setUp() { |
||||
|
ctx = new CalculatedFieldCtx(getCalculatedField(), tbelInvokeService); |
||||
|
ctx.init(); |
||||
|
state = new ScriptCalculatedFieldState(ctx.getArgNames()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testType() { |
||||
|
assertThat(state.getType()).isEqualTo(CalculatedFieldType.SCRIPT); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testUpdateState() { |
||||
|
state.arguments = new HashMap<>(Map.of("assetHumidity", assetHumidityArgEntry)); |
||||
|
|
||||
|
Map<String, ArgumentEntry> newArgs = Map.of("deviceTemperature", deviceTemperatureArgEntry); |
||||
|
boolean stateUpdated = state.updateState(newArgs); |
||||
|
|
||||
|
assertThat(stateUpdated).isTrue(); |
||||
|
assertThat(state.getArguments()).containsExactlyInAnyOrderEntriesOf( |
||||
|
Map.of( |
||||
|
"assetHumidity", assetHumidityArgEntry, |
||||
|
"deviceTemperature", deviceTemperatureArgEntry |
||||
|
) |
||||
|
); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testUpdateStateWhenUpdateExistingEntry() { |
||||
|
state.arguments = new HashMap<>(Map.of("deviceTemperature", deviceTemperatureArgEntry, "assetHumidity", assetHumidityArgEntry)); |
||||
|
|
||||
|
SingleValueArgumentEntry newArgEntry = new SingleValueArgumentEntry(ts, new LongDataEntry("assetHumidity", 41L), 349L); |
||||
|
Map<String, ArgumentEntry> newArgs = Map.of("assetHumidity", newArgEntry); |
||||
|
boolean stateUpdated = state.updateState(newArgs); |
||||
|
|
||||
|
assertThat(stateUpdated).isTrue(); |
||||
|
assertThat(state.getArguments()).containsExactlyInAnyOrderEntriesOf( |
||||
|
Map.of( |
||||
|
"assetHumidity", newArgEntry, |
||||
|
"deviceTemperature", deviceTemperatureArgEntry |
||||
|
) |
||||
|
); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testPerformCalculation() throws ExecutionException, InterruptedException { |
||||
|
state.arguments = new HashMap<>(Map.of("deviceTemperature", deviceTemperatureArgEntry, "assetHumidity", assetHumidityArgEntry)); |
||||
|
|
||||
|
CalculatedFieldResult result = state.performCalculation(ctx).get(); |
||||
|
|
||||
|
assertThat(result).isNotNull(); |
||||
|
Output output = getCalculatedFieldConfig().getOutput(); |
||||
|
assertThat(result.getType()).isEqualTo(output.getType()); |
||||
|
assertThat(result.getScope()).isEqualTo(output.getScope()); |
||||
|
assertThat(result.getResultMap()).isEqualTo(Map.of("averageDeviceTemperature", 13.0, "assetHumidity", 43L)); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testPerformCalculationWhenOldTelemetry() throws ExecutionException, InterruptedException { |
||||
|
TsRollingArgumentEntry argumentEntry = new TsRollingArgumentEntry(); |
||||
|
|
||||
|
TreeMap<Long, BasicKvEntry> values = new TreeMap<>(); |
||||
|
values.put(ts - 40000, new LongDataEntry("deviceTemperature", 4L));// will not be used for calculation
|
||||
|
values.put(ts - 45000, new LongDataEntry("deviceTemperature", 2L));// will not be used for calculation
|
||||
|
values.put(ts - 20, new LongDataEntry("deviceTemperature", 0L)); |
||||
|
|
||||
|
argumentEntry.setTsRecords(values); |
||||
|
|
||||
|
state.arguments = new HashMap<>(Map.of("deviceTemperature", argumentEntry, "assetHumidity", assetHumidityArgEntry)); |
||||
|
|
||||
|
CalculatedFieldResult result = state.performCalculation(ctx).get(); |
||||
|
|
||||
|
assertThat(result).isNotNull(); |
||||
|
Output output = getCalculatedFieldConfig().getOutput(); |
||||
|
assertThat(result.getType()).isEqualTo(output.getType()); |
||||
|
assertThat(result.getScope()).isEqualTo(output.getScope()); |
||||
|
assertThat(result.getResultMap()).isEqualTo(Map.of("averageDeviceTemperature", 0.0, "assetHumidity", 43L)); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testPerformCalculationWhenArgumentsMoreThanLimit() throws ExecutionException, InterruptedException { |
||||
|
TsRollingArgumentEntry argumentEntry = new TsRollingArgumentEntry(); |
||||
|
TreeMap<Long, BasicKvEntry> values = new TreeMap<>(); |
||||
|
values.put(ts - 20, new LongDataEntry("deviceTemperature", 1000L));// will not be used
|
||||
|
values.put(ts - 18, new LongDataEntry("deviceTemperature", 0L)); |
||||
|
values.put(ts - 16, new LongDataEntry("deviceTemperature", 0L)); |
||||
|
values.put(ts - 14, new LongDataEntry("deviceTemperature", 0L)); |
||||
|
values.put(ts - 12, new LongDataEntry("deviceTemperature", 0L)); |
||||
|
values.put(ts - 10, new LongDataEntry("deviceTemperature", 0L)); |
||||
|
argumentEntry.setTsRecords(values); |
||||
|
|
||||
|
state.arguments = new HashMap<>(Map.of("deviceTemperature", argumentEntry, "assetHumidity", assetHumidityArgEntry)); |
||||
|
|
||||
|
CalculatedFieldResult result = state.performCalculation(ctx).get(); |
||||
|
|
||||
|
assertThat(result).isNotNull(); |
||||
|
Output output = getCalculatedFieldConfig().getOutput(); |
||||
|
assertThat(result.getType()).isEqualTo(output.getType()); |
||||
|
assertThat(result.getScope()).isEqualTo(output.getScope()); |
||||
|
assertThat(result.getResultMap()).isEqualTo(Map.of("averageDeviceTemperature", 0.0, "assetHumidity", 43L)); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testIsReadyWhenNotAllArgPresent() { |
||||
|
assertThat(state.isReady()).isFalse(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testIsReadyWhenAllArgPresent() { |
||||
|
state.arguments = new HashMap<>(Map.of("deviceTemperature", deviceTemperatureArgEntry, "assetHumidity", assetHumidityArgEntry)); |
||||
|
|
||||
|
assertThat(state.isReady()).isTrue(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testIsReadyWhenEmptyEntryPresents() { |
||||
|
state.arguments = new HashMap<>(Map.of("deviceTemperature", TsRollingArgumentEntry.EMPTY, "assetHumidity", assetHumidityArgEntry)); |
||||
|
|
||||
|
assertThat(state.isReady()).isFalse(); |
||||
|
} |
||||
|
|
||||
|
private TsRollingArgumentEntry createRollingArgEntry() { |
||||
|
TsRollingArgumentEntry argumentEntry = new TsRollingArgumentEntry(); |
||||
|
long ts = System.currentTimeMillis(); |
||||
|
|
||||
|
TreeMap<Long, BasicKvEntry> values = new TreeMap<>(); |
||||
|
values.put(ts - 40, new LongDataEntry("deviceTemperature", 10L)); |
||||
|
values.put(ts - 30, new LongDataEntry("deviceTemperature", 12L)); |
||||
|
values.put(ts - 20, new LongDataEntry("deviceTemperature", 17L)); |
||||
|
|
||||
|
argumentEntry.setTsRecords(values); |
||||
|
return argumentEntry; |
||||
|
} |
||||
|
|
||||
|
private CalculatedField getCalculatedField() { |
||||
|
CalculatedField calculatedField = new CalculatedField(); |
||||
|
calculatedField.setTenantId(TENANT_ID); |
||||
|
calculatedField.setEntityId(ASSET_ID); |
||||
|
calculatedField.setType(CalculatedFieldType.SCRIPT); |
||||
|
calculatedField.setName("Test Calculated Field"); |
||||
|
calculatedField.setConfigurationVersion(1); |
||||
|
calculatedField.setConfiguration(getCalculatedFieldConfig()); |
||||
|
calculatedField.setVersion(1L); |
||||
|
return calculatedField; |
||||
|
} |
||||
|
|
||||
|
private CalculatedFieldConfiguration getCalculatedFieldConfig() { |
||||
|
SimpleCalculatedFieldConfiguration config = new SimpleCalculatedFieldConfiguration(); |
||||
|
|
||||
|
Argument argument1 = new Argument(); |
||||
|
argument1.setRefEntityId(DEVICE_ID); |
||||
|
ReferencedEntityKey refEntityKey1 = new ReferencedEntityKey("temperature", ArgumentType.TS_ROLLING, null); |
||||
|
argument1.setRefEntityKey(refEntityKey1); |
||||
|
argument1.setLimit(5); |
||||
|
argument1.setTimeWindow(30000); |
||||
|
|
||||
|
Argument argument2 = new Argument(); |
||||
|
ReferencedEntityKey refEntityKey2 = new ReferencedEntityKey("humidity", ArgumentType.TS_LATEST, null); |
||||
|
argument1.setRefEntityKey(refEntityKey2); |
||||
|
|
||||
|
config.setArguments(Map.of("deviceTemperature", argument1, "assetHumidity", argument2)); |
||||
|
|
||||
|
config.setExpression("var result = 0; foreach(element : deviceTemperature.entrySet()) { result += element.getValue(); } var map = {}; map.put(\"averageDeviceTemperature\", result / deviceTemperature.size()); map.put(\"assetHumidity\", assetHumidity); return map;"); |
||||
|
|
||||
|
Output output = new Output(); |
||||
|
output.setType(OutputType.ATTRIBUTES); |
||||
|
output.setScope(AttributeScope.SERVER_SCOPE); |
||||
|
|
||||
|
config.setOutput(output); |
||||
|
|
||||
|
return config; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,216 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf.ctx.state; |
||||
|
|
||||
|
import org.junit.jupiter.api.BeforeEach; |
||||
|
import org.junit.jupiter.api.Test; |
||||
|
import org.thingsboard.server.common.data.AttributeScope; |
||||
|
import org.thingsboard.server.common.data.cf.CalculatedField; |
||||
|
import org.thingsboard.server.common.data.cf.CalculatedFieldType; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.Argument; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.ArgumentType; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.CalculatedFieldConfiguration; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.Output; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.OutputType; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.ReferencedEntityKey; |
||||
|
import org.thingsboard.server.common.data.cf.configuration.SimpleCalculatedFieldConfiguration; |
||||
|
import org.thingsboard.server.common.data.id.AssetId; |
||||
|
import org.thingsboard.server.common.data.id.DeviceId; |
||||
|
import org.thingsboard.server.common.data.id.TenantId; |
||||
|
import org.thingsboard.server.common.data.kv.LongDataEntry; |
||||
|
import org.thingsboard.server.common.data.kv.StringDataEntry; |
||||
|
import org.thingsboard.server.service.cf.CalculatedFieldResult; |
||||
|
|
||||
|
import java.util.HashMap; |
||||
|
import java.util.Map; |
||||
|
import java.util.UUID; |
||||
|
import java.util.concurrent.ExecutionException; |
||||
|
|
||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
import static org.assertj.core.api.Assertions.assertThatThrownBy; |
||||
|
|
||||
|
public class SimpleCalculatedFieldStateTest { |
||||
|
|
||||
|
private final TenantId TENANT_ID = TenantId.fromUUID(UUID.fromString("5b18e321-3327-4290-b996-d72a65e90382")); |
||||
|
private final DeviceId DEVICE_ID = new DeviceId(UUID.fromString("5512071d-5abc-411d-a907-4cdb6539c2eb")); |
||||
|
private final AssetId ASSET_ID = new AssetId(UUID.fromString("5bc010ae-bcfd-46c8-98b9-8ee8c8955a76")); |
||||
|
|
||||
|
private final SingleValueArgumentEntry key1ArgEntry = new SingleValueArgumentEntry(System.currentTimeMillis() - 10, new LongDataEntry("key1", 11L), 145L); |
||||
|
private final SingleValueArgumentEntry key2ArgEntry = new SingleValueArgumentEntry(System.currentTimeMillis() - 6, new LongDataEntry("key2", 15L), 165L); |
||||
|
private final SingleValueArgumentEntry key3ArgEntry = new SingleValueArgumentEntry(System.currentTimeMillis() - 3, new LongDataEntry("key3", 23L), 184L); |
||||
|
|
||||
|
private SimpleCalculatedFieldState state; |
||||
|
private CalculatedFieldCtx ctx; |
||||
|
|
||||
|
@BeforeEach |
||||
|
void setUp() { |
||||
|
ctx = new CalculatedFieldCtx(getCalculatedField(), null); |
||||
|
ctx.init(); |
||||
|
state = new SimpleCalculatedFieldState(ctx.getArgNames()); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testType() { |
||||
|
assertThat(state.getType()).isEqualTo(CalculatedFieldType.SIMPLE); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testUpdateState() { |
||||
|
state.arguments = new HashMap<>(Map.of( |
||||
|
"key1", key1ArgEntry, |
||||
|
"key2", key2ArgEntry |
||||
|
)); |
||||
|
|
||||
|
Map<String, ArgumentEntry> newArgs = Map.of("key3", key3ArgEntry); |
||||
|
boolean stateUpdated = state.updateState(newArgs); |
||||
|
|
||||
|
assertThat(stateUpdated).isTrue(); |
||||
|
assertThat(state.getArguments()).containsExactlyInAnyOrderEntriesOf( |
||||
|
Map.of( |
||||
|
"key1", key1ArgEntry, |
||||
|
"key2", key2ArgEntry, |
||||
|
"key3", key3ArgEntry |
||||
|
) |
||||
|
); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testUpdateStateWhenUpdateExistingEntry() { |
||||
|
state.arguments = new HashMap<>(Map.of("key1", key1ArgEntry)); |
||||
|
|
||||
|
SingleValueArgumentEntry newArgEntry = new SingleValueArgumentEntry(System.currentTimeMillis(), new LongDataEntry("key1", 18L), 190L); |
||||
|
Map<String, ArgumentEntry> newArgs = Map.of("key1", newArgEntry); |
||||
|
boolean stateUpdated = state.updateState(newArgs); |
||||
|
|
||||
|
assertThat(stateUpdated).isTrue(); |
||||
|
assertThat(state.getArguments()).containsExactlyInAnyOrderEntriesOf(Map.of("key1", newArgEntry)); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testUpdateStateWhenRollingEntryPassed() { |
||||
|
state.arguments = new HashMap<>(Map.of( |
||||
|
"key1", key1ArgEntry, |
||||
|
"key2", key2ArgEntry |
||||
|
)); |
||||
|
|
||||
|
Map<String, ArgumentEntry> newArgs = Map.of("key3", TsRollingArgumentEntry.EMPTY); |
||||
|
assertThatThrownBy(() -> state.updateState(newArgs)) |
||||
|
.isInstanceOf(IllegalArgumentException.class) |
||||
|
.hasMessage("Rolling argument entry is not supported for simple calculated fields."); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testPerformCalculation() throws ExecutionException, InterruptedException { |
||||
|
state.arguments = new HashMap<>(Map.of( |
||||
|
"key1", key1ArgEntry, |
||||
|
"key2", key2ArgEntry, |
||||
|
"key3", key3ArgEntry |
||||
|
)); |
||||
|
|
||||
|
CalculatedFieldResult result = state.performCalculation(ctx).get(); |
||||
|
|
||||
|
assertThat(result).isNotNull(); |
||||
|
Output output = getCalculatedFieldConfig().getOutput(); |
||||
|
assertThat(result.getType()).isEqualTo(output.getType()); |
||||
|
assertThat(result.getScope()).isEqualTo(output.getScope()); |
||||
|
assertThat(result.getResultMap()).isEqualTo(Map.of("output", 49.0)); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testPerformCalculationWhenPassedNotNumber() { |
||||
|
state.arguments = new HashMap<>(Map.of( |
||||
|
"key1", key1ArgEntry, |
||||
|
"key2", new SingleValueArgumentEntry(System.currentTimeMillis() - 9, new StringDataEntry("key2", "string"), 124L), |
||||
|
"key3", key3ArgEntry |
||||
|
)); |
||||
|
|
||||
|
assertThatThrownBy(() -> state.performCalculation(ctx)) |
||||
|
.isInstanceOf(IllegalArgumentException.class) |
||||
|
.hasMessage("Argument 'key2' is not a number."); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testIsReadyWhenNotAllArgPresent() { |
||||
|
assertThat(state.isReady()).isFalse(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testIsReadyWhenAllArgPresent() { |
||||
|
state.arguments = new HashMap<>(Map.of( |
||||
|
"key1", key1ArgEntry, |
||||
|
"key2", key2ArgEntry, |
||||
|
"key3", key3ArgEntry |
||||
|
)); |
||||
|
|
||||
|
assertThat(state.isReady()).isTrue(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testIsReadyWhenEmptyEntryPresents() { |
||||
|
state.arguments = new HashMap<>(Map.of( |
||||
|
"key1", key1ArgEntry, |
||||
|
"key2", key2ArgEntry |
||||
|
)); |
||||
|
state.getArguments().put("key3", SingleValueArgumentEntry.EMPTY); |
||||
|
|
||||
|
assertThat(state.isReady()).isFalse(); |
||||
|
} |
||||
|
|
||||
|
private CalculatedField getCalculatedField() { |
||||
|
CalculatedField calculatedField = new CalculatedField(); |
||||
|
calculatedField.setTenantId(TENANT_ID); |
||||
|
calculatedField.setEntityId(DEVICE_ID); |
||||
|
calculatedField.setType(CalculatedFieldType.SIMPLE); |
||||
|
calculatedField.setName("Test Calculated Field"); |
||||
|
calculatedField.setConfigurationVersion(1); |
||||
|
calculatedField.setConfiguration(getCalculatedFieldConfig()); |
||||
|
calculatedField.setVersion(1L); |
||||
|
return calculatedField; |
||||
|
} |
||||
|
|
||||
|
private CalculatedFieldConfiguration getCalculatedFieldConfig() { |
||||
|
SimpleCalculatedFieldConfiguration config = new SimpleCalculatedFieldConfiguration(); |
||||
|
|
||||
|
Argument argument1 = new Argument(); |
||||
|
argument1.setRefEntityId(ASSET_ID); |
||||
|
ReferencedEntityKey refEntityKey1 = new ReferencedEntityKey("temp1", ArgumentType.TS_LATEST, null); |
||||
|
argument1.setRefEntityKey(refEntityKey1); |
||||
|
|
||||
|
Argument argument2 = new Argument(); |
||||
|
argument2.setRefEntityId(ASSET_ID); |
||||
|
ReferencedEntityKey refEntityKey2 = new ReferencedEntityKey("temp2", ArgumentType.ATTRIBUTE, null); |
||||
|
argument2.setRefEntityKey(refEntityKey2); |
||||
|
|
||||
|
Argument argument3 = new Argument(); |
||||
|
argument3.setRefEntityId(ASSET_ID); |
||||
|
ReferencedEntityKey refEntityKey3 = new ReferencedEntityKey("temp3", ArgumentType.TS_LATEST, null); |
||||
|
argument3.setRefEntityKey(refEntityKey3); |
||||
|
|
||||
|
config.setArguments(Map.of("key1", argument1, "key2", argument2, "key3", argument3)); |
||||
|
|
||||
|
config.setExpression("key1 + key2 + key3"); |
||||
|
|
||||
|
Output output = new Output(); |
||||
|
output.setName("output"); |
||||
|
output.setType(OutputType.ATTRIBUTES); |
||||
|
output.setScope(AttributeScope.SERVER_SCOPE); |
||||
|
|
||||
|
config.setOutput(output); |
||||
|
|
||||
|
return config; |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,72 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf.ctx.state; |
||||
|
|
||||
|
import org.junit.jupiter.api.BeforeEach; |
||||
|
import org.junit.jupiter.api.Test; |
||||
|
import org.thingsboard.server.common.data.kv.LongDataEntry; |
||||
|
|
||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
import static org.assertj.core.api.Assertions.assertThatThrownBy; |
||||
|
|
||||
|
public class SingleValueArgumentEntryTest { |
||||
|
|
||||
|
private SingleValueArgumentEntry entry; |
||||
|
|
||||
|
private final long ts = System.currentTimeMillis(); |
||||
|
|
||||
|
@BeforeEach |
||||
|
void setUp() { |
||||
|
entry = new SingleValueArgumentEntry(ts, new LongDataEntry("key", 11L), 363L); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testArgumentEntryType() { |
||||
|
assertThat(entry.getType()).isEqualTo(ArgumentEntryType.SINGLE_VALUE); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testUpdateEntryWhenRollingEntryPassed() { |
||||
|
assertThatThrownBy(() -> entry.updateEntry(TsRollingArgumentEntry.EMPTY)) |
||||
|
.isInstanceOf(IllegalArgumentException.class) |
||||
|
.hasMessage("Unsupported argument entry type for single value argument entry: " + ArgumentEntryType.TS_ROLLING); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testUpdateEntryWithThaSameTs() { |
||||
|
assertThat(entry.updateEntry(new SingleValueArgumentEntry(ts, new LongDataEntry("key", 13L), 363L))).isFalse(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testUpdateEntryWhenNewVersionIsNull() { |
||||
|
assertThat(entry.updateEntry(new SingleValueArgumentEntry(ts + 16, new LongDataEntry("key", 13L), null))).isTrue(); |
||||
|
assertThat(entry.getValue()).isEqualTo(13L); |
||||
|
assertThat(entry.getVersion()).isNull(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testUpdateEntryWhenNewVersionIsGreaterThanCurrent() { |
||||
|
assertThat(entry.updateEntry(new SingleValueArgumentEntry(ts + 18, new LongDataEntry("key", 18L), 369L))).isTrue(); |
||||
|
assertThat(entry.getValue()).isEqualTo(18L); |
||||
|
assertThat(entry.getVersion()).isEqualTo(369L); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testUpdateEntryWhenNewVersionIsLessThanCurrent() { |
||||
|
assertThat(entry.updateEntry(new SingleValueArgumentEntry(ts + 18, new LongDataEntry("key", 18L), 234L))).isFalse(); |
||||
|
} |
||||
|
|
||||
|
} |
||||
@ -0,0 +1,96 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2024 The Thingsboard Authors |
||||
|
* |
||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||
|
* you may not use this file except in compliance with the License. |
||||
|
* You may obtain a copy of the License at |
||||
|
* |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* |
||||
|
* Unless required by applicable law or agreed to in writing, software |
||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||
|
* See the License for the specific language governing permissions and |
||||
|
* limitations under the License. |
||||
|
*/ |
||||
|
package org.thingsboard.server.service.cf.ctx.state; |
||||
|
|
||||
|
import org.junit.jupiter.api.BeforeEach; |
||||
|
import org.junit.jupiter.api.Test; |
||||
|
import org.thingsboard.server.common.data.kv.BasicKvEntry; |
||||
|
import org.thingsboard.server.common.data.kv.DoubleDataEntry; |
||||
|
import org.thingsboard.server.common.data.kv.StringDataEntry; |
||||
|
|
||||
|
import java.util.Map; |
||||
|
import java.util.TreeMap; |
||||
|
|
||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||
|
import static org.assertj.core.api.Assertions.assertThatThrownBy; |
||||
|
|
||||
|
public class TsRollingArgumentEntryTest { |
||||
|
|
||||
|
private TsRollingArgumentEntry entry; |
||||
|
|
||||
|
private final long ts = System.currentTimeMillis(); |
||||
|
|
||||
|
@BeforeEach |
||||
|
void setUp() { |
||||
|
TreeMap<Long, BasicKvEntry> values = new TreeMap<>(); |
||||
|
values.put(ts - 40, new DoubleDataEntry("key", 10.0)); |
||||
|
values.put(ts - 30, new DoubleDataEntry("key", 12.0)); |
||||
|
values.put(ts - 20, new DoubleDataEntry("key", 17.0)); |
||||
|
|
||||
|
entry = new TsRollingArgumentEntry(values); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testArgumentEntryType() { |
||||
|
assertThat(entry.getType()).isEqualTo(ArgumentEntryType.TS_ROLLING); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testUpdateEntryWhenSingleValueEntryPassed() { |
||||
|
SingleValueArgumentEntry newEntry = new SingleValueArgumentEntry(ts - 10, new DoubleDataEntry("key", 23.0), 123L); |
||||
|
|
||||
|
assertThat(entry.updateEntry(newEntry)).isTrue(); |
||||
|
assertThat(entry.getTsRecords()).hasSize(4); |
||||
|
assertThat(entry.getTsRecords().get(ts - 10).getValue()).isEqualTo(23.0); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testUpdateEntryWhenSingleValueEntryWithTheSameTsPassed() { |
||||
|
SingleValueArgumentEntry newEntry = new SingleValueArgumentEntry(ts - 20, new DoubleDataEntry("key", 23.0), 123L); |
||||
|
|
||||
|
assertThat(entry.updateEntry(newEntry)).isFalse(); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testUpdateEntryWhenRollingEntryPassed() { |
||||
|
TsRollingArgumentEntry newEntry = new TsRollingArgumentEntry(); |
||||
|
TreeMap<Long, BasicKvEntry> values = new TreeMap<>(); |
||||
|
values.put(ts - 20, new DoubleDataEntry("key", 16.0)); |
||||
|
values.put(ts - 10, new DoubleDataEntry("key", 7.0)); |
||||
|
values.put(ts - 5, new DoubleDataEntry("key", 1.0)); |
||||
|
newEntry.setTsRecords(values); |
||||
|
|
||||
|
assertThat(entry.updateEntry(newEntry)).isTrue(); |
||||
|
assertThat(entry.getTsRecords()).hasSize(5); |
||||
|
assertThat(entry.getTsRecords()).isEqualTo(Map.of( |
||||
|
ts - 40, new DoubleDataEntry("key", 10.0), |
||||
|
ts - 30, new DoubleDataEntry("key", 12.0), |
||||
|
ts - 20, new DoubleDataEntry("key", 17.0), |
||||
|
ts - 10, new DoubleDataEntry("key", 7.0), |
||||
|
ts - 5, new DoubleDataEntry("key", 1.0) |
||||
|
)); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testUpdateEntryWhenValueIsNotNumber() { |
||||
|
SingleValueArgumentEntry newEntry = new SingleValueArgumentEntry(ts - 10, new StringDataEntry("key", "string"), 123L); |
||||
|
|
||||
|
assertThatThrownBy(() -> entry.updateEntry(newEntry)) |
||||
|
.isInstanceOf(IllegalArgumentException.class) |
||||
|
.hasMessage("Argument type " + ArgumentEntryType.TS_ROLLING + " only supports numeric values."); |
||||
|
} |
||||
|
|
||||
|
} |
||||
Loading…
Reference in new issue