From 5c1248afd2d8c00e11ae418e6d9cbdf149f35332 Mon Sep 17 00:00:00 2001 From: Ali Ghaznavi Date: Thu, 11 Jul 2024 18:29:42 -0400 Subject: [PATCH] Fixed linting issues --- app/core/models.py | 5 +++-- app/core/tests/test_models.py | 9 +++++---- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/app/core/models.py b/app/core/models.py index b61f8f7..c607005 100755 --- a/app/core/models.py +++ b/app/core/models.py @@ -8,11 +8,11 @@ class UserManager(BaseUserManager): - def create_user(self, email, password = None, **extrafields): + def create_user(self, email, password=None, **extrafields): if not email: raise ValueError('Email address is required to create a new User.') - user = self.model(email = self.normalize_email(email), **extrafields) + user = self.model(email=self.normalize_email(email), **extrafields) user.set_password(password) user.save(using=self._db) @@ -27,6 +27,7 @@ def create_superuser(self, email, password): return superuser + class User(AbstractBaseUser, PermissionsMixin): email = models.EmailField(max_length=255, unique=True) name = models.CharField(max_length=255) diff --git a/app/core/tests/test_models.py b/app/core/tests/test_models.py index 9cd4c03..827a35d 100644 --- a/app/core/tests/test_models.py +++ b/app/core/tests/test_models.py @@ -1,6 +1,7 @@ from django.test import TestCase from django.contrib.auth import get_user_model + class ModelTests(TestCase): def test_create_user_with_email_successful(self): @@ -8,8 +9,8 @@ def test_create_user_with_email_successful(self): password = 'testpass123' user = get_user_model().objects.create_user( - email = email, - password = password + email=email, + password=password ) self.assertEqual(user.email, email) @@ -36,8 +37,8 @@ def test_create_superuser(self): password = 'testpass123' user = get_user_model().objects.create_superuser( - email = email, - password = password + email=email, + password=password ) self.assertTrue(user.is_superuser)