Using assertj assertions

testng assertions have been replaced with assertj
assertions.
Use assertj extraction feature to shorten test