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