Added basic test of get_next/previous.

master
Brandon Taylor 2014-09-06 14:53:11 -04:00
parent 25ef724e15
commit 8df54f43e1
1 changed files with 19 additions and 2 deletions

View File

@ -2,7 +2,6 @@ import httplib
import json import json
from django.contrib.auth.models import User from django.contrib.auth.models import User
from django.core.urlresolvers import reverse
from django.db import models from django.db import models
from django.test import TestCase from django.test import TestCase
from django.test.client import Client, RequestFactory from django.test.client import Client, RequestFactory
@ -10,7 +9,7 @@ from django.test.client import Client, RequestFactory
from adminsortable.fields import SortableForeignKey from adminsortable.fields import SortableForeignKey
from adminsortable.models import Sortable from adminsortable.models import Sortable
from adminsortable.utils import get_is_sortable from adminsortable.utils import get_is_sortable
from app.models import Category, Credit, Note from app.models import Category, Credit, Note, Person
class BadSortableModel(models.Model): class BadSortableModel(models.Model):
@ -26,6 +25,8 @@ class TestSortableModel(Sortable):
class SortableTestCase(TestCase): class SortableTestCase(TestCase):
fixtures = ['initial_data.json']
def setUp(self): def setUp(self):
self.client = Client() self.client = Client()
self.factory = RequestFactory() self.factory = RequestFactory()
@ -36,6 +37,10 @@ class SortableTestCase(TestCase):
self.user.is_superuser = True self.user.is_superuser = True
self.user.save() self.user.save()
people = Person.objects.all()
self.first_person = people[0]
self.second_person = people[1]
def create_category(self, title='Category 1'): def create_category(self, title='Category 1'):
category = Category.objects.create(title=title) category = Category.objects.create(title=title)
return category return category
@ -152,3 +157,15 @@ class SortableTestCase(TestCase):
'Third category returned should have order == 3') 'Third category returned should have order == 3')
self.assertEqual(cat3.pk, 1, self.assertEqual(cat3.pk, 1,
'Category ID 1 should have been third in queryset') 'Category ID 1 should have been third in queryset')
def test_get_next(self):
result = self.first_person.get_next()
self.assertEqual(self.second_person, result, 'Next person should '
'be "{}"'.format(self.second_person))
def test_get_previous(self):
result = self.second_person.get_previous()
self.assertEqual(self.first_person, result, 'Previous person should '
'be "{}"'.format(self.first_person))