Using assertj assertions

testng assertions have been replaced with assertj
assertions.