Add more test for attribute mapping