feat(refactoring): refactor tests