""" Test for models. """ from decimal import Decimal from django.test import TestCase from django.contrib.auth import get_user_model from core import models def create_user(email='user@example.com', password='testpass123'): """Create and return a new user.""" return get_user_model().objects.create_user(email, password) class ModelTests(TestCase): """Test models.""" def test_create_user_with_email_sucessfull(self): """Test creating a user with an email is sucessfull.""" email = 'test@example.com' password = 'testpass123' user = create_user(email, password) self.assertEqual(user.email, email) self.assertTrue(user.check_password(password)) def test_new_user_email_normalized(self): """Test email is normalized for new users.""" sample_emails = [ ['test1@EXAMPLE.com', 'test1@example.com'], ['test2@Example.com', 'test2@example.com'], ['TEST3@EXAMPLE.COM', 'TEST3@example.com'], ['test4@example.COM', 'test4@example.com'], ] for email, expected in sample_emails: user = create_user(email=email) self.assertEqual(user.email, expected) def test_new_user_withouth_email_raises_error(self): """Test that creating a user withouth an email raises a ValueError.""" with self.assertRaises(ValueError): create_user('', 'test123') def test_create_superuser(self): """Test creating a superuser.""" user = get_user_model().objects.create_superuser( 'test@example.com', 'test123', ) self.assertTrue(user.is_superuser) self.assertTrue(user.is_staff) def test_create_recipe(self): """Test creating a recipe is successful.""" user = create_user() recipe = models.Recipe.objects.create( user=user, title='Nombre receta ejemplo', time_minutes=5, price=Decimal('5.50'), description='DescripciĆ³n de la receta de ejemplo' ) self.assertEqual(str(recipe), recipe.title) def test_create_tag(self): """Test creating a tag is sucessfull.""" user = create_user() tag = models.Tag.objects.create(user=user, name='Tag1') self.assertEqual(str(tag), tag.name)