Migrate TestNG to JUnit