Refactor common code across tests