def test_saving_and_retrieving_items(self): list_ = List() list_.save() first_item = Item() first_item.text = 'The first (ever) list item' first_item.saving_list = list_ first_item.save() second_item = Item() second_item.text = 'The second list item' second_item.saving_list = list_ second_item.save() saved_list = List.objects.first() self.assertEqual(saved_list, list_) saved_items = Item.objects.all() self.assertEqual(saved_items.count(), 2) first = saved_items[0] second = saved_items[1] self.assertEqual(first.text, first_item.text) self.assertEqual(first.saving_list, list_) self.assertEqual(second.text, second_item.text) self.assertEqual(second.saving_list, list_)
def test_item_is_related_to_list(self): list_ = List.objects.create() item = Item() item.saving_list = list_ item.save() self.assertIn(item, list_.item_set.all())