Add notifications tests