unit test refactoring

1 file changed