Minor changes

Add UnitTests for Iterable examples
This commit is contained in:
apeterlic
2022-04-18 10:21:53 +02:00
parent fd7e41320b
commit dfe12799e9
2 changed files with 64 additions and 26 deletions

View File

@@ -1,48 +1,32 @@
package com.baeldung.collections.iterable;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
class IterableExample {
public static void iterateUsingIterator() {
List<Integer> numbers = getNumbers();
public class IterableExample {
public void iterateUsingIterator(List<Integer> numbers) {
Iterator<Integer> iterator = numbers.iterator();
while (iterator.hasNext()) {
System.out.println(iterator.next());
}
}
public static void removeElementsUsingIterator() {
List<Integer> numbers = getNumbers();
Iterator<Integer> iterator = numbers.iterator();
while (iterator.hasNext()) {
iterator.remove();
}
}
public static void iterateUsingEnhancedForLoop() {
List<Integer> numbers = getNumbers();
public void iterateUsingEnhancedForLoop(List<Integer> numbers) {
for (Integer number : numbers) {
System.out.println(number);
}
}
public static void iterateUsingForEachLoop() {
List<Integer> numbers = getNumbers();
public void iterateUsingForEachLoop(List<Integer> numbers) {
numbers.forEach(System.out::println);
}
private static List<Integer> getNumbers() {
List<Integer> numbers = new ArrayList<>();
numbers.add(10);
numbers.add(20);
numbers.add(30);
numbers.add(40);
return numbers;
public void removeElementsUsingIterator(List<Integer> numbers) {
Iterator<Integer> iterator = numbers.iterator();
while (iterator.hasNext()) {
iterator.next();
iterator.remove();
}
}
}

View File

@@ -0,0 +1,54 @@
package com.baeldung.collections.iterable;
import com.baeldung.collections.iterable.IterableExample;
import org.junit.jupiter.api.Test;
import java.util.ArrayList;
import java.util.List;
import static org.junit.jupiter.api.Assertions.assertEquals;
class IterableUnitTest {
private static List<Integer> getNumbers() {
List<Integer> numbers = new ArrayList<>();
numbers.add(10);
numbers.add(20);
numbers.add(30);
numbers.add(40);
return numbers;
}
@Test
void givenNumbers_whenUsingIterator_thenCorrectSize() {
List<Integer> numbers = getNumbers();
IterableExample iterableExample = new IterableExample();
iterableExample.iterateUsingIterator(numbers);
assertEquals(4, numbers.size());
}
@Test
void givenNumbers_whenRemoveElements_thenEmptyList() {
List<Integer> numbers = getNumbers();
IterableExample iterableExample = new IterableExample();
iterableExample.removeElementsUsingIterator(numbers);
assertEquals(0, numbers.size());
}
@Test
void givenNumbers_whenIterateUsingEnhancedForLoop_thenCorrectSize() {
List<Integer> numbers = getNumbers();
IterableExample iterableExample = new IterableExample();
iterableExample.iterateUsingEnhancedForLoop(numbers);
assertEquals(4, numbers.size());
}
@Test
void givenNumbers_whenIterateUsingForEachLoop_thenCorrectSize() {
List<Integer> numbers = getNumbers();
IterableExample iterableExample = new IterableExample();
iterableExample.iterateUsingForEachLoop(numbers);
assertEquals(4, numbers.size());
}
}