Initial Test cases for Vector

Author: deepak-malik

src/com/deepak/data/structures/Vector/CustomVector.java

@@ -4,6 +4,7 @@
  */
 package com.deepak.data.structures.Vector;
 
+import java.util.Arrays;
 import java.util.Enumeration;
 import java.util.NoSuchElementException;
 
@@ -267,12 +268,12 @@ public synchronized T get(int index) {
  * @param newSize
  */
 private void ensureCapacity(int newSize) {
-if (newSize >= size) {
+if (newSize >= capacity()) {
 /* If capacity increment is given, use that else double the size */
 if (capacityIncrement > 0) {
-System.arraycopy(elements, 0, elements, size, size + capacityIncrement);
+elements = Arrays.copyOf(elements, capacity() + capacityIncrement);
 } else {
-System.arraycopy(elements, 0, elements, size, size * 2);
+elements = Arrays.copyOf(elements, size * 2);
 }
 }
 }

test/com/deepak/data/structures/Vector/CustomVectorTest.java

@@ -4,12 +4,55 @@
  */
 package com.deepak.data.structures.Vector;
 
+import org.junit.Assert;
+import org.junit.Test;
+
 /**
  * Test cases for custom vector
  * 
  * @author Deepak
  */
 public class CustomVectorTest {
 
-
+/**
+ * Test case for vector creation when no capacity is defined
+ */
+@Test
+public void testVecorCreationWhenNoCapacityIsDefined() {
+CustomVector<String> vector = new CustomVector<>();
+Assert.assertEquals(vector.capacity(), 10);
+}
+
+/**
+ * Test case for vector creation when capacity is defined
+ */
+@Test
+public void testVectorCreationWhenCapacityIsDefined() {
+CustomVector<String> vector = new CustomVector<>(3);
+Assert.assertEquals(vector.capacity(), 3);
+vector.addElement("A");
+vector.addElement("B");
+vector.addElement("C");
+vector.addElement("D");
+/* When 4th element is added, capacity should double
+ * since we have not specified increment capacity */
+Assert.assertEquals(vector.capacity(), 6);
+}
+
+/**
+ * Test case for vector creation when capacity increases
+ * and increment capacity is defined
+ */
+@Test
+public void testVectorCreationWhenCapacityIncreases() {
+CustomVector<String> vector = new CustomVector<>(3, 10);
+Assert.assertEquals(vector.capacity(), 3);
+vector.addElement("A");
+vector.addElement("B");
+vector.addElement("C");
+vector.addElement("D");
+/* When 4th element is added, capacity should increase
+ * taking into consideration, the increment capacity */
+Assert.assertEquals(vector.capacity(), 13);
+}
 }