from django.test import TestCase from django.contrib.auth.models import User from recipes.models import Recipe from ingredients.models import Ingredient from user.models import UserProfile # Create your tests here. class RecipeTestCase(TestCase): def setUp(self): user = User.objects.create_user('testuser') <<<<<<< HEAD profile = UserProfile.get_or_create_profile(user) ======= profile = UserProfile.create_or_get_profile(user) >>>>>>> dev Ingredient.objects.bulk_create([Ingredient(name='Vodka', category='alcohol'), Ingredient(name='Rum', category='alcohol'), Ingredient(name='Gin', category='alcohol'), Ingredient(name='Orange Juice', category='juice'), Ingredient(name='Pineapple Juice', category='juice'), Ingredient(name='White Rum', category='alcohol'), Ingredient(name='Coconut Cream', category='milk')]) self.vodka_id = Ingredient.objects.get(name='Vodka').id self.oj_id = Ingredient.objects.get(name='Orange Juice').id self.gin_id = Ingredient.objects.get(name='Gin').id self.rum_id = Ingredient.objects.get(name='Rum').id self.pineapple_juice_id = Ingredient.objects.get(name='Pineapple Juice').id self.white_rum_id = Ingredient.objects.get(name='White Rum').id
RecipeIngredients.objects.bulk_create(drink_ingredients) create_drink('screwdriver', 120, 40) create_drink('gin_and_vodka', 10, 40) add_ingredients_to_drink(self.screwdriver, [(Ingredient.objects.get(name='Vodka'), 1), (Ingredient.objects.get(name='Orange Juice'), 2)]) add_ingredients_to_drink(self.gin_and_vodka, [(Ingredient.objects.get(name='Vodka'), 2), (Ingredient.objects.get(name='Gin'), 2)]) def test_create_or_get_profile(self): <<<<<<< HEAD profile = UserProfile.get_or_create_profile(self.user) ======= profile = UserProfile.create_or_get_profile(self.user) >>>>>>> dev self.assertEqual(self.profile, profile) self.assertEqual('testuser', profile.user.username) self.assertEqual('*****@*****.**', profile.user.email) self.assertNotEqual('password', profile.user.password) def test_set_favorites(self): # Test set 1 favorite <<<<<<< HEAD # noinspection PyUnresolvedReferences ======= >>>>>>> dev def set_one_favorite(): favorites = self.profile.set_favorites(self.screwdriver.id) self.assertIn(self.screwdriver, favorites)
def get(self,request): user = self.request.user if (user.is_authenticated()): profile = UserProfile.create_or_get_profile(user)