chore(tests): refactor