Обработка добавления null в MyTreeSet

This commit is contained in:
2025-02-03 17:45:04 +03:00
parent 65f7586e24
commit 0b5f8e6aba
2 changed files with 10 additions and 0 deletions

View File

@@ -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;

View File

@@ -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));
} }
} }