Add tests for success and failure responses in AIMGatewayTest