|
| 1 | + |
| 2 | +from django.contrib.auth import get_user_model |
| 3 | +from rest_framework.test import APITestCase |
| 4 | + |
| 5 | + |
| 6 | +class TestBase(APITestCase): |
| 7 | + """ |
| 8 | + Test base class to setup a couple users. |
| 9 | + """ |
| 10 | + def setUp(self): |
| 11 | + """ |
| 12 | + Create those users |
| 13 | + """ |
| 14 | + super(TestBase, self).setUp() |
| 15 | + self.create_users() |
| 16 | + |
| 17 | + def create_user(self, username, email, password="pw", |
| 18 | + first_name='', last_name=''): |
| 19 | + """ |
| 20 | + Helper method to create a user |
| 21 | + """ |
| 22 | + User = get_user_model() |
| 23 | + |
| 24 | + user = User.objects.create_user( |
| 25 | + username, email, password=password |
| 26 | + ) |
| 27 | + if first_name or last_name: |
| 28 | + user.first_name = first_name |
| 29 | + user.last_name = last_name |
| 30 | + user.save() |
| 31 | + return user |
| 32 | + |
| 33 | + def create_users(self): |
| 34 | + """ |
| 35 | + Create a couple users |
| 36 | + """ |
| 37 | + self.john = self.create_user( |
| 38 | + 'trane', 'john@example.com', |
| 39 | + first_name='John', last_name="Coltrane") |
| 40 | + self.miles = self.create_user( |
| 41 | + 'miles', 'miles@example.com', |
| 42 | + first_name="Miles", last_name="Davis") |
| 43 | + |
0 commit comments