21
21
import com .navercorp .pinpoint .common .server .util .time .Range ;
22
22
import com .navercorp .pinpoint .web .alarm .DataCollectorCategory ;
23
23
import com .navercorp .pinpoint .web .dao .stat .AgentStatDao ;
24
+ import org .assertj .core .api .Assertions ;
24
25
import org .junit .jupiter .api .Test ;
25
26
26
- import java .util .ArrayList ;
27
27
import java .util .List ;
28
28
import java .util .Map ;
29
29
30
- import static org .junit .jupiter .api .Assertions .assertEquals ;
31
30
import static org .mockito .Mockito .mock ;
32
31
import static org .mockito .Mockito .when ;
33
32
@@ -41,36 +40,37 @@ public void collect() {
41
40
String agentId1 = "testAgent1" ;
42
41
String agentId2 = "testAgent2" ;
43
42
44
- List <String > agentList = new ArrayList <>();
45
- agentList .add (agentId1 );
46
- agentList .add (agentId2 );
43
+ List <String > agentList = List .of (agentId1 , agentId2 );
47
44
48
45
AgentStatDao <FileDescriptorBo > fileDescriptorDao = (AgentStatDao <FileDescriptorBo >) mock (AgentStatDao .class );
49
46
long timeStamp = 1558936971494L ;
50
47
Range range = Range .newUncheckedRange (timeStamp - DataCollectorFactory .SLOT_INTERVAL_FIVE_MIN , timeStamp );
51
48
52
- List <FileDescriptorBo > fileDescriptorBoList1 = new ArrayList <>();
53
49
FileDescriptorBo fileDescriptorBo1_1 = new FileDescriptorBo ();
54
50
fileDescriptorBo1_1 .setOpenFileDescriptorCount (200 );
55
51
FileDescriptorBo fileDescriptorBo1_2 = new FileDescriptorBo ();
56
52
fileDescriptorBo1_2 .setOpenFileDescriptorCount (300 );
57
53
FileDescriptorBo fileDescriptorBo1_3 = new FileDescriptorBo ();
58
54
fileDescriptorBo1_3 .setOpenFileDescriptorCount (400 );
59
- fileDescriptorBoList1 .add (fileDescriptorBo1_1 );
60
- fileDescriptorBoList1 .add (fileDescriptorBo1_2 );
61
- fileDescriptorBoList1 .add (fileDescriptorBo1_3 );
55
+
56
+ List <FileDescriptorBo > fileDescriptorBoList1 = List .of (
57
+ fileDescriptorBo1_1 ,
58
+ fileDescriptorBo1_2 ,
59
+ fileDescriptorBo1_3
60
+ );
62
61
when (fileDescriptorDao .getAgentStatList (agentId1 , range )).thenReturn (fileDescriptorBoList1 );
63
62
64
- List <FileDescriptorBo > fileDescriptorBoList2 = new ArrayList <>();
65
63
FileDescriptorBo fileDescriptorBo2_1 = new FileDescriptorBo ();
66
64
fileDescriptorBo2_1 .setOpenFileDescriptorCount (250 );
67
65
FileDescriptorBo fileDescriptorBo2_2 = new FileDescriptorBo ();
68
66
fileDescriptorBo2_2 .setOpenFileDescriptorCount (350 );
69
67
FileDescriptorBo fileDescriptorBo2_3 = new FileDescriptorBo ();
70
68
fileDescriptorBo2_3 .setOpenFileDescriptorCount (450 );
71
- fileDescriptorBoList2 .add (fileDescriptorBo2_1 );
72
- fileDescriptorBoList2 .add (fileDescriptorBo2_2 );
73
- fileDescriptorBoList2 .add (fileDescriptorBo2_3 );
69
+ List <FileDescriptorBo > fileDescriptorBoList2 = List .of (
70
+ fileDescriptorBo2_1 ,
71
+ fileDescriptorBo2_2 ,
72
+ fileDescriptorBo2_3
73
+ );
74
74
when (fileDescriptorDao .getAgentStatList (agentId2 , range )).thenReturn (fileDescriptorBoList2 );
75
75
76
76
FileDescriptorDataCollector fileDescriptorDataCollector = new FileDescriptorDataCollector (
@@ -83,8 +83,9 @@ public void collect() {
83
83
84
84
fileDescriptorDataCollector .collect ();
85
85
Map <String , Long > fileDescriptorCount = fileDescriptorDataCollector .getFileDescriptorCount ();
86
- assertEquals (fileDescriptorCount .size (), 2 );
87
- assertEquals (fileDescriptorCount .get (agentId1 ), Long .valueOf (300L ));
88
- assertEquals (fileDescriptorCount .get (agentId2 ), Long .valueOf (350L ));
86
+ Assertions .assertThat (fileDescriptorCount )
87
+ .hasSize (2 )
88
+ .containsEntry (agentId1 , 300L )
89
+ .containsEntry (agentId2 , 350L );
89
90
}
90
91
}
0 commit comments