[TASK] Adds unit test