Add test case for custom attributes