1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534 |
- import datetime
- import pickle
- import unittest
- import uuid
- from copy import deepcopy
- from django.core.exceptions import FieldError
- from django.db import DatabaseError, connection, models, transaction
- from django.db.models import CharField, Q, TimeField, UUIDField
- from django.db.models.aggregates import (
- Avg, Count, Max, Min, StdDev, Sum, Variance,
- )
- from django.db.models.expressions import (
- Case, Col, Combinable, Exists, Expression, ExpressionList,
- ExpressionWrapper, F, Func, OrderBy, OuterRef, Random, RawSQL, Ref,
- Subquery, Value, When,
- )
- from django.db.models.functions import (
- Coalesce, Concat, Length, Lower, Substr, Upper,
- )
- from django.db.models.sql import constants
- from django.db.models.sql.datastructures import Join
- from django.test import SimpleTestCase, TestCase, skipUnlessDBFeature
- from django.test.utils import Approximate
- from .models import (
- UUID, UUIDPK, Company, Employee, Experiment, Number, Result, SimulationRun,
- Time,
- )
- class BasicExpressionsTests(TestCase):
- @classmethod
- def setUpTestData(cls):
- cls.example_inc = Company.objects.create(
- name="Example Inc.", num_employees=2300, num_chairs=5,
- ceo=Employee.objects.create(firstname="Joe", lastname="Smith", salary=10)
- )
- cls.foobar_ltd = Company.objects.create(
- name="Foobar Ltd.", num_employees=3, num_chairs=4,
- ceo=Employee.objects.create(firstname="Frank", lastname="Meyer", salary=20)
- )
- cls.max = Employee.objects.create(firstname='Max', lastname='Mustermann', salary=30)
- cls.gmbh = Company.objects.create(name='Test GmbH', num_employees=32, num_chairs=1, ceo=cls.max)
- def setUp(self):
- self.company_query = Company.objects.values(
- "name", "num_employees", "num_chairs"
- ).order_by(
- "name", "num_employees", "num_chairs"
- )
- def test_annotate_values_aggregate(self):
- companies = Company.objects.annotate(
- salaries=F('ceo__salary'),
- ).values('num_employees', 'salaries').aggregate(
- result=Sum(
- F('salaries') + F('num_employees'),
- output_field=models.IntegerField()
- ),
- )
- self.assertEqual(companies['result'], 2395)
- def test_annotate_values_filter(self):
- companies = Company.objects.annotate(
- foo=RawSQL('%s', ['value']),
- ).filter(foo='value').order_by('name')
- self.assertQuerysetEqual(
- companies,
- ['<Company: Example Inc.>', '<Company: Foobar Ltd.>', '<Company: Test GmbH>'],
- )
- @unittest.skipIf(connection.vendor == 'oracle', "Oracle doesn't support using boolean type in SELECT")
- def test_filtering_on_annotate_that_uses_q(self):
- self.assertEqual(
- Company.objects.annotate(
- num_employees_check=ExpressionWrapper(Q(num_employees__gt=3), output_field=models.BooleanField())
- ).filter(num_employees_check=True).count(),
- 2,
- )
- def test_filter_inter_attribute(self):
-
-
-
- self.assertSequenceEqual(
- self.company_query.filter(num_employees__gt=F("num_chairs")), [
- {
- "num_chairs": 5,
- "name": "Example Inc.",
- "num_employees": 2300,
- },
- {
- "num_chairs": 1,
- "name": "Test GmbH",
- "num_employees": 32
- },
- ],
- )
- def test_update(self):
-
-
- self.company_query.update(num_chairs=F("num_employees"))
- self.assertSequenceEqual(
- self.company_query, [
- {
- "num_chairs": 2300,
- "name": "Example Inc.",
- "num_employees": 2300
- },
- {
- "num_chairs": 3,
- "name": "Foobar Ltd.",
- "num_employees": 3
- },
- {
- "num_chairs": 32,
- "name": "Test GmbH",
- "num_employees": 32
- }
- ],
- )
- def test_arithmetic(self):
-
-
- self.company_query.update(num_chairs=F("num_employees") + 2)
- self.assertSequenceEqual(
- self.company_query, [
- {
- 'num_chairs': 2302,
- 'name': 'Example Inc.',
- 'num_employees': 2300
- },
- {
- 'num_chairs': 5,
- 'name': 'Foobar Ltd.',
- 'num_employees': 3
- },
- {
- 'num_chairs': 34,
- 'name': 'Test GmbH',
- 'num_employees': 32
- }
- ],
- )
- def test_order_of_operations(self):
-
- self.company_query.update(num_chairs=F('num_employees') + 2 * F('num_employees'))
- self.assertSequenceEqual(
- self.company_query, [
- {
- 'num_chairs': 6900,
- 'name': 'Example Inc.',
- 'num_employees': 2300
- },
- {
- 'num_chairs': 9,
- 'name': 'Foobar Ltd.',
- 'num_employees': 3
- },
- {
- 'num_chairs': 96,
- 'name': 'Test GmbH',
- 'num_employees': 32
- }
- ],
- )
- def test_parenthesis_priority(self):
-
- self.company_query.update(num_chairs=(F('num_employees') + 2) * F('num_employees'))
- self.assertSequenceEqual(
- self.company_query, [
- {
- 'num_chairs': 5294600,
- 'name': 'Example Inc.',
- 'num_employees': 2300
- },
- {
- 'num_chairs': 15,
- 'name': 'Foobar Ltd.',
- 'num_employees': 3
- },
- {
- 'num_chairs': 1088,
- 'name': 'Test GmbH',
- 'num_employees': 32
- }
- ],
- )
- def test_update_with_fk(self):
-
- self.assertEqual(Company.objects.update(point_of_contact=F('ceo')), 3)
- self.assertQuerysetEqual(
- Company.objects.all(),
- ['Joe Smith', 'Frank Meyer', 'Max Mustermann'],
- lambda c: str(c.point_of_contact),
- ordered=False
- )
- def test_update_with_none(self):
- Number.objects.create(integer=1, float=1.0)
- Number.objects.create(integer=2)
- Number.objects.filter(float__isnull=False).update(float=Value(None))
- self.assertQuerysetEqual(
- Number.objects.all(),
- [None, None],
- lambda n: n.float,
- ordered=False
- )
- def test_filter_with_join(self):
-
- Company.objects.update(point_of_contact=F('ceo'))
- c = Company.objects.first()
- c.point_of_contact = Employee.objects.create(firstname="Guido", lastname="van Rossum")
- c.save()
- self.assertQuerysetEqual(
- Company.objects.filter(ceo__firstname=F('point_of_contact__firstname')),
- ['Foobar Ltd.', 'Test GmbH'],
- lambda c: c.name,
- ordered=False
- )
- Company.objects.exclude(
- ceo__firstname=F("point_of_contact__firstname")
- ).update(name="foo")
- self.assertEqual(
- Company.objects.exclude(
- ceo__firstname=F('point_of_contact__firstname')
- ).get().name,
- "foo",
- )
- with transaction.atomic():
- msg = "Joined field references are not permitted in this query"
- with self.assertRaisesMessage(FieldError, msg):
- Company.objects.exclude(
- ceo__firstname=F('point_of_contact__firstname')
- ).update(name=F('point_of_contact__lastname'))
- def test_object_update(self):
-
- self.gmbh.num_employees = F('num_employees') + 4
- self.gmbh.save()
- self.gmbh.refresh_from_db()
- self.assertEqual(self.gmbh.num_employees, 36)
- def test_new_object_save(self):
-
- test_co = Company(name=Lower(Value('UPPER')), num_employees=32, num_chairs=1, ceo=self.max)
- test_co.save()
- test_co.refresh_from_db()
- self.assertEqual(test_co.name, "upper")
- def test_new_object_create(self):
- test_co = Company.objects.create(name=Lower(Value('UPPER')), num_employees=32, num_chairs=1, ceo=self.max)
- test_co.refresh_from_db()
- self.assertEqual(test_co.name, "upper")
- def test_object_create_with_aggregate(self):
-
- with self.assertRaisesMessage(FieldError, 'Aggregate functions are not allowed in this query'):
- Company.objects.create(
- name='Company', num_employees=Max(Value(1)), num_chairs=1,
- ceo=Employee.objects.create(firstname="Just", lastname="Doit", salary=30),
- )
- def test_object_update_fk(self):
-
-
- test_gmbh = Company.objects.get(pk=self.gmbh.pk)
- msg = 'F(ceo)": "Company.point_of_contact" must be a "Employee" instance.'
- with self.assertRaisesMessage(ValueError, msg):
- test_gmbh.point_of_contact = F('ceo')
- test_gmbh.point_of_contact = self.gmbh.ceo
- test_gmbh.save()
- test_gmbh.name = F('ceo__last_name')
- msg = 'Joined field references are not permitted in this query'
- with self.assertRaisesMessage(FieldError, msg):
- test_gmbh.save()
- def test_object_update_unsaved_objects(self):
-
-
- acme = Company(name='The Acme Widget Co.', num_employees=12, num_chairs=5, ceo=self.max)
- acme.num_employees = F("num_employees") + 16
- msg = (
- 'Failed to insert expression "Col(expressions_company, '
- 'expressions.Company.num_employees) + Value(16)" on '
- 'expressions.Company.num_employees. F() expressions can only be '
- 'used to update, not to insert.'
- )
- with self.assertRaisesMessage(ValueError, msg):
- acme.save()
- acme.num_employees = 12
- acme.name = Lower(F('name'))
- msg = (
- 'Failed to insert expression "Lower(Col(expressions_company, '
- 'expressions.Company.name))" on expressions.Company.name. F() '
- 'expressions can only be used to update, not to insert.'
- )
- with self.assertRaisesMessage(ValueError, msg):
- acme.save()
- def test_ticket_11722_iexact_lookup(self):
- Employee.objects.create(firstname="John", lastname="Doe")
- Employee.objects.create(firstname="Test", lastname="test")
- queryset = Employee.objects.filter(firstname__iexact=F('lastname'))
- self.assertQuerysetEqual(queryset, ["<Employee: Test test>"])
- def test_ticket_16731_startswith_lookup(self):
- Employee.objects.create(firstname="John", lastname="Doe")
- e2 = Employee.objects.create(firstname="Jack", lastname="Jackson")
- e3 = Employee.objects.create(firstname="Jack", lastname="jackson")
- self.assertSequenceEqual(
- Employee.objects.filter(lastname__startswith=F('firstname')),
- [e2, e3] if connection.features.has_case_insensitive_like else [e2]
- )
- qs = Employee.objects.filter(lastname__istartswith=F('firstname')).order_by('pk')
- self.assertSequenceEqual(qs, [e2, e3])
- def test_ticket_18375_join_reuse(self):
-
-
-
- qs = Employee.objects.filter(company_ceo_set__num_chairs=F('company_ceo_set__num_employees'))
- self.assertEqual(str(qs.query).count('JOIN'), 1)
- def test_ticket_18375_kwarg_ordering(self):
-
-
-
-
- qs = Employee.objects.filter(
- company_ceo_set__num_chairs=F('company_ceo_set__num_employees'),
- company_ceo_set__num_chairs__gte=1,
- )
- self.assertEqual(str(qs.query).count('JOIN'), 1)
- def test_ticket_18375_kwarg_ordering_2(self):
-
-
-
-
-
- qs = Employee.objects.filter(
- company_ceo_set__num_employees=F('pk'),
- pk=F('company_ceo_set__num_employees')
- )
- self.assertEqual(str(qs.query).count('JOIN'), 1)
- def test_ticket_18375_chained_filters(self):
-
- qs = Employee.objects.filter(
- company_ceo_set__num_employees=F('pk')
- ).filter(
- company_ceo_set__num_employees=F('company_ceo_set__num_employees')
- )
- self.assertEqual(str(qs.query).count('JOIN'), 2)
- def test_order_by_exists(self):
- mary = Employee.objects.create(firstname='Mary', lastname='Mustermann', salary=20)
- mustermanns_by_seniority = Employee.objects.filter(lastname='Mustermann').order_by(
-
- Exists(Company.objects.filter(ceo=OuterRef('pk'))).desc()
- )
- self.assertSequenceEqual(mustermanns_by_seniority, [self.max, mary])
- def test_outerref(self):
- inner = Company.objects.filter(point_of_contact=OuterRef('pk'))
- msg = (
- 'This queryset contains a reference to an outer query and may only '
- 'be used in a subquery.'
- )
- with self.assertRaisesMessage(ValueError, msg):
- inner.exists()
- outer = Employee.objects.annotate(is_point_of_contact=Exists(inner))
- self.assertIs(outer.exists(), True)
- def test_exist_single_field_output_field(self):
- queryset = Company.objects.values('pk')
- self.assertIsInstance(Exists(queryset).output_field, models.BooleanField)
- def test_subquery(self):
- Company.objects.filter(name='Example Inc.').update(
- point_of_contact=Employee.objects.get(firstname='Joe', lastname='Smith'),
- ceo=self.max,
- )
- Employee.objects.create(firstname='Bob', lastname='Brown', salary=40)
- qs = Employee.objects.annotate(
- is_point_of_contact=Exists(Company.objects.filter(point_of_contact=OuterRef('pk'))),
- is_not_point_of_contact=~Exists(Company.objects.filter(point_of_contact=OuterRef('pk'))),
- is_ceo_of_small_company=Exists(Company.objects.filter(num_employees__lt=200, ceo=OuterRef('pk'))),
- is_ceo_small_2=~~Exists(Company.objects.filter(num_employees__lt=200, ceo=OuterRef('pk'))),
- largest_company=Subquery(Company.objects.order_by('-num_employees').filter(
- models.Q(ceo=OuterRef('pk')) | models.Q(point_of_contact=OuterRef('pk'))
- ).values('name')[:1], output_field=models.CharField())
- ).values(
- 'firstname',
- 'is_point_of_contact',
- 'is_not_point_of_contact',
- 'is_ceo_of_small_company',
- 'is_ceo_small_2',
- 'largest_company',
- ).order_by('firstname')
- results = list(qs)
-
-
- bob = results[0]
- if bob['largest_company'] == '' and connection.features.interprets_empty_strings_as_nulls:
- bob['largest_company'] = None
- self.assertEqual(results, [
- {
- 'firstname': 'Bob',
- 'is_point_of_contact': False,
- 'is_not_point_of_contact': True,
- 'is_ceo_of_small_company': False,
- 'is_ceo_small_2': False,
- 'largest_company': None,
- },
- {
- 'firstname': 'Frank',
- 'is_point_of_contact': False,
- 'is_not_point_of_contact': True,
- 'is_ceo_of_small_company': True,
- 'is_ceo_small_2': True,
- 'largest_company': 'Foobar Ltd.',
- },
- {
- 'firstname': 'Joe',
- 'is_point_of_contact': True,
- 'is_not_point_of_contact': False,
- 'is_ceo_of_small_company': False,
- 'is_ceo_small_2': False,
- 'largest_company': 'Example Inc.',
- },
- {
- 'firstname': 'Max',
- 'is_point_of_contact': False,
- 'is_not_point_of_contact': True,
- 'is_ceo_of_small_company': True,
- 'is_ceo_small_2': True,
- 'largest_company': 'Example Inc.'
- }
- ])
-
-
-
- self.assertCountEqual(
- qs.filter(is_point_of_contact=True).values('pk'),
- Employee.objects.exclude(company_point_of_contact_set=None).values('pk')
- )
- def test_in_subquery(self):
-
-
- small_companies = Company.objects.filter(num_employees__lt=200).values('pk')
- subquery_test = Company.objects.filter(pk__in=Subquery(small_companies))
- self.assertCountEqual(subquery_test, [self.foobar_ltd, self.gmbh])
- subquery_test2 = Company.objects.filter(pk=Subquery(small_companies.filter(num_employees=3)))
- self.assertCountEqual(subquery_test2, [self.foobar_ltd])
- def test_uuid_pk_subquery(self):
- u = UUIDPK.objects.create()
- UUID.objects.create(uuid_fk=u)
- qs = UUIDPK.objects.filter(id__in=Subquery(UUID.objects.values('uuid_fk__id')))
- self.assertCountEqual(qs, [u])
- def test_nested_subquery(self):
- inner = Company.objects.filter(point_of_contact=OuterRef('pk'))
- outer = Employee.objects.annotate(is_point_of_contact=Exists(inner))
- contrived = Employee.objects.annotate(
- is_point_of_contact=Subquery(
- outer.filter(pk=OuterRef('pk')).values('is_point_of_contact'),
- output_field=models.BooleanField(),
- ),
- )
- self.assertCountEqual(contrived.values_list(), outer.values_list())
- def test_nested_subquery_outer_ref_2(self):
- first = Time.objects.create(time='09:00')
- second = Time.objects.create(time='17:00')
- third = Time.objects.create(time='21:00')
- SimulationRun.objects.bulk_create([
- SimulationRun(start=first, end=second, midpoint='12:00'),
- SimulationRun(start=first, end=third, midpoint='15:00'),
- SimulationRun(start=second, end=first, midpoint='00:00'),
- ])
- inner = Time.objects.filter(time=OuterRef(OuterRef('time')), pk=OuterRef('start')).values('time')
- middle = SimulationRun.objects.annotate(other=Subquery(inner)).values('other')[:1]
- outer = Time.objects.annotate(other=Subquery(middle, output_field=models.TimeField()))
-
- self.assertCountEqual(outer, [first, second, third])
- def test_nested_subquery_outer_ref_with_autofield(self):
- first = Time.objects.create(time='09:00')
- second = Time.objects.create(time='17:00')
- SimulationRun.objects.create(start=first, end=second, midpoint='12:00')
- inner = SimulationRun.objects.filter(start=OuterRef(OuterRef('pk'))).values('start')
- middle = Time.objects.annotate(other=Subquery(inner)).values('other')[:1]
- outer = Time.objects.annotate(other=Subquery(middle, output_field=models.IntegerField()))
-
- self.assertCountEqual(outer, [first, second])
- def test_annotations_within_subquery(self):
- Company.objects.filter(num_employees__lt=50).update(ceo=Employee.objects.get(firstname='Frank'))
- inner = Company.objects.filter(
- ceo=OuterRef('pk')
- ).values('ceo').annotate(total_employees=models.Sum('num_employees')).values('total_employees')
- outer = Employee.objects.annotate(total_employees=Subquery(inner)).filter(salary__lte=Subquery(inner))
- self.assertSequenceEqual(
- outer.order_by('-total_employees').values('salary', 'total_employees'),
- [{'salary': 10, 'total_employees': 2300}, {'salary': 20, 'total_employees': 35}],
- )
- def test_subquery_references_joined_table_twice(self):
- inner = Company.objects.filter(
- num_chairs__gte=OuterRef('ceo__salary'),
- num_employees__gte=OuterRef('point_of_contact__salary'),
- )
-
- outer = Company.objects.filter(pk__in=Subquery(inner.values('pk')))
- self.assertFalse(outer.exists())
- def test_explicit_output_field(self):
- class FuncA(Func):
- output_field = models.CharField()
- class FuncB(Func):
- pass
- expr = FuncB(FuncA())
- self.assertEqual(expr.output_field, FuncA.output_field)
- def test_outerref_mixed_case_table_name(self):
- inner = Result.objects.filter(result_time__gte=OuterRef('experiment__assigned'))
- outer = Result.objects.filter(pk__in=Subquery(inner.values('pk')))
- self.assertFalse(outer.exists())
- def test_outerref_with_operator(self):
- inner = Company.objects.filter(num_employees=OuterRef('ceo__salary') + 2)
- outer = Company.objects.filter(pk__in=Subquery(inner.values('pk')))
- self.assertEqual(outer.get().name, 'Test GmbH')
- def test_pickle_expression(self):
- expr = Value(1, output_field=models.IntegerField())
- expr.convert_value
- self.assertEqual(pickle.loads(pickle.dumps(expr)), expr)
- def test_incorrect_field_in_F_expression(self):
- with self.assertRaisesMessage(FieldError, "Cannot resolve keyword 'nope' into field."):
- list(Employee.objects.filter(firstname=F('nope')))
- def test_incorrect_joined_field_in_F_expression(self):
- with self.assertRaisesMessage(FieldError, "Cannot resolve keyword 'nope' into field."):
- list(Company.objects.filter(ceo__pk=F('point_of_contact__nope')))
- class IterableLookupInnerExpressionsTests(TestCase):
- @classmethod
- def setUpTestData(cls):
- ceo = Employee.objects.create(firstname='Just', lastname='Doit', salary=30)
-
-
-
- Company.objects.create(name='5020 Ltd', num_employees=50, num_chairs=20, ceo=ceo)
- Company.objects.create(name='5040 Ltd', num_employees=50, num_chairs=40, ceo=ceo)
- Company.objects.create(name='5050 Ltd', num_employees=50, num_chairs=50, ceo=ceo)
- Company.objects.create(name='5060 Ltd', num_employees=50, num_chairs=60, ceo=ceo)
- Company.objects.create(name='99300 Ltd', num_employees=99, num_chairs=300, ceo=ceo)
- def test_in_lookup_allows_F_expressions_and_expressions_for_integers(self):
-
- queryset = Company.objects.filter(num_employees__in=([F('num_chairs') - 10]))
- self.assertQuerysetEqual(queryset, ['<Company: 5060 Ltd>'], ordered=False)
- self.assertQuerysetEqual(
- Company.objects.filter(num_employees__in=([F('num_chairs') - 10, F('num_chairs') + 10])),
- ['<Company: 5040 Ltd>', '<Company: 5060 Ltd>'],
- ordered=False
- )
- self.assertQuerysetEqual(
- Company.objects.filter(
- num_employees__in=([F('num_chairs') - 10, F('num_chairs'), F('num_chairs') + 10])
- ),
- ['<Company: 5040 Ltd>', '<Company: 5050 Ltd>', '<Company: 5060 Ltd>'],
- ordered=False
- )
- def test_expressions_in_lookups_join_choice(self):
- midpoint = datetime.time(13, 0)
- t1 = Time.objects.create(time=datetime.time(12, 0))
- t2 = Time.objects.create(time=datetime.time(14, 0))
- SimulationRun.objects.create(start=t1, end=t2, midpoint=midpoint)
- SimulationRun.objects.create(start=t1, end=None, midpoint=midpoint)
- SimulationRun.objects.create(start=None, end=t2, midpoint=midpoint)
- SimulationRun.objects.create(start=None, end=None, midpoint=midpoint)
- queryset = SimulationRun.objects.filter(midpoint__range=[F('start__time'), F('end__time')])
- self.assertQuerysetEqual(
- queryset,
- ['<SimulationRun: 13:00:00 (12:00:00 to 14:00:00)>'],
- ordered=False
- )
- for alias in queryset.query.alias_map.values():
- if isinstance(alias, Join):
- self.assertEqual(alias.join_type, constants.INNER)
- queryset = SimulationRun.objects.exclude(midpoint__range=[F('start__time'), F('end__time')])
- self.assertQuerysetEqual(queryset, [], ordered=False)
- for alias in queryset.query.alias_map.values():
- if isinstance(alias, Join):
- self.assertEqual(alias.join_type, constants.LOUTER)
- def test_range_lookup_allows_F_expressions_and_expressions_for_integers(self):
-
- Company.objects.filter(num_employees__exact=F("num_chairs"))
- self.assertQuerysetEqual(
- Company.objects.filter(num_employees__range=(F('num_chairs'), 100)),
- ['<Company: 5020 Ltd>', '<Company: 5040 Ltd>', '<Company: 5050 Ltd>'],
- ordered=False
- )
- self.assertQuerysetEqual(
- Company.objects.filter(num_employees__range=(F('num_chairs') - 10, F('num_chairs') + 10)),
- ['<Company: 5040 Ltd>', '<Company: 5050 Ltd>', '<Company: 5060 Ltd>'],
- ordered=False
- )
- self.assertQuerysetEqual(
- Company.objects.filter(num_employees__range=(F('num_chairs') - 10, 100)),
- ['<Company: 5020 Ltd>', '<Company: 5040 Ltd>', '<Company: 5050 Ltd>', '<Company: 5060 Ltd>'],
- ordered=False
- )
- self.assertQuerysetEqual(
- Company.objects.filter(num_employees__range=(1, 100)),
- [
- '<Company: 5020 Ltd>', '<Company: 5040 Ltd>', '<Company: 5050 Ltd>',
- '<Company: 5060 Ltd>', '<Company: 99300 Ltd>',
- ],
- ordered=False
- )
- @unittest.skipUnless(connection.vendor == 'sqlite',
- "This defensive test only works on databases that don't validate parameter types")
- def test_complex_expressions_do_not_introduce_sql_injection_via_untrusted_string_inclusion(self):
- """
- This tests that SQL injection isn't possible using compilation of
- expressions in iterable filters, as their compilation happens before
- the main query compilation. It's limited to SQLite, as PostgreSQL,
- Oracle and other vendors have defense in depth against this by type
- checking. Testing against SQLite (the most permissive of the built-in
- databases) demonstrates that the problem doesn't exist while keeping
- the test simple.
- """
- queryset = Company.objects.filter(name__in=[F('num_chairs') + '1)) OR ((1==1'])
- self.assertQuerysetEqual(queryset, [], ordered=False)
- def test_in_lookup_allows_F_expressions_and_expressions_for_datetimes(self):
- start = datetime.datetime(2016, 2, 3, 15, 0, 0)
- end = datetime.datetime(2016, 2, 5, 15, 0, 0)
- experiment_1 = Experiment.objects.create(
- name='Integrity testing',
- assigned=start.date(),
- start=start,
- end=end,
- completed=end.date(),
- estimated_time=end - start,
- )
- experiment_2 = Experiment.objects.create(
- name='Taste testing',
- assigned=start.date(),
- start=start,
- end=end,
- completed=end.date(),
- estimated_time=end - start,
- )
- Result.objects.create(
- experiment=experiment_1,
- result_time=datetime.datetime(2016, 2, 4, 15, 0, 0),
- )
- Result.objects.create(
- experiment=experiment_1,
- result_time=datetime.datetime(2016, 3, 10, 2, 0, 0),
- )
- Result.objects.create(
- experiment=experiment_2,
- result_time=datetime.datetime(2016, 1, 8, 5, 0, 0),
- )
- within_experiment_time = [F('experiment__start'), F('experiment__end')]
- queryset = Result.objects.filter(result_time__range=within_experiment_time)
- self.assertQuerysetEqual(queryset, ["<Result: Result at 2016-02-04 15:00:00>"])
- within_experiment_time = [F('experiment__start'), F('experiment__end')]
- queryset = Result.objects.filter(result_time__range=within_experiment_time)
- self.assertQuerysetEqual(queryset, ["<Result: Result at 2016-02-04 15:00:00>"])
- class FTests(SimpleTestCase):
- def test_deepcopy(self):
- f = F("foo")
- g = deepcopy(f)
- self.assertEqual(f.name, g.name)
- def test_deconstruct(self):
- f = F('name')
- path, args, kwargs = f.deconstruct()
- self.assertEqual(path, 'django.db.models.expressions.F')
- self.assertEqual(args, (f.name,))
- self.assertEqual(kwargs, {})
- def test_equal(self):
- f = F('name')
- same_f = F('name')
- other_f = F('username')
- self.assertEqual(f, same_f)
- self.assertNotEqual(f, other_f)
- def test_hash(self):
- d = {F('name'): 'Bob'}
- self.assertIn(F('name'), d)
- self.assertEqual(d[F('name')], 'Bob')
- def test_not_equal_Value(self):
- f = F('name')
- value = Value('name')
- self.assertNotEqual(f, value)
- self.assertNotEqual(value, f)
- class ExpressionsTests(TestCase):
- def test_F_reuse(self):
- f = F('id')
- n = Number.objects.create(integer=-1)
- c = Company.objects.create(
- name="Example Inc.", num_employees=2300, num_chairs=5,
- ceo=Employee.objects.create(firstname="Joe", lastname="Smith")
- )
- c_qs = Company.objects.filter(id=f)
- self.assertEqual(c_qs.get(), c)
-
- n_qs = Number.objects.filter(id=f)
- self.assertEqual(n_qs.get(), n)
-
- self.assertEqual(c_qs.get(), c)
- def test_patterns_escape(self):
- r"""
- Special characters (e.g. %, _ and \) stored in database are
- properly escaped when using a pattern lookup with an expression
- refs #16731
- """
- Employee.objects.bulk_create([
- Employee(firstname="%Joh\\nny", lastname="%Joh\\n"),
- Employee(firstname="Johnny", lastname="%John"),
- Employee(firstname="Jean-Claude", lastname="Claud_"),
- Employee(firstname="Jean-Claude", lastname="Claude"),
- Employee(firstname="Jean-Claude", lastname="Claude%"),
- Employee(firstname="Johnny", lastname="Joh\\n"),
- Employee(firstname="Johnny", lastname="John"),
- Employee(firstname="Johnny", lastname="_ohn"),
- ])
- self.assertQuerysetEqual(
- Employee.objects.filter(firstname__contains=F('lastname')),
- ["<Employee: %Joh\\nny %Joh\\n>", "<Employee: Jean-Claude Claude>", "<Employee: Johnny John>"],
- ordered=False,
- )
- self.assertQuerysetEqual(
- Employee.objects.filter(firstname__startswith=F('lastname')),
- ["<Employee: %Joh\\nny %Joh\\n>", "<Employee: Johnny John>"],
- ordered=False,
- )
- self.assertQuerysetEqual(
- Employee.objects.filter(firstname__endswith=F('lastname')),
- ["<Employee: Jean-Claude Claude>"],
- ordered=False,
- )
- def test_insensitive_patterns_escape(self):
- r"""
- Special characters (e.g. %, _ and \) stored in database are
- properly escaped when using a case insensitive pattern lookup with an
- expression -- refs #16731
- """
- Employee.objects.bulk_create([
- Employee(firstname="%Joh\\nny", lastname="%joh\\n"),
- Employee(firstname="Johnny", lastname="%john"),
- Employee(firstname="Jean-Claude", lastname="claud_"),
- Employee(firstname="Jean-Claude", lastname="claude"),
- Employee(firstname="Jean-Claude", lastname="claude%"),
- Employee(firstname="Johnny", lastname="joh\\n"),
- Employee(firstname="Johnny", lastname="john"),
- Employee(firstname="Johnny", lastname="_ohn"),
- ])
- self.assertQuerysetEqual(
- Employee.objects.filter(firstname__icontains=F('lastname')),
- ["<Employee: %Joh\\nny %joh\\n>", "<Employee: Jean-Claude claude>", "<Employee: Johnny john>"],
- ordered=False,
- )
- self.assertQuerysetEqual(
- Employee.objects.filter(firstname__istartswith=F('lastname')),
- ["<Employee: %Joh\\nny %joh\\n>", "<Employee: Johnny john>"],
- ordered=False,
- )
- self.assertQuerysetEqual(
- Employee.objects.filter(firstname__iendswith=F('lastname')),
- ["<Employee: Jean-Claude claude>"],
- ordered=False,
- )
- class SimpleExpressionTests(SimpleTestCase):
- def test_equal(self):
- self.assertEqual(Expression(), Expression())
- self.assertEqual(
- Expression(models.IntegerField()),
- Expression(output_field=models.IntegerField())
- )
- self.assertNotEqual(
- Expression(models.IntegerField()),
- Expression(models.CharField())
- )
- def test_hash(self):
- self.assertEqual(hash(Expression()), hash(Expression()))
- self.assertEqual(
- hash(Expression(models.IntegerField())),
- hash(Expression(output_field=models.IntegerField()))
- )
- self.assertNotEqual(
- hash(Expression(models.IntegerField())),
- hash(Expression(models.CharField())),
- )
- class ExpressionsNumericTests(TestCase):
- @classmethod
- def setUpTestData(cls):
- Number(integer=-1).save()
- Number(integer=42).save()
- Number(integer=1337).save()
- Number.objects.update(float=F('integer'))
- def test_fill_with_value_from_same_object(self):
- """
- We can fill a value in all objects with an other value of the
- same object.
- """
- self.assertQuerysetEqual(
- Number.objects.all(),
- ['<Number: -1, -1.000>', '<Number: 42, 42.000>', '<Number: 1337, 1337.000>'],
- ordered=False
- )
- def test_increment_value(self):
- """
- We can increment a value of all objects in a query set.
- """
- self.assertEqual(Number.objects.filter(integer__gt=0).update(integer=F('integer') + 1), 2)
- self.assertQuerysetEqual(
- Number.objects.all(),
- ['<Number: -1, -1.000>', '<Number: 43, 42.000>', '<Number: 1338, 1337.000>'],
- ordered=False
- )
- def test_filter_not_equals_other_field(self):
- """
- We can filter for objects, where a value is not equals the value
- of an other field.
- """
- self.assertEqual(Number.objects.filter(integer__gt=0).update(integer=F('integer') + 1), 2)
- self.assertQuerysetEqual(
- Number.objects.exclude(float=F('integer')),
- ['<Number: 43, 42.000>', '<Number: 1338, 1337.000>'],
- ordered=False
- )
- def test_complex_expressions(self):
- """
- Complex expressions of different connection types are possible.
- """
- n = Number.objects.create(integer=10, float=123.45)
- self.assertEqual(Number.objects.filter(pk=n.pk).update(
- float=F('integer') + F('float') * 2), 1)
- self.assertEqual(Number.objects.get(pk=n.pk).integer, 10)
- self.assertEqual(Number.objects.get(pk=n.pk).float, Approximate(256.900, places=3))
- class ExpressionOperatorTests(TestCase):
- @classmethod
- def setUpTestData(cls):
- cls.n = Number.objects.create(integer=42, float=15.5)
- cls.n1 = Number.objects.create(integer=-42, float=-15.5)
- def test_lefthand_addition(self):
-
- Number.objects.filter(pk=self.n.pk).update(
- integer=F('integer') + 15,
- float=F('float') + 42.7
- )
- self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 57)
- self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(58.200, places=3))
- def test_lefthand_subtraction(self):
-
- Number.objects.filter(pk=self.n.pk).update(integer=F('integer') - 15, float=F('float') - 42.7)
- self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 27)
- self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(-27.200, places=3))
- def test_lefthand_multiplication(self):
-
- Number.objects.filter(pk=self.n.pk).update(integer=F('integer') * 15, float=F('float') * 42.7)
- self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 630)
- self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(661.850, places=3))
- def test_lefthand_division(self):
-
- Number.objects.filter(pk=self.n.pk).update(integer=F('integer') / 2, float=F('float') / 42.7)
- self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 21)
- self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(0.363, places=3))
- def test_lefthand_modulo(self):
-
- Number.objects.filter(pk=self.n.pk).update(integer=F('integer') % 20)
- self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 2)
- self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(15.500, places=3))
- def test_lefthand_bitwise_and(self):
-
- Number.objects.filter(pk=self.n.pk).update(integer=F('integer').bitand(56))
- Number.objects.filter(pk=self.n1.pk).update(integer=F('integer').bitand(-56))
- self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 40)
- self.assertEqual(Number.objects.get(pk=self.n1.pk).integer, -64)
- self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(15.500, places=3))
- def test_lefthand_bitwise_left_shift_operator(self):
- Number.objects.update(integer=F('integer').bitleftshift(2))
- self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 168)
- self.assertEqual(Number.objects.get(pk=self.n1.pk).integer, -168)
- def test_lefthand_bitwise_right_shift_operator(self):
- Number.objects.update(integer=F('integer').bitrightshift(2))
- self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 10)
- self.assertEqual(Number.objects.get(pk=self.n1.pk).integer, -11)
- def test_lefthand_bitwise_or(self):
-
- Number.objects.update(integer=F('integer').bitor(48))
- self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 58)
- self.assertEqual(Number.objects.get(pk=self.n1.pk).integer, -10)
- self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(15.500, places=3))
- def test_lefthand_power(self):
-
- Number.objects.filter(pk=self.n.pk).update(integer=F('integer') ** 2, float=F('float') ** 1.5)
- self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 1764)
- self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(61.02, places=2))
- def test_right_hand_addition(self):
-
- Number.objects.filter(pk=self.n.pk).update(integer=15 + F('integer'), float=42.7 + F('float'))
-
- self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 57)
- self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(58.200, places=3))
- def test_right_hand_subtraction(self):
- Number.objects.filter(pk=self.n.pk).update(integer=15 - F('integer'), float=42.7 - F('float'))
-
- self.assertEqual(Number.objects.get(pk=self.n.pk).integer, -27)
- self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(27.200, places=3))
- def test_right_hand_multiplication(self):
-
- Number.objects.filter(pk=self.n.pk).update(integer=15 * F('integer'), float=42.7 * F('float'))
- self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 630)
- self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(661.850, places=3))
- def test_right_hand_division(self):
-
- Number.objects.filter(pk=self.n.pk).update(integer=640 / F('integer'), float=42.7 / F('float'))
- self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 15)
- self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(2.755, places=3))
- def test_right_hand_modulo(self):
-
- Number.objects.filter(pk=self.n.pk).update(integer=69 % F('integer'))
- self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 27)
- self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(15.500, places=3))
- def test_righthand_power(self):
-
- Number.objects.filter(pk=self.n.pk).update(integer=2 ** F('integer'), float=1.5 ** F('float'))
- self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 4398046511104)
- self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(536.308, places=3))
- class FTimeDeltaTests(TestCase):
- @classmethod
- def setUpTestData(cls):
- cls.sday = sday = datetime.date(2010, 6, 25)
- cls.stime = stime = datetime.datetime(2010, 6, 25, 12, 15, 30, 747000)
- midnight = datetime.time(0)
- delta0 = datetime.timedelta(0)
- delta1 = datetime.timedelta(microseconds=253000)
- delta2 = datetime.timedelta(seconds=44)
- delta3 = datetime.timedelta(hours=21, minutes=8)
- delta4 = datetime.timedelta(days=10)
- delta5 = datetime.timedelta(days=90)
-
-
- cls.deltas = []
- cls.delays = []
- cls.days_long = []
-
- end = stime + delta0
- e0 = Experiment.objects.create(
- name='e0', assigned=sday, start=stime, end=end,
- completed=end.date(), estimated_time=delta0,
- )
- cls.deltas.append(delta0)
- cls.delays.append(e0.start - datetime.datetime.combine(e0.assigned, midnight))
- cls.days_long.append(e0.completed - e0.assigned)
-
-
-
- delay = datetime.timedelta(1)
- end = stime + delay + delta1
- e1 = Experiment.objects.create(
- name='e1', assigned=sday, start=stime + delay, end=end,
- completed=end.date(), estimated_time=delta1,
- )
- cls.deltas.append(delta1)
- cls.delays.append(e1.start - datetime.datetime.combine(e1.assigned, midnight))
- cls.days_long.append(e1.completed - e1.assigned)
-
- end = stime + delta2
- e2 = Experiment.objects.create(
- name='e2', assigned=sday - datetime.timedelta(3), start=stime,
- end=end, completed=end.date(), estimated_time=datetime.timedelta(hours=1),
- )
- cls.deltas.append(delta2)
- cls.delays.append(e2.start - datetime.datetime.combine(e2.assigned, midnight))
- cls.days_long.append(e2.completed - e2.assigned)
-
- delay = datetime.timedelta(4)
- end = stime + delay + delta3
- e3 = Experiment.objects.create(
- name='e3', assigned=sday, start=stime + delay, end=end,
- completed=end.date(), estimated_time=delta3,
- )
- cls.deltas.append(delta3)
- cls.delays.append(e3.start - datetime.datetime.combine(e3.assigned, midnight))
- cls.days_long.append(e3.completed - e3.assigned)
-
- end = stime + delta4
- e4 = Experiment.objects.create(
- name='e4', assigned=sday - datetime.timedelta(10), start=stime,
- end=end, completed=end.date(), estimated_time=delta4 - datetime.timedelta(1),
- )
- cls.deltas.append(delta4)
- cls.delays.append(e4.start - datetime.datetime.combine(e4.assigned, midnight))
- cls.days_long.append(e4.completed - e4.assigned)
-
- delay = datetime.timedelta(30)
- end = stime + delay + delta5
- e5 = Experiment.objects.create(
- name='e5', assigned=sday, start=stime + delay, end=end,
- completed=end.date(), estimated_time=delta5,
- )
- cls.deltas.append(delta5)
- cls.delays.append(e5.start - datetime.datetime.combine(e5.assigned, midnight))
- cls.days_long.append(e5.completed - e5.assigned)
- cls.expnames = [e.name for e in Experiment.objects.all()]
- def test_multiple_query_compilation(self):
-
- queryset = Experiment.objects.filter(end__lt=F('start') + datetime.timedelta(hours=1))
- q1 = str(queryset.query)
- q2 = str(queryset.query)
- self.assertEqual(q1, q2)
- def test_query_clone(self):
-
- qs = Experiment.objects.filter(end__lt=F('start') + datetime.timedelta(hours=1))
- qs2 = qs.all()
- list(qs)
- list(qs2)
-
- def test_delta_add(self):
- for i, delta in enumerate(self.deltas):
- test_set = [e.name for e in Experiment.objects.filter(end__lt=F('start') + delta)]
- self.assertEqual(test_set, self.expnames[:i])
- test_set = [e.name for e in Experiment.objects.filter(end__lt=delta + F('start'))]
- self.assertEqual(test_set, self.expnames[:i])
- test_set = [e.name for e in Experiment.objects.filter(end__lte=F('start') + delta)]
- self.assertEqual(test_set, self.expnames[:i + 1])
- def test_delta_subtract(self):
- for i, delta in enumerate(self.deltas):
- test_set = [e.name for e in Experiment.objects.filter(start__gt=F('end') - delta)]
- self.assertEqual(test_set, self.expnames[:i])
- test_set = [e.name for e in Experiment.objects.filter(start__gte=F('end') - delta)]
- self.assertEqual(test_set, self.expnames[:i + 1])
- def test_exclude(self):
- for i, delta in enumerate(self.deltas):
- test_set = [e.name for e in Experiment.objects.exclude(end__lt=F('start') + delta)]
- self.assertEqual(test_set, self.expnames[i:])
- test_set = [e.name for e in Experiment.objects.exclude(end__lte=F('start') + delta)]
- self.assertEqual(test_set, self.expnames[i + 1:])
- def test_date_comparison(self):
- for i, days in enumerate(self.days_long):
- test_set = [e.name for e in Experiment.objects.filter(completed__lt=F('assigned') + days)]
- self.assertEqual(test_set, self.expnames[:i])
- test_set = [e.name for e in Experiment.objects.filter(completed__lte=F('assigned') + days)]
- self.assertEqual(test_set, self.expnames[:i + 1])
- @skipUnlessDBFeature("supports_mixed_date_datetime_comparisons")
- def test_mixed_comparisons1(self):
- for i, delay in enumerate(self.delays):
- test_set = [e.name for e in Experiment.objects.filter(assigned__gt=F('start') - delay)]
- self.assertEqual(test_set, self.expnames[:i])
- test_set = [e.name for e in Experiment.objects.filter(assigned__gte=F('start') - delay)]
- self.assertEqual(test_set, self.expnames[:i + 1])
- def test_mixed_comparisons2(self):
- for i, delay in enumerate(self.delays):
- delay = datetime.timedelta(delay.days)
- test_set = [e.name for e in Experiment.objects.filter(start__lt=F('assigned') + delay)]
- self.assertEqual(test_set, self.expnames[:i])
- test_set = [
- e.name for e in Experiment.objects.filter(start__lte=F('assigned') + delay + datetime.timedelta(1))
- ]
- self.assertEqual(test_set, self.expnames[:i + 1])
- def test_delta_update(self):
- for delta in self.deltas:
- exps = Experiment.objects.all()
- expected_durations = [e.duration() for e in exps]
- expected_starts = [e.start + delta for e in exps]
- expected_ends = [e.end + delta for e in exps]
- Experiment.objects.update(start=F('start') + delta, end=F('end') + delta)
- exps = Experiment.objects.all()
- new_starts = [e.start for e in exps]
- new_ends = [e.end for e in exps]
- new_durations = [e.duration() for e in exps]
- self.assertEqual(expected_starts, new_starts)
- self.assertEqual(expected_ends, new_ends)
- self.assertEqual(expected_durations, new_durations)
- def test_invalid_operator(self):
- with self.assertRaises(DatabaseError):
- list(Experiment.objects.filter(start=F('start') * datetime.timedelta(0)))
- def test_durationfield_add(self):
- zeros = [e.name for e in Experiment.objects.filter(start=F('start') + F('estimated_time'))]
- self.assertEqual(zeros, ['e0'])
- end_less = [e.name for e in Experiment.objects.filter(end__lt=F('start') + F('estimated_time'))]
- self.assertEqual(end_less, ['e2'])
- delta_math = [
- e.name for e in
- Experiment.objects.filter(end__gte=F('start') + F('estimated_time') + datetime.timedelta(hours=1))
- ]
- self.assertEqual(delta_math, ['e4'])
- queryset = Experiment.objects.annotate(shifted=ExpressionWrapper(
- F('start') + Value(None, output_field=models.DurationField()),
- output_field=models.DateTimeField(),
- ))
- self.assertIsNone(queryset.first().shifted)
- @skipUnlessDBFeature('supports_temporal_subtraction')
- def test_date_subtraction(self):
- queryset = Experiment.objects.annotate(
- completion_duration=ExpressionWrapper(
- F('completed') - F('assigned'), output_field=models.DurationField()
- )
- )
- at_least_5_days = {e.name for e in queryset.filter(completion_duration__gte=datetime.timedelta(days=5))}
- self.assertEqual(at_least_5_days, {'e3', 'e4', 'e5'})
- at_least_120_days = {e.name for e in queryset.filter(completion_duration__gte=datetime.timedelta(days=120))}
- self.assertEqual(at_least_120_days, {'e5'})
- less_than_5_days = {e.name for e in queryset.filter(completion_duration__lt=datetime.timedelta(days=5))}
- self.assertEqual(less_than_5_days, {'e0', 'e1', 'e2'})
- queryset = Experiment.objects.annotate(difference=ExpressionWrapper(
- F('completed') - Value(None, output_field=models.DateField()),
- output_field=models.DurationField(),
- ))
- self.assertIsNone(queryset.first().difference)
- queryset = Experiment.objects.annotate(shifted=ExpressionWrapper(
- F('completed') - Value(None, output_field=models.DurationField()),
- output_field=models.DateField(),
- ))
- self.assertIsNone(queryset.first().shifted)
- @skipUnlessDBFeature('supports_temporal_subtraction')
- def test_time_subtraction(self):
- Time.objects.create(time=datetime.time(12, 30, 15, 2345))
- queryset = Time.objects.annotate(
- difference=ExpressionWrapper(
- F('time') - Value(datetime.time(11, 15, 0), output_field=models.TimeField()),
- output_field=models.DurationField(),
- )
- )
- self.assertEqual(
- queryset.get().difference,
- datetime.timedelta(hours=1, minutes=15, seconds=15, microseconds=2345)
- )
- queryset = Time.objects.annotate(difference=ExpressionWrapper(
- F('time') - Value(None, output_field=models.TimeField()),
- output_field=models.DurationField(),
- ))
- self.assertIsNone(queryset.first().difference)
- queryset = Time.objects.annotate(shifted=ExpressionWrapper(
- F('time') - Value(None, output_field=models.DurationField()),
- output_field=models.TimeField(),
- ))
- self.assertIsNone(queryset.first().shifted)
- @skipUnlessDBFeature('supports_temporal_subtraction')
- def test_datetime_subtraction(self):
- under_estimate = [
- e.name for e in Experiment.objects.filter(estimated_time__gt=F('end') - F('start'))
- ]
- self.assertEqual(under_estimate, ['e2'])
- over_estimate = [
- e.name for e in Experiment.objects.filter(estimated_time__lt=F('end') - F('start'))
- ]
- self.assertEqual(over_estimate, ['e4'])
- queryset = Experiment.objects.annotate(difference=ExpressionWrapper(
- F('start') - Value(None, output_field=models.DateTimeField()),
- output_field=models.DurationField(),
- ))
- self.assertIsNone(queryset.first().difference)
- queryset = Experiment.objects.annotate(shifted=ExpressionWrapper(
- F('start') - Value(None, output_field=models.DurationField()),
- output_field=models.DateTimeField(),
- ))
- self.assertIsNone(queryset.first().shifted)
- @skipUnlessDBFeature('supports_temporal_subtraction')
- def test_datetime_subtraction_microseconds(self):
- delta = datetime.timedelta(microseconds=8999999999999999)
- Experiment.objects.update(end=F('start') + delta)
- qs = Experiment.objects.annotate(
- delta=ExpressionWrapper(F('end') - F('start'), output_field=models.DurationField())
- )
- for e in qs:
- self.assertEqual(e.delta, delta)
- def test_duration_with_datetime(self):
-
-
-
- over_estimate = Experiment.objects.exclude(name='e1').filter(
- completed__gt=self.stime + F('estimated_time'),
- ).order_by('name')
- self.assertQuerysetEqual(over_estimate, ['e3', 'e4', 'e5'], lambda e: e.name)
- def test_duration_with_datetime_microseconds(self):
- delta = datetime.timedelta(microseconds=8999999999999999)
- qs = Experiment.objects.annotate(dt=ExpressionWrapper(
- F('start') + delta,
- output_field=models.DateTimeField(),
- ))
- for e in qs:
- self.assertEqual(e.dt, e.start + delta)
- def test_date_minus_duration(self):
- more_than_4_days = Experiment.objects.filter(
- assigned__lt=F('completed') - Value(datetime.timedelta(days=4), output_field=models.DurationField())
- )
- self.assertQuerysetEqual(more_than_4_days, ['e3', 'e4', 'e5'], lambda e: e.name)
- def test_negative_timedelta_update(self):
-
- experiments = Experiment.objects.filter(name='e0').annotate(
- start_sub_seconds=F('start') + datetime.timedelta(seconds=-30),
- ).annotate(
- start_sub_minutes=F('start_sub_seconds') + datetime.timedelta(minutes=-30),
- ).annotate(
- start_sub_hours=F('start_sub_minutes') + datetime.timedelta(hours=-2),
- ).annotate(
- new_start=F('start_sub_hours') + datetime.timedelta(days=-2),
- )
- expected_start = datetime.datetime(2010, 6, 23, 9, 45, 0)
-
- experiments = experiments.annotate(new_start=F('new_start') + datetime.timedelta(microseconds=-30))
- expected_start += datetime.timedelta(microseconds=+746970)
- experiments.update(start=F('new_start'))
- e0 = Experiment.objects.get(name='e0')
- self.assertEqual(e0.start, expected_start)
- class ValueTests(TestCase):
- def test_update_TimeField_using_Value(self):
- Time.objects.create()
- Time.objects.update(time=Value(datetime.time(1), output_field=TimeField()))
- self.assertEqual(Time.objects.get().time, datetime.time(1))
- def test_update_UUIDField_using_Value(self):
- UUID.objects.create()
- UUID.objects.update(uuid=Value(uuid.UUID('12345678901234567890123456789012'), output_field=UUIDField()))
- self.assertEqual(UUID.objects.get().uuid, uuid.UUID('12345678901234567890123456789012'))
- def test_deconstruct(self):
- value = Value('name')
- path, args, kwargs = value.deconstruct()
- self.assertEqual(path, 'django.db.models.expressions.Value')
- self.assertEqual(args, (value.value,))
- self.assertEqual(kwargs, {})
- def test_deconstruct_output_field(self):
- value = Value('name', output_field=CharField())
- path, args, kwargs = value.deconstruct()
- self.assertEqual(path, 'django.db.models.expressions.Value')
- self.assertEqual(args, (value.value,))
- self.assertEqual(len(kwargs), 1)
- self.assertEqual(kwargs['output_field'].deconstruct(), CharField().deconstruct())
- def test_equal(self):
- value = Value('name')
- self.assertEqual(value, Value('name'))
- self.assertNotEqual(value, Value('username'))
- def test_hash(self):
- d = {Value('name'): 'Bob'}
- self.assertIn(Value('name'), d)
- self.assertEqual(d[Value('name')], 'Bob')
- def test_equal_output_field(self):
- value = Value('name', output_field=CharField())
- same_value = Value('name', output_field=CharField())
- other_value = Value('name', output_field=TimeField())
- no_output_field = Value('name')
- self.assertEqual(value, same_value)
- self.assertNotEqual(value, other_value)
- self.assertNotEqual(value, no_output_field)
- def test_raise_empty_expressionlist(self):
- msg = 'ExpressionList requires at least one expression'
- with self.assertRaisesMessage(ValueError, msg):
- ExpressionList()
- class FieldTransformTests(TestCase):
- @classmethod
- def setUpTestData(cls):
- cls.sday = sday = datetime.date(2010, 6, 25)
- cls.stime = stime = datetime.datetime(2010, 6, 25, 12, 15, 30, 747000)
- cls.ex1 = Experiment.objects.create(
- name='Experiment 1',
- assigned=sday,
- completed=sday + datetime.timedelta(2),
- estimated_time=datetime.timedelta(2),
- start=stime,
- end=stime + datetime.timedelta(2),
- )
- def test_month_aggregation(self):
- self.assertEqual(
- Experiment.objects.aggregate(month_count=Count('assigned__month')),
- {'month_count': 1}
- )
- def test_transform_in_values(self):
- self.assertQuerysetEqual(
- Experiment.objects.values('assigned__month'),
- ["{'assigned__month': 6}"]
- )
- def test_multiple_transforms_in_values(self):
- self.assertQuerysetEqual(
- Experiment.objects.values('end__date__month'),
- ["{'end__date__month': 6}"]
- )
- class ReprTests(SimpleTestCase):
- def test_expressions(self):
- self.assertEqual(
- repr(Case(When(a=1))),
- "<Case: CASE WHEN <Q: (AND: ('a', 1))> THEN Value(None), ELSE Value(None)>"
- )
- self.assertEqual(
- repr(When(Q(age__gte=18), then=Value('legal'))),
- "<When: WHEN <Q: (AND: ('age__gte', 18))> THEN Value(legal)>"
- )
- self.assertEqual(repr(Col('alias', 'field')), "Col(alias, field)")
- self.assertEqual(repr(F('published')), "F(published)")
- self.assertEqual(repr(F('cost') + F('tax')), "<CombinedExpression: F(cost) + F(tax)>")
- self.assertEqual(
- repr(ExpressionWrapper(F('cost') + F('tax'), models.IntegerField())),
- "ExpressionWrapper(F(cost) + F(tax))"
- )
- self.assertEqual(repr(Func('published', function='TO_CHAR')), "Func(F(published), function=TO_CHAR)")
- self.assertEqual(repr(OrderBy(Value(1))), 'OrderBy(Value(1), descending=False)')
- self.assertEqual(repr(Random()), "Random()")
- self.assertEqual(repr(RawSQL('table.col', [])), "RawSQL(table.col, [])")
- self.assertEqual(repr(Ref('sum_cost', Sum('cost'))), "Ref(sum_cost, Sum(F(cost)))")
- self.assertEqual(repr(Value(1)), "Value(1)")
- self.assertEqual(
- repr(ExpressionList(F('col'), F('anothercol'))),
- 'ExpressionList(F(col), F(anothercol))'
- )
- self.assertEqual(
- repr(ExpressionList(OrderBy(F('col'), descending=False))),
- 'ExpressionList(OrderBy(F(col), descending=False))'
- )
- def test_functions(self):
- self.assertEqual(repr(Coalesce('a', 'b')), "Coalesce(F(a), F(b))")
- self.assertEqual(repr(Concat('a', 'b')), "Concat(ConcatPair(F(a), F(b)))")
- self.assertEqual(repr(Length('a')), "Length(F(a))")
- self.assertEqual(repr(Lower('a')), "Lower(F(a))")
- self.assertEqual(repr(Substr('a', 1, 3)), "Substr(F(a), Value(1), Value(3))")
- self.assertEqual(repr(Upper('a')), "Upper(F(a))")
- def test_aggregates(self):
- self.assertEqual(repr(Avg('a')), "Avg(F(a))")
- self.assertEqual(repr(Count('a')), "Count(F(a))")
- self.assertEqual(repr(Count('*')), "Count('*')")
- self.assertEqual(repr(Max('a')), "Max(F(a))")
- self.assertEqual(repr(Min('a')), "Min(F(a))")
- self.assertEqual(repr(StdDev('a')), "StdDev(F(a), sample=False)")
- self.assertEqual(repr(Sum('a')), "Sum(F(a))")
- self.assertEqual(repr(Variance('a', sample=True)), "Variance(F(a), sample=True)")
- def test_distinct_aggregates(self):
- self.assertEqual(repr(Count('a', distinct=True)), "Count(F(a), distinct=True)")
- self.assertEqual(repr(Count('*', distinct=True)), "Count('*', distinct=True)")
- def test_filtered_aggregates(self):
- filter = Q(a=1)
- self.assertEqual(repr(Avg('a', filter=filter)), "Avg(F(a), filter=(AND: ('a', 1)))")
- self.assertEqual(repr(Count('a', filter=filter)), "Count(F(a), filter=(AND: ('a', 1)))")
- self.assertEqual(repr(Max('a', filter=filter)), "Max(F(a), filter=(AND: ('a', 1)))")
- self.assertEqual(repr(Min('a', filter=filter)), "Min(F(a), filter=(AND: ('a', 1)))")
- self.assertEqual(repr(StdDev('a', filter=filter)), "StdDev(F(a), filter=(AND: ('a', 1)), sample=False)")
- self.assertEqual(repr(Sum('a', filter=filter)), "Sum(F(a), filter=(AND: ('a', 1)))")
- self.assertEqual(
- repr(Variance('a', sample=True, filter=filter)),
- "Variance(F(a), filter=(AND: ('a', 1)), sample=True)"
- )
- self.assertEqual(
- repr(Count('a', filter=filter, distinct=True)), "Count(F(a), distinct=True, filter=(AND: ('a', 1)))"
- )
- class CombinableTests(SimpleTestCase):
- bitwise_msg = 'Use .bitand() and .bitor() for bitwise logical operations.'
- def test_negation(self):
- c = Combinable()
- self.assertEqual(-c, c * -1)
- def test_and(self):
- with self.assertRaisesMessage(NotImplementedError, self.bitwise_msg):
- Combinable() & Combinable()
- def test_or(self):
- with self.assertRaisesMessage(NotImplementedError, self.bitwise_msg):
- Combinable() | Combinable()
- def test_reversed_and(self):
- with self.assertRaisesMessage(NotImplementedError, self.bitwise_msg):
- object() & Combinable()
- def test_reversed_or(self):
- with self.assertRaisesMessage(NotImplementedError, self.bitwise_msg):
- object() | Combinable()
|