__init__.py

fromdjango.contrib.authimportget_user_model
fromrest_framework.testimportAPITestCase


classTestBase(APITestCase):
"""
 Test base class to setup a couple users.
 """
defsetUp(self):
"""
 Create those users
 """
super(TestBase, self).setUp()
self.create_users()

defcreate_user(self, username, email, password="pw",
first_name='', last_name=''):
"""
 Helper method to create a user
 """
User=get_user_model()
user=User.objects.create_user(
username, email, password=password
 )
iffirst_nameorlast_name:
user.first_name=first_name
user.last_name=last_name
user.save()
returnuser

defcreate_users(self):
"""
 Create a couple users
 """
self.john=self.create_user(
'trane', 'john@example.com',
first_name='John', last_name="Coltrane")
self.miles=self.create_user(
'miles', 'miles@example.com',
first_name="Miles", last_name="Davis")