Move to JUnit to get rid of external TestNG dependency.