def test_aggregate_join(self):
counter = 1
counters = []
for _ in range(0, DBTestCase.NUMBER_OF_RECORDS):
test_related_model = TestModelRelation(field_1=99)
test_related_model.save()
test_model = TestModel(field_10=counter, field_19=test_related_model)
test_model.save()
counters.append(counter)
counter += 1
count_result_1 = TestModel.objects.filter(field_19__field_1=99).aggregate(Count('field_10'))
count_result_2 = TestModel.objects.filter(field_19__field_1=0).aggregate(Count('field_10'))
avg_result_1 = TestModel.objects.filter(field_19__field_1=99).aggregate(Avg('field_10'))
avg_result_2 = TestModel.objects.filter(field_19__field_1=0).aggregate(Avg('field_10'))
min_result_1 = TestModel.objects.filter(field_19__field_1=99).aggregate(Min('field_10'))
min_result_2 = TestModel.objects.filter(field_19__field_1=0).aggregate(Min('field_10'))
max_result_1 = TestModel.objects.filter(field_19__field_1=99).aggregate(Max('field_10'))
max_result_2 = TestModel.objects.filter(field_19__field_1=0).aggregate(Max('field_10'))
sum_result_1 = TestModel.objects.filter(field_19__field_1=99).aggregate(Sum('field_10'))
sum_result_2 = TestModel.objects.filter(field_19__field_1=0).aggregate(Sum('field_10'))
self.assertEqual(count_result_1, {'field_10__count': DBTestCase.NUMBER_OF_RECORDS})
self.assertEqual(count_result_2, {'field_10__count': 0})
self.assertEqual(avg_result_1, {'field_10__avg': float(sum(counters))/len(counters)})
self.assertEqual(avg_result_2, {'field_10__avg': None})
self.assertEqual(min_result_1, {'field_10__min': 1})
self.assertEqual(min_result_2, {'field_10__min': None})
self.assertEqual(max_result_1, {'field_10__max': DBTestCase.NUMBER_OF_RECORDS})
self.assertEqual(max_result_2, {'field_10__max': None})
self.assertEqual(sum_result_1, {'field_10__sum': sum(counters)})
self.assertEqual(sum_result_2, {'field_10__sum': None})
def test_inserting_relations(self):
test_relation = TestModelRelation()
test_relation.save()
for _ in range(0, DBTestCase.NUMBER_OF_RELATIONS):
test_model = TestModel()
test_model.save()
test_relation.test_models.add(test_model)
self.assertEqual(test_relation.test_models.all()[0].field_4, 'abc')
self.assertEqual(test_relation.test_models.all().count(), DBTestCase.NUMBER_OF_RELATIONS)
def test_join_lookup(self):
for _ in range(0, DBTestCase.NUMBER_OF_RECORDS):
test_model = TestModel()
test_model.save()
test_relation = TestModelRelation()
test_relation.save()
test_relation.test_models.add(test_model)
self.assertEqual(TestModel.objects.filter(field_19__field_1=1234).count(), DBTestCase.NUMBER_OF_RECORDS)
self.assertEqual(TestModel.objects.filter(field_19__field_1=0).count(), 0)
def test_inserting_relations(self):
test_relation = TestModelRelation()
test_relation.save()
for _ in range(0, DBTestCase.NUMBER_OF_RELATIONS):
test_model = TestModel()
test_model.save()
test_relation.test_models.add(test_model)
self.assertEqual(test_relation.test_models.all()[0].field_4, 'abc')
self.assertEqual(test_relation.test_models.all().count(),
DBTestCase.NUMBER_OF_RELATIONS)
def test_limit_related_lookup(self):
for _ in range(0, DBTestCase.NUMBER_OF_RECORDS):
test_model = TestModel()
test_model.save()
test_relation = TestModelRelation()
test_relation.save()
test_relation.test_models.add(test_model)
test_models = TestModel.objects.select_related('field_19').all()[2:4]
self.assertEqual(len(test_models), 2)
def test_limit_related_lookup(self):
for _ in range(0, DBTestCase.NUMBER_OF_RECORDS):
test_model = TestModel()
test_model.save()
test_relation = TestModelRelation()
test_relation.save()
test_relation.test_models.add(test_model)
test_models = TestModel.objects.select_related('field_19').all()[2:4]
self.assertEqual(len(test_models), 2)
def test_related_lookup(self):
test_model = None
test_relation = TestModelRelation()
test_relation.save()
for _ in range(0, DBTestCase.NUMBER_OF_RELATIONS):
test_model = TestModel()
test_model.save()
test_relation.test_models.add(test_model)
test_model = TestModel.objects.select_related('field_19').get(id=test_model.id)
self.assertEqual(test_model.field_19.field_1, 1234)
def test_aggregate_join(self):
counter = 1
counters = []
for _ in range(0, DBTestCase.NUMBER_OF_RECORDS):
test_related_model = TestModelRelation(field_1=99)
test_related_model.save()
test_model = TestModel(field_10=counter,
field_19=test_related_model)
test_model.save()
counters.append(counter)
counter += 1
count_result_1 = TestModel.objects.filter(
field_19__field_1=99).aggregate(Count('field_10'))
count_result_2 = TestModel.objects.filter(
field_19__field_1=0).aggregate(Count('field_10'))
avg_result_1 = TestModel.objects.filter(
field_19__field_1=99).aggregate(Avg('field_10'))
avg_result_2 = TestModel.objects.filter(field_19__field_1=0).aggregate(
Avg('field_10'))
min_result_1 = TestModel.objects.filter(
field_19__field_1=99).aggregate(Min('field_10'))
min_result_2 = TestModel.objects.filter(field_19__field_1=0).aggregate(
Min('field_10'))
max_result_1 = TestModel.objects.filter(
field_19__field_1=99).aggregate(Max('field_10'))
max_result_2 = TestModel.objects.filter(field_19__field_1=0).aggregate(
Max('field_10'))
sum_result_1 = TestModel.objects.filter(
field_19__field_1=99).aggregate(Sum('field_10'))
sum_result_2 = TestModel.objects.filter(field_19__field_1=0).aggregate(
Sum('field_10'))
self.assertEqual(count_result_1,
{'field_10__count': DBTestCase.NUMBER_OF_RECORDS})
self.assertEqual(count_result_2, {'field_10__count': 0})
self.assertEqual(
avg_result_1,
{'field_10__avg': float(sum(counters)) / len(counters)})
self.assertEqual(avg_result_2, {'field_10__avg': None})
self.assertEqual(min_result_1, {'field_10__min': 1})
self.assertEqual(min_result_2, {'field_10__min': None})
self.assertEqual(max_result_1,
{'field_10__max': DBTestCase.NUMBER_OF_RECORDS})
self.assertEqual(max_result_2, {'field_10__max': None})
self.assertEqual(sum_result_1, {'field_10__sum': sum(counters)})
self.assertEqual(sum_result_2, {'field_10__sum': None})
def test_related_lookup(self):
test_model = None
test_relation = TestModelRelation()
test_relation.save()
for _ in range(0, DBTestCase.NUMBER_OF_RELATIONS):
test_model = TestModel()
test_model.save()
test_relation.test_models.add(test_model)
test_model = TestModel.objects.select_related('field_19').get(
id=test_model.id)
self.assertEqual(test_model.field_19.field_1, 1234)
def test_join_lookup(self):
for _ in range(0, DBTestCase.NUMBER_OF_RECORDS):
test_model = TestModel()
test_model.save()
test_relation = TestModelRelation()
test_relation.save()
test_relation.test_models.add(test_model)
self.assertEqual(
TestModel.objects.filter(field_19__field_1=1234).count(),
DBTestCase.NUMBER_OF_RECORDS)
self.assertEqual(
TestModel.objects.filter(field_19__field_1=0).count(), 0)
