Fix tests

application/src/main/java/org/thingsboard/server/service/edqs/DefaultEdqsService.java

@@ -55,9 +55,9 @@ import org.thingsboard.server.common.data.kv.KvEntry;
 import org.thingsboard.server.common.msg.edqs.EdqsService;
 import org.thingsboard.server.common.msg.queue.ServiceType;
 import org.thingsboard.server.dao.attributes.AttributesService;
-import org.thingsboard.server.edqs.processor.EdqsConverter;
+import org.thingsboard.server.edqs.util.EdqsConverter;
 import org.thingsboard.server.edqs.processor.EdqsProducer;
-import org.thingsboard.server.edqs.util.EdqsPartitionService;
+import org.thingsboard.server.edqs.state.EdqsPartitionService;
 import org.thingsboard.server.gen.transport.TransportProtos;
 import org.thingsboard.server.gen.transport.TransportProtos.EdqsEventMsg;
 import org.thingsboard.server.gen.transport.TransportProtos.EdqsRequestMsg;

common/data/src/main/java/org/thingsboard/server/common/data/edqs/AttributeKv.java

@@ -20,6 +20,7 @@ import lombok.Builder;
 import lombok.Data;
 import lombok.NoArgsConstructor;
 import org.thingsboard.server.common.data.AttributeScope;
+import org.thingsboard.server.common.data.ObjectType;
 import org.thingsboard.server.common.data.id.EntityId;
 import org.thingsboard.server.common.data.kv.AttributeKvEntry;
 import org.thingsboard.server.common.data.kv.KvEntry;
@@ -66,4 +67,9 @@ public class AttributeKv implements EdqsObject {
         return version;
     }
 
+    @Override
+    public ObjectType type() {
+        return ObjectType.ATTRIBUTE_KV;
+    }
+
 }

common/data/src/main/java/org/thingsboard/server/common/data/edqs/EdqsObject.java

@@ -16,6 +16,7 @@
 package org.thingsboard.server.common.data.edqs;
 
 import com.fasterxml.jackson.annotation.JsonIgnore;
+import org.thingsboard.server.common.data.ObjectType;
 
 public interface EdqsObject {
 
@@ -25,4 +26,7 @@ public interface EdqsObject {
     @JsonIgnore
     Long version();
 
+    @JsonIgnore
+    ObjectType type();
+
 }

common/data/src/main/java/org/thingsboard/server/common/data/edqs/Entity.java

@@ -20,6 +20,7 @@ import com.fasterxml.jackson.annotation.JsonTypeInfo;
 import lombok.Data;
 import lombok.NoArgsConstructor;
 import org.thingsboard.server.common.data.EntityType;
+import org.thingsboard.server.common.data.ObjectType;
 import org.thingsboard.server.common.data.edqs.fields.EntityFields;
 import org.thingsboard.server.common.data.edqs.fields.EntityIdFields;
 
@@ -59,4 +60,9 @@ public class Entity implements EdqsObject {
         return fields.getVersion();
     }
 
+    @Override
+    public ObjectType type() {
+        return ObjectType.fromEntityType(type);
+    }
+
 }

common/data/src/main/java/org/thingsboard/server/common/data/edqs/LatestTsKv.java

@@ -19,6 +19,7 @@ import lombok.AllArgsConstructor;
 import lombok.Builder;
 import lombok.Data;
 import lombok.NoArgsConstructor;
+import org.thingsboard.server.common.data.ObjectType;
 import org.thingsboard.server.common.data.id.EntityId;
 import org.thingsboard.server.common.data.kv.KvEntry;
 import org.thingsboard.server.common.data.kv.TsKvEntry;
@@ -61,4 +62,9 @@ public class LatestTsKv implements EdqsObject {
         return version;
     }
 
+    @Override
+    public ObjectType type() {
+        return ObjectType.LATEST_TS_KV;
+    }
+
 }

common/data/src/main/java/org/thingsboard/server/common/data/relation/EntityRelation.java

@@ -25,6 +25,7 @@ import lombok.ToString;
 import lombok.extern.slf4j.Slf4j;
 import org.thingsboard.server.common.data.BaseDataWithAdditionalInfo;
 import org.thingsboard.server.common.data.HasVersion;
+import org.thingsboard.server.common.data.ObjectType;
 import org.thingsboard.server.common.data.edqs.EdqsObject;
 import org.thingsboard.server.common.data.id.EntityId;
 import org.thingsboard.server.common.data.validation.Length;
@@ -127,4 +128,9 @@ public class EntityRelation implements HasVersion, Serializable, EdqsObject {
         return version;
     }
 
+    @Override
+    public ObjectType type() {
+        return ObjectType.RELATION;
+    }
+
 }

common/edqs/src/main/java/org/thingsboard/server/edqs/processor/EdqsProcessor.java

@@ -49,7 +49,8 @@ import org.thingsboard.server.common.msg.queue.ServiceType;
 import org.thingsboard.server.common.msg.queue.TopicPartitionInfo;
 import org.thingsboard.server.edqs.repo.EdqRepository;
 import org.thingsboard.server.edqs.state.EdqsStateService;
