✅ Add tests for survey return code

✅ Add tests for survey return code #242