diff --git a/src/apps/dashboard/tests/test_models.py b/src/apps/dashboard/tests/test_models.py new file mode 100644 index 0000000..89347a6 --- /dev/null +++ b/src/apps/dashboard/tests/test_models.py @@ -0,0 +1,31 @@ +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) + \ No newline at end of file diff --git a/src/apps/dashboard/tests/test_all.py b/src/apps/dashboard/tests/test_views.py similarity index 77% rename from src/apps/dashboard/tests/test_all.py rename to src/apps/dashboard/tests/test_views.py index aa3ea94..111f574 100644 --- a/src/apps/dashboard/tests/test_all.py +++ b/src/apps/dashboard/tests/test_views.py @@ -1,5 +1,5 @@ from django.test import TestCase -from .models import Item, List +from ..models import Item, List import lxml.html class HomePageTest(TestCase): @@ -15,34 +15,6 @@ class HomePageTest(TestCase): inputs = form.cssselect('input') self.assertIn('item_text', [input.get('name') for input in inputs]) -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) - class DashViewTest(TestCase): def test_uses_list_template(self): mylist = List.objects.create()