def get_user_profile(user):
"""
This function should be used to
'get or create' a user's profile
"""
try:
profile = user.get_profile()
except ObjectDoesNotExist:
profile = UserProfile()
profile.user = request.user
profile.save()
return profile
def get_user_profile(user):
"""
This function should be used to
'get or create' a user's profile
"""
try:
profile = user.get_profile()
except ObjectDoesNotExist:
profile = UserProfile()
profile.user = request.user
profile.save()
return profile
def setUp(self):
"""
Required Setup for testing:
-Create a user with permission to add favourites
-Create a user with default premissions
-Create a show with dummy data
-Create blank user profile so favourites can be added
"""
user = User.objects.create_user('john', '*****@*****.**',
'password')
perm = Permission.objects.get(codename="can_have_favs")
user.save()
user.user_permissions.add(perm)
profile = UserProfile()
profile.user = user
profile.save()
user = User.objects.create_user('jane', '*****@*****.**',
'password')
user.save()
profile = UserProfile()
profile.user = user
profile.save()
new_show = Show()
new_show.name = "Some test show"
new_show.id = 1234
new_show.save()
new_entry = GuideEntry()
#Compile a date/time stamp for the show
year = datetime.now().strftime("%Y")
month = datetime.now().strftime("%m")
day = datetime.now().strftime("%d")
date_object = datetime.strptime(
year + '-' + month + '-' + day + ' 5:00 pm', '%Y-%m-%d %I:%M %p')
new_entry.start = date_object
new_entry.network = "SPIKE TV"
new_entry.show = new_show
new_entry.save()
def setUp(self):
"""
Required Setup for testing:
-Create a user with permission to add favourites
-Create a user with default premissions
-Create a show with dummy data
-Create blank user profile so favourites can be added
"""
user = User.objects.create_user('john', '*****@*****.**',
'password')
perm = Permission.objects.get(codename="can_have_favs")
user.save()
user.user_permissions.add(perm)
profile = UserProfile()
profile.user = user
profile.save()
user = User.objects.create_user('jane', '*****@*****.**',
'password')
user.save()
profile = UserProfile()
profile.user = user
profile.save()
new_show = Show()
new_show.name = "Some test show"
new_show.id = 1234
new_show.save()
new_entry = GuideEntry()
#Compile a date/time stamp for the show
year = datetime.now().strftime("%Y")
month = datetime.now().strftime("%m")
day = datetime.now().strftime("%d")
date_object = datetime.strptime(year + '-' + month
+ '-' + day + ' 5:00 pm', '%Y-%m-%d %I:%M %p')
new_entry.start = date_object
new_entry.network = "SPIKE TV"
new_entry.show = new_show
new_entry.save()
