Add tests to check proxy feature