Refactor test to make it more given-when-then like