Adding mocking to the notification tests.