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

test relations: print all relations added as sql procedure to reproduce exact data in the PostgreSQL

Browse Source
This commit is contained in:
Sergey Matvienko 2021-09-06 13:00:34 +03:00
parent d65b02df19
commit f2f3468077
1 changed files with 46 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
dao/src/test/java/org/thingsboard/server/dao/service/BaseEntityServiceTest.java
View File

@ -70,6 +70,7 @@ import org.thingsboard.server.common.data.relation.RelationEntityTypeFilter;
import org.thingsboard.server.common.data.relation.RelationTypeGroup; import org.thingsboard.server.common.data.relation.RelationTypeGroup;
import org.thingsboard.server.dao.attributes.AttributesService; import org.thingsboard.server.dao.attributes.AttributesService;
import org.thingsboard.server.dao.model.sqlts.ts.TsKvEntity; import org.thingsboard.server.dao.model.sqlts.ts.TsKvEntity;
import org.thingsboard.server.dao.sql.relation.RelationRepository;
import org.thingsboard.server.dao.timeseries.TimeseriesService; import org.thingsboard.server.dao.timeseries.TimeseriesService;
import java.util.ArrayList; import java.util.ArrayList;
@ -98,6 +99,9 @@ public abstract class BaseEntityServiceTest extends AbstractServiceTest {
@Autowired @Autowired
private JdbcTemplate template; private JdbcTemplate template;
@Autowired
private RelationRepository relationRepository;
@Before @Before
public void before() { public void before() {
Tenant tenant = new Tenant(); Tenant tenant = new Tenant();
@ -571,6 +575,47 @@ public abstract class BaseEntityServiceTest extends AbstractServiceTest {
createLoopRelations(tenantId, "Loop-Tnt-Ast-Dev", tenantId, assets.get(0).getId(), devices.get(0).getId()); createLoopRelations(tenantId, "Loop-Tnt-Ast-Dev", tenantId, assets.get(0).getId(), devices.get(0).getId());
createLoopRelations(tenantId, "Loop-Tnt-Ast", tenantId, assets.get(1).getId()); createLoopRelations(tenantId, "Loop-Tnt-Ast", tenantId, assets.get(1).getId());
createLoopRelations(tenantId, "Loop-Ast-Tnt-Ast", assets.get(2).getId(), tenantId, assets.get(3).getId()); createLoopRelations(tenantId, "Loop-Ast-Tnt-Ast", assets.get(2).getId(), tenantId, assets.get(3).getId());
printAllRelations();
}
/*
* This useful to reproduce exact data in the PostgreSQL and play around with pgadmin query and analyze tool
* */
private void printAllRelations() {
System.out.println("" +
"DO\n" +
"$$\n" +
" DECLARE\n" +
" someint integer;\n" +
" BEGIN\n" +
" DROP TABLE IF EXISTS relation_test;\n" +
" CREATE TABLE IF NOT EXISTS relation_test\n" +
" (\n" +
" from_id uuid,\n" +
" from_type varchar(255),\n" +
" to_id uuid,\n" +
" to_type varchar(255),\n" +
" relation_type_group varchar(255),\n" +
" relation_type varchar(255),\n" +
" additional_info varchar,\n" +
" CONSTRAINT relation_test_pkey PRIMARY KEY (from_id, from_type, relation_type_group, relation_type, to_id, to_type)\n" +
" );");
relationRepository.findAll().forEach(r ->
System.out.printf("INSERT INTO relation_test (from_id, from_type, to_id, to_type, relation_type_group, relation_type, additional_info)" +
" VALUES (%s, %s, %s, %s, %s, %s, %s);\n",
quote(r.getFromId()), quote(r.getFromType()), quote(r.getToId()), quote(r.getToType()),
quote(r.getRelationTypeGroup()), quote(r.getRelationType()), quote(r.getAdditionalInfo()))
);
System.out.println("" +
" END\n" +
"$$;");
}
private String quote(Object s) {
return s == null ? null : "'" + s + "'";
} }
void createLoopRelations(TenantId tenantId, String type, EntityId... ids) { void createLoopRelations(TenantId tenantId, String type, EntityId... ids) {