Add unit tests for notifications