55import java .util .concurrent .CountDownLatch ;
66import java .util .concurrent .Semaphore ;
77import org .junit .Assert ;
8- import org .testng . annotations .Test ;
8+ import org .junit .Test ;
99import org .tron .common .utils .ReflectUtils ;
1010import org .tron .core .services .ratelimiter .adapter .GlobalPreemptibleAdapter ;
1111import org .tron .core .services .ratelimiter .adapter .IPQPSRateLimiterAdapter ;
@@ -23,20 +23,19 @@ public void testStrategy() {
2323 IPQpsStrategy strategy1 = (IPQpsStrategy ) ReflectUtils .getFieldObject (adapter1 ,
2424 "strategy" );
2525
26- Assert .assertTrue ( Double .valueOf (
26+ Assert .assertEquals ( 5.0d , Double .parseDouble (
2727 ReflectUtils .getFieldValue (strategy1 .getMapParams ().get ("qps" ),
28- "value" ).toString ())
29- == 5.0d );
30- Assert .assertTrue (strategy1 .getMapParams ().get ("notExist" ) == null );
28+ "value" ).toString ()), 0.0 );
29+ Assert .assertNull (strategy1 .getMapParams ().get ("notExist" ));
3130
3231 String paramString2 = "qps=5xyz" ;
3332 IPQPSRateLimiterAdapter adapter2 = new IPQPSRateLimiterAdapter (paramString2 );
3433 IPQpsStrategy strategy2 = (IPQpsStrategy ) ReflectUtils .getFieldObject (adapter2 ,
3534 "strategy" );
3635
37- Assert .assertTrue ( Double .valueOf (
36+ Assert .assertEquals ( IPQpsStrategy . DEFAULT_IPQPS , Double .valueOf (
3837 ReflectUtils .getFieldValue (strategy2 .getMapParams ().get ("qps" ),
39- "value" ).toString ()). equals ( IPQpsStrategy . DEFAULT_IPQPS ) );
38+ "value" ).toString ()));
4039 }
4140
4241 @ Test
@@ -46,9 +45,9 @@ public void testIPQPSRateLimiterAdapter() {
4645
4746 IPQpsStrategy strategy = (IPQpsStrategy ) ReflectUtils .getFieldObject (adapter ,
4847 "strategy" );
49- Assert .assertTrue ( Double
50- .valueOf (ReflectUtils .getFieldValue (strategy .getMapParams ().get ("qps" ),
51- "value" ).toString ()) == 5.0d );
48+ Assert .assertEquals ( 5.0d , Double
49+ .parseDouble (ReflectUtils .getFieldValue (strategy .getMapParams ().get ("qps" ),
50+ "value" ).toString ()), 0.0 );
5251
5352 long t0 = System .currentTimeMillis ();
5453 for (int i = 0 ; i < 20 ; i ++) {
@@ -61,15 +60,15 @@ public void testIPQPSRateLimiterAdapter() {
6160 for (int i = 0 ; i < 20 ; i ++) {
6261 if (i % 2 == 0 ) {
6362 strategy .acquire ("1.2.3.4" );
64- } else if ( i % 2 == 1 ) {
63+ } else {
6564 strategy .acquire ("4.3.2.1" );
6665 }
6766 }
6867 t1 = System .currentTimeMillis ();
6968 Assert .assertTrue (t1 - t0 > 1500 );
7069 Cache <String , RateLimiter > ipLimiter = (Cache <String , RateLimiter >) ReflectUtils
7170 .getFieldObject (strategy , "ipLimiter" );
72- Assert .assertTrue ( ipLimiter .size () == 2 );
71+ Assert .assertEquals ( 2 , ipLimiter .size ());
7372 }
7473
7574 @ Test
@@ -78,9 +77,9 @@ public void testGlobalPreemptibleAdapter() {
7877 GlobalPreemptibleAdapter adapter1 = new GlobalPreemptibleAdapter (paramString1 );
7978 GlobalPreemptibleStrategy strategy1 = (GlobalPreemptibleStrategy ) ReflectUtils
8079 .getFieldObject (adapter1 , "strategy" );
81- Assert .assertTrue ( Integer .valueOf (
80+ Assert .assertEquals ( 1 , Integer .parseInt (
8281 ReflectUtils .getFieldValue (strategy1 .getMapParams ().get ("permit" ),
83- "value" ).toString ()) == 1 );
82+ "value" ).toString ()));
8483 boolean first = strategy1 .acquire ();
8584 Assert .assertTrue (first );
8685
@@ -95,9 +94,9 @@ public void testGlobalPreemptibleAdapter() {
9594 GlobalPreemptibleAdapter adapter2 = new GlobalPreemptibleAdapter (paramString2 );
9695 GlobalPreemptibleStrategy strategy2 = (GlobalPreemptibleStrategy ) ReflectUtils
9796 .getFieldObject (adapter2 , "strategy" );
98- Assert .assertTrue ( Integer .valueOf (
97+ Assert .assertEquals ( 3 , Integer .parseInt (
9998 ReflectUtils .getFieldValue (strategy2 .getMapParams ().get ("permit" ),
100- "value" ).toString ()) == 3 );
99+ "value" ).toString ()));
101100
102101 first = strategy2 .acquire ();
103102 Assert .assertTrue (first );
@@ -114,11 +113,11 @@ public void testGlobalPreemptibleAdapter() {
114113 Assert .assertTrue (fourAfterOneRelease );
115114
116115 Semaphore sp = (Semaphore ) ReflectUtils .getFieldObject (strategy2 , "sp" );
117- Assert .assertTrue ( sp .availablePermits () == 0 );
116+ Assert .assertEquals ( 0 , sp .availablePermits ());
118117 strategy2 .release ();
119118 strategy2 .release ();
120119 strategy2 .release ();
121- Assert .assertTrue ( sp .availablePermits () == 3 );
120+ Assert .assertEquals ( 3 , sp .availablePermits ());
122121
123122 }
124123
@@ -128,9 +127,9 @@ public void testQpsRateLimiterAdapter() {
128127 QpsRateLimiterAdapter adapter = new QpsRateLimiterAdapter (paramString );
129128
130129 QpsStrategy strategy = (QpsStrategy ) ReflectUtils .getFieldObject (adapter , "strategy" );
131- Assert .assertTrue ( Double
132- .valueOf (ReflectUtils .getFieldValue (strategy .getMapParams ().get ("qps" ),
133- "value" ).toString ()) == 5 );
130+ Assert .assertEquals ( 5 , Double
131+ .parseDouble (ReflectUtils .getFieldValue (strategy .getMapParams ().get ("qps" ),
132+ "value" ).toString ()), 0.0 );
134133 strategy .acquire ();
135134
136135 long t0 = System .currentTimeMillis ();
0 commit comments