Add tests for associations by belongs_to