Replace using testNG with jUnit