Refactoring a big test into smaller tests