Add tests for the refactoring models