[TASK] refactoring Unit Test