def test_get_icon_method_gets_default_icon(self): user_link = UserLinkFactory(url='http://noiconurl.com') icon = user_link.get_icon() self.assertEqual(icon, 'fa-globe')
def test_get_icon_method_gets_correct_icon(self): user_link = UserLinkFactory(url='http://github.com/nlh-kabu') icon = user_link.get_icon() self.assertEqual(icon, 'fa-github')