def test_get_multiple_objects(self):
predicate = OrmP(int_value__in=[1, 2])
with self.assertRaises(MultipleObjectsReturned):
TestObj.objects.get(predicate)
with self.assertRaises(MultipleObjectsReturned):
predicate.get(self.objects)
def test_get_return_value(self):
predicate = OrmP(int_value=1)
self.assertEqual(TestObj.objects.get(predicate), self.obj1)
self.assertEqual(predicate.get(self.objects), self.obj1)
def test_get_no_objects(self):
predicate = OrmP(int_value=3)
with self.assertRaises(ObjectDoesNotExist):
TestObj.objects.get(predicate)
with self.assertRaises(ObjectDoesNotExist):
predicate.get(self.objects)
def test_get_multiple_objects(self):
predicate = OrmP(int_value__in=[1, 2])
with self.assertRaises(MultipleObjectsReturned):
TestObj.objects.get(predicate)
with self.assertRaises(MultipleObjectsReturned):
predicate.get(self.objects)
def test_get_return_value(self):
predicate = OrmP(int_value=1)
self.assertEqual(TestObj.objects.get(predicate), self.obj1)
self.assertEqual(predicate.get(self.objects), self.obj1)
def test_get_no_objects(self):
predicate = OrmP(int_value=3)
with self.assertRaises(ObjectDoesNotExist):
TestObj.objects.get(predicate)
with self.assertRaises(ObjectDoesNotExist):
predicate.get(self.objects)
