Migrate all tests to TestNG