[TASK] Update unit tests to match search packages refactoring