Browse Source

fixed entity service registry test

pull/12009/head
IrynaMatveieva 2 years ago
parent
commit
258ba4807b
  1. 3
      dao/src/main/java/org/thingsboard/server/dao/entity/DefaultEntityServiceRegistry.java
  2. 5
      dao/src/test/java/org/thingsboard/server/dao/service/EntityServiceRegistryTest.java

3
dao/src/main/java/org/thingsboard/server/dao/entity/DefaultEntityServiceRegistry.java

@ -43,6 +43,9 @@ public class DefaultEntityServiceRegistry implements EntityServiceRegistry {
if (EntityType.RULE_CHAIN.equals(entityType)) {
entityDaoServicesMap.put(EntityType.RULE_NODE, entityDaoService);
}
if (EntityType.CALCULATED_FIELD.equals(entityType)) {
entityDaoServicesMap.put(EntityType.CALCULATED_FIELD_LINK, entityDaoService);
}
});
log.debug("Initialized EntityServiceRegistry total [{}] entries", entityDaoServicesMap.size());
}

5
dao/src/test/java/org/thingsboard/server/dao/service/EntityServiceRegistryTest.java

@ -20,6 +20,7 @@ import org.junit.Assert;
import org.junit.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.thingsboard.server.common.data.EntityType;
import org.thingsboard.server.dao.calculated_field.CalculatedFieldService;
import org.thingsboard.server.dao.entity.EntityDaoService;
import org.thingsboard.server.dao.entity.EntityServiceRegistry;
import org.thingsboard.server.dao.rule.RuleChainService;
@ -44,4 +45,8 @@ public class EntityServiceRegistryTest extends AbstractServiceTest {
Assert.assertTrue(entityServiceRegistry.getServiceByEntityType(EntityType.RULE_NODE) instanceof RuleChainService);
}
@Test
public void givenCalculatedFieldLinkEntityType_whenGetServiceByEntityTypeCalled_thenReturnedCalculatedFieldService() {
Assert.assertTrue(entityServiceRegistry.getServiceByEntityType(EntityType.CALCULATED_FIELD_LINK) instanceof CalculatedFieldService);
}
}

Loading…
Cancel
Save