Add tests for customizing AR relation