66import java .util .Iterator ;
77import java .util .List ;
88
9- import static org .junit . jupiter .api .Assertions .* ;
9+ import static org .assertj . core .api .Assertions .assertThat ;
1010
1111public class RemoveElementFromAnArrayListUnitTest {
1212 private List <String > getSportList () {
@@ -22,12 +22,10 @@ private List<String> getSportList() {
2222 @ Test
2323 void whenRemoveByIndex_thenCorrect () {
2424 List <String > sports = getSportList ();
25- assertEquals (5 , sports .size ());
26- assertTrue (sports .contains ("Basketball" ));
25+ assertThat (sports ).hasSize (5 ).contains ("Basketball" );
2726
2827 sports .remove (1 ); //removing Basketball
29- assertEquals (4 , sports .size ());
30- assertFalse (sports .contains ("Basketball" ));
28+ assertThat (sports ).hasSize (4 ).doesNotContain ("Basketball" );
3129 }
3230
3331 @ Test
@@ -36,7 +34,7 @@ void whenRemoveLastByIndex_thenCorrect() {
3634 List <String > expected = List .of ("Football" , "Basketball" , "Baseball" , "Boxing" );
3735
3836 sports .remove (sports .size () - 1 );
39- assertEquals ( expected , sports );
37+ assertThat ( sports ). isEqualTo ( expected );
4038 }
4139
4240 @ Test
@@ -45,26 +43,23 @@ void whenUsingJDK21RemoveLast_thenCorrect() {
4543 List <String > expected = List .of ("Football" , "Basketball" , "Baseball" , "Boxing" );
4644
4745 sports .removeLast ();
48- assertEquals ( expected , sports );
46+ assertThat ( sports ). isEqualTo ( expected );
4947 }
5048
5149
5250 @ Test
5351 void whenRemoveByElement_thenCorrect () {
5452 List <String > sports = getSportList ();
55- assertEquals (5 , sports .size ());
56- assertTrue (sports .contains ("Basketball" ));
53+ assertThat (sports ).hasSize (5 ).contains ("Basketball" );
5754
5855 sports .remove ("Basketball" );
59- assertEquals (4 , sports .size ());
60- assertFalse (sports .contains ("Basketball" ));
56+ assertThat (sports ).hasSize (4 ).doesNotContain ("Basketball" );
6157 }
6258
6359 @ Test
6460 void whenRemovingWhileIterating_thenCorrect () {
6561 List <String > sports = getSportList ();
66- assertEquals (5 , sports .size ());
67- assertTrue (sports .contains ("Basketball" ));
62+ assertThat (sports ).hasSize (5 ).contains ("Basketball" );
6863
6964 Iterator <String > iterator = sports .iterator ();
7065 while (iterator .hasNext ()) {
@@ -73,19 +68,16 @@ void whenRemovingWhileIterating_thenCorrect() {
7368 break ;
7469 }
7570 }
76- assertEquals (4 , sports .size ());
77- assertFalse (sports .contains ("Basketball" ));
71+ assertThat (sports ).hasSize (4 ).doesNotContain ("Basketball" );
7872 }
7973
8074 @ Test
8175 void whenRemoveIf_thenCorrect () {
8276 List <String > sports = getSportList ();
83- assertEquals (5 , sports .size ());
84- assertTrue (sports .contains ("Basketball" ));
77+ assertThat (sports ).hasSize (5 ).contains ("Basketball" );
8578
8679 sports .removeIf (p -> p .equals ("Basketball" ));
8780
88- assertEquals (4 , sports .size ());
89- assertFalse (sports .contains ("Basketball" ));
81+ assertThat (sports ).hasSize (4 ).doesNotContain ("Basketball" );
9082 }
9183}
0 commit comments