Add tests for notification service

This adds tests to the notification service