REFACTOR: add mock objects to the test