Add test directory and JUnit test