def test_get_current_user_url(self): url = Customer.get_url_by_user(self.sudhir_gmail) self.assertEqual(url, 'sudhirurl') url = Customer.get_url_by_user(users.User('*****@*****.**')) self.assertFalse(url) url = Customer.get_url_by_user(None) self.assertFalse(url)
def get_current_user_url(user): return Customer.get_url_by_user(user)