Added tests for callbacks mocking