Add tests for flash messages