31 lines
1.0 KiB
Python
31 lines
1.0 KiB
Python
|
from django.test import TestCase
|
||
|
|
from ..models import Item, List
|
||
|
|
|
||
|
|
class ListAndItemModelsTest(TestCase):
|
||
|
|
def test_saving_and_retrieving_items(self):
|
||
|
|
mylist = List()
|
||
|
|
mylist.save()
|
||
|
|
|
||
|
|
first_item = Item()
|
||
|
|
first_item.text = "The first (ever) list item"
|
||
|
|
first_item.list = mylist
|
||
|
|
first_item.save()
|
||
|
|
|
||
|
|
second_item = Item()
|
||
|
|
second_item.text = "A sequel somehow better than the first"
|
||
|
|
second_item.list = mylist
|
||
|
|
second_item.save()
|
||
|
|
|
||
|
|
saved_list = List.objects.get()
|
||
|
|
self.assertEqual(saved_list, mylist)
|
||
|
|
|
||
|
|
saved_items = Item.objects.all()
|
||
|
|
self.assertEqual(saved_items.count(), 2)
|
||
|
|
|
||
|
|
first_saved_item = saved_items[0]
|
||
|
|
second_saved_item = saved_items[1]
|
||
|
|
self.assertEqual(first_saved_item.text, "The first (ever) list item")
|
||
|
|
self.assertEqual(first_saved_item.list, mylist)
|
||
|
|
self.assertEqual(second_saved_item.text, "A sequel somehow better than the first")
|
||
|
|
self.assertEqual(second_saved_item.list, mylist)
|
||
|
|
|