diff --git a/core-java-arrays/src/main/java/com/baeldung/array/AddElementToEndOfArray.java b/core-java-arrays/src/main/java/com/baeldung/array/AddElementToEndOfArray.java index 1dc6600600..dcd61cdfa7 100644 --- a/core-java-arrays/src/main/java/com/baeldung/array/AddElementToEndOfArray.java +++ b/core-java-arrays/src/main/java/com/baeldung/array/AddElementToEndOfArray.java @@ -8,7 +8,7 @@ public class AddElementToEndOfArray { public Integer[] addElementUsingArraysCopyOf(Integer[] srcArray, int elementToAdd) { Integer[] destArray = Arrays.copyOf(srcArray, srcArray.length + 1); - destArray[destArray.length-1] = elementToAdd; + destArray[destArray.length - 1] = elementToAdd; return destArray; } @@ -26,9 +26,9 @@ public class AddElementToEndOfArray { System.arraycopy(srcArray, 0, destArray, 0, srcArray.length); - destArray[destArray.length-1] = elementToAdd; + destArray[destArray.length - 1] = elementToAdd; return destArray; } -} \ No newline at end of file +} diff --git a/core-java-arrays/src/test/java/com/baeldung/array/AddElementToEndOfArrayUnitTest.java b/core-java-arrays/src/test/java/com/baeldung/array/AddElementToEndOfArrayUnitTest.java index beeed1a313..f6f1f954f6 100644 --- a/core-java-arrays/src/test/java/com/baeldung/array/AddElementToEndOfArrayUnitTest.java +++ b/core-java-arrays/src/test/java/com/baeldung/array/AddElementToEndOfArrayUnitTest.java @@ -3,44 +3,47 @@ package com.baeldung.array; import org.junit.Before; import org.junit.Test; -import static org.junit.Assert.assertEquals; +import static org.junit.Assert.assertArrayEquals; public class AddElementToEndOfArrayUnitTest { AddElementToEndOfArray addElementToEndOfArray; @Before - public void init(){ + public void init() { addElementToEndOfArray = new AddElementToEndOfArray(); } @Test - public void givenSourceArrayAndElement_whenAddElementUsingArraysCopyIsInvoked_thenNewElementMustBeAdded(){ - Integer[] sourceArray = {1,2,3,4}; + public void givenSourceArrayAndElement_whenAddElementUsingArraysCopyIsInvoked_thenNewElementMustBeAdded() { + Integer[] sourceArray = {1, 2, 3, 4}; int elementToAdd = 5; Integer[] destArray = addElementToEndOfArray.addElementUsingArraysCopyOf(sourceArray, elementToAdd); - assertEquals(elementToAdd, destArray[destArray.length-1].intValue()); + Integer[] expectedArray = {1, 2, 3, 4, 5}; + assertArrayEquals(expectedArray, destArray); } @Test - public void givenSourceArrayAndElement_whenAddElementUsingArrayListIsInvoked_thenNewElementMustBeAdded(){ - Integer[] sourceArray = {1,2,3,4}; + public void givenSourceArrayAndElement_whenAddElementUsingArrayListIsInvoked_thenNewElementMustBeAdded() { + Integer[] sourceArray = {1, 2, 3, 4}; int elementToAdd = 5; Integer[] destArray = addElementToEndOfArray.addElementUsingArrayList(sourceArray, elementToAdd); - assertEquals(elementToAdd, destArray[destArray.length-1].intValue()); + Integer[] expectedArray = {1, 2, 3, 4, 5}; + assertArrayEquals(expectedArray, destArray); } @Test - public void givenSourceArrayAndElement_whenAddElementUsingSystemArrayCopyIsInvoked_thenNewElementMustBeAdded(){ - Integer[] sourceArray = {1,2,3,4}; + public void givenSourceArrayAndElement_whenAddElementUsingSystemArrayCopyIsInvoked_thenNewElementMustBeAdded() { + Integer[] sourceArray = {1, 2, 3, 4}; int elementToAdd = 5; Integer[] destArray = addElementToEndOfArray.addElementUsingSystemArrayCopy(sourceArray, elementToAdd); - assertEquals(elementToAdd, destArray[destArray.length-1].intValue()); + Integer[] expectedArray = {1, 2, 3, 4, 5}; + assertArrayEquals(expectedArray, destArray); } }