1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796 |
- from __future__ import unicode_literals
- import datetime
- import pickle
- import unittest
- from collections import OrderedDict
- from operator import attrgetter
- from django.core.exceptions import FieldError
- from django.db import DEFAULT_DB_ALIAS, connection
- from django.db.models import F, Q, Count
- from django.db.models.sql.constants import LOUTER
- from django.db.models.sql.datastructures import EmptyResultSet
- from django.db.models.sql.where import NothingNode, WhereNode
- from django.test import TestCase, skipUnlessDBFeature
- from django.test.utils import CaptureQueriesContext
- from django.utils import six
- from django.utils.six.moves import range
- from .models import (
- FK1, X, Annotation, Article, Author, BaseA, Book, CategoryItem,
- CategoryRelationship, Celebrity, Channel, Chapter, Child, ChildObjectA,
- Classroom, Company, Cover, CustomPk, CustomPkTag, Detail, DumbCategory,
- Eaten, Employment, ExtraInfo, Fan, Food, Identifier, Individual, Item, Job,
- JobResponsibilities, Join, LeafA, LeafB, LoopX, LoopZ, ManagedModel,
- Member, ModelA, ModelB, ModelC, ModelD, MyObject, NamedCategory, Node,
- Note, NullableName, Number, ObjectA, ObjectB, ObjectC, OneToOneCategory,
- Order, OrderItem, Page, Paragraph, Person, Plaything, PointerA, Program,
- ProxyCategory, ProxyObjectA, ProxyObjectB, Ranking, Related,
- RelatedIndividual, RelatedObject, Report, ReservedName, Responsibility,
- School, SharedConnection, SimpleCategory, SingleObject, SpecialCategory,
- Staff, StaffUser, Student, Tag, Task, Ticket21203Child, Ticket21203Parent,
- Ticket23605A, Ticket23605B, Ticket23605C, TvChef, Valid,
- )
- class BaseQuerysetTest(TestCase):
- def assertValueQuerysetEqual(self, qs, values):
- return self.assertQuerysetEqual(qs, values, transform=lambda x: x)
- class Queries1Tests(BaseQuerysetTest):
- @classmethod
- def setUpTestData(cls):
- generic = NamedCategory.objects.create(name="Generic")
- cls.t1 = Tag.objects.create(name='t1', category=generic)
- cls.t2 = Tag.objects.create(name='t2', parent=cls.t1, category=generic)
- cls.t3 = Tag.objects.create(name='t3', parent=cls.t1)
- t4 = Tag.objects.create(name='t4', parent=cls.t3)
- cls.t5 = Tag.objects.create(name='t5', parent=cls.t3)
- cls.n1 = Note.objects.create(note='n1', misc='foo', id=1)
- n2 = Note.objects.create(note='n2', misc='bar', id=2)
- cls.n3 = Note.objects.create(note='n3', misc='foo', id=3)
- ann1 = Annotation.objects.create(name='a1', tag=cls.t1)
- ann1.notes.add(cls.n1)
- ann2 = Annotation.objects.create(name='a2', tag=t4)
- ann2.notes.add(n2, cls.n3)
- # Create these out of order so that sorting by 'id' will be different to sorting
- # by 'info'. Helps detect some problems later.
- cls.e2 = ExtraInfo.objects.create(info='e2', note=n2, value=41)
- e1 = ExtraInfo.objects.create(info='e1', note=cls.n1, value=42)
- cls.a1 = Author.objects.create(name='a1', num=1001, extra=e1)
- cls.a2 = Author.objects.create(name='a2', num=2002, extra=e1)
- a3 = Author.objects.create(name='a3', num=3003, extra=cls.e2)
- cls.a4 = Author.objects.create(name='a4', num=4004, extra=cls.e2)
- cls.time1 = datetime.datetime(2007, 12, 19, 22, 25, 0)
- cls.time2 = datetime.datetime(2007, 12, 19, 21, 0, 0)
- time3 = datetime.datetime(2007, 12, 20, 22, 25, 0)
- time4 = datetime.datetime(2007, 12, 20, 21, 0, 0)
- cls.i1 = Item.objects.create(name='one', created=cls.time1, modified=cls.time1, creator=cls.a1, note=cls.n3)
- cls.i1.tags = [cls.t1, cls.t2]
- cls.i2 = Item.objects.create(name='two', created=cls.time2, creator=cls.a2, note=n2)
- cls.i2.tags = [cls.t1, cls.t3]
- cls.i3 = Item.objects.create(name='three', created=time3, creator=cls.a2, note=cls.n3)
- i4 = Item.objects.create(name='four', created=time4, creator=cls.a4, note=cls.n3)
- i4.tags = [t4]
- cls.r1 = Report.objects.create(name='r1', creator=cls.a1)
- Report.objects.create(name='r2', creator=a3)
- Report.objects.create(name='r3')
- # Ordering by 'rank' gives us rank2, rank1, rank3. Ordering by the Meta.ordering
- # will be rank3, rank2, rank1.
- cls.rank1 = Ranking.objects.create(rank=2, author=cls.a2)
- Cover.objects.create(title="first", item=i4)
- Cover.objects.create(title="second", item=cls.i2)
- def test_subquery_condition(self):
- qs1 = Tag.objects.filter(pk__lte=0)
- qs2 = Tag.objects.filter(parent__in=qs1)
- qs3 = Tag.objects.filter(parent__in=qs2)
- self.assertEqual(qs3.query.subq_aliases, {'T', 'U', 'V'})
- self.assertIn('v0', str(qs3.query).lower())
- qs4 = qs3.filter(parent__in=qs1)
- self.assertEqual(qs4.query.subq_aliases, {'T', 'U', 'V'})
- # It is possible to reuse U for the second subquery, no need to use W.
- self.assertNotIn('w0', str(qs4.query).lower())
- # So, 'U0."id"' is referenced twice.
- self.assertTrue(str(qs4.query).lower().count('u0'), 2)
- def test_ticket1050(self):
- self.assertQuerysetEqual(
- Item.objects.filter(tags__isnull=True),
- ['<Item: three>']
- )
- self.assertQuerysetEqual(
- Item.objects.filter(tags__id__isnull=True),
- ['<Item: three>']
- )
- def test_ticket1801(self):
- self.assertQuerysetEqual(
- Author.objects.filter(item=self.i2),
- ['<Author: a2>']
- )
- self.assertQuerysetEqual(
- Author.objects.filter(item=self.i3),
- ['<Author: a2>']
- )
- self.assertQuerysetEqual(
- Author.objects.filter(item=self.i2) & Author.objects.filter(item=self.i3),
- ['<Author: a2>']
- )
- def test_ticket2306(self):
- # Checking that no join types are "left outer" joins.
- query = Item.objects.filter(tags=self.t2).query
- self.assertNotIn(LOUTER, [x.join_type for x in query.alias_map.values()])
- self.assertQuerysetEqual(
- Item.objects.filter(Q(tags=self.t1)).order_by('name'),
- ['<Item: one>', '<Item: two>']
- )
- self.assertQuerysetEqual(
- Item.objects.filter(Q(tags=self.t1)).filter(Q(tags=self.t2)),
- ['<Item: one>']
- )
- self.assertQuerysetEqual(
- Item.objects.filter(Q(tags=self.t1)).filter(Q(creator__name='fred') | Q(tags=self.t2)),
- ['<Item: one>']
- )
- # Each filter call is processed "at once" against a single table, so this is
- # different from the previous example as it tries to find tags that are two
- # things at once (rather than two tags).
- self.assertQuerysetEqual(
- Item.objects.filter(Q(tags=self.t1) & Q(tags=self.t2)),
- []
- )
- self.assertQuerysetEqual(
- Item.objects.filter(Q(tags=self.t1), Q(creator__name='fred') | Q(tags=self.t2)),
- []
- )
- qs = Author.objects.filter(ranking__rank=2, ranking__id=self.rank1.id)
- self.assertQuerysetEqual(list(qs), ['<Author: a2>'])
- self.assertEqual(2, qs.query.count_active_tables(), 2)
- qs = Author.objects.filter(ranking__rank=2).filter(ranking__id=self.rank1.id)
- self.assertEqual(qs.query.count_active_tables(), 3)
- def test_ticket4464(self):
- self.assertQuerysetEqual(
- Item.objects.filter(tags=self.t1).filter(tags=self.t2),
- ['<Item: one>']
- )
- self.assertQuerysetEqual(
- Item.objects.filter(tags__in=[self.t1, self.t2]).distinct().order_by('name'),
- ['<Item: one>', '<Item: two>']
- )
- self.assertQuerysetEqual(
- Item.objects.filter(tags__in=[self.t1, self.t2]).filter(tags=self.t3),
- ['<Item: two>']
- )
- # Make sure .distinct() works with slicing (this was broken in Oracle).
- self.assertQuerysetEqual(
- Item.objects.filter(tags__in=[self.t1, self.t2]).order_by('name')[:3],
- ['<Item: one>', '<Item: one>', '<Item: two>']
- )
- self.assertQuerysetEqual(
- Item.objects.filter(tags__in=[self.t1, self.t2]).distinct().order_by('name')[:3],
- ['<Item: one>', '<Item: two>']
- )
- def test_tickets_2080_3592(self):
- self.assertQuerysetEqual(
- Author.objects.filter(item__name='one') | Author.objects.filter(name='a3'),
- ['<Author: a1>', '<Author: a3>']
- )
- self.assertQuerysetEqual(
- Author.objects.filter(Q(item__name='one') | Q(name='a3')),
- ['<Author: a1>', '<Author: a3>']
- )
- self.assertQuerysetEqual(
- Author.objects.filter(Q(name='a3') | Q(item__name='one')),
- ['<Author: a1>', '<Author: a3>']
- )
- self.assertQuerysetEqual(
- Author.objects.filter(Q(item__name='three') | Q(report__name='r3')),
- ['<Author: a2>']
- )
- def test_ticket6074(self):
- # Merging two empty result sets shouldn't leave a queryset with no constraints
- # (which would match everything).
- self.assertQuerysetEqual(Author.objects.filter(Q(id__in=[])), [])
- self.assertQuerysetEqual(
- Author.objects.filter(Q(id__in=[]) | Q(id__in=[])),
- []
- )
- def test_tickets_1878_2939(self):
- self.assertEqual(Item.objects.values('creator').distinct().count(), 3)
- # Create something with a duplicate 'name' so that we can test multi-column
- # cases (which require some tricky SQL transformations under the covers).
- xx = Item(name='four', created=self.time1, creator=self.a2, note=self.n1)
- xx.save()
- self.assertEqual(
- Item.objects.exclude(name='two').values('creator', 'name').distinct().count(),
- 4
- )
- self.assertEqual(
- Item.objects.exclude(name='two').extra(select={'foo': '%s'}, select_params=(1,)).values('creator', 'name', 'foo').distinct().count(),
- 4
- )
- self.assertEqual(
- Item.objects.exclude(name='two').extra(select={'foo': '%s'}, select_params=(1,)).values('creator', 'name').distinct().count(),
- 4
- )
- xx.delete()
- def test_ticket7323(self):
- self.assertEqual(Item.objects.values('creator', 'name').count(), 4)
- def test_ticket2253(self):
- q1 = Item.objects.order_by('name')
- q2 = Item.objects.filter(id=self.i1.id)
- self.assertQuerysetEqual(
- q1,
- ['<Item: four>', '<Item: one>', '<Item: three>', '<Item: two>']
- )
- self.assertQuerysetEqual(q2, ['<Item: one>'])
- self.assertQuerysetEqual(
- (q1 | q2).order_by('name'),
- ['<Item: four>', '<Item: one>', '<Item: three>', '<Item: two>']
- )
- self.assertQuerysetEqual((q1 & q2).order_by('name'), ['<Item: one>'])
- q1 = Item.objects.filter(tags=self.t1)
- q2 = Item.objects.filter(note=self.n3, tags=self.t2)
- q3 = Item.objects.filter(creator=self.a4)
- self.assertQuerysetEqual(
- ((q1 & q2) | q3).order_by('name'),
- ['<Item: four>', '<Item: one>']
- )
- def test_order_by_tables(self):
- q1 = Item.objects.order_by('name')
- q2 = Item.objects.filter(id=self.i1.id)
- list(q2)
- combined_query = (q1 & q2).order_by('name').query
- self.assertEqual(len([
- t for t in combined_query.tables if combined_query.alias_refcount[t]
- ]), 1)
- def test_order_by_join_unref(self):
- """
- This test is related to the above one, testing that there aren't
- old JOINs in the query.
- """
- qs = Celebrity.objects.order_by('greatest_fan__fan_of')
- self.assertIn('OUTER JOIN', str(qs.query))
- qs = qs.order_by('id')
- self.assertNotIn('OUTER JOIN', str(qs.query))
- def test_get_clears_ordering(self):
- """
- get() should clear ordering for optimization purposes.
- """
- with CaptureQueriesContext(connection) as captured_queries:
- Author.objects.order_by('name').get(pk=self.a1.pk)
- self.assertNotIn('order by', captured_queries[0]['sql'].lower())
- def test_tickets_4088_4306(self):
- self.assertQuerysetEqual(
- Report.objects.filter(creator=1001),
- ['<Report: r1>']
- )
- self.assertQuerysetEqual(
- Report.objects.filter(creator__num=1001),
- ['<Report: r1>']
- )
- self.assertQuerysetEqual(Report.objects.filter(creator__id=1001), [])
- self.assertQuerysetEqual(
- Report.objects.filter(creator__id=self.a1.id),
- ['<Report: r1>']
- )
- self.assertQuerysetEqual(
- Report.objects.filter(creator__name='a1'),
- ['<Report: r1>']
- )
- def test_ticket4510(self):
- self.assertQuerysetEqual(
- Author.objects.filter(report__name='r1'),
- ['<Author: a1>']
- )
- def test_ticket7378(self):
- self.assertQuerysetEqual(self.a1.report_set.all(), ['<Report: r1>'])
- def test_tickets_5324_6704(self):
- self.assertQuerysetEqual(
- Item.objects.filter(tags__name='t4'),
- ['<Item: four>']
- )
- self.assertQuerysetEqual(
- Item.objects.exclude(tags__name='t4').order_by('name').distinct(),
- ['<Item: one>', '<Item: three>', '<Item: two>']
- )
- self.assertQuerysetEqual(
- Item.objects.exclude(tags__name='t4').order_by('name').distinct().reverse(),
- ['<Item: two>', '<Item: three>', '<Item: one>']
- )
- self.assertQuerysetEqual(
- Author.objects.exclude(item__name='one').distinct().order_by('name'),
- ['<Author: a2>', '<Author: a3>', '<Author: a4>']
- )
- # Excluding across a m2m relation when there is more than one related
- # object associated was problematic.
- self.assertQuerysetEqual(
- Item.objects.exclude(tags__name='t1').order_by('name'),
- ['<Item: four>', '<Item: three>']
- )
- self.assertQuerysetEqual(
- Item.objects.exclude(tags__name='t1').exclude(tags__name='t4'),
- ['<Item: three>']
- )
- # Excluding from a relation that cannot be NULL should not use outer joins.
- query = Item.objects.exclude(creator__in=[self.a1, self.a2]).query
- self.assertNotIn(LOUTER, [x.join_type for x in query.alias_map.values()])
- # Similarly, when one of the joins cannot possibly, ever, involve NULL
- # values (Author -> ExtraInfo, in the following), it should never be
- # promoted to a left outer join. So the following query should only
- # involve one "left outer" join (Author -> Item is 0-to-many).
- qs = Author.objects.filter(id=self.a1.id).filter(Q(extra__note=self.n1) | Q(item__note=self.n3))
- self.assertEqual(
- len([x for x in qs.query.alias_map.values() if x.join_type == LOUTER and qs.query.alias_refcount[x.table_alias]]),
- 1
- )
- # The previous changes shouldn't affect nullable foreign key joins.
- self.assertQuerysetEqual(
- Tag.objects.filter(parent__isnull=True).order_by('name'),
- ['<Tag: t1>']
- )
- self.assertQuerysetEqual(
- Tag.objects.exclude(parent__isnull=True).order_by('name'),
- ['<Tag: t2>', '<Tag: t3>', '<Tag: t4>', '<Tag: t5>']
- )
- self.assertQuerysetEqual(
- Tag.objects.exclude(Q(parent__name='t1') | Q(parent__isnull=True)).order_by('name'),
- ['<Tag: t4>', '<Tag: t5>']
- )
- self.assertQuerysetEqual(
- Tag.objects.exclude(Q(parent__isnull=True) | Q(parent__name='t1')).order_by('name'),
- ['<Tag: t4>', '<Tag: t5>']
- )
- self.assertQuerysetEqual(
- Tag.objects.exclude(Q(parent__parent__isnull=True)).order_by('name'),
- ['<Tag: t4>', '<Tag: t5>']
- )
- self.assertQuerysetEqual(
- Tag.objects.filter(~Q(parent__parent__isnull=True)).order_by('name'),
- ['<Tag: t4>', '<Tag: t5>']
- )
- def test_ticket2091(self):
- t = Tag.objects.get(name='t4')
- self.assertQuerysetEqual(
- Item.objects.filter(tags__in=[t]),
- ['<Item: four>']
- )
- def test_avoid_infinite_loop_on_too_many_subqueries(self):
- x = Tag.objects.filter(pk=1)
- local_recursion_limit = 127
- msg = 'Maximum recursion depth exceeded: too many subqueries.'
- with self.assertRaisesMessage(RuntimeError, msg):
- for i in six.moves.range(local_recursion_limit * 2):
- x = Tag.objects.filter(pk__in=x)
- def test_reasonable_number_of_subq_aliases(self):
- x = Tag.objects.filter(pk=1)
- for _ in range(20):
- x = Tag.objects.filter(pk__in=x)
- self.assertEqual(
- x.query.subq_aliases, {
- 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'AA', 'AB', 'AC', 'AD',
- 'AE', 'AF', 'AG', 'AH', 'AI', 'AJ', 'AK', 'AL', 'AM', 'AN',
- }
- )
- def test_heterogeneous_qs_combination(self):
- # Combining querysets built on different models should behave in a well-defined
- # fashion. We raise an error.
- self.assertRaisesMessage(
- AssertionError,
- 'Cannot combine queries on two different base models.',
- lambda: Author.objects.all() & Tag.objects.all()
- )
- self.assertRaisesMessage(
- AssertionError,
- 'Cannot combine queries on two different base models.',
- lambda: Author.objects.all() | Tag.objects.all()
- )
- def test_ticket3141(self):
- self.assertEqual(Author.objects.extra(select={'foo': '1'}).count(), 4)
- self.assertEqual(
- Author.objects.extra(select={'foo': '%s'}, select_params=(1,)).count(),
- 4
- )
- def test_ticket2400(self):
- self.assertQuerysetEqual(
- Author.objects.filter(item__isnull=True),
- ['<Author: a3>']
- )
- self.assertQuerysetEqual(
- Tag.objects.filter(item__isnull=True),
- ['<Tag: t5>']
- )
- def test_ticket2496(self):
- self.assertQuerysetEqual(
- Item.objects.extra(tables=['queries_author']).select_related().order_by('name')[:1],
- ['<Item: four>']
- )
- def test_error_raised_on_filter_with_dictionary(self):
- with self.assertRaisesMessage(FieldError, 'Cannot parse keyword query as dict'):
- Note.objects.filter({'note': 'n1', 'misc': 'foo'})
- def test_tickets_2076_7256(self):
- # Ordering on related tables should be possible, even if the table is
- # not otherwise involved.
- self.assertQuerysetEqual(
- Item.objects.order_by('note__note', 'name'),
- ['<Item: two>', '<Item: four>', '<Item: one>', '<Item: three>']
- )
- # Ordering on a related field should use the remote model's default
- # ordering as a final step.
- self.assertQuerysetEqual(
- Author.objects.order_by('extra', '-name'),
- ['<Author: a2>', '<Author: a1>', '<Author: a4>', '<Author: a3>']
- )
- # Using remote model default ordering can span multiple models (in this
- # case, Cover is ordered by Item's default, which uses Note's default).
- self.assertQuerysetEqual(
- Cover.objects.all(),
- ['<Cover: first>', '<Cover: second>']
- )
- # If the remote model does not have a default ordering, we order by its 'id'
- # field.
- self.assertQuerysetEqual(
- Item.objects.order_by('creator', 'name'),
- ['<Item: one>', '<Item: three>', '<Item: two>', '<Item: four>']
- )
- # Ordering by a many-valued attribute (e.g. a many-to-many or reverse
- # ForeignKey) is legal, but the results might not make sense. That
- # isn't Django's problem. Garbage in, garbage out.
- self.assertQuerysetEqual(
- Item.objects.filter(tags__isnull=False).order_by('tags', 'id'),
- ['<Item: one>', '<Item: two>', '<Item: one>', '<Item: two>', '<Item: four>']
- )
- # If we replace the default ordering, Django adjusts the required
- # tables automatically. Item normally requires a join with Note to do
- # the default ordering, but that isn't needed here.
- qs = Item.objects.order_by('name')
- self.assertQuerysetEqual(
- qs,
- ['<Item: four>', '<Item: one>', '<Item: three>', '<Item: two>']
- )
- self.assertEqual(len(qs.query.tables), 1)
- def test_tickets_2874_3002(self):
- qs = Item.objects.select_related().order_by('note__note', 'name')
- self.assertQuerysetEqual(
- qs,
- ['<Item: two>', '<Item: four>', '<Item: one>', '<Item: three>']
- )
- # This is also a good select_related() test because there are multiple
- # Note entries in the SQL. The two Note items should be different.
- self.assertTrue(repr(qs[0].note), '<Note: n2>')
- self.assertEqual(repr(qs[0].creator.extra.note), '<Note: n1>')
- def test_ticket3037(self):
- self.assertQuerysetEqual(
- Item.objects.filter(Q(creator__name='a3', name='two') | Q(creator__name='a4', name='four')),
- ['<Item: four>']
- )
- def test_tickets_5321_7070(self):
- # Ordering columns must be included in the output columns. Note that
- # this means results that might otherwise be distinct are not (if there
- # are multiple values in the ordering cols), as in this example. This
- # isn't a bug; it's a warning to be careful with the selection of
- # ordering columns.
- self.assertValueQuerysetEqual(
- Note.objects.values('misc').distinct().order_by('note', '-misc'),
- [{'misc': 'foo'}, {'misc': 'bar'}, {'misc': 'foo'}]
- )
- def test_ticket4358(self):
- # If you don't pass any fields to values(), relation fields are
- # returned as "foo_id" keys, not "foo". For consistency, you should be
- # able to pass "foo_id" in the fields list and have it work, too. We
- # actually allow both "foo" and "foo_id".
- # The *_id version is returned by default.
- self.assertIn('note_id', ExtraInfo.objects.values()[0])
- # You can also pass it in explicitly.
- self.assertValueQuerysetEqual(
- ExtraInfo.objects.values('note_id'),
- [{'note_id': 1}, {'note_id': 2}]
- )
- # ...or use the field name.
- self.assertValueQuerysetEqual(
- ExtraInfo.objects.values('note'),
- [{'note': 1}, {'note': 2}]
- )
- def test_ticket2902(self):
- # Parameters can be given to extra_select, *if* you use an OrderedDict.
- # (First we need to know which order the keys fall in "naturally" on
- # your system, so we can put things in the wrong way around from
- # normal. A normal dict would thus fail.)
- s = [('a', '%s'), ('b', '%s')]
- params = ['one', 'two']
- if {'a': 1, 'b': 2}.keys() == ['a', 'b']:
- s.reverse()
- params.reverse()
- # This slightly odd comparison works around the fact that PostgreSQL will
- # return 'one' and 'two' as strings, not Unicode objects. It's a side-effect of
- # using constants here and not a real concern.
- d = Item.objects.extra(select=OrderedDict(s), select_params=params).values('a', 'b')[0]
- self.assertEqual(d, {'a': 'one', 'b': 'two'})
- # Order by the number of tags attached to an item.
- l = Item.objects.extra(select={'count': 'select count(*) from queries_item_tags where queries_item_tags.item_id = queries_item.id'}).order_by('-count')
- self.assertEqual([o.count for o in l], [2, 2, 1, 0])
- def test_ticket6154(self):
- # Multiple filter statements are joined using "AND" all the time.
- self.assertQuerysetEqual(
- Author.objects.filter(id=self.a1.id).filter(Q(extra__note=self.n1) | Q(item__note=self.n3)),
- ['<Author: a1>']
- )
- self.assertQuerysetEqual(
- Author.objects.filter(Q(extra__note=self.n1) | Q(item__note=self.n3)).filter(id=self.a1.id),
- ['<Author: a1>']
- )
- def test_ticket6981(self):
- self.assertQuerysetEqual(
- Tag.objects.select_related('parent').order_by('name'),
- ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>', '<Tag: t4>', '<Tag: t5>']
- )
- def test_ticket9926(self):
- self.assertQuerysetEqual(
- Tag.objects.select_related("parent", "category").order_by('name'),
- ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>', '<Tag: t4>', '<Tag: t5>']
- )
- self.assertQuerysetEqual(
- Tag.objects.select_related('parent', "parent__category").order_by('name'),
- ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>', '<Tag: t4>', '<Tag: t5>']
- )
- def test_tickets_6180_6203(self):
- # Dates with limits and/or counts
- self.assertEqual(Item.objects.count(), 4)
- self.assertEqual(Item.objects.datetimes('created', 'month').count(), 1)
- self.assertEqual(Item.objects.datetimes('created', 'day').count(), 2)
- self.assertEqual(len(Item.objects.datetimes('created', 'day')), 2)
- self.assertEqual(Item.objects.datetimes('created', 'day')[0], datetime.datetime(2007, 12, 19, 0, 0))
- def test_tickets_7087_12242(self):
- # Dates with extra select columns
- self.assertQuerysetEqual(
- Item.objects.datetimes('created', 'day').extra(select={'a': 1}),
- ['datetime.datetime(2007, 12, 19, 0, 0)', 'datetime.datetime(2007, 12, 20, 0, 0)']
- )
- self.assertQuerysetEqual(
- Item.objects.extra(select={'a': 1}).datetimes('created', 'day'),
- ['datetime.datetime(2007, 12, 19, 0, 0)', 'datetime.datetime(2007, 12, 20, 0, 0)']
- )
- name = "one"
- self.assertQuerysetEqual(
- Item.objects.datetimes('created', 'day').extra(where=['name=%s'], params=[name]),
- ['datetime.datetime(2007, 12, 19, 0, 0)']
- )
- self.assertQuerysetEqual(
- Item.objects.extra(where=['name=%s'], params=[name]).datetimes('created', 'day'),
- ['datetime.datetime(2007, 12, 19, 0, 0)']
- )
- def test_ticket7155(self):
- # Nullable dates
- self.assertQuerysetEqual(
- Item.objects.datetimes('modified', 'day'),
- ['datetime.datetime(2007, 12, 19, 0, 0)']
- )
- def test_ticket7098(self):
- # Make sure semi-deprecated ordering by related models syntax still
- # works.
- self.assertValueQuerysetEqual(
- Item.objects.values('note__note').order_by('queries_note.note', 'id'),
- [{'note__note': 'n2'}, {'note__note': 'n3'}, {'note__note': 'n3'}, {'note__note': 'n3'}]
- )
- def test_ticket7096(self):
- # Make sure exclude() with multiple conditions continues to work.
- self.assertQuerysetEqual(
- Tag.objects.filter(parent=self.t1, name='t3').order_by('name'),
- ['<Tag: t3>']
- )
- self.assertQuerysetEqual(
- Tag.objects.exclude(parent=self.t1, name='t3').order_by('name'),
- ['<Tag: t1>', '<Tag: t2>', '<Tag: t4>', '<Tag: t5>']
- )
- self.assertQuerysetEqual(
- Item.objects.exclude(tags__name='t1', name='one').order_by('name').distinct(),
- ['<Item: four>', '<Item: three>', '<Item: two>']
- )
- self.assertQuerysetEqual(
- Item.objects.filter(name__in=['three', 'four']).exclude(tags__name='t1').order_by('name'),
- ['<Item: four>', '<Item: three>']
- )
- # More twisted cases, involving nested negations.
- self.assertQuerysetEqual(
- Item.objects.exclude(~Q(tags__name='t1', name='one')),
- ['<Item: one>']
- )
- self.assertQuerysetEqual(
- Item.objects.filter(~Q(tags__name='t1', name='one'), name='two'),
- ['<Item: two>']
- )
- self.assertQuerysetEqual(
- Item.objects.exclude(~Q(tags__name='t1', name='one'), name='two'),
- ['<Item: four>', '<Item: one>', '<Item: three>']
- )
- def test_tickets_7204_7506(self):
- # Make sure querysets with related fields can be pickled. If this
- # doesn't crash, it's a Good Thing.
- pickle.dumps(Item.objects.all())
- def test_ticket7813(self):
- # We should also be able to pickle things that use select_related().
- # The only tricky thing here is to ensure that we do the related
- # selections properly after unpickling.
- qs = Item.objects.select_related()
- query = qs.query.get_compiler(qs.db).as_sql()[0]
- query2 = pickle.loads(pickle.dumps(qs.query))
- self.assertEqual(
- query2.get_compiler(qs.db).as_sql()[0],
- query
- )
- def test_deferred_load_qs_pickling(self):
- # Check pickling of deferred-loading querysets
- qs = Item.objects.defer('name', 'creator')
- q2 = pickle.loads(pickle.dumps(qs))
- self.assertEqual(list(qs), list(q2))
- q3 = pickle.loads(pickle.dumps(qs, pickle.HIGHEST_PROTOCOL))
- self.assertEqual(list(qs), list(q3))
- def test_ticket7277(self):
- self.assertQuerysetEqual(
- self.n1.annotation_set.filter(Q(tag=self.t5) | Q(tag__children=self.t5) | Q(tag__children__children=self.t5)),
- ['<Annotation: a1>']
- )
- def test_tickets_7448_7707(self):
- # Complex objects should be converted to strings before being used in
- # lookups.
- self.assertQuerysetEqual(
- Item.objects.filter(created__in=[self.time1, self.time2]),
- ['<Item: one>', '<Item: two>']
- )
- def test_ticket7235(self):
- # An EmptyQuerySet should not raise exceptions if it is filtered.
- Eaten.objects.create(meal='m')
- q = Eaten.objects.none()
- with self.assertNumQueries(0):
- self.assertQuerysetEqual(q.all(), [])
- self.assertQuerysetEqual(q.filter(meal='m'), [])
- self.assertQuerysetEqual(q.exclude(meal='m'), [])
- self.assertQuerysetEqual(q.complex_filter({'pk': 1}), [])
- self.assertQuerysetEqual(q.select_related('food'), [])
- self.assertQuerysetEqual(q.annotate(Count('food')), [])
- self.assertQuerysetEqual(q.order_by('meal', 'food'), [])
- self.assertQuerysetEqual(q.distinct(), [])
- self.assertQuerysetEqual(
- q.extra(select={'foo': "1"}),
- []
- )
- q.query.low_mark = 1
- self.assertRaisesMessage(
- AssertionError,
- 'Cannot change a query once a slice has been taken',
- q.extra, select={'foo': "1"}
- )
- self.assertQuerysetEqual(q.reverse(), [])
- self.assertQuerysetEqual(q.defer('meal'), [])
- self.assertQuerysetEqual(q.only('meal'), [])
- def test_ticket7791(self):
- # There were "issues" when ordering and distinct-ing on fields related
- # via ForeignKeys.
- self.assertEqual(
- len(Note.objects.order_by('extrainfo__info').distinct()),
- 3
- )
- # Pickling of DateQuerySets used to fail
- qs = Item.objects.datetimes('created', 'month')
- pickle.loads(pickle.dumps(qs))
- def test_ticket9997(self):
- # If a ValuesList or Values queryset is passed as an inner query, we
- # make sure it's only requesting a single value and use that as the
- # thing to select.
- self.assertQuerysetEqual(
- Tag.objects.filter(name__in=Tag.objects.filter(parent=self.t1).values('name')),
- ['<Tag: t2>', '<Tag: t3>']
- )
- # Multi-valued values() and values_list() querysets should raise errors.
- self.assertRaisesMessage(
- TypeError,
- 'Cannot use multi-field values as a filter value.',
- lambda: Tag.objects.filter(name__in=Tag.objects.filter(parent=self.t1).values('name', 'id'))
- )
- self.assertRaisesMessage(
- TypeError,
- 'Cannot use multi-field values as a filter value.',
- lambda: Tag.objects.filter(name__in=Tag.objects.filter(parent=self.t1).values_list('name', 'id'))
- )
- def test_ticket9985(self):
- # qs.values_list(...).values(...) combinations should work.
- self.assertValueQuerysetEqual(
- Note.objects.values_list("note", flat=True).values("id").order_by("id"),
- [{'id': 1}, {'id': 2}, {'id': 3}]
- )
- self.assertQuerysetEqual(
- Annotation.objects.filter(notes__in=Note.objects.filter(note="n1").values_list('note').values('id')),
- ['<Annotation: a1>']
- )
- def test_ticket10205(self):
- # When bailing out early because of an empty "__in" filter, we need
- # to set things up correctly internally so that subqueries can continue properly.
- self.assertEqual(Tag.objects.filter(name__in=()).update(name="foo"), 0)
- def test_ticket10432(self):
- # Testing an empty "__in" filter with a generator as the value.
- def f():
- return iter([])
- n_obj = Note.objects.all()[0]
- def g():
- for i in [n_obj.pk]:
- yield i
- self.assertQuerysetEqual(Note.objects.filter(pk__in=f()), [])
- self.assertEqual(list(Note.objects.filter(pk__in=g())), [n_obj])
- def test_ticket10742(self):
- # Queries used in an __in clause don't execute subqueries
- subq = Author.objects.filter(num__lt=3000)
- qs = Author.objects.filter(pk__in=subq)
- self.assertQuerysetEqual(qs, ['<Author: a1>', '<Author: a2>'])
- # The subquery result cache should not be populated
- self.assertIsNone(subq._result_cache)
- subq = Author.objects.filter(num__lt=3000)
- qs = Author.objects.exclude(pk__in=subq)
- self.assertQuerysetEqual(qs, ['<Author: a3>', '<Author: a4>'])
- # The subquery result cache should not be populated
- self.assertIsNone(subq._result_cache)
- subq = Author.objects.filter(num__lt=3000)
- self.assertQuerysetEqual(
- Author.objects.filter(Q(pk__in=subq) & Q(name='a1')),
- ['<Author: a1>']
- )
- # The subquery result cache should not be populated
- self.assertIsNone(subq._result_cache)
- def test_ticket7076(self):
- # Excluding shouldn't eliminate NULL entries.
- self.assertQuerysetEqual(
- Item.objects.exclude(modified=self.time1).order_by('name'),
- ['<Item: four>', '<Item: three>', '<Item: two>']
- )
- self.assertQuerysetEqual(
- Tag.objects.exclude(parent__name=self.t1.name),
- ['<Tag: t1>', '<Tag: t4>', '<Tag: t5>']
- )
- def test_ticket7181(self):
- # Ordering by related tables should accommodate nullable fields (this
- # test is a little tricky, since NULL ordering is database dependent.
- # Instead, we just count the number of results).
- self.assertEqual(len(Tag.objects.order_by('parent__name')), 5)
- # Empty querysets can be merged with others.
- self.assertQuerysetEqual(
- Note.objects.none() | Note.objects.all(),
- ['<Note: n1>', '<Note: n2>', '<Note: n3>']
- )
- self.assertQuerysetEqual(
- Note.objects.all() | Note.objects.none(),
- ['<Note: n1>', '<Note: n2>', '<Note: n3>']
- )
- self.assertQuerysetEqual(Note.objects.none() & Note.objects.all(), [])
- self.assertQuerysetEqual(Note.objects.all() & Note.objects.none(), [])
- def test_ticket9411(self):
- # Make sure bump_prefix() (an internal Query method) doesn't (re-)break. It's
- # sufficient that this query runs without error.
- qs = Tag.objects.values_list('id', flat=True).order_by('id')
- qs.query.bump_prefix(qs.query)
- first = qs[0]
- self.assertEqual(list(qs), list(range(first, first + 5)))
- def test_ticket8439(self):
- # Complex combinations of conjunctions, disjunctions and nullable
- # relations.
- self.assertQuerysetEqual(
- Author.objects.filter(Q(item__note__extrainfo=self.e2) | Q(report=self.r1, name='xyz')),
- ['<Author: a2>']
- )
- self.assertQuerysetEqual(
- Author.objects.filter(Q(report=self.r1, name='xyz') | Q(item__note__extrainfo=self.e2)),
- ['<Author: a2>']
- )
- self.assertQuerysetEqual(
- Annotation.objects.filter(Q(tag__parent=self.t1) | Q(notes__note='n1', name='a1')),
- ['<Annotation: a1>']
- )
- xx = ExtraInfo.objects.create(info='xx', note=self.n3)
- self.assertQuerysetEqual(
- Note.objects.filter(Q(extrainfo__author=self.a1) | Q(extrainfo=xx)),
- ['<Note: n1>', '<Note: n3>']
- )
- q = Note.objects.filter(Q(extrainfo__author=self.a1) | Q(extrainfo=xx)).query
- self.assertEqual(
- len([x for x in q.alias_map.values() if x.join_type == LOUTER and q.alias_refcount[x.table_alias]]),
- 1
- )
- def test_ticket17429(self):
- """
- Ensure that Meta.ordering=None works the same as Meta.ordering=[]
- """
- original_ordering = Tag._meta.ordering
- Tag._meta.ordering = None
- try:
- self.assertQuerysetEqual(
- Tag.objects.all(),
- ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>', '<Tag: t4>', '<Tag: t5>'],
- ordered=False
- )
- finally:
- Tag._meta.ordering = original_ordering
- def test_exclude(self):
- self.assertQuerysetEqual(
- Item.objects.exclude(tags__name='t4'),
- [repr(i) for i in Item.objects.filter(~Q(tags__name='t4'))])
- self.assertQuerysetEqual(
- Item.objects.exclude(Q(tags__name='t4') | Q(tags__name='t3')),
- [repr(i) for i in Item.objects.filter(~(Q(tags__name='t4') | Q(tags__name='t3')))])
- self.assertQuerysetEqual(
- Item.objects.exclude(Q(tags__name='t4') | ~Q(tags__name='t3')),
- [repr(i) for i in Item.objects.filter(~(Q(tags__name='t4') | ~Q(tags__name='t3')))])
- def test_nested_exclude(self):
- self.assertQuerysetEqual(
- Item.objects.exclude(~Q(tags__name='t4')),
- [repr(i) for i in Item.objects.filter(~~Q(tags__name='t4'))])
- def test_double_exclude(self):
- self.assertQuerysetEqual(
- Item.objects.filter(Q(tags__name='t4')),
- [repr(i) for i in Item.objects.filter(~~Q(tags__name='t4'))])
- self.assertQuerysetEqual(
- Item.objects.filter(Q(tags__name='t4')),
- [repr(i) for i in Item.objects.filter(~Q(~Q(tags__name='t4')))])
- def test_exclude_in(self):
- self.assertQuerysetEqual(
- Item.objects.exclude(Q(tags__name__in=['t4', 't3'])),
- [repr(i) for i in Item.objects.filter(~Q(tags__name__in=['t4', 't3']))])
- self.assertQuerysetEqual(
- Item.objects.filter(Q(tags__name__in=['t4', 't3'])),
- [repr(i) for i in Item.objects.filter(~~Q(tags__name__in=['t4', 't3']))])
- def test_ticket_10790_1(self):
- # Querying direct fields with isnull should trim the left outer join.
- # It also should not create INNER JOIN.
- q = Tag.objects.filter(parent__isnull=True)
- self.assertQuerysetEqual(q, ['<Tag: t1>'])
- self.assertNotIn('JOIN', str(q.query))
- q = Tag.objects.filter(parent__isnull=False)
- self.assertQuerysetEqual(
- q,
- ['<Tag: t2>', '<Tag: t3>', '<Tag: t4>', '<Tag: t5>'],
- )
- self.assertNotIn('JOIN', str(q.query))
- q = Tag.objects.exclude(parent__isnull=True)
- self.assertQuerysetEqual(
- q,
- ['<Tag: t2>', '<Tag: t3>', '<Tag: t4>', '<Tag: t5>'],
- )
- self.assertNotIn('JOIN', str(q.query))
- q = Tag.objects.exclude(parent__isnull=False)
- self.assertQuerysetEqual(q, ['<Tag: t1>'])
- self.assertNotIn('JOIN', str(q.query))
- q = Tag.objects.exclude(parent__parent__isnull=False)
- self.assertQuerysetEqual(
- q,
- ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>'],
- )
- self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 1)
- self.assertNotIn('INNER JOIN', str(q.query))
- def test_ticket_10790_2(self):
- # Querying across several tables should strip only the last outer join,
- # while preserving the preceding inner joins.
- q = Tag.objects.filter(parent__parent__isnull=False)
- self.assertQuerysetEqual(
- q,
- ['<Tag: t4>', '<Tag: t5>'],
- )
- self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 0)
- self.assertEqual(str(q.query).count('INNER JOIN'), 1)
- # Querying without isnull should not convert anything to left outer join.
- q = Tag.objects.filter(parent__parent=self.t1)
- self.assertQuerysetEqual(
- q,
- ['<Tag: t4>', '<Tag: t5>'],
- )
- self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 0)
- self.assertEqual(str(q.query).count('INNER JOIN'), 1)
- def test_ticket_10790_3(self):
- # Querying via indirect fields should populate the left outer join
- q = NamedCategory.objects.filter(tag__isnull=True)
- self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 1)
- # join to dumbcategory ptr_id
- self.assertEqual(str(q.query).count('INNER JOIN'), 1)
- self.assertQuerysetEqual(q, [])
- # Querying across several tables should strip only the last join, while
- # preserving the preceding left outer joins.
- q = NamedCategory.objects.filter(tag__parent__isnull=True)
- self.assertEqual(str(q.query).count('INNER JOIN'), 1)
- self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 1)
- self.assertQuerysetEqual(q, ['<NamedCategory: Generic>'])
- def test_ticket_10790_4(self):
- # Querying across m2m field should not strip the m2m table from join.
- q = Author.objects.filter(item__tags__isnull=True)
- self.assertQuerysetEqual(
- q,
- ['<Author: a2>', '<Author: a3>'],
- )
- self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 2)
- self.assertNotIn('INNER JOIN', str(q.query))
- q = Author.objects.filter(item__tags__parent__isnull=True)
- self.assertQuerysetEqual(
- q,
- ['<Author: a1>', '<Author: a2>', '<Author: a2>', '<Author: a3>'],
- )
- self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 3)
- self.assertNotIn('INNER JOIN', str(q.query))
- def test_ticket_10790_5(self):
- # Querying with isnull=False across m2m field should not create outer joins
- q = Author.objects.filter(item__tags__isnull=False)
- self.assertQuerysetEqual(
- q,
- ['<Author: a1>', '<Author: a1>', '<Author: a2>', '<Author: a2>', '<Author: a4>']
- )
- self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 0)
- self.assertEqual(str(q.query).count('INNER JOIN'), 2)
- q = Author.objects.filter(item__tags__parent__isnull=False)
- self.assertQuerysetEqual(
- q,
- ['<Author: a1>', '<Author: a2>', '<Author: a4>']
- )
- self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 0)
- self.assertEqual(str(q.query).count('INNER JOIN'), 3)
- q = Author.objects.filter(item__tags__parent__parent__isnull=False)
- self.assertQuerysetEqual(
- q,
- ['<Author: a4>']
- )
- self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 0)
- self.assertEqual(str(q.query).count('INNER JOIN'), 4)
- def test_ticket_10790_6(self):
- # Querying with isnull=True across m2m field should not create inner joins
- # and strip last outer join
- q = Author.objects.filter(item__tags__parent__parent__isnull=True)
- self.assertQuerysetEqual(
- q,
- ['<Author: a1>', '<Author: a1>', '<Author: a2>', '<Author: a2>',
- '<Author: a2>', '<Author: a3>']
- )
- self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 4)
- self.assertEqual(str(q.query).count('INNER JOIN'), 0)
- q = Author.objects.filter(item__tags__parent__isnull=True)
- self.assertQuerysetEqual(
- q,
- ['<Author: a1>', '<Author: a2>', '<Author: a2>', '<Author: a3>']
- )
- self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 3)
- self.assertEqual(str(q.query).count('INNER JOIN'), 0)
- def test_ticket_10790_7(self):
- # Reverse querying with isnull should not strip the join
- q = Author.objects.filter(item__isnull=True)
- self.assertQuerysetEqual(
- q,
- ['<Author: a3>']
- )
- self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 1)
- self.assertEqual(str(q.query).count('INNER JOIN'), 0)
- q = Author.objects.filter(item__isnull=False)
- self.assertQuerysetEqual(
- q,
- ['<Author: a1>', '<Author: a2>', '<Author: a2>', '<Author: a4>']
- )
- self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 0)
- self.assertEqual(str(q.query).count('INNER JOIN'), 1)
- def test_ticket_10790_8(self):
- # Querying with combined q-objects should also strip the left outer join
- q = Tag.objects.filter(Q(parent__isnull=True) | Q(parent=self.t1))
- self.assertQuerysetEqual(
- q,
- ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>']
- )
- self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 0)
- self.assertEqual(str(q.query).count('INNER JOIN'), 0)
- def test_ticket_10790_combine(self):
- # Combining queries should not re-populate the left outer join
- q1 = Tag.objects.filter(parent__isnull=True)
- q2 = Tag.objects.filter(parent__isnull=False)
- q3 = q1 | q2
- self.assertQuerysetEqual(
- q3,
- ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>', '<Tag: t4>', '<Tag: t5>'],
- )
- self.assertEqual(str(q3.query).count('LEFT OUTER JOIN'), 0)
- self.assertEqual(str(q3.query).count('INNER JOIN'), 0)
- q3 = q1 & q2
- self.assertQuerysetEqual(q3, [])
- self.assertEqual(str(q3.query).count('LEFT OUTER JOIN'), 0)
- self.assertEqual(str(q3.query).count('INNER JOIN'), 0)
- q2 = Tag.objects.filter(parent=self.t1)
- q3 = q1 | q2
- self.assertQuerysetEqual(
- q3,
- ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>']
- )
- self.assertEqual(str(q3.query).count('LEFT OUTER JOIN'), 0)
- self.assertEqual(str(q3.query).count('INNER JOIN'), 0)
- q3 = q2 | q1
- self.assertQuerysetEqual(
- q3,
- ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>']
- )
- self.assertEqual(str(q3.query).count('LEFT OUTER JOIN'), 0)
- self.assertEqual(str(q3.query).count('INNER JOIN'), 0)
- q1 = Tag.objects.filter(parent__isnull=True)
- q2 = Tag.objects.filter(parent__parent__isnull=True)
- q3 = q1 | q2
- self.assertQuerysetEqual(
- q3,
- ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>']
- )
- self.assertEqual(str(q3.query).count('LEFT OUTER JOIN'), 1)
- self.assertEqual(str(q3.query).count('INNER JOIN'), 0)
- q3 = q2 | q1
- self.assertQuerysetEqual(
- q3,
- ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>']
- )
- self.assertEqual(str(q3.query).count('LEFT OUTER JOIN'), 1)
- self.assertEqual(str(q3.query).count('INNER JOIN'), 0)
- def test_ticket19672(self):
- self.assertQuerysetEqual(
- Report.objects.filter(Q(creator__isnull=False) &
- ~Q(creator__extra__value=41)),
- ['<Report: r1>']
- )
- def test_ticket_20250(self):
- # A negated Q along with an annotated queryset failed in Django 1.4
- qs = Author.objects.annotate(Count('item'))
- qs = qs.filter(~Q(extra__value=0))
- self.assertIn('SELECT', str(qs.query))
- self.assertQuerysetEqual(
- qs,
- ['<Author: a1>', '<Author: a2>', '<Author: a3>', '<Author: a4>']
- )
- def test_lookup_constraint_fielderror(self):
- msg = (
- "Cannot resolve keyword 'unknown_field' into field. Choices are: "
- "annotation, category, category_id, children, id, item, "
- "managedmodel, name, parent, parent_id"
- )
- with self.assertRaisesMessage(FieldError, msg):
- Tag.objects.filter(unknown_field__name='generic')
- class Queries2Tests(TestCase):
- @classmethod
- def setUpTestData(cls):
- Number.objects.create(num=4)
- Number.objects.create(num=8)
- Number.objects.create(num=12)
- def test_ticket4289(self):
- # A slight variation on the restricting the filtering choices by the
- # lookup constraints.
- self.assertQuerysetEqual(Number.objects.filter(num__lt=4), [])
- self.assertQuerysetEqual(Number.objects.filter(num__gt=8, num__lt=12), [])
- self.assertQuerysetEqual(
- Number.objects.filter(num__gt=8, num__lt=13),
- ['<Number: 12>']
- )
- self.assertQuerysetEqual(
- Number.objects.filter(Q(num__lt=4) | Q(num__gt=8, num__lt=12)),
- []
- )
- self.assertQuerysetEqual(
- Number.objects.filter(Q(num__gt=8, num__lt=12) | Q(num__lt=4)),
- []
- )
- self.assertQuerysetEqual(
- Number.objects.filter(Q(num__gt=8) & Q(num__lt=12) | Q(num__lt=4)),
- []
- )
- self.assertQuerysetEqual(
- Number.objects.filter(Q(num__gt=7) & Q(num__lt=12) | Q(num__lt=4)),
- ['<Number: 8>']
- )
- def test_ticket12239(self):
- # Float was being rounded to integer on gte queries on integer field. Tests
- # show that gt, lt, gte, and lte work as desired. Note that the fix changes
- # get_prep_lookup for gte and lt queries only.
- self.assertQuerysetEqual(
- Number.objects.filter(num__gt=11.9),
- ['<Number: 12>']
- )
- self.assertQuerysetEqual(Number.objects.filter(num__gt=12), [])
- self.assertQuerysetEqual(Number.objects.filter(num__gt=12.0), [])
- self.assertQuerysetEqual(Number.objects.filter(num__gt=12.1), [])
- self.assertQuerysetEqual(
- Number.objects.filter(num__lt=12),
- ['<Number: 4>', '<Number: 8>'],
- ordered=False
- )
- self.assertQuerysetEqual(
- Number.objects.filter(num__lt=12.0),
- ['<Number: 4>', '<Number: 8>'],
- ordered=False
- )
- self.assertQuerysetEqual(
- Number.objects.filter(num__lt=12.1),
- ['<Number: 4>', '<Number: 8>', '<Number: 12>'],
- ordered=False
- )
- self.assertQuerysetEqual(
- Number.objects.filter(num__gte=11.9),
- ['<Number: 12>']
- )
- self.assertQuerysetEqual(
- Number.objects.filter(num__gte=12),
- ['<Number: 12>']
- )
- self.assertQuerysetEqual(
- Number.objects.filter(num__gte=12.0),
- ['<Number: 12>']
- )
- self.assertQuerysetEqual(Number.objects.filter(num__gte=12.1), [])
- self.assertQuerysetEqual(Number.objects.filter(num__gte=12.9), [])
- self.assertQuerysetEqual(
- Number.objects.filter(num__lte=11.9),
- ['<Number: 4>', '<Number: 8>'],
- ordered=False
- )
- self.assertQuerysetEqual(
- Number.objects.filter(num__lte=12),
- ['<Number: 4>', '<Number: 8>', '<Number: 12>'],
- ordered=False
- )
- self.assertQuerysetEqual(
- Number.objects.filter(num__lte=12.0),
- ['<Number: 4>', '<Number: 8>', '<Number: 12>'],
- ordered=False
- )
- self.assertQuerysetEqual(
- Number.objects.filter(num__lte=12.1),
- ['<Number: 4>', '<Number: 8>', '<Number: 12>'],
- ordered=False
- )
- self.assertQuerysetEqual(
- Number.objects.filter(num__lte=12.9),
- ['<Number: 4>', '<Number: 8>', '<Number: 12>'],
- ordered=False
- )
- def test_ticket7759(self):
- # Count should work with a partially read result set.
- count = Number.objects.count()
- qs = Number.objects.all()
- def run():
- for obj in qs:
- return qs.count() == count
- self.assertTrue(run())
- class Queries3Tests(BaseQuerysetTest):
- def test_ticket7107(self):
- # This shouldn't create an infinite loop.
- self.assertQuerysetEqual(Valid.objects.all(), [])
- def test_ticket8683(self):
- # Raise proper error when a DateQuerySet gets passed a wrong type of
- # field
- self.assertRaisesMessage(
- AssertionError,
- "'name' isn't a DateTimeField.",
- Item.objects.datetimes, 'name', 'month'
- )
- def test_ticket22023(self):
- with self.assertRaisesMessage(TypeError,
- "Cannot call only() after .values() or .values_list()"):
- Valid.objects.values().only()
- with self.assertRaisesMessage(TypeError,
- "Cannot call defer() after .values() or .values_list()"):
- Valid.objects.values().defer()
- class Queries4Tests(BaseQuerysetTest):
- @classmethod
- def setUpTestData(cls):
- generic = NamedCategory.objects.create(name="Generic")
- cls.t1 = Tag.objects.create(name='t1', category=generic)
- n1 = Note.objects.create(note='n1', misc='foo', id=1)
- n2 = Note.objects.create(note='n2', misc='bar', id=2)
- e1 = ExtraInfo.objects.create(info='e1', note=n1)
- e2 = ExtraInfo.objects.create(info='e2', note=n2)
- cls.a1 = Author.objects.create(name='a1', num=1001, extra=e1)
- cls.a3 = Author.objects.create(name='a3', num=3003, extra=e2)
- cls.r1 = Report.objects.create(name='r1', creator=cls.a1)
- cls.r2 = Report.objects.create(name='r2', creator=cls.a3)
- cls.r3 = Report.objects.create(name='r3')
- Item.objects.create(name='i1', created=datetime.datetime.now(), note=n1, creator=cls.a1)
- Item.objects.create(name='i2', created=datetime.datetime.now(), note=n1, creator=cls.a3)
- def test_ticket11811(self):
- unsaved_category = NamedCategory(name="Other")
- with six.assertRaisesRegex(self, ValueError,
- 'Unsaved model instance <NamedCategory: Other> '
- 'cannot be used in an ORM query.'):
- Tag.objects.filter(pk=self.t1.pk).update(category=unsaved_category)
- def test_ticket14876(self):
- # Note: when combining the query we need to have information available
- # about the join type of the trimmed "creator__isnull" join. If we
- # don't have that information, then the join is created as INNER JOIN
- # and results will be incorrect.
- q1 = Report.objects.filter(Q(creator__isnull=True) | Q(creator__extra__info='e1'))
- q2 = Report.objects.filter(Q(creator__isnull=True)) | Report.objects.filter(Q(creator__extra__info='e1'))
- self.assertQuerysetEqual(q1, ["<Report: r1>", "<Report: r3>"], ordered=False)
- self.assertEqual(str(q1.query), str(q2.query))
- q1 = Report.objects.filter(Q(creator__extra__info='e1') | Q(creator__isnull=True))
- q2 = Report.objects.filter(Q(creator__extra__info='e1')) | Report.objects.filter(Q(creator__isnull=True))
- self.assertQuerysetEqual(q1, ["<Report: r1>", "<Report: r3>"], ordered=False)
- self.assertEqual(str(q1.query), str(q2.query))
- q1 = Item.objects.filter(Q(creator=self.a1) | Q(creator__report__name='r1')).order_by()
- q2 = Item.objects.filter(Q(creator=self.a1)).order_by() | Item.objects.filter(Q(creator__report__name='r1')).order_by()
- self.assertQuerysetEqual(q1, ["<Item: i1>"])
- self.assertEqual(str(q1.query), str(q2.query))
- q1 = Item.objects.filter(Q(creator__report__name='e1') | Q(creator=self.a1)).order_by()
- q2 = Item.objects.filter(Q(creator__report__name='e1')).order_by() | Item.objects.filter(Q(creator=self.a1)).order_by()
- self.assertQuerysetEqual(q1, ["<Item: i1>"])
- self.assertEqual(str(q1.query), str(q2.query))
- def test_combine_join_reuse(self):
- # Test that we correctly recreate joins having identical connections
- # in the rhs query, in case the query is ORed together. Related to
- # ticket #18748
- Report.objects.create(name='r4', creator=self.a1)
- q1 = Author.objects.filter(report__name='r5')
- q2 = Author.objects.filter(report__name='r4').filter(report__name='r1')
- combined = q1 | q2
- self.assertEqual(str(combined.query).count('JOIN'), 2)
- self.assertEqual(len(combined), 1)
- self.assertEqual(combined[0].name, 'a1')
- def test_ticket7095(self):
- # Updates that are filtered on the model being updated are somewhat
- # tricky in MySQL. This exercises that case.
- ManagedModel.objects.create(data='mm1', tag=self.t1, public=True)
- self.assertEqual(ManagedModel.objects.update(data='mm'), 1)
- # A values() or values_list() query across joined models must use outer
- # joins appropriately.
- # Note: In Oracle, we expect a null CharField to return '' instead of
- # None.
- if connection.features.interprets_empty_strings_as_nulls:
- expected_null_charfield_repr = ''
- else:
- expected_null_charfield_repr = None
- self.assertValueQuerysetEqual(
- Report.objects.values_list("creator__extra__info", flat=True).order_by("name"),
- ['e1', 'e2', expected_null_charfield_repr],
- )
- # Similarly for select_related(), joins beyond an initial nullable join
- # must use outer joins so that all results are included.
- self.assertQuerysetEqual(
- Report.objects.select_related("creator", "creator__extra").order_by("name"),
- ['<Report: r1>', '<Report: r2>', '<Report: r3>']
- )
- # When there are multiple paths to a table from another table, we have
- # to be careful not to accidentally reuse an inappropriate join when
- # using select_related(). We used to return the parent's Detail record
- # here by mistake.
- d1 = Detail.objects.create(data="d1")
- d2 = Detail.objects.create(data="d2")
- m1 = Member.objects.create(name="m1", details=d1)
- m2 = Member.objects.create(name="m2", details=d2)
- Child.objects.create(person=m2, parent=m1)
- obj = m1.children.select_related("person__details")[0]
- self.assertEqual(obj.person.details.data, 'd2')
- def test_order_by_resetting(self):
- # Calling order_by() with no parameters removes any existing ordering on the
- # model. But it should still be possible to add new ordering after that.
- qs = Author.objects.order_by().order_by('name')
- self.assertIn('ORDER BY', qs.query.get_compiler(qs.db).as_sql()[0])
- def test_order_by_reverse_fk(self):
- # It is possible to order by reverse of foreign key, although that can lead
- # to duplicate results.
- c1 = SimpleCategory.objects.create(name="category1")
- c2 = SimpleCategory.objects.create(name="category2")
- CategoryItem.objects.create(category=c1)
- CategoryItem.objects.create(category=c2)
- CategoryItem.objects.create(category=c1)
- self.assertQuerysetEqual(
- SimpleCategory.objects.order_by('categoryitem', 'pk'),
- [c1, c2, c1], lambda x: x)
- def test_ticket10181(self):
- # Avoid raising an EmptyResultSet if an inner query is probably
- # empty (and hence, not executed).
- self.assertQuerysetEqual(
- Tag.objects.filter(id__in=Tag.objects.filter(id__in=[])),
- []
- )
- def test_ticket15316_filter_false(self):
- c1 = SimpleCategory.objects.create(name="category1")
- c2 = SpecialCategory.objects.create(name="named category1",
- special_name="special1")
- c3 = SpecialCategory.objects.create(name="named category2",
- special_name="special2")
- CategoryItem.objects.create(category=c1)
- ci2 = CategoryItem.objects.create(category=c2)
- ci3 = CategoryItem.objects.create(category=c3)
- qs = CategoryItem.objects.filter(category__specialcategory__isnull=False)
- self.assertEqual(qs.count(), 2)
- self.assertQuerysetEqual(qs, [ci2.pk, ci3.pk], lambda x: x.pk, False)
- def test_ticket15316_exclude_false(self):
- c1 = SimpleCategory.objects.create(name="category1")
- c2 = SpecialCategory.objects.create(name="named category1",
- special_name="special1")
- c3 = SpecialCategory.objects.create(name="named category2",
- special_name="special2")
- ci1 = CategoryItem.objects.create(category=c1)
- CategoryItem.objects.create(category=c2)
- CategoryItem.objects.create(category=c3)
- qs = CategoryItem.objects.exclude(category__specialcategory__isnull=False)
- self.assertEqual(qs.count(), 1)
- self.assertQuerysetEqual(qs, [ci1.pk], lambda x: x.pk)
- def test_ticket15316_filter_true(self):
- c1 = SimpleCategory.objects.create(name="category1")
- c2 = SpecialCategory.objects.create(name="named category1",
- special_name="special1")
- c3 = SpecialCategory.objects.create(name="named category2",
- special_name="special2")
- ci1 = CategoryItem.objects.create(category=c1)
- CategoryItem.objects.create(category=c2)
- CategoryItem.objects.create(category=c3)
- qs = CategoryItem.objects.filter(category__specialcategory__isnull=True)
- self.assertEqual(qs.count(), 1)
- self.assertQuerysetEqual(qs, [ci1.pk], lambda x: x.pk)
- def test_ticket15316_exclude_true(self):
- c1 = SimpleCategory.objects.create(name="category1")
- c2 = SpecialCategory.objects.create(name="named category1",
- special_name="special1")
- c3 = SpecialCategory.objects.create(name="named category2",
- special_name="special2")
- CategoryItem.objects.create(category=c1)
- ci2 = CategoryItem.objects.create(category=c2)
- ci3 = CategoryItem.objects.create(category=c3)
- qs = CategoryItem.objects.exclude(category__specialcategory__isnull=True)
- self.assertEqual(qs.count(), 2)
- self.assertQuerysetEqual(qs, [ci2.pk, ci3.pk], lambda x: x.pk, False)
- def test_ticket15316_one2one_filter_false(self):
- c = SimpleCategory.objects.create(name="cat")
- c0 = SimpleCategory.objects.create(name="cat0")
- c1 = SimpleCategory.objects.create(name="category1")
- OneToOneCategory.objects.create(category=c1, new_name="new1")
- OneToOneCategory.objects.create(category=c0, new_name="new2")
- CategoryItem.objects.create(category=c)
- ci2 = CategoryItem.objects.create(category=c0)
- ci3 = CategoryItem.objects.create(category=c1)
- qs = CategoryItem.objects.filter(category__onetoonecategory__isnull=False)
- self.assertEqual(qs.count(), 2)
- self.assertQuerysetEqual(qs, [ci2.pk, ci3.pk], lambda x: x.pk, False)
- def test_ticket15316_one2one_exclude_false(self):
- c = SimpleCategory.objects.create(name="cat")
- c0 = SimpleCategory.objects.create(name="cat0")
- c1 = SimpleCategory.objects.create(name="category1")
- OneToOneCategory.objects.create(category=c1, new_name="new1")
- OneToOneCategory.objects.create(category=c0, new_name="new2")
- ci1 = CategoryItem.objects.create(category=c)
- CategoryItem.objects.create(category=c0)
- CategoryItem.objects.create(category=c1)
- qs = CategoryItem.objects.exclude(category__onetoonecategory__isnull=False)
- self.assertEqual(qs.count(), 1)
- self.assertQuerysetEqual(qs, [ci1.pk], lambda x: x.pk)
- def test_ticket15316_one2one_filter_true(self):
- c = SimpleCategory.objects.create(name="cat")
- c0 = SimpleCategory.objects.create(name="cat0")
- c1 = SimpleCategory.objects.create(name="category1")
- OneToOneCategory.objects.create(category=c1, new_name="new1")
- OneToOneCategory.objects.create(category=c0, new_name="new2")
- ci1 = CategoryItem.objects.create(category=c)
- CategoryItem.objects.create(category=c0)
- CategoryItem.objects.create(category=c1)
- qs = CategoryItem.objects.filter(category__onetoonecategory__isnull=True)
- self.assertEqual(qs.count(), 1)
- self.assertQuerysetEqual(qs, [ci1.pk], lambda x: x.pk)
- def test_ticket15316_one2one_exclude_true(self):
- c = SimpleCategory.objects.create(name="cat")
- c0 = SimpleCategory.objects.create(name="cat0")
- c1 = SimpleCategory.objects.create(name="category1")
- OneToOneCategory.objects.create(category=c1, new_name="new1")
- OneToOneCategory.objects.create(category=c0, new_name="new2")
- CategoryItem.objects.create(category=c)
- ci2 = CategoryItem.objects.create(category=c0)
- ci3 = CategoryItem.objects.create(category=c1)
- qs = CategoryItem.objects.exclude(category__onetoonecategory__isnull=True)
- self.assertEqual(qs.count(), 2)
- self.assertQuerysetEqual(qs, [ci2.pk, ci3.pk], lambda x: x.pk, False)
- class Queries5Tests(TestCase):
- @classmethod
- def setUpTestData(cls):
- # Ordering by 'rank' gives us rank2, rank1, rank3. Ordering by the
- # Meta.ordering will be rank3, rank2, rank1.
- n1 = Note.objects.create(note='n1', misc='foo', id=1)
- n2 = Note.objects.create(note='n2', misc='bar', id=2)
- e1 = ExtraInfo.objects.create(info='e1', note=n1)
- e2 = ExtraInfo.objects.create(info='e2', note=n2)
- a1 = Author.objects.create(name='a1', num=1001, extra=e1)
- a2 = Author.objects.create(name='a2', num=2002, extra=e1)
- a3 = Author.objects.create(name='a3', num=3003, extra=e2)
- cls.rank1 = Ranking.objects.create(rank=2, author=a2)
- Ranking.objects.create(rank=1, author=a3)
- Ranking.objects.create(rank=3, author=a1)
- def test_ordering(self):
- # Cross model ordering is possible in Meta, too.
- self.assertQuerysetEqual(
- Ranking.objects.all(),
- ['<Ranking: 3: a1>', '<Ranking: 2: a2>', '<Ranking: 1: a3>']
- )
- self.assertQuerysetEqual(
- Ranking.objects.all().order_by('rank'),
- ['<Ranking: 1: a3>', '<Ranking: 2: a2>', '<Ranking: 3: a1>']
- )
- # Ordering of extra() pieces is possible, too and you can mix extra
- # fields and model fields in the ordering.
- self.assertQuerysetEqual(
- Ranking.objects.extra(tables=['django_site'], order_by=['-django_site.id', 'rank']),
- ['<Ranking: 1: a3>', '<Ranking: 2: a2>', '<Ranking: 3: a1>']
- )
- qs = Ranking.objects.extra(select={'good': 'case when rank > 2 then 1 else 0 end'})
- self.assertEqual(
- [o.good for o in qs.extra(order_by=('-good',))],
- [True, False, False]
- )
- self.assertQuerysetEqual(
- qs.extra(order_by=('-good', 'id')),
- ['<Ranking: 3: a1>', '<Ranking: 2: a2>', '<Ranking: 1: a3>']
- )
- # Despite having some extra aliases in the query, we can still omit
- # them in a values() query.
- dicts = qs.values('id', 'rank').order_by('id')
- self.assertEqual(
- [d['rank'] for d in dicts],
- [2, 1, 3]
- )
- def test_ticket7256(self):
- # An empty values() call includes all aliases, including those from an
- # extra()
- qs = Ranking.objects.extra(select={'good': 'case when rank > 2 then 1 else 0 end'})
- dicts = qs.values().order_by('id')
- for d in dicts:
- del d['id']
- del d['author_id']
- self.assertEqual(
- [sorted(d.items()) for d in dicts],
- [[('good', 0), ('rank', 2)], [('good', 0), ('rank', 1)], [('good', 1), ('rank', 3)]]
- )
- def test_ticket7045(self):
- # Extra tables used to crash SQL construction on the second use.
- qs = Ranking.objects.extra(tables=['django_site'])
- qs.query.get_compiler(qs.db).as_sql()
- # test passes if this doesn't raise an exception.
- qs.query.get_compiler(qs.db).as_sql()
- def test_ticket9848(self):
- # Make sure that updates which only filter on sub-tables don't
- # inadvertently update the wrong records (bug #9848).
- # Make sure that the IDs from different tables don't happen to match.
- self.assertQuerysetEqual(
- Ranking.objects.filter(author__name='a1'),
- ['<Ranking: 3: a1>']
- )
- self.assertEqual(
- Ranking.objects.filter(author__name='a1').update(rank='4'),
- 1
- )
- r = Ranking.objects.filter(author__name='a1')[0]
- self.assertNotEqual(r.id, r.author.id)
- self.assertEqual(r.rank, 4)
- r.rank = 3
- r.save()
- self.assertQuerysetEqual(
- Ranking.objects.all(),
- ['<Ranking: 3: a1>', '<Ranking: 2: a2>', '<Ranking: 1: a3>']
- )
- def test_ticket5261(self):
- # Test different empty excludes.
- self.assertQuerysetEqual(
- Note.objects.exclude(Q()),
- ['<Note: n1>', '<Note: n2>']
- )
- self.assertQuerysetEqual(
- Note.objects.filter(~Q()),
- ['<Note: n1>', '<Note: n2>']
- )
- self.assertQuerysetEqual(
- Note.objects.filter(~Q() | ~Q()),
- ['<Note: n1>', '<Note: n2>']
- )
- self.assertQuerysetEqual(
- Note.objects.exclude(~Q() & ~Q()),
- ['<Note: n1>', '<Note: n2>']
- )
- def test_extra_select_literal_percent_s(self):
- # Allow %%s to escape select clauses
- self.assertEqual(
- Note.objects.extra(select={'foo': "'%%s'"})[0].foo,
- '%s'
- )
- self.assertEqual(
- Note.objects.extra(select={'foo': "'%%s bar %%s'"})[0].foo,
- '%s bar %s'
- )
- self.assertEqual(
- Note.objects.extra(select={'foo': "'bar %%s'"})[0].foo,
- 'bar %s'
- )
- class SelectRelatedTests(TestCase):
- def test_tickets_3045_3288(self):
- # Once upon a time, select_related() with circular relations would loop
- # infinitely if you forgot to specify "depth". Now we set an arbitrary
- # default upper bound.
- self.assertQuerysetEqual(X.objects.all(), [])
- self.assertQuerysetEqual(X.objects.select_related(), [])
- class SubclassFKTests(TestCase):
- def test_ticket7778(self):
- # Model subclasses could not be deleted if a nullable foreign key
- # relates to a model that relates back.
- num_celebs = Celebrity.objects.count()
- tvc = TvChef.objects.create(name="Huey")
- self.assertEqual(Celebrity.objects.count(), num_celebs + 1)
- Fan.objects.create(fan_of=tvc)
- Fan.objects.create(fan_of=tvc)
- tvc.delete()
- # The parent object should have been deleted as well.
- self.assertEqual(Celebrity.objects.count(), num_celebs)
- class CustomPkTests(TestCase):
- def test_ticket7371(self):
- self.assertQuerysetEqual(Related.objects.order_by('custom'), [])
- class NullableRelOrderingTests(TestCase):
- def test_ticket10028(self):
- # Ordering by model related to nullable relations(!) should use outer
- # joins, so that all results are included.
- Plaything.objects.create(name="p1")
- self.assertQuerysetEqual(
- Plaything.objects.all(),
- ['<Plaything: p1>']
- )
- def test_join_already_in_query(self):
- # Ordering by model related to nullable relations should not change
- # the join type of already existing joins.
- Plaything.objects.create(name="p1")
- s = SingleObject.objects.create(name='s')
- r = RelatedObject.objects.create(single=s, f=1)
- Plaything.objects.create(name="p2", others=r)
- qs = Plaything.objects.all().filter(others__isnull=False).order_by('pk')
- self.assertNotIn('JOIN', str(qs.query))
- qs = Plaything.objects.all().filter(others__f__isnull=False).order_by('pk')
- self.assertIn('INNER', str(qs.query))
- qs = qs.order_by('others__single__name')
- # The ordering by others__single__pk will add one new join (to single)
- # and that join must be LEFT join. The already existing join to related
- # objects must be kept INNER. So, we have both an INNER and a LEFT join
- # in the query.
- self.assertEqual(str(qs.query).count('LEFT'), 1)
- self.assertEqual(str(qs.query).count('INNER'), 1)
- self.assertQuerysetEqual(
- qs,
- ['<Plaything: p2>']
- )
- class DisjunctiveFilterTests(TestCase):
- @classmethod
- def setUpTestData(cls):
- cls.n1 = Note.objects.create(note='n1', misc='foo', id=1)
- ExtraInfo.objects.create(info='e1', note=cls.n1)
- def test_ticket7872(self):
- # Another variation on the disjunctive filtering theme.
- # For the purposes of this regression test, it's important that there is no
- # Join object related to the LeafA we create.
- LeafA.objects.create(data='first')
- self.assertQuerysetEqual(LeafA.objects.all(), ['<LeafA: first>'])
- self.assertQuerysetEqual(
- LeafA.objects.filter(Q(data='first') | Q(join__b__data='second')),
- ['<LeafA: first>']
- )
- def test_ticket8283(self):
- # Checking that applying filters after a disjunction works correctly.
- self.assertQuerysetEqual(
- (ExtraInfo.objects.filter(note=self.n1) | ExtraInfo.objects.filter(info='e2')).filter(note=self.n1),
- ['<ExtraInfo: e1>']
- )
- self.assertQuerysetEqual(
- (ExtraInfo.objects.filter(info='e2') | ExtraInfo.objects.filter(note=self.n1)).filter(note=self.n1),
- ['<ExtraInfo: e1>']
- )
- class Queries6Tests(TestCase):
- @classmethod
- def setUpTestData(cls):
- generic = NamedCategory.objects.create(name="Generic")
- t1 = Tag.objects.create(name='t1', category=generic)
- Tag.objects.create(name='t2', parent=t1, category=generic)
- t3 = Tag.objects.create(name='t3', parent=t1)
- t4 = Tag.objects.create(name='t4', parent=t3)
- Tag.objects.create(name='t5', parent=t3)
- n1 = Note.objects.create(note='n1', misc='foo', id=1)
- ann1 = Annotation.objects.create(name='a1', tag=t1)
- ann1.notes.add(n1)
- Annotation.objects.create(name='a2', tag=t4)
- def test_parallel_iterators(self):
- # Test that parallel iterators work.
- qs = Tag.objects.all()
- i1, i2 = iter(qs), iter(qs)
- self.assertEqual(repr(next(i1)), '<Tag: t1>')
- self.assertEqual(repr(next(i1)), '<Tag: t2>')
- self.assertEqual(repr(next(i2)), '<Tag: t1>')
- self.assertEqual(repr(next(i2)), '<Tag: t2>')
- self.assertEqual(repr(next(i2)), '<Tag: t3>')
- self.assertEqual(repr(next(i1)), '<Tag: t3>')
- qs = X.objects.all()
- self.assertEqual(bool(qs), False)
- self.assertEqual(bool(qs), False)
- def test_nested_queries_sql(self):
- # Nested queries should not evaluate the inner query as part of constructing the
- # SQL (so we should see a nested query here, indicated by two "SELECT" calls).
- qs = Annotation.objects.filter(notes__in=Note.objects.filter(note="xyzzy"))
- self.assertEqual(
- qs.query.get_compiler(qs.db).as_sql()[0].count('SELECT'),
- 2
- )
- def test_tickets_8921_9188(self):
- # Incorrect SQL was being generated for certain types of exclude()
- # queries that crossed multi-valued relations (#8921, #9188 and some
- # pre-emptively discovered cases).
- self.assertQuerysetEqual(
- PointerA.objects.filter(connection__pointerb__id=1),
- []
- )
- self.assertQuerysetEqual(
- PointerA.objects.exclude(connection__pointerb__id=1),
- []
- )
- self.assertQuerysetEqual(
- Tag.objects.exclude(children=None),
- ['<Tag: t1>', '<Tag: t3>']
- )
- # This example is tricky because the parent could be NULL, so only checking
- # parents with annotations omits some results (tag t1, in this case).
- self.assertQuerysetEqual(
- Tag.objects.exclude(parent__annotation__name="a1"),
- ['<Tag: t1>', '<Tag: t4>', '<Tag: t5>']
- )
- # The annotation->tag link is single values and tag->children links is
- # multi-valued. So we have to split the exclude filter in the middle
- # and then optimize the inner query without losing results.
- self.assertQuerysetEqual(
- Annotation.objects.exclude(tag__children__name="t2"),
- ['<Annotation: a2>']
- )
- # Nested queries are possible (although should be used with care, since
- # they have performance problems on backends like MySQL.
- self.assertQuerysetEqual(
- Annotation.objects.filter(notes__in=Note.objects.filter(note="n1")),
- ['<Annotation: a1>']
- )
- def test_ticket3739(self):
- # The all() method on querysets returns a copy of the queryset.
- q1 = Tag.objects.order_by('name')
- self.assertIsNot(q1, q1.all())
- def test_ticket_11320(self):
- qs = Tag.objects.exclude(category=None).exclude(category__name='foo')
- self.assertEqual(str(qs.query).count(' INNER JOIN '), 1)
- class RawQueriesTests(TestCase):
- def setUp(self):
- Note.objects.create(note='n1', misc='foo', id=1)
- def test_ticket14729(self):
- # Test representation of raw query with one or few parameters passed as list
- query = "SELECT * FROM queries_note WHERE note = %s"
- params = ['n1']
- qs = Note.objects.raw(query, params=params)
- self.assertEqual(repr(qs), "<RawQuerySet: SELECT * FROM queries_note WHERE note = n1>")
- query = "SELECT * FROM queries_note WHERE note = %s and misc = %s"
- params = ['n1', 'foo']
- qs = Note.objects.raw(query, params=params)
- self.assertEqual(repr(qs), "<RawQuerySet: SELECT * FROM queries_note WHERE note = n1 and misc = foo>")
- class GeneratorExpressionTests(TestCase):
- def test_ticket10432(self):
- # Using an empty generator expression as the rvalue for an "__in"
- # lookup is legal.
- self.assertQuerysetEqual(
- Note.objects.filter(pk__in=(x for x in ())),
- []
- )
- class ComparisonTests(TestCase):
- def setUp(self):
- self.n1 = Note.objects.create(note='n1', misc='foo', id=1)
- e1 = ExtraInfo.objects.create(info='e1', note=self.n1)
- self.a2 = Author.objects.create(name='a2', num=2002, extra=e1)
- def test_ticket8597(self):
- # Regression tests for case-insensitive comparisons
- Item.objects.create(name="a_b", created=datetime.datetime.now(), creator=self.a2, note=self.n1)
- Item.objects.create(name="x%y", created=datetime.datetime.now(), creator=self.a2, note=self.n1)
- self.assertQuerysetEqual(
- Item.objects.filter(name__iexact="A_b"),
- ['<Item: a_b>']
- )
- self.assertQuerysetEqual(
- Item.objects.filter(name__iexact="x%Y"),
- ['<Item: x%y>']
- )
- self.assertQuerysetEqual(
- Item.objects.filter(name__istartswith="A_b"),
- ['<Item: a_b>']
- )
- self.assertQuerysetEqual(
- Item.objects.filter(name__iendswith="A_b"),
- ['<Item: a_b>']
- )
- class ExistsSql(TestCase):
- def test_exists(self):
- with CaptureQueriesContext(connection) as captured_queries:
- self.assertFalse(Tag.objects.exists())
- # Ok - so the exist query worked - but did it include too many columns?
- self.assertEqual(len(captured_queries), 1)
- qstr = captured_queries[0]['sql']
- id, name = connection.ops.quote_name('id'), connection.ops.quote_name('name')
- self.assertNotIn(id, qstr)
- self.assertNotIn(name, qstr)
- def test_ticket_18414(self):
- Article.objects.create(name='one', created=datetime.datetime.now())
- Article.objects.create(name='one', created=datetime.datetime.now())
- Article.objects.create(name='two', created=datetime.datetime.now())
- self.assertTrue(Article.objects.exists())
- self.assertTrue(Article.objects.distinct().exists())
- self.assertTrue(Article.objects.distinct()[1:3].exists())
- self.assertFalse(Article.objects.distinct()[1:1].exists())
- @skipUnlessDBFeature('can_distinct_on_fields')
- def test_ticket_18414_distinct_on(self):
- Article.objects.create(name='one', created=datetime.datetime.now())
- Article.objects.create(name='one', created=datetime.datetime.now())
- Article.objects.create(name='two', created=datetime.datetime.now())
- self.assertTrue(Article.objects.distinct('name').exists())
- self.assertTrue(Article.objects.distinct('name')[1:2].exists())
- self.assertFalse(Article.objects.distinct('name')[2:3].exists())
- class QuerysetOrderedTests(unittest.TestCase):
- """
- Tests for the Queryset.ordered attribute.
- """
- def test_no_default_or_explicit_ordering(self):
- self.assertEqual(Annotation.objects.all().ordered, False)
- def test_cleared_default_ordering(self):
- self.assertEqual(Tag.objects.all().ordered, True)
- self.assertEqual(Tag.objects.all().order_by().ordered, False)
- def test_explicit_ordering(self):
- self.assertEqual(Annotation.objects.all().order_by('id').ordered, True)
- def test_order_by_extra(self):
- self.assertEqual(Annotation.objects.all().extra(order_by=['id']).ordered, True)
- def test_annotated_ordering(self):
- qs = Annotation.objects.annotate(num_notes=Count('notes'))
- self.assertEqual(qs.ordered, False)
- self.assertEqual(qs.order_by('num_notes').ordered, True)
- @skipUnlessDBFeature('allow_sliced_subqueries')
- class SubqueryTests(TestCase):
- @classmethod
- def setUpTestData(cls):
- DumbCategory.objects.create(id=1)
- DumbCategory.objects.create(id=2)
- DumbCategory.objects.create(id=3)
- DumbCategory.objects.create(id=4)
- def test_ordered_subselect(self):
- "Subselects honor any manual ordering"
- query = DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[0:2])
- self.assertEqual(set(query.values_list('id', flat=True)), {3, 4})
- query = DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[:2])
- self.assertEqual(set(query.values_list('id', flat=True)), {3, 4})
- query = DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[1:2])
- self.assertEqual(set(query.values_list('id', flat=True)), {3})
- query = DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[2:])
- self.assertEqual(set(query.values_list('id', flat=True)), {1, 2})
- def test_slice_subquery_and_query(self):
- """
- Slice a query that has a sliced subquery
- """
- query = DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[0:2])[0:2]
- self.assertEqual({x.id for x in query}, {3, 4})
- query = DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[1:3])[1:3]
- self.assertEqual({x.id for x in query}, {3})
- query = DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[2:])[1:]
- self.assertEqual({x.id for x in query}, {2})
- def test_related_sliced_subquery(self):
- """
- Related objects constraints can safely contain sliced subqueries.
- refs #22434
- """
- generic = NamedCategory.objects.create(id=5, name="Generic")
- t1 = Tag.objects.create(name='t1', category=generic)
- t2 = Tag.objects.create(name='t2', category=generic)
- ManagedModel.objects.create(data='mm1', tag=t1, public=True)
- mm2 = ManagedModel.objects.create(data='mm2', tag=t2, public=True)
- query = ManagedModel.normal_manager.filter(
- tag__in=Tag.objects.order_by('-id')[:1]
- )
- self.assertEqual({x.id for x in query}, {mm2.id})
- def test_sliced_delete(self):
- "Delete queries can safely contain sliced subqueries"
- DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[0:1]).delete()
- self.assertEqual(set(DumbCategory.objects.values_list('id', flat=True)), {1, 2, 3})
- DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[1:2]).delete()
- self.assertEqual(set(DumbCategory.objects.values_list('id', flat=True)), {1, 3})
- DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[1:]).delete()
- self.assertEqual(set(DumbCategory.objects.values_list('id', flat=True)), {3})
- class CloneTests(TestCase):
- def test_evaluated_queryset_as_argument(self):
- "#13227 -- If a queryset is already evaluated, it can still be used as a query arg"
- n = Note(note='Test1', misc='misc')
- n.save()
- e = ExtraInfo(info='good', note=n)
- e.save()
- n_list = Note.objects.all()
- # Evaluate the Note queryset, populating the query cache
- list(n_list)
- # Use the note queryset in a query, and evaluate
- # that query in a way that involves cloning.
- self.assertEqual(ExtraInfo.objects.filter(note__in=n_list)[0].info, 'good')
- def test_no_model_options_cloning(self):
- """
- Test that cloning a queryset does not get out of hand. While complete
- testing is impossible, this is a sanity check against invalid use of
- deepcopy. refs #16759.
- """
- opts_class = type(Note._meta)
- note_deepcopy = getattr(opts_class, "__deepcopy__", None)
- opts_class.__deepcopy__ = lambda obj, memo: self.fail("Model options shouldn't be cloned.")
- try:
- Note.objects.filter(pk__lte=F('pk') + 1).all()
- finally:
- if note_deepcopy is None:
- delattr(opts_class, "__deepcopy__")
- else:
- opts_class.__deepcopy__ = note_deepcopy
- def test_no_fields_cloning(self):
- """
- Test that cloning a queryset does not get out of hand. While complete
- testing is impossible, this is a sanity check against invalid use of
- deepcopy. refs #16759.
- """
- opts_class = type(Note._meta.get_field("misc"))
- note_deepcopy = getattr(opts_class, "__deepcopy__", None)
- opts_class.__deepcopy__ = lambda obj, memo: self.fail("Model fields shouldn't be cloned")
- try:
- Note.objects.filter(note=F('misc')).all()
- finally:
- if note_deepcopy is None:
- delattr(opts_class, "__deepcopy__")
- else:
- opts_class.__deepcopy__ = note_deepcopy
- class EmptyQuerySetTests(TestCase):
- def test_emptyqueryset_values(self):
- # #14366 -- Calling .values() on an empty QuerySet and then cloning
- # that should not cause an error
- self.assertQuerysetEqual(
- Number.objects.none().values('num').order_by('num'), []
- )
- def test_values_subquery(self):
- self.assertQuerysetEqual(
- Number.objects.filter(pk__in=Number.objects.none().values("pk")),
- []
- )
- self.assertQuerysetEqual(
- Number.objects.filter(pk__in=Number.objects.none().values_list("pk")),
- []
- )
- def test_ticket_19151(self):
- # #19151 -- Calling .values() or .values_list() on an empty QuerySet
- # should return an empty QuerySet and not cause an error.
- q = Author.objects.none()
- self.assertQuerysetEqual(q.values(), [])
- self.assertQuerysetEqual(q.values_list(), [])
- class ValuesQuerysetTests(BaseQuerysetTest):
- @classmethod
- def setUpTestData(cls):
- Number.objects.create(num=72)
- cls.identity = staticmethod(lambda x: x)
- def test_flat_values_list(self):
- qs = Number.objects.values_list("num")
- qs = qs.values_list("num", flat=True)
- self.assertValueQuerysetEqual(qs, [72])
- def test_extra_values(self):
- # testing for ticket 14930 issues
- qs = Number.objects.extra(select=OrderedDict([('value_plus_x', 'num+%s'),
- ('value_minus_x', 'num-%s')]),
- select_params=(1, 2))
- qs = qs.order_by('value_minus_x')
- qs = qs.values('num')
- self.assertQuerysetEqual(qs, [{'num': 72}], self.identity)
- def test_extra_values_order_twice(self):
- # testing for ticket 14930 issues
- qs = Number.objects.extra(select={'value_plus_one': 'num+1', 'value_minus_one': 'num-1'})
- qs = qs.order_by('value_minus_one').order_by('value_plus_one')
- qs = qs.values('num')
- self.assertQuerysetEqual(qs, [{'num': 72}], self.identity)
- def test_extra_values_order_multiple(self):
- # Postgres doesn't allow constants in order by, so check for that.
- qs = Number.objects.extra(select={
- 'value_plus_one': 'num+1',
- 'value_minus_one': 'num-1',
- 'constant_value': '1'
- })
- qs = qs.order_by('value_plus_one', 'value_minus_one', 'constant_value')
- qs = qs.values('num')
- self.assertQuerysetEqual(qs, [{'num': 72}], self.identity)
- def test_extra_values_order_in_extra(self):
- # testing for ticket 14930 issues
- qs = Number.objects.extra(
- select={'value_plus_one': 'num+1', 'value_minus_one': 'num-1'},
- order_by=['value_minus_one'])
- qs = qs.values('num')
- def test_extra_select_params_values_order_in_extra(self):
- # testing for 23259 issue
- qs = Number.objects.extra(
- select={'value_plus_x': 'num+%s'},
- select_params=[1],
- order_by=['value_plus_x'])
- qs = qs.filter(num=72)
- qs = qs.values('num')
- self.assertQuerysetEqual(qs, [{'num': 72}], self.identity)
- def test_extra_multiple_select_params_values_order_by(self):
- # testing for 23259 issue
- qs = Number.objects.extra(select=OrderedDict([('value_plus_x', 'num+%s'),
- ('value_minus_x', 'num-%s')]),
- select_params=(72, 72))
- qs = qs.order_by('value_minus_x')
- qs = qs.filter(num=1)
- qs = qs.values('num')
- self.assertQuerysetEqual(qs, [], self.identity)
- def test_extra_values_list(self):
- # testing for ticket 14930 issues
- qs = Number.objects.extra(select={'value_plus_one': 'num+1'})
- qs = qs.order_by('value_plus_one')
- qs = qs.values_list('num')
- self.assertQuerysetEqual(qs, [(72,)], self.identity)
- def test_flat_extra_values_list(self):
- # testing for ticket 14930 issues
- qs = Number.objects.extra(select={'value_plus_one': 'num+1'})
- qs = qs.order_by('value_plus_one')
- qs = qs.values_list('num', flat=True)
- self.assertQuerysetEqual(qs, [72], self.identity)
- def test_field_error_values_list(self):
- # see #23443
- with self.assertRaisesMessage(FieldError,
- "Cannot resolve keyword %r into field."
- " Join on 'name' not permitted." % 'foo'):
- Tag.objects.values_list('name__foo')
- class QuerySetSupportsPythonIdioms(TestCase):
- @classmethod
- def setUpTestData(cls):
- some_date = datetime.datetime(2014, 5, 16, 12, 1)
- for i in range(1, 8):
- Article.objects.create(
- name="Article {}".format(i), created=some_date)
- def get_ordered_articles(self):
- return Article.objects.all().order_by('name')
- def test_can_get_items_using_index_and_slice_notation(self):
- self.assertEqual(self.get_ordered_articles()[0].name, 'Article 1')
- self.assertQuerysetEqual(self.get_ordered_articles()[1:3],
- ["<Article: Article 2>", "<Article: Article 3>"])
- def test_slicing_with_steps_can_be_used(self):
- self.assertQuerysetEqual(self.get_ordered_articles()[::2],
- ["<Article: Article 1>",
- "<Article: Article 3>",
- "<Article: Article 5>",
- "<Article: Article 7>"])
- @unittest.skipUnless(six.PY2, "Python 2 only -- Python 3 doesn't have longs.")
- def test_slicing_works_with_longs(self):
- # NOQA: long undefined on PY3
- self.assertEqual(self.get_ordered_articles()[long(0)].name, 'Article 1') # NOQA
- self.assertQuerysetEqual(self.get_ordered_articles()[long(1):long(3)], # NOQA
- ["<Article: Article 2>", "<Article: Article 3>"])
- self.assertQuerysetEqual(self.get_ordered_articles()[::long(2)], # NOQA
- ["<Article: Article 1>",
- "<Article: Article 3>",
- "<Article: Article 5>",
- "<Article: Article 7>"])
- # And can be mixed with ints.
- self.assertQuerysetEqual(self.get_ordered_articles()[1:long(3)], # NOQA
- ["<Article: Article 2>", "<Article: Article 3>"])
- def test_slicing_without_step_is_lazy(self):
- with self.assertNumQueries(0):
- self.get_ordered_articles()[0:5]
- def test_slicing_with_tests_is_not_lazy(self):
- with self.assertNumQueries(1):
- self.get_ordered_articles()[0:5:3]
- def test_slicing_can_slice_again_after_slicing(self):
- self.assertQuerysetEqual(self.get_ordered_articles()[0:5][0:2],
- ["<Article: Article 1>",
- "<Article: Article 2>"])
- self.assertQuerysetEqual(self.get_ordered_articles()[0:5][4:],
- ["<Article: Article 5>"])
- self.assertQuerysetEqual(self.get_ordered_articles()[0:5][5:], [])
- # Some more tests!
- self.assertQuerysetEqual(self.get_ordered_articles()[2:][0:2],
- ["<Article: Article 3>", "<Article: Article 4>"])
- self.assertQuerysetEqual(self.get_ordered_articles()[2:][:2],
- ["<Article: Article 3>", "<Article: Article 4>"])
- self.assertQuerysetEqual(self.get_ordered_articles()[2:][2:3],
- ["<Article: Article 5>"])
- # Using an offset without a limit is also possible.
- self.assertQuerysetEqual(self.get_ordered_articles()[5:],
- ["<Article: Article 6>",
- "<Article: Article 7>"])
- def test_slicing_cannot_filter_queryset_once_sliced(self):
- six.assertRaisesRegex(
- self,
- AssertionError,
- "Cannot filter a query once a slice has been taken.",
- Article.objects.all()[0:5].filter,
- id=1,
- )
- def test_slicing_cannot_reorder_queryset_once_sliced(self):
- six.assertRaisesRegex(
- self,
- AssertionError,
- "Cannot reorder a query once a slice has been taken.",
- Article.objects.all()[0:5].order_by,
- 'id',
- )
- def test_slicing_cannot_combine_queries_once_sliced(self):
- six.assertRaisesRegex(
- self,
- AssertionError,
- "Cannot combine queries once a slice has been taken.",
- lambda: Article.objects.all()[0:1] & Article.objects.all()[4:5]
- )
- def test_slicing_negative_indexing_not_supported_for_single_element(self):
- """hint: inverting your ordering might do what you need"""
- six.assertRaisesRegex(
- self,
- AssertionError,
- "Negative indexing is not supported.",
- lambda: Article.objects.all()[-1]
- )
- def test_slicing_negative_indexing_not_supported_for_range(self):
- """hint: inverting your ordering might do what you need"""
- six.assertRaisesRegex(
- self,
- AssertionError,
- "Negative indexing is not supported.",
- lambda: Article.objects.all()[0:-5]
- )
- def test_can_get_number_of_items_in_queryset_using_standard_len(self):
- self.assertEqual(len(Article.objects.filter(name__exact='Article 1')), 1)
- def test_can_combine_queries_using_and_and_or_operators(self):
- s1 = Article.objects.filter(name__exact='Article 1')
- s2 = Article.objects.filter(name__exact='Article 2')
- self.assertQuerysetEqual((s1 | s2).order_by('name'),
- ["<Article: Article 1>",
- "<Article: Article 2>"])
- self.assertQuerysetEqual(s1 & s2, [])
- class WeirdQuerysetSlicingTests(BaseQuerysetTest):
- @classmethod
- def setUpTestData(cls):
- Number.objects.create(num=1)
- Number.objects.create(num=2)
- Article.objects.create(name='one', created=datetime.datetime.now())
- Article.objects.create(name='two', created=datetime.datetime.now())
- Article.objects.create(name='three', created=datetime.datetime.now())
- Article.objects.create(name='four', created=datetime.datetime.now())
- def test_tickets_7698_10202(self):
- # People like to slice with '0' as the high-water mark.
- self.assertQuerysetEqual(Article.objects.all()[0:0], [])
- self.assertQuerysetEqual(Article.objects.all()[0:0][:10], [])
- self.assertEqual(Article.objects.all()[:0].count(), 0)
- self.assertRaisesMessage(
- AssertionError,
- 'Cannot change a query once a slice has been taken.',
- Article.objects.all()[:0].latest, 'created'
- )
- def test_empty_resultset_sql(self):
- # ticket #12192
- self.assertNumQueries(0, lambda: list(Number.objects.all()[1:1]))
- class EscapingTests(TestCase):
- def test_ticket_7302(self):
- # Reserved names are appropriately escaped
- ReservedName.objects.create(name='a', order=42)
- ReservedName.objects.create(name='b', order=37)
- self.assertQuerysetEqual(
- ReservedName.objects.all().order_by('order'),
- ['<ReservedName: b>', '<ReservedName: a>']
- )
- self.assertQuerysetEqual(
- ReservedName.objects.extra(select={'stuff': 'name'}, order_by=('order', 'stuff')),
- ['<ReservedName: b>', '<ReservedName: a>']
- )
- class ToFieldTests(TestCase):
- def test_in_query(self):
- apple = Food.objects.create(name="apple")
- pear = Food.objects.create(name="pear")
- lunch = Eaten.objects.create(food=apple, meal="lunch")
- dinner = Eaten.objects.create(food=pear, meal="dinner")
- self.assertEqual(
- set(Eaten.objects.filter(food__in=[apple, pear])),
- {lunch, dinner},
- )
- def test_reverse_in(self):
- apple = Food.objects.create(name="apple")
- pear = Food.objects.create(name="pear")
- lunch_apple = Eaten.objects.create(food=apple, meal="lunch")
- lunch_pear = Eaten.objects.create(food=pear, meal="dinner")
- self.assertEqual(
- set(Food.objects.filter(eaten__in=[lunch_apple, lunch_pear])),
- {apple, pear}
- )
- def test_single_object(self):
- apple = Food.objects.create(name="apple")
- lunch = Eaten.objects.create(food=apple, meal="lunch")
- dinner = Eaten.objects.create(food=apple, meal="dinner")
- self.assertEqual(
- set(Eaten.objects.filter(food=apple)),
- {lunch, dinner}
- )
- def test_single_object_reverse(self):
- apple = Food.objects.create(name="apple")
- lunch = Eaten.objects.create(food=apple, meal="lunch")
- self.assertEqual(
- set(Food.objects.filter(eaten=lunch)),
- {apple}
- )
- def test_recursive_fk(self):
- node1 = Node.objects.create(num=42)
- node2 = Node.objects.create(num=1, parent=node1)
- self.assertEqual(
- list(Node.objects.filter(parent=node1)),
- [node2]
- )
- def test_recursive_fk_reverse(self):
- node1 = Node.objects.create(num=42)
- node2 = Node.objects.create(num=1, parent=node1)
- self.assertEqual(
- list(Node.objects.filter(node=node2)),
- [node1]
- )
- class ConditionalTests(BaseQuerysetTest):
- """Tests whose execution depend on different environment conditions like
- Python version or DB backend features"""
- @classmethod
- def setUpTestData(cls):
- generic = NamedCategory.objects.create(name="Generic")
- t1 = Tag.objects.create(name='t1', category=generic)
- Tag.objects.create(name='t2', parent=t1, category=generic)
- t3 = Tag.objects.create(name='t3', parent=t1)
- Tag.objects.create(name='t4', parent=t3)
- Tag.objects.create(name='t5', parent=t3)
- def test_infinite_loop(self):
- # If you're not careful, it's possible to introduce infinite loops via
- # default ordering on foreign keys in a cycle. We detect that.
- self.assertRaisesMessage(
- FieldError,
- 'Infinite loop caused by ordering.',
- lambda: list(LoopX.objects.all()) # Force queryset evaluation with list()
- )
- self.assertRaisesMessage(
- FieldError,
- 'Infinite loop caused by ordering.',
- lambda: list(LoopZ.objects.all()) # Force queryset evaluation with list()
- )
- # Note that this doesn't cause an infinite loop, since the default
- # ordering on the Tag model is empty (and thus defaults to using "id"
- # for the related field).
- self.assertEqual(len(Tag.objects.order_by('parent')), 5)
- # ... but you can still order in a non-recursive fashion among linked
- # fields (the previous test failed because the default ordering was
- # recursive).
- self.assertQuerysetEqual(
- LoopX.objects.all().order_by('y__x__y__x__id'),
- []
- )
- # When grouping without specifying ordering, we add an explicit "ORDER BY NULL"
- # portion in MySQL to prevent unnecessary sorting.
- @skipUnlessDBFeature('requires_explicit_null_ordering_when_grouping')
- def test_null_ordering_added(self):
- query = Tag.objects.values_list('parent_id', flat=True).order_by().query
- query.group_by = ['parent_id']
- sql = query.get_compiler(DEFAULT_DB_ALIAS).as_sql()[0]
- fragment = "ORDER BY "
- pos = sql.find(fragment)
- self.assertEqual(sql.find(fragment, pos + 1), -1)
- self.assertEqual(sql.find("NULL", pos + len(fragment)), pos + len(fragment))
- # Sqlite 3 does not support passing in more than 1000 parameters except by
- # changing a parameter at compilation time.
- @skipUnlessDBFeature('supports_1000_query_parameters')
- def test_ticket14244(self):
- # Test that the "in" lookup works with lists of 1000 items or more.
- # The numbers amount is picked to force three different IN batches
- # for Oracle, yet to be less than 2100 parameter limit for MSSQL.
- numbers = list(range(2050))
- Number.objects.all().delete()
- Number.objects.bulk_create(Number(num=num) for num in numbers)
- self.assertEqual(
- Number.objects.filter(num__in=numbers[:1000]).count(),
- 1000
- )
- self.assertEqual(
- Number.objects.filter(num__in=numbers[:1001]).count(),
- 1001
- )
- self.assertEqual(
- Number.objects.filter(num__in=numbers[:2000]).count(),
- 2000
- )
- self.assertEqual(
- Number.objects.filter(num__in=numbers).count(),
- len(numbers)
- )
- class UnionTests(unittest.TestCase):
- """
- Tests for the union of two querysets. Bug #12252.
- """
- @classmethod
- def setUpTestData(cls):
- objectas = []
- objectbs = []
- objectcs = []
- a_info = ['one', 'two', 'three']
- for name in a_info:
- o = ObjectA(name=name)
- o.save()
- objectas.append(o)
- b_info = [('un', 1, objectas[0]), ('deux', 2, objectas[0]), ('trois', 3, objectas[2])]
- for name, number, objecta in b_info:
- o = ObjectB(name=name, num=number, objecta=objecta)
- o.save()
- objectbs.append(o)
- c_info = [('ein', objectas[2], objectbs[2]), ('zwei', objectas[1], objectbs[1])]
- for name, objecta, objectb in c_info:
- o = ObjectC(name=name, objecta=objecta, objectb=objectb)
- o.save()
- objectcs.append(o)
- def check_union(self, model, Q1, Q2):
- filter = model.objects.filter
- self.assertEqual(set(filter(Q1) | filter(Q2)), set(filter(Q1 | Q2)))
- self.assertEqual(set(filter(Q2) | filter(Q1)), set(filter(Q1 | Q2)))
- def test_A_AB(self):
- Q1 = Q(name='two')
- Q2 = Q(objectb__name='deux')
- self.check_union(ObjectA, Q1, Q2)
- def test_A_AB2(self):
- Q1 = Q(name='two')
- Q2 = Q(objectb__name='deux', objectb__num=2)
- self.check_union(ObjectA, Q1, Q2)
- def test_AB_ACB(self):
- Q1 = Q(objectb__name='deux')
- Q2 = Q(objectc__objectb__name='deux')
- self.check_union(ObjectA, Q1, Q2)
- def test_BAB_BAC(self):
- Q1 = Q(objecta__objectb__name='deux')
- Q2 = Q(objecta__objectc__name='ein')
- self.check_union(ObjectB, Q1, Q2)
- def test_BAB_BACB(self):
- Q1 = Q(objecta__objectb__name='deux')
- Q2 = Q(objecta__objectc__objectb__name='trois')
- self.check_union(ObjectB, Q1, Q2)
- def test_BA_BCA__BAB_BAC_BCA(self):
- Q1 = Q(objecta__name='one', objectc__objecta__name='two')
- Q2 = Q(objecta__objectc__name='ein', objectc__objecta__name='three', objecta__objectb__name='trois')
- self.check_union(ObjectB, Q1, Q2)
- class DefaultValuesInsertTest(TestCase):
- def test_no_extra_params(self):
- # Ticket #17056 -- affects Oracle
- try:
- DumbCategory.objects.create()
- except TypeError:
- self.fail("Creation of an instance of a model with only the PK field shouldn't error out after bulk insert refactoring (#17056)")
- class ExcludeTests(TestCase):
- @classmethod
- def setUpTestData(cls):
- f1 = Food.objects.create(name='apples')
- Food.objects.create(name='oranges')
- Eaten.objects.create(food=f1, meal='dinner')
- j1 = Job.objects.create(name='Manager')
- r1 = Responsibility.objects.create(description='Playing golf')
- j2 = Job.objects.create(name='Programmer')
- r2 = Responsibility.objects.create(description='Programming')
- JobResponsibilities.objects.create(job=j1, responsibility=r1)
- JobResponsibilities.objects.create(job=j2, responsibility=r2)
- def test_to_field(self):
- self.assertQuerysetEqual(
- Food.objects.exclude(eaten__meal='dinner'),
- ['<Food: oranges>'])
- self.assertQuerysetEqual(
- Job.objects.exclude(responsibilities__description='Playing golf'),
- ['<Job: Programmer>'])
- self.assertQuerysetEqual(
- Responsibility.objects.exclude(jobs__name='Manager'),
- ['<Responsibility: Programming>'])
- def test_ticket14511(self):
- alex = Person.objects.get_or_create(name='Alex')[0]
- jane = Person.objects.get_or_create(name='Jane')[0]
- oracle = Company.objects.get_or_create(name='Oracle')[0]
- google = Company.objects.get_or_create(name='Google')[0]
- microsoft = Company.objects.get_or_create(name='Microsoft')[0]
- intel = Company.objects.get_or_create(name='Intel')[0]
- def employ(employer, employee, title):
- Employment.objects.get_or_create(employee=employee, employer=employer, title=title)
- employ(oracle, alex, 'Engineer')
- employ(oracle, alex, 'Developer')
- employ(google, alex, 'Engineer')
- employ(google, alex, 'Manager')
- employ(microsoft, alex, 'Manager')
- employ(intel, alex, 'Manager')
- employ(microsoft, jane, 'Developer')
- employ(intel, jane, 'Manager')
- alex_tech_employers = alex.employers.filter(
- employment__title__in=('Engineer', 'Developer')).distinct().order_by('name')
- self.assertQuerysetEqual(alex_tech_employers, [google, oracle], lambda x: x)
- alex_nontech_employers = alex.employers.exclude(
- employment__title__in=('Engineer', 'Developer')).distinct().order_by('name')
- self.assertQuerysetEqual(alex_nontech_employers, [google, intel, microsoft], lambda x: x)
- class ExcludeTest17600(TestCase):
- """
- Some regressiontests for ticket #17600. Some of these likely duplicate
- other existing tests.
- """
- @classmethod
- def setUpTestData(cls):
- # Create a few Orders.
- cls.o1 = Order.objects.create(pk=1)
- cls.o2 = Order.objects.create(pk=2)
- cls.o3 = Order.objects.create(pk=3)
- # Create some OrderItems for the first order with homogeneous
- # status_id values
- cls.oi1 = OrderItem.objects.create(order=cls.o1, status=1)
- cls.oi2 = OrderItem.objects.create(order=cls.o1, status=1)
- cls.oi3 = OrderItem.objects.create(order=cls.o1, status=1)
- # Create some OrderItems for the second order with heterogeneous
- # status_id values
- cls.oi4 = OrderItem.objects.create(order=cls.o2, status=1)
- cls.oi5 = OrderItem.objects.create(order=cls.o2, status=2)
- cls.oi6 = OrderItem.objects.create(order=cls.o2, status=3)
- # Create some OrderItems for the second order with heterogeneous
- # status_id values
- cls.oi7 = OrderItem.objects.create(order=cls.o3, status=2)
- cls.oi8 = OrderItem.objects.create(order=cls.o3, status=3)
- cls.oi9 = OrderItem.objects.create(order=cls.o3, status=4)
- def test_exclude_plain(self):
- """
- This should exclude Orders which have some items with status 1
- """
- self.assertQuerysetEqual(
- Order.objects.exclude(items__status=1),
- ['<Order: 3>'])
- def test_exclude_plain_distinct(self):
- """
- This should exclude Orders which have some items with status 1
- """
- self.assertQuerysetEqual(
- Order.objects.exclude(items__status=1).distinct(),
- ['<Order: 3>'])
- def test_exclude_with_q_object_distinct(self):
- """
- This should exclude Orders which have some items with status 1
- """
- self.assertQuerysetEqual(
- Order.objects.exclude(Q(items__status=1)).distinct(),
- ['<Order: 3>'])
- def test_exclude_with_q_object_no_distinct(self):
- """
- This should exclude Orders which have some items with status 1
- """
- self.assertQuerysetEqual(
- Order.objects.exclude(Q(items__status=1)),
- ['<Order: 3>'])
- def test_exclude_with_q_is_equal_to_plain_exclude(self):
- """
- Using exclude(condition) and exclude(Q(condition)) should
- yield the same QuerySet
- """
- self.assertEqual(
- list(Order.objects.exclude(items__status=1).distinct()),
- list(Order.objects.exclude(Q(items__status=1)).distinct()))
- def test_exclude_with_q_is_equal_to_plain_exclude_variation(self):
- """
- Using exclude(condition) and exclude(Q(condition)) should
- yield the same QuerySet
- """
- self.assertEqual(
- list(Order.objects.exclude(items__status=1)),
- list(Order.objects.exclude(Q(items__status=1)).distinct()))
- @unittest.expectedFailure
- def test_only_orders_with_all_items_having_status_1(self):
- """
- This should only return orders having ALL items set to status 1, or
- those items not having any orders at all. The correct way to write
- this query in SQL seems to be using two nested subqueries.
- """
- self.assertQuerysetEqual(
- Order.objects.exclude(~Q(items__status=1)).distinct(),
- ['<Order: 1>'])
- class Exclude15786(TestCase):
- """Regression test for #15786"""
- def test_ticket15786(self):
- c1 = SimpleCategory.objects.create(name='c1')
- c2 = SimpleCategory.objects.create(name='c2')
- OneToOneCategory.objects.create(category=c1)
- OneToOneCategory.objects.create(category=c2)
- rel = CategoryRelationship.objects.create(first=c1, second=c2)
- self.assertEqual(
- CategoryRelationship.objects.exclude(
- first__onetoonecategory=F('second__onetoonecategory')
- ).get(), rel
- )
- class NullInExcludeTest(TestCase):
- @classmethod
- def setUpTestData(cls):
- NullableName.objects.create(name='i1')
- NullableName.objects.create()
- def test_null_in_exclude_qs(self):
- none_val = '' if connection.features.interprets_empty_strings_as_nulls else None
- self.assertQuerysetEqual(
- NullableName.objects.exclude(name__in=[]),
- ['i1', none_val], attrgetter('name'))
- self.assertQuerysetEqual(
- NullableName.objects.exclude(name__in=['i1']),
- [none_val], attrgetter('name'))
- self.assertQuerysetEqual(
- NullableName.objects.exclude(name__in=['i3']),
- ['i1', none_val], attrgetter('name'))
- inner_qs = NullableName.objects.filter(name='i1').values_list('name')
- self.assertQuerysetEqual(
- NullableName.objects.exclude(name__in=inner_qs),
- [none_val], attrgetter('name'))
- # Check that the inner queryset wasn't executed - it should be turned
- # into subquery above
- self.assertIs(inner_qs._result_cache, None)
- @unittest.expectedFailure
- def test_col_not_in_list_containing_null(self):
- """
- The following case is not handled properly because
- SQL's COL NOT IN (list containing null) handling is too weird to
- abstract away.
- """
- self.assertQuerysetEqual(
- NullableName.objects.exclude(name__in=[None]),
- ['i1'], attrgetter('name'))
- def test_double_exclude(self):
- self.assertEqual(
- list(NullableName.objects.filter(~~Q(name='i1'))),
- list(NullableName.objects.filter(Q(name='i1'))))
- self.assertNotIn(
- 'IS NOT NULL',
- str(NullableName.objects.filter(~~Q(name='i1')).query))
- class EmptyStringsAsNullTest(TestCase):
- """
- Test that filtering on non-null character fields works as expected.
- The reason for these tests is that Oracle treats '' as NULL, and this
- can cause problems in query construction. Refs #17957.
- """
- @classmethod
- def setUpTestData(cls):
- cls.nc = NamedCategory.objects.create(name='')
- def test_direct_exclude(self):
- self.assertQuerysetEqual(
- NamedCategory.objects.exclude(name__in=['nonexisting']),
- [self.nc.pk], attrgetter('pk')
- )
- def test_joined_exclude(self):
- self.assertQuerysetEqual(
- DumbCategory.objects.exclude(namedcategory__name__in=['nonexisting']),
- [self.nc.pk], attrgetter('pk')
- )
- def test_21001(self):
- foo = NamedCategory.objects.create(name='foo')
- self.assertQuerysetEqual(
- NamedCategory.objects.exclude(name=''),
- [foo.pk], attrgetter('pk')
- )
- class ProxyQueryCleanupTest(TestCase):
- def test_evaluated_proxy_count(self):
- """
- Test that generating the query string doesn't alter the query's state
- in irreversible ways. Refs #18248.
- """
- ProxyCategory.objects.create()
- qs = ProxyCategory.objects.all()
- self.assertEqual(qs.count(), 1)
- str(qs.query)
- self.assertEqual(qs.count(), 1)
- class WhereNodeTest(TestCase):
- class DummyNode(object):
- def as_sql(self, compiler, connection):
- return 'dummy', []
- class MockCompiler(object):
- def compile(self, node):
- return node.as_sql(self, connection)
- def __call__(self, name):
- return connection.ops.quote_name(name)
- def test_empty_full_handling_conjunction(self):
- compiler = WhereNodeTest.MockCompiler()
- w = WhereNode(children=[NothingNode()])
- self.assertRaises(EmptyResultSet, w.as_sql, compiler, connection)
- w.negate()
- self.assertEqual(w.as_sql(compiler, connection), ('', []))
- w = WhereNode(children=[self.DummyNode(), self.DummyNode()])
- self.assertEqual(w.as_sql(compiler, connection), ('(dummy AND dummy)', []))
- w.negate()
- self.assertEqual(w.as_sql(compiler, connection), ('NOT (dummy AND dummy)', []))
- w = WhereNode(children=[NothingNode(), self.DummyNode()])
- self.assertRaises(EmptyResultSet, w.as_sql, compiler, connection)
- w.negate()
- self.assertEqual(w.as_sql(compiler, connection), ('', []))
- def test_empty_full_handling_disjunction(self):
- compiler = WhereNodeTest.MockCompiler()
- w = WhereNode(children=[NothingNode()], connector='OR')
- self.assertRaises(EmptyResultSet, w.as_sql, compiler, connection)
- w.negate()
- self.assertEqual(w.as_sql(compiler, connection), ('', []))
- w = WhereNode(children=[self.DummyNode(), self.DummyNode()], connector='OR')
- self.assertEqual(w.as_sql(compiler, connection), ('(dummy OR dummy)', []))
- w.negate()
- self.assertEqual(w.as_sql(compiler, connection), ('NOT (dummy OR dummy)', []))
- w = WhereNode(children=[NothingNode(), self.DummyNode()], connector='OR')
- self.assertEqual(w.as_sql(compiler, connection), ('dummy', []))
- w.negate()
- self.assertEqual(w.as_sql(compiler, connection), ('NOT (dummy)', []))
- def test_empty_nodes(self):
- compiler = WhereNodeTest.MockCompiler()
- empty_w = WhereNode()
- w = WhereNode(children=[empty_w, empty_w])
- self.assertEqual(w.as_sql(compiler, connection), ('', []))
- w.negate()
- with self.assertRaises(EmptyResultSet):
- w.as_sql(compiler, connection)
- w.connector = 'OR'
- with self.assertRaises(EmptyResultSet):
- w.as_sql(compiler, connection)
- w.negate()
- self.assertEqual(w.as_sql(compiler, connection), ('', []))
- w = WhereNode(children=[empty_w, NothingNode()], connector='OR')
- self.assertEqual(w.as_sql(compiler, connection), ('', []))
- w = WhereNode(children=[empty_w, NothingNode()], connector='AND')
- with self.assertRaises(EmptyResultSet):
- w.as_sql(compiler, connection)
- class IteratorExceptionsTest(TestCase):
- def test_iter_exceptions(self):
- qs = ExtraInfo.objects.only('author')
- with self.assertRaises(AttributeError):
- list(qs)
- def test_invalid_qs_list(self):
- # Test for #19895 - second iteration over invalid queryset
- # raises errors.
- qs = Article.objects.order_by('invalid_column')
- self.assertRaises(FieldError, list, qs)
- self.assertRaises(FieldError, list, qs)
- class NullJoinPromotionOrTest(TestCase):
- @classmethod
- def setUpTestData(cls):
- cls.d1 = ModelD.objects.create(name='foo')
- d2 = ModelD.objects.create(name='bar')
- cls.a1 = ModelA.objects.create(name='a1', d=cls.d1)
- c = ModelC.objects.create(name='c')
- b = ModelB.objects.create(name='b', c=c)
- cls.a2 = ModelA.objects.create(name='a2', b=b, d=d2)
- def test_ticket_17886(self):
- # The first Q-object is generating the match, the rest of the filters
- # should not remove the match even if they do not match anything. The
- # problem here was that b__name generates a LOUTER JOIN, then
- # b__c__name generates join to c, which the ORM tried to promote but
- # failed as that join isn't nullable.
- q_obj = (
- Q(d__name='foo') |
- Q(b__name='foo') |
- Q(b__c__name='foo')
- )
- qset = ModelA.objects.filter(q_obj)
- self.assertEqual(list(qset), [self.a1])
- # We generate one INNER JOIN to D. The join is direct and not nullable
- # so we can use INNER JOIN for it. However, we can NOT use INNER JOIN
- # for the b->c join, as a->b is nullable.
- self.assertEqual(str(qset.query).count('INNER JOIN'), 1)
- def test_isnull_filter_promotion(self):
- qs = ModelA.objects.filter(Q(b__name__isnull=True))
- self.assertEqual(str(qs.query).count('LEFT OUTER'), 1)
- self.assertEqual(list(qs), [self.a1])
- qs = ModelA.objects.filter(~Q(b__name__isnull=True))
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- self.assertEqual(list(qs), [self.a2])
- qs = ModelA.objects.filter(~~Q(b__name__isnull=True))
- self.assertEqual(str(qs.query).count('LEFT OUTER'), 1)
- self.assertEqual(list(qs), [self.a1])
- qs = ModelA.objects.filter(Q(b__name__isnull=False))
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- self.assertEqual(list(qs), [self.a2])
- qs = ModelA.objects.filter(~Q(b__name__isnull=False))
- self.assertEqual(str(qs.query).count('LEFT OUTER'), 1)
- self.assertEqual(list(qs), [self.a1])
- qs = ModelA.objects.filter(~~Q(b__name__isnull=False))
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- self.assertEqual(list(qs), [self.a2])
- def test_null_join_demotion(self):
- qs = ModelA.objects.filter(Q(b__name__isnull=False) & Q(b__name__isnull=True))
- self.assertIn(' INNER JOIN ', str(qs.query))
- qs = ModelA.objects.filter(Q(b__name__isnull=True) & Q(b__name__isnull=False))
- self.assertIn(' INNER JOIN ', str(qs.query))
- qs = ModelA.objects.filter(Q(b__name__isnull=False) | Q(b__name__isnull=True))
- self.assertIn(' LEFT OUTER JOIN ', str(qs.query))
- qs = ModelA.objects.filter(Q(b__name__isnull=True) | Q(b__name__isnull=False))
- self.assertIn(' LEFT OUTER JOIN ', str(qs.query))
- def test_ticket_21366(self):
- n = Note.objects.create(note='n', misc='m')
- e = ExtraInfo.objects.create(info='info', note=n)
- a = Author.objects.create(name='Author1', num=1, extra=e)
- Ranking.objects.create(rank=1, author=a)
- r1 = Report.objects.create(name='Foo', creator=a)
- r2 = Report.objects.create(name='Bar')
- Report.objects.create(name='Bar', creator=a)
- qs = Report.objects.filter(
- Q(creator__ranking__isnull=True) |
- Q(creator__ranking__rank=1, name='Foo')
- )
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 2)
- self.assertEqual(str(qs.query).count(' JOIN '), 2)
- self.assertQuerysetEqual(
- qs.order_by('name'), [r2, r1], lambda x: x)
- def test_ticket_21748(self):
- i1 = Identifier.objects.create(name='i1')
- i2 = Identifier.objects.create(name='i2')
- i3 = Identifier.objects.create(name='i3')
- Program.objects.create(identifier=i1)
- Channel.objects.create(identifier=i1)
- Program.objects.create(identifier=i2)
- self.assertQuerysetEqual(
- Identifier.objects.filter(program=None, channel=None),
- [i3], lambda x: x)
- self.assertQuerysetEqual(
- Identifier.objects.exclude(program=None, channel=None).order_by('name'),
- [i1, i2], lambda x: x)
- def test_ticket_21748_double_negated_and(self):
- i1 = Identifier.objects.create(name='i1')
- i2 = Identifier.objects.create(name='i2')
- Identifier.objects.create(name='i3')
- p1 = Program.objects.create(identifier=i1)
- c1 = Channel.objects.create(identifier=i1)
- Program.objects.create(identifier=i2)
- # Check the ~~Q() (or equivalently .exclude(~Q)) works like Q() for
- # join promotion.
- qs1_doubleneg = Identifier.objects.exclude(~Q(program__id=p1.id, channel__id=c1.id)).order_by('pk')
- qs1_filter = Identifier.objects.filter(program__id=p1.id, channel__id=c1.id).order_by('pk')
- self.assertQuerysetEqual(qs1_doubleneg, qs1_filter, lambda x: x)
- self.assertEqual(str(qs1_filter.query).count('JOIN'),
- str(qs1_doubleneg.query).count('JOIN'))
- self.assertEqual(2, str(qs1_doubleneg.query).count('INNER JOIN'))
- self.assertEqual(str(qs1_filter.query).count('INNER JOIN'),
- str(qs1_doubleneg.query).count('INNER JOIN'))
- def test_ticket_21748_double_negated_or(self):
- i1 = Identifier.objects.create(name='i1')
- i2 = Identifier.objects.create(name='i2')
- Identifier.objects.create(name='i3')
- p1 = Program.objects.create(identifier=i1)
- c1 = Channel.objects.create(identifier=i1)
- p2 = Program.objects.create(identifier=i2)
- # Test OR + doubleneq. The expected result is that channel is LOUTER
- # joined, program INNER joined
- qs1_filter = Identifier.objects.filter(
- Q(program__id=p2.id, channel__id=c1.id)
- | Q(program__id=p1.id)
- ).order_by('pk')
- qs1_doubleneg = Identifier.objects.exclude(
- ~Q(Q(program__id=p2.id, channel__id=c1.id)
- | Q(program__id=p1.id))
- ).order_by('pk')
- self.assertQuerysetEqual(qs1_doubleneg, qs1_filter, lambda x: x)
- self.assertEqual(str(qs1_filter.query).count('JOIN'),
- str(qs1_doubleneg.query).count('JOIN'))
- self.assertEqual(1, str(qs1_doubleneg.query).count('INNER JOIN'))
- self.assertEqual(str(qs1_filter.query).count('INNER JOIN'),
- str(qs1_doubleneg.query).count('INNER JOIN'))
- def test_ticket_21748_complex_filter(self):
- i1 = Identifier.objects.create(name='i1')
- i2 = Identifier.objects.create(name='i2')
- Identifier.objects.create(name='i3')
- p1 = Program.objects.create(identifier=i1)
- c1 = Channel.objects.create(identifier=i1)
- p2 = Program.objects.create(identifier=i2)
- # Finally, a more complex case, one time in a way where each
- # NOT is pushed to lowest level in the boolean tree, and
- # another query where this isn't done.
- qs1 = Identifier.objects.filter(
- ~Q(~Q(program__id=p2.id, channel__id=c1.id)
- & Q(program__id=p1.id))).order_by('pk')
- qs2 = Identifier.objects.filter(
- Q(Q(program__id=p2.id, channel__id=c1.id)
- | ~Q(program__id=p1.id))).order_by('pk')
- self.assertQuerysetEqual(qs1, qs2, lambda x: x)
- self.assertEqual(str(qs1.query).count('JOIN'),
- str(qs2.query).count('JOIN'))
- self.assertEqual(0, str(qs1.query).count('INNER JOIN'))
- self.assertEqual(str(qs1.query).count('INNER JOIN'),
- str(qs2.query).count('INNER JOIN'))
- class ReverseJoinTrimmingTest(TestCase):
- def test_reverse_trimming(self):
- # Check that we don't accidentally trim reverse joins - we can't know
- # if there is anything on the other side of the join, so trimming
- # reverse joins can't be done, ever.
- t = Tag.objects.create()
- qs = Tag.objects.filter(annotation__tag=t.pk)
- self.assertIn('INNER JOIN', str(qs.query))
- self.assertEqual(list(qs), [])
- class JoinReuseTest(TestCase):
- """
- Test that the queries reuse joins sensibly (for example, direct joins
- are always reused).
- """
- def test_fk_reuse(self):
- qs = Annotation.objects.filter(tag__name='foo').filter(tag__name='bar')
- self.assertEqual(str(qs.query).count('JOIN'), 1)
- def test_fk_reuse_select_related(self):
- qs = Annotation.objects.filter(tag__name='foo').select_related('tag')
- self.assertEqual(str(qs.query).count('JOIN'), 1)
- def test_fk_reuse_annotation(self):
- qs = Annotation.objects.filter(tag__name='foo').annotate(cnt=Count('tag__name'))
- self.assertEqual(str(qs.query).count('JOIN'), 1)
- def test_fk_reuse_disjunction(self):
- qs = Annotation.objects.filter(Q(tag__name='foo') | Q(tag__name='bar'))
- self.assertEqual(str(qs.query).count('JOIN'), 1)
- def test_fk_reuse_order_by(self):
- qs = Annotation.objects.filter(tag__name='foo').order_by('tag__name')
- self.assertEqual(str(qs.query).count('JOIN'), 1)
- def test_revo2o_reuse(self):
- qs = Detail.objects.filter(member__name='foo').filter(member__name='foo')
- self.assertEqual(str(qs.query).count('JOIN'), 1)
- def test_revfk_noreuse(self):
- qs = Author.objects.filter(report__name='r4').filter(report__name='r1')
- self.assertEqual(str(qs.query).count('JOIN'), 2)
- class DisjunctionPromotionTests(TestCase):
- def test_disjuction_promotion_select_related(self):
- fk1 = FK1.objects.create(f1='f1', f2='f2')
- basea = BaseA.objects.create(a=fk1)
- qs = BaseA.objects.filter(Q(a=fk1) | Q(b=2))
- self.assertEqual(str(qs.query).count(' JOIN '), 0)
- qs = qs.select_related('a', 'b')
- self.assertEqual(str(qs.query).count(' INNER JOIN '), 0)
- self.assertEqual(str(qs.query).count(' LEFT OUTER JOIN '), 2)
- with self.assertNumQueries(1):
- self.assertQuerysetEqual(qs, [basea], lambda x: x)
- self.assertEqual(qs[0].a, fk1)
- self.assertIs(qs[0].b, None)
- def test_disjunction_promotion1(self):
- # Pre-existing join, add two ORed filters to the same join,
- # all joins can be INNER JOINS.
- qs = BaseA.objects.filter(a__f1='foo')
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- qs = qs.filter(Q(b__f1='foo') | Q(b__f2='foo'))
- self.assertEqual(str(qs.query).count('INNER JOIN'), 2)
- # Reverse the order of AND and OR filters.
- qs = BaseA.objects.filter(Q(b__f1='foo') | Q(b__f2='foo'))
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- qs = qs.filter(a__f1='foo')
- self.assertEqual(str(qs.query).count('INNER JOIN'), 2)
- def test_disjunction_promotion2(self):
- qs = BaseA.objects.filter(a__f1='foo')
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- # Now we have two different joins in an ORed condition, these
- # must be OUTER joins. The pre-existing join should remain INNER.
- qs = qs.filter(Q(b__f1='foo') | Q(c__f2='foo'))
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 2)
- # Reverse case.
- qs = BaseA.objects.filter(Q(b__f1='foo') | Q(c__f2='foo'))
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 2)
- qs = qs.filter(a__f1='foo')
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 2)
- def test_disjunction_promotion3(self):
- qs = BaseA.objects.filter(a__f2='bar')
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- # The ANDed a__f2 filter allows us to use keep using INNER JOIN
- # even inside the ORed case. If the join to a__ returns nothing,
- # the ANDed filter for a__f2 can't be true.
- qs = qs.filter(Q(a__f1='foo') | Q(b__f2='foo'))
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 1)
- def test_disjunction_promotion3_demote(self):
- # This one needs demotion logic: the first filter causes a to be
- # outer joined, the second filter makes it inner join again.
- qs = BaseA.objects.filter(
- Q(a__f1='foo') | Q(b__f2='foo')).filter(a__f2='bar')
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 1)
- def test_disjunction_promotion4_demote(self):
- qs = BaseA.objects.filter(Q(a=1) | Q(a=2))
- self.assertEqual(str(qs.query).count('JOIN'), 0)
- # Demote needed for the "a" join. It is marked as outer join by
- # above filter (even if it is trimmed away).
- qs = qs.filter(a__f1='foo')
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- def test_disjunction_promotion4(self):
- qs = BaseA.objects.filter(a__f1='foo')
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- qs = qs.filter(Q(a=1) | Q(a=2))
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- def test_disjunction_promotion5_demote(self):
- qs = BaseA.objects.filter(Q(a=1) | Q(a=2))
- # Note that the above filters on a force the join to an
- # inner join even if it is trimmed.
- self.assertEqual(str(qs.query).count('JOIN'), 0)
- qs = qs.filter(Q(a__f1='foo') | Q(b__f1='foo'))
- # So, now the a__f1 join doesn't need promotion.
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- # But b__f1 does.
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 1)
- qs = BaseA.objects.filter(Q(a__f1='foo') | Q(b__f1='foo'))
- # Now the join to a is created as LOUTER
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 2)
- qs = qs.filter(Q(a=1) | Q(a=2))
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 1)
- def test_disjunction_promotion6(self):
- qs = BaseA.objects.filter(Q(a=1) | Q(a=2))
- self.assertEqual(str(qs.query).count('JOIN'), 0)
- qs = BaseA.objects.filter(Q(a__f1='foo') & Q(b__f1='foo'))
- self.assertEqual(str(qs.query).count('INNER JOIN'), 2)
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 0)
- qs = BaseA.objects.filter(Q(a__f1='foo') & Q(b__f1='foo'))
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 0)
- self.assertEqual(str(qs.query).count('INNER JOIN'), 2)
- qs = qs.filter(Q(a=1) | Q(a=2))
- self.assertEqual(str(qs.query).count('INNER JOIN'), 2)
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 0)
- def test_disjunction_promotion7(self):
- qs = BaseA.objects.filter(Q(a=1) | Q(a=2))
- self.assertEqual(str(qs.query).count('JOIN'), 0)
- qs = BaseA.objects.filter(Q(a__f1='foo') | (Q(b__f1='foo') & Q(a__f1='bar')))
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 1)
- qs = BaseA.objects.filter(
- (Q(a__f1='foo') | Q(b__f1='foo')) & (Q(a__f1='bar') | Q(c__f1='foo'))
- )
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 3)
- self.assertEqual(str(qs.query).count('INNER JOIN'), 0)
- qs = BaseA.objects.filter(
- (Q(a__f1='foo') | (Q(a__f1='bar')) & (Q(b__f1='bar') | Q(c__f1='foo')))
- )
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 2)
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- def test_disjunction_promotion_fexpression(self):
- qs = BaseA.objects.filter(Q(a__f1=F('b__f1')) | Q(b__f1='foo'))
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 1)
- self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
- qs = BaseA.objects.filter(Q(a__f1=F('c__f1')) | Q(b__f1='foo'))
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 3)
- qs = BaseA.objects.filter(Q(a__f1=F('b__f1')) | Q(a__f2=F('b__f2')) | Q(c__f1='foo'))
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 3)
- qs = BaseA.objects.filter(Q(a__f1=F('c__f1')) | (Q(pk=1) & Q(pk=2)))
- self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 2)
- self.assertEqual(str(qs.query).count('INNER JOIN'), 0)
- class ManyToManyExcludeTest(TestCase):
- def test_exclude_many_to_many(self):
- Identifier.objects.create(name='extra')
- program = Program.objects.create(identifier=Identifier.objects.create(name='program'))
- channel = Channel.objects.create(identifier=Identifier.objects.create(name='channel'))
- channel.programs.add(program)
- # channel contains 'program1', so all Identifiers except that one
- # should be returned
- self.assertQuerysetEqual(
- Identifier.objects.exclude(program__channel=channel).order_by('name'),
- ['<Identifier: channel>', '<Identifier: extra>']
- )
- self.assertQuerysetEqual(
- Identifier.objects.exclude(program__channel=None).order_by('name'),
- ['<Identifier: program>']
- )
- def test_ticket_12823(self):
- pg3 = Page.objects.create(text='pg3')
- pg2 = Page.objects.create(text='pg2')
- pg1 = Page.objects.create(text='pg1')
- pa1 = Paragraph.objects.create(text='pa1')
- pa1.page = [pg1, pg2]
- pa2 = Paragraph.objects.create(text='pa2')
- pa2.page = [pg2, pg3]
- pa3 = Paragraph.objects.create(text='pa3')
- ch1 = Chapter.objects.create(title='ch1', paragraph=pa1)
- ch2 = Chapter.objects.create(title='ch2', paragraph=pa2)
- ch3 = Chapter.objects.create(title='ch3', paragraph=pa3)
- b1 = Book.objects.create(title='b1', chapter=ch1)
- b2 = Book.objects.create(title='b2', chapter=ch2)
- b3 = Book.objects.create(title='b3', chapter=ch3)
- q = Book.objects.exclude(chapter__paragraph__page__text='pg1')
- self.assertNotIn('IS NOT NULL', str(q.query))
- self.assertEqual(len(q), 2)
- self.assertNotIn(b1, q)
- self.assertIn(b2, q)
- self.assertIn(b3, q)
- class RelabelCloneTest(TestCase):
- def test_ticket_19964(self):
- my1 = MyObject.objects.create(data='foo')
- my1.parent = my1
- my1.save()
- my2 = MyObject.objects.create(data='bar', parent=my1)
- parents = MyObject.objects.filter(parent=F('id'))
- children = MyObject.objects.filter(parent__in=parents).exclude(parent=F('id'))
- self.assertEqual(list(parents), [my1])
- # Evaluating the children query (which has parents as part of it) does
- # not change results for the parents query.
- self.assertEqual(list(children), [my2])
- self.assertEqual(list(parents), [my1])
- class Ticket20101Tests(TestCase):
- def test_ticket_20101(self):
- """
- Tests QuerySet ORed combining in exclude subquery case.
- """
- t = Tag.objects.create(name='foo')
- a1 = Annotation.objects.create(tag=t, name='a1')
- a2 = Annotation.objects.create(tag=t, name='a2')
- a3 = Annotation.objects.create(tag=t, name='a3')
- n = Note.objects.create(note='foo', misc='bar')
- qs1 = Note.objects.exclude(annotation__in=[a1, a2])
- qs2 = Note.objects.filter(annotation__in=[a3])
- self.assertIn(n, qs1)
- self.assertNotIn(n, qs2)
- self.assertIn(n, (qs1 | qs2))
- class EmptyStringPromotionTests(TestCase):
- def test_empty_string_promotion(self):
- qs = RelatedObject.objects.filter(single__name='')
- if connection.features.interprets_empty_strings_as_nulls:
- self.assertIn('LEFT OUTER JOIN', str(qs.query))
- else:
- self.assertNotIn('LEFT OUTER JOIN', str(qs.query))
- class ValuesSubqueryTests(TestCase):
- def test_values_in_subquery(self):
- # Check that if a values() queryset is used, then the given values
- # will be used instead of forcing use of the relation's field.
- o1 = Order.objects.create(id=-2)
- o2 = Order.objects.create(id=-1)
- oi1 = OrderItem.objects.create(order=o1, status=0)
- oi1.status = oi1.pk
- oi1.save()
- OrderItem.objects.create(order=o2, status=0)
- # The query below should match o1 as it has related order_item
- # with id == status.
- self.assertQuerysetEqual(
- Order.objects.filter(items__in=OrderItem.objects.values_list('status')),
- [o1.pk], lambda x: x.pk)
- class DoubleInSubqueryTests(TestCase):
- def test_double_subquery_in(self):
- lfa1 = LeafA.objects.create(data='foo')
- lfa2 = LeafA.objects.create(data='bar')
- lfb1 = LeafB.objects.create(data='lfb1')
- lfb2 = LeafB.objects.create(data='lfb2')
- Join.objects.create(a=lfa1, b=lfb1)
- Join.objects.create(a=lfa2, b=lfb2)
- leaf_as = LeafA.objects.filter(data='foo').values_list('pk', flat=True)
- joins = Join.objects.filter(a__in=leaf_as).values_list('b__id', flat=True)
- qs = LeafB.objects.filter(pk__in=joins)
- self.assertQuerysetEqual(
- qs, [lfb1], lambda x: x)
- class Ticket18785Tests(TestCase):
- def test_ticket_18785(self):
- # Test join trimming from ticket18785
- qs = Item.objects.exclude(
- note__isnull=False
- ).filter(
- name='something', creator__extra__isnull=True
- ).order_by()
- self.assertEqual(1, str(qs.query).count('INNER JOIN'))
- self.assertEqual(0, str(qs.query).count('OUTER JOIN'))
- class Ticket20788Tests(TestCase):
- def test_ticket_20788(self):
- Paragraph.objects.create()
- paragraph = Paragraph.objects.create()
- page = paragraph.page.create()
- chapter = Chapter.objects.create(paragraph=paragraph)
- Book.objects.create(chapter=chapter)
- paragraph2 = Paragraph.objects.create()
- Page.objects.create()
- chapter2 = Chapter.objects.create(paragraph=paragraph2)
- book2 = Book.objects.create(chapter=chapter2)
- sentences_not_in_pub = Book.objects.exclude(
- chapter__paragraph__page=page)
- self.assertQuerysetEqual(
- sentences_not_in_pub, [book2], lambda x: x)
- class Ticket12807Tests(TestCase):
- def test_ticket_12807(self):
- p1 = Paragraph.objects.create()
- p2 = Paragraph.objects.create()
- # The ORed condition below should have no effect on the query - the
- # ~Q(pk__in=[]) will always be True.
- qs = Paragraph.objects.filter((Q(pk=p2.pk) | ~Q(pk__in=[])) & Q(pk=p1.pk))
- self.assertQuerysetEqual(qs, [p1], lambda x: x)
- class RelatedLookupTypeTests(TestCase):
- error = 'Cannot query "%s": Must be "%s" instance.'
- @classmethod
- def setUpTestData(cls):
- cls.oa = ObjectA.objects.create(name="oa")
- cls.poa = ProxyObjectA.objects.get(name="oa")
- cls.coa = ChildObjectA.objects.create(name="coa")
- cls.wrong_type = Order.objects.create(id=cls.oa.pk)
- cls.ob = ObjectB.objects.create(name="ob", objecta=cls.oa, num=1)
- ProxyObjectB.objects.create(name="pob", objecta=cls.oa, num=2)
- cls.pob = ProxyObjectB.objects.all()
- ObjectC.objects.create(childobjecta=cls.coa)
- def test_wrong_type_lookup(self):
- """
- A ValueError is raised when the incorrect object type is passed to a
- query lookup.
- """
- # Passing incorrect object type
- with self.assertRaisesMessage(ValueError,
- self.error % (self.wrong_type, ObjectA._meta.object_name)):
- ObjectB.objects.get(objecta=self.wrong_type)
- with self.assertRaisesMessage(ValueError,
- self.error % (self.wrong_type, ObjectA._meta.object_name)):
- ObjectB.objects.filter(objecta__in=[self.wrong_type])
- with self.assertRaisesMessage(ValueError,
- self.error % (self.wrong_type, ObjectA._meta.object_name)):
- ObjectB.objects.filter(objecta=self.wrong_type)
- with self.assertRaisesMessage(ValueError,
- self.error % (self.wrong_type, ObjectB._meta.object_name)):
- ObjectA.objects.filter(objectb__in=[self.wrong_type, self.ob])
- # Passing an object of the class on which query is done.
- with self.assertRaisesMessage(ValueError,
- self.error % (self.ob, ObjectA._meta.object_name)):
- ObjectB.objects.filter(objecta__in=[self.poa, self.ob])
- with self.assertRaisesMessage(ValueError,
- self.error % (self.ob, ChildObjectA._meta.object_name)):
- ObjectC.objects.exclude(childobjecta__in=[self.coa, self.ob])
- def test_wrong_backward_lookup(self):
- """
- A ValueError is raised when the incorrect object type is passed to a
- query lookup for backward relations.
- """
- with self.assertRaisesMessage(ValueError,
- self.error % (self.oa, ObjectB._meta.object_name)):
- ObjectA.objects.filter(objectb__in=[self.oa, self.ob])
- with self.assertRaisesMessage(ValueError,
- self.error % (self.oa, ObjectB._meta.object_name)):
- ObjectA.objects.exclude(objectb=self.oa)
- with self.assertRaisesMessage(ValueError,
- self.error % (self.wrong_type, ObjectB._meta.object_name)):
- ObjectA.objects.get(objectb=self.wrong_type)
- def test_correct_lookup(self):
- """
- When passing proxy model objects, child objects, or parent objects,
- lookups work fine.
- """
- out_a = ['<ObjectA: oa>', ]
- out_b = ['<ObjectB: ob>', '<ObjectB: pob>']
- out_c = ['<ObjectC: >']
- # proxy model objects
- self.assertQuerysetEqual(ObjectB.objects.filter(objecta=self.poa).order_by('name'), out_b)
- self.assertQuerysetEqual(ObjectA.objects.filter(objectb__in=self.pob).order_by('pk'), out_a * 2)
- # child objects
- self.assertQuerysetEqual(ObjectB.objects.filter(objecta__in=[self.coa]), [])
- self.assertQuerysetEqual(ObjectB.objects.filter(objecta__in=[self.poa, self.coa]).order_by('name'), out_b)
- self.assertQuerysetEqual(ObjectB.objects.filter(objecta__in=iter([self.poa, self.coa])).order_by('name'), out_b)
- # parent objects
- self.assertQuerysetEqual(ObjectC.objects.exclude(childobjecta=self.oa), out_c)
- # QuerySet related object type checking shouldn't issue queries
- # (the querysets aren't evaluated here, hence zero queries) (#23266).
- with self.assertNumQueries(0):
- ObjectB.objects.filter(objecta__in=ObjectA.objects.all())
- def test_values_queryset_lookup(self):
- """
- #23396 - Ensure ValueQuerySets are not checked for compatibility with the lookup field
- """
- # Make sure the num and objecta field values match.
- ob = ObjectB.objects.get(name='ob')
- ob.num = ob.objecta.pk
- ob.save()
- pob = ObjectB.objects.get(name='pob')
- pob.num = pob.objecta.pk
- pob.save()
- self.assertQuerysetEqual(ObjectB.objects.filter(
- objecta__in=ObjectB.objects.all().values_list('num')
- ).order_by('pk'), ['<ObjectB: ob>', '<ObjectB: pob>'])
- class Ticket14056Tests(TestCase):
- def test_ticket_14056(self):
- s1 = SharedConnection.objects.create(data='s1')
- s2 = SharedConnection.objects.create(data='s2')
- s3 = SharedConnection.objects.create(data='s3')
- PointerA.objects.create(connection=s2)
- expected_ordering = (
- [s1, s3, s2] if connection.features.nulls_order_largest
- else [s2, s1, s3]
- )
- self.assertQuerysetEqual(
- SharedConnection.objects.order_by('-pointera__connection', 'pk'),
- expected_ordering, lambda x: x
- )
- class Ticket20955Tests(TestCase):
- def test_ticket_20955(self):
- jack = Staff.objects.create(name='jackstaff')
- jackstaff = StaffUser.objects.create(staff=jack)
- jill = Staff.objects.create(name='jillstaff')
- jillstaff = StaffUser.objects.create(staff=jill)
- task = Task.objects.create(creator=jackstaff, owner=jillstaff, title="task")
- task_get = Task.objects.get(pk=task.pk)
- # Load data so that assertNumQueries doesn't complain about the get
- # version's queries.
- task_get.creator.staffuser.staff
- task_get.owner.staffuser.staff
- qs = Task.objects.select_related(
- 'creator__staffuser__staff', 'owner__staffuser__staff')
- self.assertEqual(str(qs.query).count(' JOIN '), 6)
- task_select_related = qs.get(pk=task.pk)
- with self.assertNumQueries(0):
- self.assertEqual(task_select_related.creator.staffuser.staff,
- task_get.creator.staffuser.staff)
- self.assertEqual(task_select_related.owner.staffuser.staff,
- task_get.owner.staffuser.staff)
- class Ticket21203Tests(TestCase):
- def test_ticket_21203(self):
- p = Ticket21203Parent.objects.create(parent_bool=True)
- c = Ticket21203Child.objects.create(parent=p)
- qs = Ticket21203Child.objects.select_related('parent').defer('parent__created')
- self.assertQuerysetEqual(qs, [c], lambda x: x)
- self.assertIs(qs[0].parent.parent_bool, True)
- class ValuesJoinPromotionTests(TestCase):
- def test_values_no_promotion_for_existing(self):
- qs = Node.objects.filter(parent__parent__isnull=False)
- self.assertIn(' INNER JOIN ', str(qs.query))
- qs = qs.values('parent__parent__id')
- self.assertIn(' INNER JOIN ', str(qs.query))
- # Make sure there is a left outer join without the filter.
- qs = Node.objects.values('parent__parent__id')
- self.assertIn(' LEFT OUTER JOIN ', str(qs.query))
- def test_non_nullable_fk_not_promoted(self):
- qs = ObjectB.objects.values('objecta__name')
- self.assertIn(' INNER JOIN ', str(qs.query))
- def test_ticket_21376(self):
- a = ObjectA.objects.create()
- ObjectC.objects.create(objecta=a)
- qs = ObjectC.objects.filter(
- Q(objecta=a) | Q(objectb__objecta=a),
- )
- qs = qs.filter(
- Q(objectb=1) | Q(objecta=a),
- )
- self.assertEqual(qs.count(), 1)
- tblname = connection.ops.quote_name(ObjectB._meta.db_table)
- self.assertIn(' LEFT OUTER JOIN %s' % tblname, str(qs.query))
- class ForeignKeyToBaseExcludeTests(TestCase):
- def test_ticket_21787(self):
- sc1 = SpecialCategory.objects.create(special_name='sc1', name='sc1')
- sc2 = SpecialCategory.objects.create(special_name='sc2', name='sc2')
- sc3 = SpecialCategory.objects.create(special_name='sc3', name='sc3')
- c1 = CategoryItem.objects.create(category=sc1)
- CategoryItem.objects.create(category=sc2)
- self.assertQuerysetEqual(
- SpecialCategory.objects.exclude(
- categoryitem__id=c1.pk).order_by('name'),
- [sc2, sc3], lambda x: x
- )
- self.assertQuerysetEqual(
- SpecialCategory.objects.filter(categoryitem__id=c1.pk),
- [sc1], lambda x: x
- )
- class ReverseM2MCustomPkTests(TestCase):
- def test_ticket_21879(self):
- cpt1 = CustomPkTag.objects.create(id='cpt1', tag='cpt1')
- cp1 = CustomPk.objects.create(name='cp1', extra='extra')
- cp1.custompktag_set.add(cpt1)
- self.assertQuerysetEqual(
- CustomPk.objects.filter(custompktag=cpt1), [cp1],
- lambda x: x)
- self.assertQuerysetEqual(
- CustomPkTag.objects.filter(custom_pk=cp1), [cpt1],
- lambda x: x)
- class Ticket22429Tests(TestCase):
- def test_ticket_22429(self):
- sc1 = School.objects.create()
- st1 = Student.objects.create(school=sc1)
- sc2 = School.objects.create()
- st2 = Student.objects.create(school=sc2)
- cr = Classroom.objects.create(school=sc1)
- cr.students.add(st1)
- queryset = Student.objects.filter(~Q(classroom__school=F('school')))
- self.assertQuerysetEqual(queryset, [st2], lambda x: x)
- class Ticket23605Tests(TestCase):
- def test_ticket_23605(self):
- # Test filtering on a complicated q-object from ticket's report.
- # The query structure is such that we have multiple nested subqueries.
- # The original problem was that the inner queries weren't relabeled
- # correctly.
- # See also #24090.
- a1 = Ticket23605A.objects.create()
- a2 = Ticket23605A.objects.create()
- c1 = Ticket23605C.objects.create(field_c0=10000.0)
- Ticket23605B.objects.create(
- field_b0=10000.0, field_b1=True,
- modelc_fk=c1, modela_fk=a1)
- complex_q = Q(pk__in=Ticket23605A.objects.filter(
- Q(
- # True for a1 as field_b0 = 10000, field_c0=10000
- # False for a2 as no ticket23605b found
- ticket23605b__field_b0__gte=1000000 /
- F("ticket23605b__modelc_fk__field_c0")
- ) &
- # True for a1 (field_b1=True)
- Q(ticket23605b__field_b1=True) &
- ~Q(ticket23605b__pk__in=Ticket23605B.objects.filter(
- ~(
- # Same filters as above commented filters, but
- # double-negated (one for Q() above, one for
- # parentheses). So, again a1 match, a2 not.
- Q(field_b1=True) &
- Q(field_b0__gte=1000000 / F("modelc_fk__field_c0"))
- )
- ))).filter(ticket23605b__field_b1=True))
- qs1 = Ticket23605A.objects.filter(complex_q)
- self.assertQuerysetEqual(qs1, [a1], lambda x: x)
- qs2 = Ticket23605A.objects.exclude(complex_q)
- self.assertQuerysetEqual(qs2, [a2], lambda x: x)
- class TestTicket24279(TestCase):
- def test_ticket_24278(self):
- School.objects.create()
- qs = School.objects.filter(Q(pk__in=()) | Q())
- self.assertQuerysetEqual(qs, [])
- class TestInvalidValuesRelation(TestCase):
- def test_invalid_values(self):
- with self.assertRaises(ValueError):
- Annotation.objects.filter(tag='abc')
- with self.assertRaises(ValueError):
- Annotation.objects.filter(tag__in=[123, 'abc'])
- class TestTicket24605(TestCase):
- def test_ticket_24605(self):
- """
- Subquery table names should be quoted.
- """
- i1 = Individual.objects.create(alive=True)
- RelatedIndividual.objects.create(related=i1)
- i2 = Individual.objects.create(alive=False)
- RelatedIndividual.objects.create(related=i2)
- i3 = Individual.objects.create(alive=True)
- i4 = Individual.objects.create(alive=False)
- self.assertQuerysetEqual(
- Individual.objects.filter(Q(alive=False), Q(related_individual__isnull=True)),
- [i4], lambda x: x
- )
- self.assertQuerysetEqual(
- Individual.objects.exclude(
- Q(alive=False), Q(related_individual__isnull=True)
- ).order_by('pk'),
- [i1, i2, i3], lambda x: x
- )
- class Ticket23622Tests(TestCase):
- @skipUnlessDBFeature('can_distinct_on_fields')
- def test_ticket_23622(self):
- """
- Make sure __pk__in and __in work the same for related fields when
- using a distinct on subquery.
- """
- a1 = Ticket23605A.objects.create()
- a2 = Ticket23605A.objects.create()
- c1 = Ticket23605C.objects.create(field_c0=0.0)
- Ticket23605B.objects.create(
- modela_fk=a1, field_b0=123,
- field_b1=datetime.date(2013, 1, 6),
- modelc_fk=c1,
- )
- Ticket23605B.objects.create(
- modela_fk=a1, field_b0=23,
- field_b1=datetime.date(2011, 6, 6),
- modelc_fk=c1,
- )
- Ticket23605B.objects.create(
- modela_fk=a1, field_b0=234,
- field_b1=datetime.date(2011, 9, 2),
- modelc_fk=c1,
- )
- Ticket23605B.objects.create(
- modela_fk=a1, field_b0=12,
- field_b1=datetime.date(2012, 9, 15),
- modelc_fk=c1,
- )
- Ticket23605B.objects.create(
- modela_fk=a2, field_b0=567,
- field_b1=datetime.date(2014, 3, 1),
- modelc_fk=c1,
- )
- Ticket23605B.objects.create(
- modela_fk=a2, field_b0=76,
- field_b1=datetime.date(2011, 3, 3),
- modelc_fk=c1,
- )
- Ticket23605B.objects.create(
- modela_fk=a2, field_b0=7,
- field_b1=datetime.date(2012, 10, 20),
- modelc_fk=c1,
- )
- Ticket23605B.objects.create(
- modela_fk=a2, field_b0=56,
- field_b1=datetime.date(2011, 1, 27),
- modelc_fk=c1,
- )
- qx = (
- Q(ticket23605b__pk__in=Ticket23605B.objects.order_by('modela_fk', '-field_b1').distinct('modela_fk'))
- & Q(ticket23605b__field_b0__gte=300)
- )
- qy = (
- Q(ticket23605b__in=Ticket23605B.objects.order_by('modela_fk', '-field_b1').distinct('modela_fk'))
- & Q(ticket23605b__field_b0__gte=300)
- )
- self.assertEqual(
- set(Ticket23605A.objects.filter(qx).values_list('pk', flat=True)),
- set(Ticket23605A.objects.filter(qy).values_list('pk', flat=True))
- )
- self.assertQuerysetEqual(
- Ticket23605A.objects.filter(qx),
- [a2], lambda x: x
- )
|