-import org.thingsboard.server.edqs.util.EdqsPartitionService;
+import org.thingsboard.server.edqs.util.EdqsConverter;
+import org.thingsboard.server.edqs.state.EdqsPartitionService;
 import org.thingsboard.server.edqs.util.VersionsStore;
 import org.thingsboard.server.gen.transport.TransportProtos;
 import org.thingsboard.server.gen.transport.TransportProtos.EdqsEventMsg;

common/edqs/src/main/java/org/thingsboard/server/edqs/processor/EdqsProducer.java

@@ -21,7 +21,7 @@ import org.apache.kafka.common.errors.RecordTooLargeException;
 import org.thingsboard.server.common.data.ObjectType;
 import org.thingsboard.server.common.data.id.TenantId;
 import org.thingsboard.server.common.msg.queue.TopicPartitionInfo;
-import org.thingsboard.server.edqs.util.EdqsPartitionService;
+import org.thingsboard.server.edqs.state.EdqsPartitionService;
 import org.thingsboard.server.gen.transport.TransportProtos.ToEdqsMsg;
 import org.thingsboard.server.queue.TbQueueCallback;
 import org.thingsboard.server.queue.TbQueueMsgMetadata;

common/edqs/src/main/java/org/thingsboard/server/edqs/util/EdqsPartitionService.java → common/edqs/src/main/java/org/thingsboard/server/edqs/state/EdqsPartitionService.java

@@ -13,7 +13,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package org.thingsboard.server.edqs.util;
+package org.thingsboard.server.edqs.state;
 
 import lombok.RequiredArgsConstructor;
 import org.springframework.stereotype.Service;

common/edqs/src/main/java/org/thingsboard/server/edqs/state/KafkaEdqsStateService.java

@@ -30,7 +30,6 @@ import org.thingsboard.server.common.msg.queue.ServiceType;
 import org.thingsboard.server.common.msg.queue.TopicPartitionInfo;
 import org.thingsboard.server.edqs.processor.EdqsProcessor;
 import org.thingsboard.server.edqs.processor.EdqsProducer;
-import org.thingsboard.server.edqs.util.EdqsPartitionService;
 import org.thingsboard.server.edqs.util.VersionsStore;
 import org.thingsboard.server.gen.transport.TransportProtos.EdqsEventMsg;
 import org.thingsboard.server.gen.transport.TransportProtos.ToEdqsMsg;

common/edqs/src/main/java/org/thingsboard/server/edqs/processor/EdqsConverter.java → common/edqs/src/main/java/org/thingsboard/server/edqs/util/EdqsConverter.java

@@ -13,7 +13,7 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-package org.thingsboard.server.edqs.processor;
+package org.thingsboard.server.edqs.util;
 
 import com.google.protobuf.ByteString;
 import lombok.RequiredArgsConstructor;

edqs/src/test/java/org/thingsboard/server/edqs/repo/AbstractEDQTest.java

@@ -59,7 +59,7 @@ import org.thingsboard.server.common.data.query.KeyFilter;
 import org.thingsboard.server.common.data.query.StringFilterPredicate;
 import org.thingsboard.server.common.data.relation.EntityRelation;
 import org.thingsboard.server.common.data.relation.RelationTypeGroup;
-import org.thingsboard.server.edqs.processor.EdqsConverter;
+import org.thingsboard.server.edqs.util.EdqsConverter;
 
 import java.util.Collections;
 import java.util.List;
@@ -67,7 +67,7 @@ import java.util.UUID;
 
 @RunWith(SpringRunner.class)
 @Configuration
-@ComponentScan({"org.thingsboard.server.edqs.repo"})
+@ComponentScan({"org.thingsboard.server.edqs.repo", "org.thingsboard.server.edqs.util"})
 @EntityScan("org.thingsboard.server.edqs")
 @TestPropertySource(locations = {"classpath:edq-test.properties"})
 @TestExecutionListeners({
@@ -77,6 +77,8 @@ public abstract class AbstractEDQTest {
 
     @Autowired
     protected InMemoryEdqRepository repository;
+    @Autowired
+    protected EdqsConverter edqsConverter;
 
     protected final TenantId tenantId = TenantId.fromUUID(UUID.randomUUID());
     protected final CustomerId customerId = new CustomerId(UUID.randomUUID());
@@ -218,7 +220,7 @@ public abstract class AbstractEDQTest {
     }
 
     protected void createRelation(EntityType fromType, UUID fromId, EntityType toType, UUID toId, RelationTypeGroup group, String type) {
-        repository.get(tenantId).addOrUpdate(new EntityRelation(EntityIdFactory.getByTypeAndUuid(fromType, fromId), EntityIdFactory.getByTypeAndUuid(toType, toId), type, group));
+        addOrUpdate(new EntityRelation(EntityIdFactory.getByTypeAndUuid(fromType, fromId), EntityIdFactory.getByTypeAndUuid(toType, toId), type, group));
     }
 
 
@@ -243,6 +245,8 @@ public abstract class AbstractEDQTest {
     }
 
     protected void addOrUpdate(EdqsObject edqsObject) {
+        byte[] serialized = edqsConverter.serialize(edqsObject.type(), edqsObject);
+        edqsObject = edqsConverter.deserialize(edqsObject.type(), serialized);
         repository.get(tenantId).addOrUpdate(edqsObject);
     }