예제 #1
0
    def test_save_and_retrieve_cities(self):
        list_ = List()
        list_.save()

        first_city = City()
        first_city.name = 'First city'
        first_city.list = list_
        first_city.save()

        second_city = City()
        second_city.name = 'Second city'
        second_city.list = list_
        second_city.save()

        saved_list = List.objects.first()
        self.assertEqual(saved_list, list_)

        saved_cities = City.objects.all()
        self.assertEqual(saved_cities.count(), 2)

        first_saved_city = saved_cities[0]
        second_saved_city = saved_cities[1]
        self.assertEqual(first_saved_city.name, 'First city')
        self.assertEqual(first_saved_city.list, list_)
        self.assertEqual(second_saved_city.name, 'Second city')
        self.assertEqual(second_saved_city.list, list_)
 def test_cannot_save_empty_city(self):
     list_ = List.objects.create()
     city = City()
     city.name = ''
     city.list = list_
     with self.assertRaises(ValidationError):
         city.save()
         city.full_clean()