Add tests for flash redirect