Skip to content

Commit aa1bc86

Browse files
committed
[impr-rm-el-list] using assertj
1 parent d481109 commit aa1bc86

File tree

1 file changed

+11
-19
lines changed

1 file changed

+11
-19
lines changed

core-java-modules/core-java-collections-array-list/src/test/java/com/baeldung/list/removeelement/RemoveElementFromAnArrayListUnitTest.java

Lines changed: 11 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
import java.util.Iterator;
77
import java.util.List;
88

9-
import static org.junit.jupiter.api.Assertions.*;
9+
import static org.assertj.core.api.Assertions.assertThat;
1010

1111
public 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

Comments
 (0)