It might make sense for simplicity to create a method that will encapsulate the two behaviors which are used in this test case. You can differentiate based on extra arguments. It is mainly for removing the code clutter.