11package io .kafbat .ui .mapper ;
22
33import static org .assertj .core .api .Assertions .assertThat ;
4- import static org .junit .jupiter .api .Assertions .*;
54
65import io .kafbat .ui .config .ClustersProperties ;
76import io .kafbat .ui .connect .model .ClusterInfo ;
@@ -26,31 +25,25 @@ class KafkaConnectMapperTest {
2625 @ Test
2726 void toKafkaConnect () {
2827 ThreadLocalRandom random = ThreadLocalRandom .current ();
29- KafkaConnectMapper mapper = new KafkaConnectMapperImpl ();
30- ClustersProperties .ConnectCluster connectCluster = ClustersProperties .ConnectCluster .builder ()
31- .name (UUID .randomUUID ().toString ())
32- .address ("http://localhost:" + random .nextInt (1000 , 5000 ))
33- .username (UUID .randomUUID ().toString ())
34- .password (UUID .randomUUID ().toString ()).build ();
3528
3629 List <InternalConnectorInfo > connectors = new ArrayList <>();
3730 int failedConnectors = 0 ;
3831 int failedTasks = 0 ;
3932 int tasksPerConnector = random .nextInt (1 , 10 );
4033
4134 for (int i = 0 ; i < 10 ; i ++) {
42- ConnectorStateDTO connectorStateDTO ;
35+ ConnectorStateDTO connectorState ;
4336 if (random .nextBoolean ()) {
44- connectorStateDTO = ConnectorStateDTO .FAILED ;
37+ connectorState = ConnectorStateDTO .FAILED ;
4538 failedConnectors ++;
4639 } else {
47- connectorStateDTO = ConnectorStateDTO .RUNNING ;
40+ connectorState = ConnectorStateDTO .RUNNING ;
4841 }
4942
50- ConnectorDTO connectorDTO = new ConnectorDTO ();
51- connectorDTO .setName (UUID .randomUUID ().toString ());
52- connectorDTO .setStatus (
53- new ConnectorStatusDTO (connectorStateDTO , UUID .randomUUID ().toString ())
43+ ConnectorDTO connectorDto = new ConnectorDTO ();
44+ connectorDto .setName (UUID .randomUUID ().toString ());
45+ connectorDto .setStatus (
46+ new ConnectorStatusDTO (connectorState , UUID .randomUUID ().toString ())
5447 );
5548
5649 List <TaskDTO > tasks = new ArrayList <>();
@@ -76,9 +69,9 @@ void toKafkaConnect() {
7669 taskIds .add (taskId );
7770 }
7871
79- connectorDTO .setTasks (taskIds );
72+ connectorDto .setTasks (taskIds );
8073 InternalConnectorInfo connector = InternalConnectorInfo .builder ()
81- .connector (connectorDTO )
74+ .connector (connectorDto )
8275 .tasks (tasks )
8376 .build ();
8477
@@ -90,17 +83,24 @@ void toKafkaConnect() {
9083 clusterInfo .setCommit (UUID .randomUUID ().toString ());
9184 clusterInfo .setKafkaClusterId (UUID .randomUUID ().toString ());
9285
93- ConnectDTO connectDTO = new ConnectDTO ();
94- connectDTO .setName (connectCluster .getName ());
95- connectDTO .setAddress (connectCluster .getAddress ());
96- connectDTO .setVersion (JsonNullable .of (clusterInfo .getVersion ()));
97- connectDTO .setCommit (JsonNullable .of (clusterInfo .getCommit ()));
98- connectDTO .setClusterId (JsonNullable .of (clusterInfo .getKafkaClusterId ()));
99- connectDTO .setConnectorsCount (JsonNullable .of (connectors .size ()));
100- connectDTO .setFailedConnectorsCount (JsonNullable .of (failedConnectors ));
101- connectDTO .setTasksCount (JsonNullable .of (connectors .size () * tasksPerConnector ));
102- connectDTO .setFailedTasksCount (JsonNullable .of (failedTasks ));
86+ ClustersProperties .ConnectCluster connectCluster = ClustersProperties .ConnectCluster .builder ()
87+ .name (UUID .randomUUID ().toString ())
88+ .address ("http://localhost:" + random .nextInt (1000 , 5000 ))
89+ .username (UUID .randomUUID ().toString ())
90+ .password (UUID .randomUUID ().toString ()).build ();
91+
92+ ConnectDTO connectDto = new ConnectDTO ();
93+ connectDto .setName (connectCluster .getName ());
94+ connectDto .setAddress (connectCluster .getAddress ());
95+ connectDto .setVersion (JsonNullable .of (clusterInfo .getVersion ()));
96+ connectDto .setCommit (JsonNullable .of (clusterInfo .getCommit ()));
97+ connectDto .setClusterId (JsonNullable .of (clusterInfo .getKafkaClusterId ()));
98+ connectDto .setConnectorsCount (JsonNullable .of (connectors .size ()));
99+ connectDto .setFailedConnectorsCount (JsonNullable .of (failedConnectors ));
100+ connectDto .setTasksCount (JsonNullable .of (connectors .size () * tasksPerConnector ));
101+ connectDto .setFailedTasksCount (JsonNullable .of (failedTasks ));
103102
103+ KafkaConnectMapper mapper = new KafkaConnectMapperImpl ();
104104 ConnectDTO kafkaConnect = mapper .toKafkaConnect (
105105 connectCluster ,
106106 connectors ,
@@ -109,7 +109,7 @@ void toKafkaConnect() {
109109 );
110110
111111 assertThat (kafkaConnect ).isNotNull ();
112- assertThat (kafkaConnect ).isEqualTo (connectDTO );
112+ assertThat (kafkaConnect ).isEqualTo (connectDto );
113113
114114 }
115115}
0 commit comments