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
from django.contrib.auth.models import User
from django.core.urlresolvers import reverse
from django.db import models
from django.test import TestCase
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.models import 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):
@ -26,6 +25,8 @@ class TestSortableModel(Sortable):
class SortableTestCase(TestCase):
fixtures = ['initial_data.json']
def setUp(self):
self.client = Client()
self.factory = RequestFactory()
@ -36,6 +37,10 @@ class SortableTestCase(TestCase):
self.user.is_superuser = True
self.user.save()
people = Person.objects.all()
self.first_person = people[0]
self.second_person = people[1]
def create_category(self, title='Category 1'):
category = Category.objects.create(title=title)
return category
@ -152,3 +157,15 @@ class SortableTestCase(TestCase):
'Third category returned should have order == 3')
self.assertEqual(cat3.pk, 1,
'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))