Обработка добавления null в MyTreeSet
This commit is contained in:
@@ -20,6 +20,10 @@ public class MyTreeSet<E extends Comparable<E>> {
|
||||
}
|
||||
|
||||
public boolean add(E element) {
|
||||
if (element == null) {
|
||||
throw new NullPointerException("Нельзя добавить null в TreeSet!");
|
||||
}
|
||||
|
||||
if (root == null) {
|
||||
root = new Node(element);
|
||||
return true;
|
||||
|
||||
@@ -18,8 +18,14 @@ class MyTreeSetTests {
|
||||
|
||||
@Test
|
||||
void testAdd() {
|
||||
assertThrows(NullPointerException.class, () -> myTreeSet.add(null));
|
||||
assertThrows(NullPointerException.class, () -> treeSet.add(null));
|
||||
|
||||
assertEquals(myTreeSet.add(150), treeSet.add(150));
|
||||
assertEquals(myTreeSet.add(200), treeSet.add(200));
|
||||
assertEquals(myTreeSet.add(150), treeSet.add(150));
|
||||
|
||||
assertThrows(NullPointerException.class, () -> myTreeSet.add(null));
|
||||
assertThrows(NullPointerException.class, () -> treeSet.add(null));
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user