OnJava8-Examples/unittesting/JUnitDemo.java

//: unittesting/JUnitDemo.java
// Simple use of JUnit to test ArrayList
// {Depends: junit.jar}
import java.util.*;
import org.junit.Test;
import org.junit.Assert.*;

// So we can see the list objects being created,
// and keep track of when they are cleaned up:
class CountedList extends ArrayList<String> {
  private static int counter = 0;
  private int id = counter++;
  public CountedList() {
    System.out.println("CountedList #" + id);
  }
  public int getId() { return id; }
}

public class JUnitDemo {
  private CountedList list = new CountedList();
  // You can use the constructor instead of setUp():
  public JUnitDemo() {
    for(int i = 0; i < 3; i++)
      list.add("" + i);
  }
  // Thus, setUp() is optional, but is run right
  // before the test:
  protected void setUp() {
    System.out.println("Set up for " + list.getId());
  }
  // tearDown() is also optional, and is called after
  // each test. setUp() and tearDown() can be either
  // protected or public:
  public void tearDown() {
    System.out.println("Tearing down " + list.getId());
  }
  // All tests have method names beginning with "test":
  @Test
  public void testInsert() {
    System.out.println("Running testInsert()");
    org.junit.Assert.assertEquals(list.size(), 3);
    list.add(1, "Insert");
    org.junit.Assert.assertEquals(list.size(), 4);
    org.junit.Assert.assertEquals(list.get(1), "Insert");
  }
  @Test
  public void testReplace() {
    System.out.println("Running testReplace()");
    org.junit.Assert.assertEquals(list.size(), 3);
    list.set(1, "Replace");
    org.junit.Assert.assertEquals(list.size(), 3);
    org.junit.Assert.assertEquals(list.get(1), "Replace");
  }
  // A "helper" method to reduce code duplication. As long
  // as the name doesn't start with "test," it will not
  // be automatically executed by JUnit.
  private void compare(ArrayList lst, String[] strs) {
    Object[] array = lst.toArray();
    org.junit.Assert.assertTrue("Arrays not the same length",
      array.length == strs.length);
    for(int i = 0; i < array.length; i++)
      org.junit.Assert.assertEquals(strs[i], (String)array[i]);
  }
  @Test
  public void testOrder() {
    System.out.println("Running testOrder()");
    compare(list, new String[] { "0", "1", "2" });
  }
  @Test
  public void testRemove() {
    System.out.println("Running testRemove()");
    org.junit.Assert.assertEquals(list.size(), 3);
    list.remove(1);
    org.junit.Assert.assertEquals(list.size(), 2);
    compare(list, new String[] { "0", "2" });
  }
  @Test
  public void testAddAll() {
    System.out.println("Running testAddAll()");
    list.addAll(Arrays.asList(new String[] {
      "An", "African", "Swallow"}));
    org.junit.Assert.assertEquals(list.size(), 6);
    compare(list, new String[] { "0", "1", "2",
       "An", "African", "Swallow" });
  }
  public static void main(String[] args) {
    // Invoke JUnit on the class:
    org.junit.runner.JUnitCore.runClasses(JUnitDemo.class);
  }
} ///:~