def test_average_rating(self):
meal = create(food_models.Meal)
review1_init = {'rating': 1, 'meal': meal}
review2_init = {'rating': 2, 'meal': meal}
create(models.Review, (review1_init, review2_init))
expected_results = '1.5 Stars'
self.assertEqual(models.Review.average_rating(meal), expected_results)
def setUp(self):
self.maxDiff = None
user = get_system_user()
self.recipe = create(Recipe, {'user': user, 'servings': 4})
self.recipe.ingredients.add(
create(Ingredient, {
'quantity': 2,
'weight': 5
}))
def _generate_reviews(self, review_count):
initial_list = list()
for _ in itertools.repeat(None, review_count):
initial_list.append({
'title': random.choice(self.REVIEW_TITLES),
'content': random.choice(self.REVIEW_CONTENTS),
'rating': random.randint(1, 5),
'user': random.choice(accounts_models.User.objects.all()),
'meal': random.choice(food_models.Meal.objects.all()),
})
create(common_models.Review, initial_list)
def test_create_with_multi_model_and_single_count_and_single_initial(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 0)
initial1 = {
'decimal_field': Decimal('10.00')
}
initial2 = {
'char_field': 'test value'
}
args = ((ModelOne, initial1), (ModelTwo, initial2))
models = create(*args)
self.assertIsInstance(models[0], ModelOne)
self.assertEqual(models[0].decimal_field, initial1['decimal_field'])
self.assertIsInstance(models[1], ModelTwo)
self.assertEqual(models[1].char_field, initial2['char_field'])
# Since ModelTwo has a FK and OneToOne to ModelOne we expect:
# 1 for the count in the factory models
# 1 for each OneToOne fields
models = ModelOne.objects.all()
self.assertEqual(len(models), 2)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 1)
def test_create_with_multi_model_and_multi_count(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 0)
count1 = 2
count2 = 3
args = ((ModelOne, count1), (ModelTwo, count2))
models = create(*args)
self.assertEqual(len(models), count1 + count2)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelOne)
self.assertIsInstance(models[2], ModelTwo)
self.assertIsInstance(models[3], ModelTwo)
self.assertIsInstance(models[4], ModelTwo)
# Since ModelTwo has a FK and OneToOne to ModelOne we expect:
# 1 for the count in the factory models (count1)
# 1 for each OneToOne fields (count2)
models = ModelOne.objects.all()
self.assertEqual(len(models), count1 + count2)
models = ModelTwo.objects.all()
self.assertEqual(len(models), count2)
def test_create_with_multi_model_and_multi_count_and_multi_initial(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 0)
initial1_1 = {'decimal_field': Decimal('10.00')}
initial1_2 = {'decimal_field': Decimal('8.00')}
initial2_1 = {'char_field': 'test value 1'}
initial2_2 = {'char_field': 'test value 2'}
args = ((ModelOne, [initial1_1,
initial1_2]), (ModelTwo, [initial2_1, initial2_2]))
models = create(*args)
self.assertEqual(len(models), 4)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelOne)
self.assertEqual(models[0].decimal_field, initial1_1['decimal_field'])
self.assertEqual(models[1].decimal_field, initial1_2['decimal_field'])
self.assertIsInstance(models[2], ModelTwo)
self.assertIsInstance(models[3], ModelTwo)
self.assertEqual(models[2].char_field, initial2_1['char_field'])
self.assertEqual(models[3].char_field, initial2_2['char_field'])
# Since ModelTwo has a FK and OneToOne to ModelOne we expect:
# 1 for the count in the factory models (count1)
# 1 for each OneToOne fields (count2)
models = ModelOne.objects.all()
self.assertEqual(len(models), 4)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 2)
def test_create_with_multi_model_and_multi_count(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 0)
count1 = 2
count2 = 3
args = ((ModelOne, count1), (ModelTwo, count2))
models = create(*args)
self.assertEqual(len(models), count1 + count2)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelOne)
self.assertIsInstance(models[2], ModelTwo)
self.assertIsInstance(models[3], ModelTwo)
self.assertIsInstance(models[4], ModelTwo)
# Since ModelTwo has a FK and OneToOne to ModelOne we expect:
# 1 for the count in the factory models (count1)
# 1 for each OneToOne fields (count2)
models = ModelOne.objects.all()
self.assertEqual(len(models), count1 + count2)
models = ModelTwo.objects.all()
self.assertEqual(len(models), count2)
def test_create_trivial(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
model = create(ModelOne)
self.assertIsInstance(model, ModelOne)
models = ModelOne.objects.all()
self.assertEqual(len(models), 1)
def test_create_trivial(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
model = create(ModelOne)
self.assertIsInstance(model, ModelOne)
models = ModelOne.objects.all()
self.assertEqual(len(models), 1)
def test_create_with_single_count(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
count = 1
args = (ModelOne, count)
model = create(*args)
self.assertIsInstance(model, ModelOne)
models = ModelOne.objects.all()
self.assertEqual(len(models), count)
def test_create_with_single_count(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
count = 1
args = (ModelOne, count)
model = create(*args)
self.assertIsInstance(model, ModelOne)
models = ModelOne.objects.all()
self.assertEqual(len(models), count)
def test_create_with_count_and_initial(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
initial = {'decimal_field': Decimal('10.00')}
args = (ModelOne, initial)
model = create(*args)
self.assertIsInstance(model, ModelOne)
self.assertEqual(model.decimal_field, initial['decimal_field'])
models = ModelOne.objects.all()
self.assertEqual(len(models), 1)
def _generate_meals(self, meal_count):
initial_list = list()
for _ in itertools.repeat(None, meal_count):
initial_list.append({
'name': random.choice(self.MEAL_NAMES),
})
meals = create(food_models.Meal, initial_list)
for meal in meals:
item_count = random.randint(
1, food_models.Item.objects.all().count())
for _ in range(0, item_count):
item = random.choice(food_models.Item.objects.all())
meal.items.add(item)
def test_create_with_multi_count(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
count = 2
args = (ModelOne, count)
models = create(*args)
self.assertEqual(len(models), count)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelOne)
self.assertNotEqual(models[0], models[1])
models = ModelOne.objects.all()
self.assertEqual(len(models), count)
def test_create_with_multi_count(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
count = 2
args = (ModelOne, count)
models = create(*args)
self.assertEqual(len(models), count)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelOne)
self.assertNotEqual(models[0], models[1])
models = ModelOne.objects.all()
self.assertEqual(len(models), count)
def test_create_with_count_and_initial(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
initial = {
'decimal_field': Decimal('10.00')
}
args = (ModelOne, initial)
model = create(*args)
self.assertIsInstance(model, ModelOne)
self.assertEqual(model.decimal_field, initial['decimal_field'])
models = ModelOne.objects.all()
self.assertEqual(len(models), 1)
def setUp(self):
super().setUp()
self.user = get_user_model().objects.create(username='******')
self.form = create(Recipe, {'user': self.user, 'title': 'test_title'})
self.client = Client()
# group and permission setup
self.group = Group(name='recipe admin')
self.group.save()
add_delete_permission = Permission.objects.get(name='Can delete recipe')
add_edit_permission = Permission.objects.get(name='Can change recipe')
self.admin_user = get_user_model().objects.create(username='******')
self.admin_user.groups.add(self.group)
self.group.permissions.add(add_delete_permission, add_edit_permission)
def test_create_with_multi_model_trivial(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 0)
models = create((ModelOne, ), (ModelTwo, ))
self.assertEqual(len(models), 2)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelTwo)
# Since ModelTwo has a FK and OneToOne to ModelOne we expect 2
models = ModelOne.objects.all()
self.assertEqual(len(models), 2)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 1)
def test_create_with_multi_model_trivial(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 0)
models = create((ModelOne, ), (ModelTwo, ))
self.assertEqual(len(models), 2)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelTwo)
# Since ModelTwo has a FK and OneToOne to ModelOne we expect 2
models = ModelOne.objects.all()
self.assertEqual(len(models), 2)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 1)
def test_create_with_multi_count_and_multi_initial(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
initial1 = {'decimal_field': Decimal('10.00')}
initial2 = {'decimal_field': Decimal('8.00')}
args = (ModelOne, [initial1, initial2])
models = create(*args)
self.assertEqual(len(models), 2)
self.assertIsInstance(models[0], ModelOne)
self.assertEqual(models[0].decimal_field, initial1['decimal_field'])
self.assertIsInstance(models[1], ModelOne)
self.assertEqual(models[1].decimal_field, initial2['decimal_field'])
models = ModelOne.objects.all()
self.assertEqual(len(models), 2)
def test_create_with_multi_model_and_single_count(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 0)
count1 = 1
count2 = 1
args = ((ModelOne, count1), (ModelTwo, count2))
models = create(*args)
self.assertEqual(len(models), 2)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelTwo)
# Since ModelTwo has a FK and OneToOne to ModelOne we expect 2
models = ModelOne.objects.all()
self.assertEqual(len(models), 2)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 1)
def test_create_with_multi_model_and_single_count(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 0)
count1 = 1
count2 = 1
args = ((ModelOne, count1), (ModelTwo, count2))
models = create(*args)
self.assertEqual(len(models), 2)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelTwo)
# Since ModelTwo has a FK and OneToOne to ModelOne we expect 2
models = ModelOne.objects.all()
self.assertEqual(len(models), 2)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 1)
def test_create_with_multi_model_and_multi_count_and_single_initial(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 0)
count1 = 2
initial1 = {
'decimal_field': Decimal('10.00')
}
initial_list1 = list()
for _ in itertools.repeat(None, count1):
initial_list1.append(initial1)
count2 = 3
initial2 = {
'char_field': 'test value'
}
initial_list2 = list()
for _ in itertools.repeat(None, count2):
initial_list2.append(initial2)
args = ((ModelOne, initial_list1), (ModelTwo, initial_list2))
models = create(*args)
self.assertEqual(len(models), count1 + count2)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelOne)
self.assertEqual(models[0].decimal_field, initial1['decimal_field'])
self.assertEqual(models[1].decimal_field, initial1['decimal_field'])
self.assertIsInstance(models[2], ModelTwo)
self.assertIsInstance(models[3], ModelTwo)
self.assertIsInstance(models[4], ModelTwo)
self.assertEqual(models[2].char_field, initial2['char_field'])
self.assertEqual(models[3].char_field, initial2['char_field'])
self.assertEqual(models[4].char_field, initial2['char_field'])
# Since ModelTwo has a FK and OneToOne to ModelOne we expect:
# 1 for the count in the factory models (count1)
# 1 for each OneToOne fields (count2)
models = ModelOne.objects.all()
self.assertEqual(len(models), count1 + count2)
models = ModelTwo.objects.all()
self.assertEqual(len(models), count2)
def test_create_with_multi_count_and_single_initial(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
count = 2
initial1 = {'decimal_field': Decimal('10.00')}
initial_list = list()
for _ in itertools.repeat(None, count):
initial_list.append(initial1)
args = (ModelOne, initial_list)
models = create(*args)
self.assertEqual(len(models), count)
self.assertIsInstance(models[0], ModelOne)
self.assertEqual(models[0].decimal_field, initial1['decimal_field'])
self.assertIsInstance(models[1], ModelOne)
self.assertEqual(models[1].decimal_field, initial1['decimal_field'])
models = ModelOne.objects.all()
self.assertEqual(len(models), count)
def test_create_with_multi_count_and_multi_initial(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
initial1 = {
'decimal_field': Decimal('10.00')
}
initial2 = {
'decimal_field': Decimal('8.00')
}
args = (ModelOne, [initial1, initial2])
models = create(*args)
self.assertEqual(len(models), 2)
self.assertIsInstance(models[0], ModelOne)
self.assertEqual(models[0].decimal_field, initial1['decimal_field'])
self.assertIsInstance(models[1], ModelOne)
self.assertEqual(models[1].decimal_field, initial2['decimal_field'])
models = ModelOne.objects.all()
self.assertEqual(len(models), 2)
def test_create_with_multi_count_and_single_initial(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
count = 2
initial1 = {
'decimal_field': Decimal('10.00')
}
initial_list = list()
for _ in itertools.repeat(None, count):
initial_list.append(initial1)
args = (ModelOne, initial_list)
models = create(*args)
self.assertEqual(len(models), count)
self.assertIsInstance(models[0], ModelOne)
self.assertEqual(models[0].decimal_field, initial1['decimal_field'])
self.assertIsInstance(models[1], ModelOne)
self.assertEqual(models[1].decimal_field, initial1['decimal_field'])
models = ModelOne.objects.all()
self.assertEqual(len(models), count)
def test_create_with_multi_model_and_multi_count_and_multi_initial(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 0)
initial1_1 = {
'decimal_field': Decimal('10.00')
}
initial1_2 = {
'decimal_field': Decimal('8.00')
}
initial2_1 = {
'char_field': 'test value 1'
}
initial2_2 = {
'char_field': 'test value 2'
}
args = ((ModelOne, [initial1_1, initial1_2]),
(ModelTwo, [initial2_1, initial2_2]))
models = create(*args)
self.assertEqual(len(models), 4)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelOne)
self.assertEqual(models[0].decimal_field, initial1_1['decimal_field'])
self.assertEqual(models[1].decimal_field, initial1_2['decimal_field'])
self.assertIsInstance(models[2], ModelTwo)
self.assertIsInstance(models[3], ModelTwo)
self.assertEqual(models[2].char_field, initial2_1['char_field'])
self.assertEqual(models[3].char_field, initial2_2['char_field'])
# Since ModelTwo has a FK and OneToOne to ModelOne we expect:
# 1 for the count in the factory models (count1)
# 1 for each OneToOne fields (count2)
models = ModelOne.objects.all()
self.assertEqual(len(models), 4)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 2)
def test_create_with_multi_model_and_multi_count_and_single_initial(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 0)
count1 = 2
initial1 = {'decimal_field': Decimal('10.00')}
initial_list1 = list()
for _ in itertools.repeat(None, count1):
initial_list1.append(initial1)
count2 = 3
initial2 = {'char_field': 'test value'}
initial_list2 = list()
for _ in itertools.repeat(None, count2):
initial_list2.append(initial2)
args = ((ModelOne, initial_list1), (ModelTwo, initial_list2))
models = create(*args)
self.assertEqual(len(models), count1 + count2)
self.assertIsInstance(models[0], ModelOne)
self.assertIsInstance(models[1], ModelOne)
self.assertEqual(models[0].decimal_field, initial1['decimal_field'])
self.assertEqual(models[1].decimal_field, initial1['decimal_field'])
self.assertIsInstance(models[2], ModelTwo)
self.assertIsInstance(models[3], ModelTwo)
self.assertIsInstance(models[4], ModelTwo)
self.assertEqual(models[2].char_field, initial2['char_field'])
self.assertEqual(models[3].char_field, initial2['char_field'])
self.assertEqual(models[4].char_field, initial2['char_field'])
# Since ModelTwo has a FK and OneToOne to ModelOne we expect:
# 1 for the count in the factory models (count1)
# 1 for each OneToOne fields (count2)
models = ModelOne.objects.all()
self.assertEqual(len(models), count1 + count2)
models = ModelTwo.objects.all()
self.assertEqual(len(models), count2)
def test_create_with_multi_model_and_single_count_and_single_initial(self):
models = ModelOne.objects.all()
self.assertEqual(len(models), 0)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 0)
initial1 = {'decimal_field': Decimal('10.00')}
initial2 = {'char_field': 'test value'}
args = ((ModelOne, initial1), (ModelTwo, initial2))
models = create(*args)
self.assertIsInstance(models[0], ModelOne)
self.assertEqual(models[0].decimal_field, initial1['decimal_field'])
self.assertIsInstance(models[1], ModelTwo)
self.assertEqual(models[1].char_field, initial2['char_field'])
# Since ModelTwo has a FK and OneToOne to ModelOne we expect:
# 1 for the count in the factory models
# 1 for each OneToOne fields
models = ModelOne.objects.all()
self.assertEqual(len(models), 2)
models = ModelTwo.objects.all()
self.assertEqual(len(models), 1)
def setUp(self):
self.user = get_user_model().objects.create(username='******')
self.comment = create(Comment)
self.recipe = create(Recipe)
def setUp(self):
self.user = get_user_model().objects.create(username='******')
self.recipe = create(Recipe, {'title': 'testing'})
self.recipe_list_url = reverse_lazy('api:v1:recipe-list')
self.recipe_detail_url = reverse_lazy('api:v1:recipe-detail',
kwargs={'pk': self.recipe.pk})
def _generate_items(self, item_count):
for item_name in self.ITEM_NAMES:
if food_models.Item.objects.all().count() >= item_count:
break
create(food_models.Item, {'name': item_name})