Add tests for flash_message