tests.py 155 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819
  1. from __future__ import unicode_literals
  2. import datetime
  3. import pickle
  4. import unittest
  5. from collections import OrderedDict
  6. from operator import attrgetter
  7. from django.core.exceptions import FieldError
  8. from django.db import DEFAULT_DB_ALIAS, connection
  9. from django.db.models import Count, F, Q
  10. from django.db.models.sql.constants import LOUTER
  11. from django.db.models.sql.datastructures import EmptyResultSet
  12. from django.db.models.sql.where import NothingNode, WhereNode
  13. from django.test import TestCase, skipUnlessDBFeature
  14. from django.test.utils import CaptureQueriesContext
  15. from django.utils import six
  16. from django.utils.six.moves import range
  17. from .models import (
  18. FK1, Annotation, Article, Author, BaseA, Book, CategoryItem,
  19. CategoryRelationship, Celebrity, Channel, Chapter, Child, ChildObjectA,
  20. Classroom, Company, Cover, CustomPk, CustomPkTag, Detail, DumbCategory,
  21. Eaten, Employment, ExtraInfo, Fan, Food, Identifier, Individual, Item, Job,
  22. JobResponsibilities, Join, LeafA, LeafB, LoopX, LoopZ, ManagedModel,
  23. Member, ModelA, ModelB, ModelC, ModelD, MyObject, NamedCategory, Node,
  24. Note, NullableName, Number, ObjectA, ObjectB, ObjectC, OneToOneCategory,
  25. Order, OrderItem, Page, Paragraph, Person, Plaything, PointerA, Program,
  26. ProxyCategory, ProxyObjectA, ProxyObjectB, Ranking, Related,
  27. RelatedIndividual, RelatedObject, Report, ReservedName, Responsibility,
  28. School, SharedConnection, SimpleCategory, SingleObject, SpecialCategory,
  29. Staff, StaffUser, Student, Tag, Task, Ticket21203Child, Ticket21203Parent,
  30. Ticket23605A, Ticket23605B, Ticket23605C, TvChef, Valid, X,
  31. )
  32. class BaseQuerysetTest(TestCase):
  33. def assertValueQuerysetEqual(self, qs, values):
  34. return self.assertQuerysetEqual(qs, values, transform=lambda x: x)
  35. class Queries1Tests(BaseQuerysetTest):
  36. @classmethod
  37. def setUpTestData(cls):
  38. generic = NamedCategory.objects.create(name="Generic")
  39. cls.t1 = Tag.objects.create(name='t1', category=generic)
  40. cls.t2 = Tag.objects.create(name='t2', parent=cls.t1, category=generic)
  41. cls.t3 = Tag.objects.create(name='t3', parent=cls.t1)
  42. t4 = Tag.objects.create(name='t4', parent=cls.t3)
  43. cls.t5 = Tag.objects.create(name='t5', parent=cls.t3)
  44. cls.n1 = Note.objects.create(note='n1', misc='foo', id=1)
  45. n2 = Note.objects.create(note='n2', misc='bar', id=2)
  46. cls.n3 = Note.objects.create(note='n3', misc='foo', id=3)
  47. ann1 = Annotation.objects.create(name='a1', tag=cls.t1)
  48. ann1.notes.add(cls.n1)
  49. ann2 = Annotation.objects.create(name='a2', tag=t4)
  50. ann2.notes.add(n2, cls.n3)
  51. # Create these out of order so that sorting by 'id' will be different to sorting
  52. # by 'info'. Helps detect some problems later.
  53. cls.e2 = ExtraInfo.objects.create(info='e2', note=n2, value=41)
  54. e1 = ExtraInfo.objects.create(info='e1', note=cls.n1, value=42)
  55. cls.a1 = Author.objects.create(name='a1', num=1001, extra=e1)
  56. cls.a2 = Author.objects.create(name='a2', num=2002, extra=e1)
  57. a3 = Author.objects.create(name='a3', num=3003, extra=cls.e2)
  58. cls.a4 = Author.objects.create(name='a4', num=4004, extra=cls.e2)
  59. cls.time1 = datetime.datetime(2007, 12, 19, 22, 25, 0)
  60. cls.time2 = datetime.datetime(2007, 12, 19, 21, 0, 0)
  61. time3 = datetime.datetime(2007, 12, 20, 22, 25, 0)
  62. time4 = datetime.datetime(2007, 12, 20, 21, 0, 0)
  63. cls.i1 = Item.objects.create(name='one', created=cls.time1, modified=cls.time1, creator=cls.a1, note=cls.n3)
  64. cls.i1.tags.set([cls.t1, cls.t2])
  65. cls.i2 = Item.objects.create(name='two', created=cls.time2, creator=cls.a2, note=n2)
  66. cls.i2.tags.set([cls.t1, cls.t3])
  67. cls.i3 = Item.objects.create(name='three', created=time3, creator=cls.a2, note=cls.n3)
  68. i4 = Item.objects.create(name='four', created=time4, creator=cls.a4, note=cls.n3)
  69. i4.tags.set([t4])
  70. cls.r1 = Report.objects.create(name='r1', creator=cls.a1)
  71. Report.objects.create(name='r2', creator=a3)
  72. Report.objects.create(name='r3')
  73. # Ordering by 'rank' gives us rank2, rank1, rank3. Ordering by the Meta.ordering
  74. # will be rank3, rank2, rank1.
  75. cls.rank1 = Ranking.objects.create(rank=2, author=cls.a2)
  76. Cover.objects.create(title="first", item=i4)
  77. Cover.objects.create(title="second", item=cls.i2)
  78. def test_subquery_condition(self):
  79. qs1 = Tag.objects.filter(pk__lte=0)
  80. qs2 = Tag.objects.filter(parent__in=qs1)
  81. qs3 = Tag.objects.filter(parent__in=qs2)
  82. self.assertEqual(qs3.query.subq_aliases, {'T', 'U', 'V'})
  83. self.assertIn('v0', str(qs3.query).lower())
  84. qs4 = qs3.filter(parent__in=qs1)
  85. self.assertEqual(qs4.query.subq_aliases, {'T', 'U', 'V'})
  86. # It is possible to reuse U for the second subquery, no need to use W.
  87. self.assertNotIn('w0', str(qs4.query).lower())
  88. # So, 'U0."id"' is referenced twice.
  89. self.assertTrue(str(qs4.query).lower().count('u0'), 2)
  90. def test_ticket1050(self):
  91. self.assertQuerysetEqual(
  92. Item.objects.filter(tags__isnull=True),
  93. ['<Item: three>']
  94. )
  95. self.assertQuerysetEqual(
  96. Item.objects.filter(tags__id__isnull=True),
  97. ['<Item: three>']
  98. )
  99. def test_ticket1801(self):
  100. self.assertQuerysetEqual(
  101. Author.objects.filter(item=self.i2),
  102. ['<Author: a2>']
  103. )
  104. self.assertQuerysetEqual(
  105. Author.objects.filter(item=self.i3),
  106. ['<Author: a2>']
  107. )
  108. self.assertQuerysetEqual(
  109. Author.objects.filter(item=self.i2) & Author.objects.filter(item=self.i3),
  110. ['<Author: a2>']
  111. )
  112. def test_ticket2306(self):
  113. # Checking that no join types are "left outer" joins.
  114. query = Item.objects.filter(tags=self.t2).query
  115. self.assertNotIn(LOUTER, [x.join_type for x in query.alias_map.values()])
  116. self.assertQuerysetEqual(
  117. Item.objects.filter(Q(tags=self.t1)).order_by('name'),
  118. ['<Item: one>', '<Item: two>']
  119. )
  120. self.assertQuerysetEqual(
  121. Item.objects.filter(Q(tags=self.t1)).filter(Q(tags=self.t2)),
  122. ['<Item: one>']
  123. )
  124. self.assertQuerysetEqual(
  125. Item.objects.filter(Q(tags=self.t1)).filter(Q(creator__name='fred') | Q(tags=self.t2)),
  126. ['<Item: one>']
  127. )
  128. # Each filter call is processed "at once" against a single table, so this is
  129. # different from the previous example as it tries to find tags that are two
  130. # things at once (rather than two tags).
  131. self.assertQuerysetEqual(
  132. Item.objects.filter(Q(tags=self.t1) & Q(tags=self.t2)),
  133. []
  134. )
  135. self.assertQuerysetEqual(
  136. Item.objects.filter(Q(tags=self.t1), Q(creator__name='fred') | Q(tags=self.t2)),
  137. []
  138. )
  139. qs = Author.objects.filter(ranking__rank=2, ranking__id=self.rank1.id)
  140. self.assertQuerysetEqual(list(qs), ['<Author: a2>'])
  141. self.assertEqual(2, qs.query.count_active_tables(), 2)
  142. qs = Author.objects.filter(ranking__rank=2).filter(ranking__id=self.rank1.id)
  143. self.assertEqual(qs.query.count_active_tables(), 3)
  144. def test_ticket4464(self):
  145. self.assertQuerysetEqual(
  146. Item.objects.filter(tags=self.t1).filter(tags=self.t2),
  147. ['<Item: one>']
  148. )
  149. self.assertQuerysetEqual(
  150. Item.objects.filter(tags__in=[self.t1, self.t2]).distinct().order_by('name'),
  151. ['<Item: one>', '<Item: two>']
  152. )
  153. self.assertQuerysetEqual(
  154. Item.objects.filter(tags__in=[self.t1, self.t2]).filter(tags=self.t3),
  155. ['<Item: two>']
  156. )
  157. # Make sure .distinct() works with slicing (this was broken in Oracle).
  158. self.assertQuerysetEqual(
  159. Item.objects.filter(tags__in=[self.t1, self.t2]).order_by('name')[:3],
  160. ['<Item: one>', '<Item: one>', '<Item: two>']
  161. )
  162. self.assertQuerysetEqual(
  163. Item.objects.filter(tags__in=[self.t1, self.t2]).distinct().order_by('name')[:3],
  164. ['<Item: one>', '<Item: two>']
  165. )
  166. def test_tickets_2080_3592(self):
  167. self.assertQuerysetEqual(
  168. Author.objects.filter(item__name='one') | Author.objects.filter(name='a3'),
  169. ['<Author: a1>', '<Author: a3>']
  170. )
  171. self.assertQuerysetEqual(
  172. Author.objects.filter(Q(item__name='one') | Q(name='a3')),
  173. ['<Author: a1>', '<Author: a3>']
  174. )
  175. self.assertQuerysetEqual(
  176. Author.objects.filter(Q(name='a3') | Q(item__name='one')),
  177. ['<Author: a1>', '<Author: a3>']
  178. )
  179. self.assertQuerysetEqual(
  180. Author.objects.filter(Q(item__name='three') | Q(report__name='r3')),
  181. ['<Author: a2>']
  182. )
  183. def test_ticket6074(self):
  184. # Merging two empty result sets shouldn't leave a queryset with no constraints
  185. # (which would match everything).
  186. self.assertQuerysetEqual(Author.objects.filter(Q(id__in=[])), [])
  187. self.assertQuerysetEqual(
  188. Author.objects.filter(Q(id__in=[]) | Q(id__in=[])),
  189. []
  190. )
  191. def test_tickets_1878_2939(self):
  192. self.assertEqual(Item.objects.values('creator').distinct().count(), 3)
  193. # Create something with a duplicate 'name' so that we can test multi-column
  194. # cases (which require some tricky SQL transformations under the covers).
  195. xx = Item(name='four', created=self.time1, creator=self.a2, note=self.n1)
  196. xx.save()
  197. self.assertEqual(
  198. Item.objects.exclude(name='two').values('creator', 'name').distinct().count(),
  199. 4
  200. )
  201. self.assertEqual(
  202. (
  203. Item.objects
  204. .exclude(name='two')
  205. .extra(select={'foo': '%s'}, select_params=(1,))
  206. .values('creator', 'name', 'foo')
  207. .distinct()
  208. .count()
  209. ),
  210. 4
  211. )
  212. self.assertEqual(
  213. (
  214. Item.objects
  215. .exclude(name='two')
  216. .extra(select={'foo': '%s'}, select_params=(1,))
  217. .values('creator', 'name')
  218. .distinct()
  219. .count()
  220. ),
  221. 4
  222. )
  223. xx.delete()
  224. def test_ticket7323(self):
  225. self.assertEqual(Item.objects.values('creator', 'name').count(), 4)
  226. def test_ticket2253(self):
  227. q1 = Item.objects.order_by('name')
  228. q2 = Item.objects.filter(id=self.i1.id)
  229. self.assertQuerysetEqual(
  230. q1,
  231. ['<Item: four>', '<Item: one>', '<Item: three>', '<Item: two>']
  232. )
  233. self.assertQuerysetEqual(q2, ['<Item: one>'])
  234. self.assertQuerysetEqual(
  235. (q1 | q2).order_by('name'),
  236. ['<Item: four>', '<Item: one>', '<Item: three>', '<Item: two>']
  237. )
  238. self.assertQuerysetEqual((q1 & q2).order_by('name'), ['<Item: one>'])
  239. q1 = Item.objects.filter(tags=self.t1)
  240. q2 = Item.objects.filter(note=self.n3, tags=self.t2)
  241. q3 = Item.objects.filter(creator=self.a4)
  242. self.assertQuerysetEqual(
  243. ((q1 & q2) | q3).order_by('name'),
  244. ['<Item: four>', '<Item: one>']
  245. )
  246. def test_order_by_tables(self):
  247. q1 = Item.objects.order_by('name')
  248. q2 = Item.objects.filter(id=self.i1.id)
  249. list(q2)
  250. combined_query = (q1 & q2).order_by('name').query
  251. self.assertEqual(len([
  252. t for t in combined_query.tables if combined_query.alias_refcount[t]
  253. ]), 1)
  254. def test_order_by_join_unref(self):
  255. """
  256. This test is related to the above one, testing that there aren't
  257. old JOINs in the query.
  258. """
  259. qs = Celebrity.objects.order_by('greatest_fan__fan_of')
  260. self.assertIn('OUTER JOIN', str(qs.query))
  261. qs = qs.order_by('id')
  262. self.assertNotIn('OUTER JOIN', str(qs.query))
  263. def test_get_clears_ordering(self):
  264. """
  265. get() should clear ordering for optimization purposes.
  266. """
  267. with CaptureQueriesContext(connection) as captured_queries:
  268. Author.objects.order_by('name').get(pk=self.a1.pk)
  269. self.assertNotIn('order by', captured_queries[0]['sql'].lower())
  270. def test_tickets_4088_4306(self):
  271. self.assertQuerysetEqual(
  272. Report.objects.filter(creator=1001),
  273. ['<Report: r1>']
  274. )
  275. self.assertQuerysetEqual(
  276. Report.objects.filter(creator__num=1001),
  277. ['<Report: r1>']
  278. )
  279. self.assertQuerysetEqual(Report.objects.filter(creator__id=1001), [])
  280. self.assertQuerysetEqual(
  281. Report.objects.filter(creator__id=self.a1.id),
  282. ['<Report: r1>']
  283. )
  284. self.assertQuerysetEqual(
  285. Report.objects.filter(creator__name='a1'),
  286. ['<Report: r1>']
  287. )
  288. def test_ticket4510(self):
  289. self.assertQuerysetEqual(
  290. Author.objects.filter(report__name='r1'),
  291. ['<Author: a1>']
  292. )
  293. def test_ticket7378(self):
  294. self.assertQuerysetEqual(self.a1.report_set.all(), ['<Report: r1>'])
  295. def test_tickets_5324_6704(self):
  296. self.assertQuerysetEqual(
  297. Item.objects.filter(tags__name='t4'),
  298. ['<Item: four>']
  299. )
  300. self.assertQuerysetEqual(
  301. Item.objects.exclude(tags__name='t4').order_by('name').distinct(),
  302. ['<Item: one>', '<Item: three>', '<Item: two>']
  303. )
  304. self.assertQuerysetEqual(
  305. Item.objects.exclude(tags__name='t4').order_by('name').distinct().reverse(),
  306. ['<Item: two>', '<Item: three>', '<Item: one>']
  307. )
  308. self.assertQuerysetEqual(
  309. Author.objects.exclude(item__name='one').distinct().order_by('name'),
  310. ['<Author: a2>', '<Author: a3>', '<Author: a4>']
  311. )
  312. # Excluding across a m2m relation when there is more than one related
  313. # object associated was problematic.
  314. self.assertQuerysetEqual(
  315. Item.objects.exclude(tags__name='t1').order_by('name'),
  316. ['<Item: four>', '<Item: three>']
  317. )
  318. self.assertQuerysetEqual(
  319. Item.objects.exclude(tags__name='t1').exclude(tags__name='t4'),
  320. ['<Item: three>']
  321. )
  322. # Excluding from a relation that cannot be NULL should not use outer joins.
  323. query = Item.objects.exclude(creator__in=[self.a1, self.a2]).query
  324. self.assertNotIn(LOUTER, [x.join_type for x in query.alias_map.values()])
  325. # Similarly, when one of the joins cannot possibly, ever, involve NULL
  326. # values (Author -> ExtraInfo, in the following), it should never be
  327. # promoted to a left outer join. So the following query should only
  328. # involve one "left outer" join (Author -> Item is 0-to-many).
  329. qs = Author.objects.filter(id=self.a1.id).filter(Q(extra__note=self.n1) | Q(item__note=self.n3))
  330. self.assertEqual(
  331. len([
  332. x for x in qs.query.alias_map.values()
  333. if x.join_type == LOUTER and qs.query.alias_refcount[x.table_alias]
  334. ]),
  335. 1
  336. )
  337. # The previous changes shouldn't affect nullable foreign key joins.
  338. self.assertQuerysetEqual(
  339. Tag.objects.filter(parent__isnull=True).order_by('name'),
  340. ['<Tag: t1>']
  341. )
  342. self.assertQuerysetEqual(
  343. Tag.objects.exclude(parent__isnull=True).order_by('name'),
  344. ['<Tag: t2>', '<Tag: t3>', '<Tag: t4>', '<Tag: t5>']
  345. )
  346. self.assertQuerysetEqual(
  347. Tag.objects.exclude(Q(parent__name='t1') | Q(parent__isnull=True)).order_by('name'),
  348. ['<Tag: t4>', '<Tag: t5>']
  349. )
  350. self.assertQuerysetEqual(
  351. Tag.objects.exclude(Q(parent__isnull=True) | Q(parent__name='t1')).order_by('name'),
  352. ['<Tag: t4>', '<Tag: t5>']
  353. )
  354. self.assertQuerysetEqual(
  355. Tag.objects.exclude(Q(parent__parent__isnull=True)).order_by('name'),
  356. ['<Tag: t4>', '<Tag: t5>']
  357. )
  358. self.assertQuerysetEqual(
  359. Tag.objects.filter(~Q(parent__parent__isnull=True)).order_by('name'),
  360. ['<Tag: t4>', '<Tag: t5>']
  361. )
  362. def test_ticket2091(self):
  363. t = Tag.objects.get(name='t4')
  364. self.assertQuerysetEqual(
  365. Item.objects.filter(tags__in=[t]),
  366. ['<Item: four>']
  367. )
  368. def test_avoid_infinite_loop_on_too_many_subqueries(self):
  369. x = Tag.objects.filter(pk=1)
  370. local_recursion_limit = 127
  371. msg = 'Maximum recursion depth exceeded: too many subqueries.'
  372. with self.assertRaisesMessage(RuntimeError, msg):
  373. for i in six.moves.range(local_recursion_limit * 2):
  374. x = Tag.objects.filter(pk__in=x)
  375. def test_reasonable_number_of_subq_aliases(self):
  376. x = Tag.objects.filter(pk=1)
  377. for _ in range(20):
  378. x = Tag.objects.filter(pk__in=x)
  379. self.assertEqual(
  380. x.query.subq_aliases, {
  381. 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'AA', 'AB', 'AC', 'AD',
  382. 'AE', 'AF', 'AG', 'AH', 'AI', 'AJ', 'AK', 'AL', 'AM', 'AN',
  383. }
  384. )
  385. def test_heterogeneous_qs_combination(self):
  386. # Combining querysets built on different models should behave in a well-defined
  387. # fashion. We raise an error.
  388. with self.assertRaisesMessage(AssertionError, 'Cannot combine queries on two different base models.'):
  389. Author.objects.all() & Tag.objects.all()
  390. with self.assertRaisesMessage(AssertionError, 'Cannot combine queries on two different base models.'):
  391. Author.objects.all() | Tag.objects.all()
  392. def test_ticket3141(self):
  393. self.assertEqual(Author.objects.extra(select={'foo': '1'}).count(), 4)
  394. self.assertEqual(
  395. Author.objects.extra(select={'foo': '%s'}, select_params=(1,)).count(),
  396. 4
  397. )
  398. def test_ticket2400(self):
  399. self.assertQuerysetEqual(
  400. Author.objects.filter(item__isnull=True),
  401. ['<Author: a3>']
  402. )
  403. self.assertQuerysetEqual(
  404. Tag.objects.filter(item__isnull=True),
  405. ['<Tag: t5>']
  406. )
  407. def test_ticket2496(self):
  408. self.assertQuerysetEqual(
  409. Item.objects.extra(tables=['queries_author']).select_related().order_by('name')[:1],
  410. ['<Item: four>']
  411. )
  412. def test_error_raised_on_filter_with_dictionary(self):
  413. with self.assertRaisesMessage(FieldError, 'Cannot parse keyword query as dict'):
  414. Note.objects.filter({'note': 'n1', 'misc': 'foo'})
  415. def test_tickets_2076_7256(self):
  416. # Ordering on related tables should be possible, even if the table is
  417. # not otherwise involved.
  418. self.assertQuerysetEqual(
  419. Item.objects.order_by('note__note', 'name'),
  420. ['<Item: two>', '<Item: four>', '<Item: one>', '<Item: three>']
  421. )
  422. # Ordering on a related field should use the remote model's default
  423. # ordering as a final step.
  424. self.assertQuerysetEqual(
  425. Author.objects.order_by('extra', '-name'),
  426. ['<Author: a2>', '<Author: a1>', '<Author: a4>', '<Author: a3>']
  427. )
  428. # Using remote model default ordering can span multiple models (in this
  429. # case, Cover is ordered by Item's default, which uses Note's default).
  430. self.assertQuerysetEqual(
  431. Cover.objects.all(),
  432. ['<Cover: first>', '<Cover: second>']
  433. )
  434. # If the remote model does not have a default ordering, we order by its 'id'
  435. # field.
  436. self.assertQuerysetEqual(
  437. Item.objects.order_by('creator', 'name'),
  438. ['<Item: one>', '<Item: three>', '<Item: two>', '<Item: four>']
  439. )
  440. # Ordering by a many-valued attribute (e.g. a many-to-many or reverse
  441. # ForeignKey) is legal, but the results might not make sense. That
  442. # isn't Django's problem. Garbage in, garbage out.
  443. self.assertQuerysetEqual(
  444. Item.objects.filter(tags__isnull=False).order_by('tags', 'id'),
  445. ['<Item: one>', '<Item: two>', '<Item: one>', '<Item: two>', '<Item: four>']
  446. )
  447. # If we replace the default ordering, Django adjusts the required
  448. # tables automatically. Item normally requires a join with Note to do
  449. # the default ordering, but that isn't needed here.
  450. qs = Item.objects.order_by('name')
  451. self.assertQuerysetEqual(
  452. qs,
  453. ['<Item: four>', '<Item: one>', '<Item: three>', '<Item: two>']
  454. )
  455. self.assertEqual(len(qs.query.tables), 1)
  456. def test_tickets_2874_3002(self):
  457. qs = Item.objects.select_related().order_by('note__note', 'name')
  458. self.assertQuerysetEqual(
  459. qs,
  460. ['<Item: two>', '<Item: four>', '<Item: one>', '<Item: three>']
  461. )
  462. # This is also a good select_related() test because there are multiple
  463. # Note entries in the SQL. The two Note items should be different.
  464. self.assertTrue(repr(qs[0].note), '<Note: n2>')
  465. self.assertEqual(repr(qs[0].creator.extra.note), '<Note: n1>')
  466. def test_ticket3037(self):
  467. self.assertQuerysetEqual(
  468. Item.objects.filter(Q(creator__name='a3', name='two') | Q(creator__name='a4', name='four')),
  469. ['<Item: four>']
  470. )
  471. def test_tickets_5321_7070(self):
  472. # Ordering columns must be included in the output columns. Note that
  473. # this means results that might otherwise be distinct are not (if there
  474. # are multiple values in the ordering cols), as in this example. This
  475. # isn't a bug; it's a warning to be careful with the selection of
  476. # ordering columns.
  477. self.assertValueQuerysetEqual(
  478. Note.objects.values('misc').distinct().order_by('note', '-misc'),
  479. [{'misc': 'foo'}, {'misc': 'bar'}, {'misc': 'foo'}]
  480. )
  481. def test_ticket4358(self):
  482. # If you don't pass any fields to values(), relation fields are
  483. # returned as "foo_id" keys, not "foo". For consistency, you should be
  484. # able to pass "foo_id" in the fields list and have it work, too. We
  485. # actually allow both "foo" and "foo_id".
  486. # The *_id version is returned by default.
  487. self.assertIn('note_id', ExtraInfo.objects.values()[0])
  488. # You can also pass it in explicitly.
  489. self.assertValueQuerysetEqual(
  490. ExtraInfo.objects.values('note_id'),
  491. [{'note_id': 1}, {'note_id': 2}]
  492. )
  493. # ...or use the field name.
  494. self.assertValueQuerysetEqual(
  495. ExtraInfo.objects.values('note'),
  496. [{'note': 1}, {'note': 2}]
  497. )
  498. def test_ticket2902(self):
  499. # Parameters can be given to extra_select, *if* you use an OrderedDict.
  500. # (First we need to know which order the keys fall in "naturally" on
  501. # your system, so we can put things in the wrong way around from
  502. # normal. A normal dict would thus fail.)
  503. s = [('a', '%s'), ('b', '%s')]
  504. params = ['one', 'two']
  505. if {'a': 1, 'b': 2}.keys() == ['a', 'b']:
  506. s.reverse()
  507. params.reverse()
  508. # This slightly odd comparison works around the fact that PostgreSQL will
  509. # return 'one' and 'two' as strings, not Unicode objects. It's a side-effect of
  510. # using constants here and not a real concern.
  511. d = Item.objects.extra(select=OrderedDict(s), select_params=params).values('a', 'b')[0]
  512. self.assertEqual(d, {'a': 'one', 'b': 'two'})
  513. # Order by the number of tags attached to an item.
  514. l = (
  515. Item.objects
  516. .extra(select={
  517. 'count': 'select count(*) from queries_item_tags where queries_item_tags.item_id = queries_item.id'
  518. })
  519. .order_by('-count')
  520. )
  521. self.assertEqual([o.count for o in l], [2, 2, 1, 0])
  522. def test_ticket6154(self):
  523. # Multiple filter statements are joined using "AND" all the time.
  524. self.assertQuerysetEqual(
  525. Author.objects.filter(id=self.a1.id).filter(Q(extra__note=self.n1) | Q(item__note=self.n3)),
  526. ['<Author: a1>']
  527. )
  528. self.assertQuerysetEqual(
  529. Author.objects.filter(Q(extra__note=self.n1) | Q(item__note=self.n3)).filter(id=self.a1.id),
  530. ['<Author: a1>']
  531. )
  532. def test_ticket6981(self):
  533. self.assertQuerysetEqual(
  534. Tag.objects.select_related('parent').order_by('name'),
  535. ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>', '<Tag: t4>', '<Tag: t5>']
  536. )
  537. def test_ticket9926(self):
  538. self.assertQuerysetEqual(
  539. Tag.objects.select_related("parent", "category").order_by('name'),
  540. ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>', '<Tag: t4>', '<Tag: t5>']
  541. )
  542. self.assertQuerysetEqual(
  543. Tag.objects.select_related('parent', "parent__category").order_by('name'),
  544. ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>', '<Tag: t4>', '<Tag: t5>']
  545. )
  546. def test_tickets_6180_6203(self):
  547. # Dates with limits and/or counts
  548. self.assertEqual(Item.objects.count(), 4)
  549. self.assertEqual(Item.objects.datetimes('created', 'month').count(), 1)
  550. self.assertEqual(Item.objects.datetimes('created', 'day').count(), 2)
  551. self.assertEqual(len(Item.objects.datetimes('created', 'day')), 2)
  552. self.assertEqual(Item.objects.datetimes('created', 'day')[0], datetime.datetime(2007, 12, 19, 0, 0))
  553. def test_tickets_7087_12242(self):
  554. # Dates with extra select columns
  555. self.assertQuerysetEqual(
  556. Item.objects.datetimes('created', 'day').extra(select={'a': 1}),
  557. ['datetime.datetime(2007, 12, 19, 0, 0)', 'datetime.datetime(2007, 12, 20, 0, 0)']
  558. )
  559. self.assertQuerysetEqual(
  560. Item.objects.extra(select={'a': 1}).datetimes('created', 'day'),
  561. ['datetime.datetime(2007, 12, 19, 0, 0)', 'datetime.datetime(2007, 12, 20, 0, 0)']
  562. )
  563. name = "one"
  564. self.assertQuerysetEqual(
  565. Item.objects.datetimes('created', 'day').extra(where=['name=%s'], params=[name]),
  566. ['datetime.datetime(2007, 12, 19, 0, 0)']
  567. )
  568. self.assertQuerysetEqual(
  569. Item.objects.extra(where=['name=%s'], params=[name]).datetimes('created', 'day'),
  570. ['datetime.datetime(2007, 12, 19, 0, 0)']
  571. )
  572. def test_ticket7155(self):
  573. # Nullable dates
  574. self.assertQuerysetEqual(
  575. Item.objects.datetimes('modified', 'day'),
  576. ['datetime.datetime(2007, 12, 19, 0, 0)']
  577. )
  578. def test_ticket7098(self):
  579. # Make sure semi-deprecated ordering by related models syntax still
  580. # works.
  581. self.assertValueQuerysetEqual(
  582. Item.objects.values('note__note').order_by('queries_note.note', 'id'),
  583. [{'note__note': 'n2'}, {'note__note': 'n3'}, {'note__note': 'n3'}, {'note__note': 'n3'}]
  584. )
  585. def test_ticket7096(self):
  586. # Make sure exclude() with multiple conditions continues to work.
  587. self.assertQuerysetEqual(
  588. Tag.objects.filter(parent=self.t1, name='t3').order_by('name'),
  589. ['<Tag: t3>']
  590. )
  591. self.assertQuerysetEqual(
  592. Tag.objects.exclude(parent=self.t1, name='t3').order_by('name'),
  593. ['<Tag: t1>', '<Tag: t2>', '<Tag: t4>', '<Tag: t5>']
  594. )
  595. self.assertQuerysetEqual(
  596. Item.objects.exclude(tags__name='t1', name='one').order_by('name').distinct(),
  597. ['<Item: four>', '<Item: three>', '<Item: two>']
  598. )
  599. self.assertQuerysetEqual(
  600. Item.objects.filter(name__in=['three', 'four']).exclude(tags__name='t1').order_by('name'),
  601. ['<Item: four>', '<Item: three>']
  602. )
  603. # More twisted cases, involving nested negations.
  604. self.assertQuerysetEqual(
  605. Item.objects.exclude(~Q(tags__name='t1', name='one')),
  606. ['<Item: one>']
  607. )
  608. self.assertQuerysetEqual(
  609. Item.objects.filter(~Q(tags__name='t1', name='one'), name='two'),
  610. ['<Item: two>']
  611. )
  612. self.assertQuerysetEqual(
  613. Item.objects.exclude(~Q(tags__name='t1', name='one'), name='two'),
  614. ['<Item: four>', '<Item: one>', '<Item: three>']
  615. )
  616. def test_tickets_7204_7506(self):
  617. # Make sure querysets with related fields can be pickled. If this
  618. # doesn't crash, it's a Good Thing.
  619. pickle.dumps(Item.objects.all())
  620. def test_ticket7813(self):
  621. # We should also be able to pickle things that use select_related().
  622. # The only tricky thing here is to ensure that we do the related
  623. # selections properly after unpickling.
  624. qs = Item.objects.select_related()
  625. query = qs.query.get_compiler(qs.db).as_sql()[0]
  626. query2 = pickle.loads(pickle.dumps(qs.query))
  627. self.assertEqual(
  628. query2.get_compiler(qs.db).as_sql()[0],
  629. query
  630. )
  631. def test_deferred_load_qs_pickling(self):
  632. # Check pickling of deferred-loading querysets
  633. qs = Item.objects.defer('name', 'creator')
  634. q2 = pickle.loads(pickle.dumps(qs))
  635. self.assertEqual(list(qs), list(q2))
  636. q3 = pickle.loads(pickle.dumps(qs, pickle.HIGHEST_PROTOCOL))
  637. self.assertEqual(list(qs), list(q3))
  638. def test_ticket7277(self):
  639. self.assertQuerysetEqual(
  640. self.n1.annotation_set.filter(
  641. Q(tag=self.t5) | Q(tag__children=self.t5) | Q(tag__children__children=self.t5)
  642. ),
  643. ['<Annotation: a1>']
  644. )
  645. def test_tickets_7448_7707(self):
  646. # Complex objects should be converted to strings before being used in
  647. # lookups.
  648. self.assertQuerysetEqual(
  649. Item.objects.filter(created__in=[self.time1, self.time2]),
  650. ['<Item: one>', '<Item: two>']
  651. )
  652. def test_ticket7235(self):
  653. # An EmptyQuerySet should not raise exceptions if it is filtered.
  654. Eaten.objects.create(meal='m')
  655. q = Eaten.objects.none()
  656. with self.assertNumQueries(0):
  657. self.assertQuerysetEqual(q.all(), [])
  658. self.assertQuerysetEqual(q.filter(meal='m'), [])
  659. self.assertQuerysetEqual(q.exclude(meal='m'), [])
  660. self.assertQuerysetEqual(q.complex_filter({'pk': 1}), [])
  661. self.assertQuerysetEqual(q.select_related('food'), [])
  662. self.assertQuerysetEqual(q.annotate(Count('food')), [])
  663. self.assertQuerysetEqual(q.order_by('meal', 'food'), [])
  664. self.assertQuerysetEqual(q.distinct(), [])
  665. self.assertQuerysetEqual(
  666. q.extra(select={'foo': "1"}),
  667. []
  668. )
  669. q.query.low_mark = 1
  670. with self.assertRaisesMessage(AssertionError, 'Cannot change a query once a slice has been taken'):
  671. q.extra(select={'foo': "1"})
  672. self.assertQuerysetEqual(q.reverse(), [])
  673. self.assertQuerysetEqual(q.defer('meal'), [])
  674. self.assertQuerysetEqual(q.only('meal'), [])
  675. def test_ticket7791(self):
  676. # There were "issues" when ordering and distinct-ing on fields related
  677. # via ForeignKeys.
  678. self.assertEqual(
  679. len(Note.objects.order_by('extrainfo__info').distinct()),
  680. 3
  681. )
  682. # Pickling of QuerySets using datetimes() should work.
  683. qs = Item.objects.datetimes('created', 'month')
  684. pickle.loads(pickle.dumps(qs))
  685. def test_ticket9997(self):
  686. # If a ValuesList or Values queryset is passed as an inner query, we
  687. # make sure it's only requesting a single value and use that as the
  688. # thing to select.
  689. self.assertQuerysetEqual(
  690. Tag.objects.filter(name__in=Tag.objects.filter(parent=self.t1).values('name')),
  691. ['<Tag: t2>', '<Tag: t3>']
  692. )
  693. # Multi-valued values() and values_list() querysets should raise errors.
  694. with self.assertRaisesMessage(TypeError, 'Cannot use multi-field values as a filter value.'):
  695. Tag.objects.filter(name__in=Tag.objects.filter(parent=self.t1).values('name', 'id'))
  696. with self.assertRaisesMessage(TypeError, 'Cannot use multi-field values as a filter value.'):
  697. Tag.objects.filter(name__in=Tag.objects.filter(parent=self.t1).values_list('name', 'id'))
  698. def test_ticket9985(self):
  699. # qs.values_list(...).values(...) combinations should work.
  700. self.assertValueQuerysetEqual(
  701. Note.objects.values_list("note", flat=True).values("id").order_by("id"),
  702. [{'id': 1}, {'id': 2}, {'id': 3}]
  703. )
  704. self.assertQuerysetEqual(
  705. Annotation.objects.filter(notes__in=Note.objects.filter(note="n1").values_list('note').values('id')),
  706. ['<Annotation: a1>']
  707. )
  708. def test_ticket10205(self):
  709. # When bailing out early because of an empty "__in" filter, we need
  710. # to set things up correctly internally so that subqueries can continue properly.
  711. self.assertEqual(Tag.objects.filter(name__in=()).update(name="foo"), 0)
  712. def test_ticket10432(self):
  713. # Testing an empty "__in" filter with a generator as the value.
  714. def f():
  715. return iter([])
  716. n_obj = Note.objects.all()[0]
  717. def g():
  718. for i in [n_obj.pk]:
  719. yield i
  720. self.assertQuerysetEqual(Note.objects.filter(pk__in=f()), [])
  721. self.assertEqual(list(Note.objects.filter(pk__in=g())), [n_obj])
  722. def test_ticket10742(self):
  723. # Queries used in an __in clause don't execute subqueries
  724. subq = Author.objects.filter(num__lt=3000)
  725. qs = Author.objects.filter(pk__in=subq)
  726. self.assertQuerysetEqual(qs, ['<Author: a1>', '<Author: a2>'])
  727. # The subquery result cache should not be populated
  728. self.assertIsNone(subq._result_cache)
  729. subq = Author.objects.filter(num__lt=3000)
  730. qs = Author.objects.exclude(pk__in=subq)
  731. self.assertQuerysetEqual(qs, ['<Author: a3>', '<Author: a4>'])
  732. # The subquery result cache should not be populated
  733. self.assertIsNone(subq._result_cache)
  734. subq = Author.objects.filter(num__lt=3000)
  735. self.assertQuerysetEqual(
  736. Author.objects.filter(Q(pk__in=subq) & Q(name='a1')),
  737. ['<Author: a1>']
  738. )
  739. # The subquery result cache should not be populated
  740. self.assertIsNone(subq._result_cache)
  741. def test_ticket7076(self):
  742. # Excluding shouldn't eliminate NULL entries.
  743. self.assertQuerysetEqual(
  744. Item.objects.exclude(modified=self.time1).order_by('name'),
  745. ['<Item: four>', '<Item: three>', '<Item: two>']
  746. )
  747. self.assertQuerysetEqual(
  748. Tag.objects.exclude(parent__name=self.t1.name),
  749. ['<Tag: t1>', '<Tag: t4>', '<Tag: t5>']
  750. )
  751. def test_ticket7181(self):
  752. # Ordering by related tables should accommodate nullable fields (this
  753. # test is a little tricky, since NULL ordering is database dependent.
  754. # Instead, we just count the number of results).
  755. self.assertEqual(len(Tag.objects.order_by('parent__name')), 5)
  756. # Empty querysets can be merged with others.
  757. self.assertQuerysetEqual(
  758. Note.objects.none() | Note.objects.all(),
  759. ['<Note: n1>', '<Note: n2>', '<Note: n3>']
  760. )
  761. self.assertQuerysetEqual(
  762. Note.objects.all() | Note.objects.none(),
  763. ['<Note: n1>', '<Note: n2>', '<Note: n3>']
  764. )
  765. self.assertQuerysetEqual(Note.objects.none() & Note.objects.all(), [])
  766. self.assertQuerysetEqual(Note.objects.all() & Note.objects.none(), [])
  767. def test_ticket9411(self):
  768. # Make sure bump_prefix() (an internal Query method) doesn't (re-)break. It's
  769. # sufficient that this query runs without error.
  770. qs = Tag.objects.values_list('id', flat=True).order_by('id')
  771. qs.query.bump_prefix(qs.query)
  772. first = qs[0]
  773. self.assertEqual(list(qs), list(range(first, first + 5)))
  774. def test_ticket8439(self):
  775. # Complex combinations of conjunctions, disjunctions and nullable
  776. # relations.
  777. self.assertQuerysetEqual(
  778. Author.objects.filter(Q(item__note__extrainfo=self.e2) | Q(report=self.r1, name='xyz')),
  779. ['<Author: a2>']
  780. )
  781. self.assertQuerysetEqual(
  782. Author.objects.filter(Q(report=self.r1, name='xyz') | Q(item__note__extrainfo=self.e2)),
  783. ['<Author: a2>']
  784. )
  785. self.assertQuerysetEqual(
  786. Annotation.objects.filter(Q(tag__parent=self.t1) | Q(notes__note='n1', name='a1')),
  787. ['<Annotation: a1>']
  788. )
  789. xx = ExtraInfo.objects.create(info='xx', note=self.n3)
  790. self.assertQuerysetEqual(
  791. Note.objects.filter(Q(extrainfo__author=self.a1) | Q(extrainfo=xx)),
  792. ['<Note: n1>', '<Note: n3>']
  793. )
  794. q = Note.objects.filter(Q(extrainfo__author=self.a1) | Q(extrainfo=xx)).query
  795. self.assertEqual(
  796. len([x for x in q.alias_map.values() if x.join_type == LOUTER and q.alias_refcount[x.table_alias]]),
  797. 1
  798. )
  799. def test_ticket17429(self):
  800. """
  801. Ensure that Meta.ordering=None works the same as Meta.ordering=[]
  802. """
  803. original_ordering = Tag._meta.ordering
  804. Tag._meta.ordering = None
  805. try:
  806. self.assertQuerysetEqual(
  807. Tag.objects.all(),
  808. ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>', '<Tag: t4>', '<Tag: t5>'],
  809. ordered=False
  810. )
  811. finally:
  812. Tag._meta.ordering = original_ordering
  813. def test_exclude(self):
  814. self.assertQuerysetEqual(
  815. Item.objects.exclude(tags__name='t4'),
  816. [repr(i) for i in Item.objects.filter(~Q(tags__name='t4'))])
  817. self.assertQuerysetEqual(
  818. Item.objects.exclude(Q(tags__name='t4') | Q(tags__name='t3')),
  819. [repr(i) for i in Item.objects.filter(~(Q(tags__name='t4') | Q(tags__name='t3')))])
  820. self.assertQuerysetEqual(
  821. Item.objects.exclude(Q(tags__name='t4') | ~Q(tags__name='t3')),
  822. [repr(i) for i in Item.objects.filter(~(Q(tags__name='t4') | ~Q(tags__name='t3')))])
  823. def test_nested_exclude(self):
  824. self.assertQuerysetEqual(
  825. Item.objects.exclude(~Q(tags__name='t4')),
  826. [repr(i) for i in Item.objects.filter(~~Q(tags__name='t4'))])
  827. def test_double_exclude(self):
  828. self.assertQuerysetEqual(
  829. Item.objects.filter(Q(tags__name='t4')),
  830. [repr(i) for i in Item.objects.filter(~~Q(tags__name='t4'))])
  831. self.assertQuerysetEqual(
  832. Item.objects.filter(Q(tags__name='t4')),
  833. [repr(i) for i in Item.objects.filter(~Q(~Q(tags__name='t4')))])
  834. def test_exclude_in(self):
  835. self.assertQuerysetEqual(
  836. Item.objects.exclude(Q(tags__name__in=['t4', 't3'])),
  837. [repr(i) for i in Item.objects.filter(~Q(tags__name__in=['t4', 't3']))])
  838. self.assertQuerysetEqual(
  839. Item.objects.filter(Q(tags__name__in=['t4', 't3'])),
  840. [repr(i) for i in Item.objects.filter(~~Q(tags__name__in=['t4', 't3']))])
  841. def test_ticket_10790_1(self):
  842. # Querying direct fields with isnull should trim the left outer join.
  843. # It also should not create INNER JOIN.
  844. q = Tag.objects.filter(parent__isnull=True)
  845. self.assertQuerysetEqual(q, ['<Tag: t1>'])
  846. self.assertNotIn('JOIN', str(q.query))
  847. q = Tag.objects.filter(parent__isnull=False)
  848. self.assertQuerysetEqual(
  849. q,
  850. ['<Tag: t2>', '<Tag: t3>', '<Tag: t4>', '<Tag: t5>'],
  851. )
  852. self.assertNotIn('JOIN', str(q.query))
  853. q = Tag.objects.exclude(parent__isnull=True)
  854. self.assertQuerysetEqual(
  855. q,
  856. ['<Tag: t2>', '<Tag: t3>', '<Tag: t4>', '<Tag: t5>'],
  857. )
  858. self.assertNotIn('JOIN', str(q.query))
  859. q = Tag.objects.exclude(parent__isnull=False)
  860. self.assertQuerysetEqual(q, ['<Tag: t1>'])
  861. self.assertNotIn('JOIN', str(q.query))
  862. q = Tag.objects.exclude(parent__parent__isnull=False)
  863. self.assertQuerysetEqual(
  864. q,
  865. ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>'],
  866. )
  867. self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 1)
  868. self.assertNotIn('INNER JOIN', str(q.query))
  869. def test_ticket_10790_2(self):
  870. # Querying across several tables should strip only the last outer join,
  871. # while preserving the preceding inner joins.
  872. q = Tag.objects.filter(parent__parent__isnull=False)
  873. self.assertQuerysetEqual(
  874. q,
  875. ['<Tag: t4>', '<Tag: t5>'],
  876. )
  877. self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 0)
  878. self.assertEqual(str(q.query).count('INNER JOIN'), 1)
  879. # Querying without isnull should not convert anything to left outer join.
  880. q = Tag.objects.filter(parent__parent=self.t1)
  881. self.assertQuerysetEqual(
  882. q,
  883. ['<Tag: t4>', '<Tag: t5>'],
  884. )
  885. self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 0)
  886. self.assertEqual(str(q.query).count('INNER JOIN'), 1)
  887. def test_ticket_10790_3(self):
  888. # Querying via indirect fields should populate the left outer join
  889. q = NamedCategory.objects.filter(tag__isnull=True)
  890. self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 1)
  891. # join to dumbcategory ptr_id
  892. self.assertEqual(str(q.query).count('INNER JOIN'), 1)
  893. self.assertQuerysetEqual(q, [])
  894. # Querying across several tables should strip only the last join, while
  895. # preserving the preceding left outer joins.
  896. q = NamedCategory.objects.filter(tag__parent__isnull=True)
  897. self.assertEqual(str(q.query).count('INNER JOIN'), 1)
  898. self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 1)
  899. self.assertQuerysetEqual(q, ['<NamedCategory: Generic>'])
  900. def test_ticket_10790_4(self):
  901. # Querying across m2m field should not strip the m2m table from join.
  902. q = Author.objects.filter(item__tags__isnull=True)
  903. self.assertQuerysetEqual(
  904. q,
  905. ['<Author: a2>', '<Author: a3>'],
  906. )
  907. self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 2)
  908. self.assertNotIn('INNER JOIN', str(q.query))
  909. q = Author.objects.filter(item__tags__parent__isnull=True)
  910. self.assertQuerysetEqual(
  911. q,
  912. ['<Author: a1>', '<Author: a2>', '<Author: a2>', '<Author: a3>'],
  913. )
  914. self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 3)
  915. self.assertNotIn('INNER JOIN', str(q.query))
  916. def test_ticket_10790_5(self):
  917. # Querying with isnull=False across m2m field should not create outer joins
  918. q = Author.objects.filter(item__tags__isnull=False)
  919. self.assertQuerysetEqual(
  920. q,
  921. ['<Author: a1>', '<Author: a1>', '<Author: a2>', '<Author: a2>', '<Author: a4>']
  922. )
  923. self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 0)
  924. self.assertEqual(str(q.query).count('INNER JOIN'), 2)
  925. q = Author.objects.filter(item__tags__parent__isnull=False)
  926. self.assertQuerysetEqual(
  927. q,
  928. ['<Author: a1>', '<Author: a2>', '<Author: a4>']
  929. )
  930. self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 0)
  931. self.assertEqual(str(q.query).count('INNER JOIN'), 3)
  932. q = Author.objects.filter(item__tags__parent__parent__isnull=False)
  933. self.assertQuerysetEqual(
  934. q,
  935. ['<Author: a4>']
  936. )
  937. self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 0)
  938. self.assertEqual(str(q.query).count('INNER JOIN'), 4)
  939. def test_ticket_10790_6(self):
  940. # Querying with isnull=True across m2m field should not create inner joins
  941. # and strip last outer join
  942. q = Author.objects.filter(item__tags__parent__parent__isnull=True)
  943. self.assertQuerysetEqual(
  944. q,
  945. ['<Author: a1>', '<Author: a1>', '<Author: a2>', '<Author: a2>',
  946. '<Author: a2>', '<Author: a3>']
  947. )
  948. self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 4)
  949. self.assertEqual(str(q.query).count('INNER JOIN'), 0)
  950. q = Author.objects.filter(item__tags__parent__isnull=True)
  951. self.assertQuerysetEqual(
  952. q,
  953. ['<Author: a1>', '<Author: a2>', '<Author: a2>', '<Author: a3>']
  954. )
  955. self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 3)
  956. self.assertEqual(str(q.query).count('INNER JOIN'), 0)
  957. def test_ticket_10790_7(self):
  958. # Reverse querying with isnull should not strip the join
  959. q = Author.objects.filter(item__isnull=True)
  960. self.assertQuerysetEqual(
  961. q,
  962. ['<Author: a3>']
  963. )
  964. self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 1)
  965. self.assertEqual(str(q.query).count('INNER JOIN'), 0)
  966. q = Author.objects.filter(item__isnull=False)
  967. self.assertQuerysetEqual(
  968. q,
  969. ['<Author: a1>', '<Author: a2>', '<Author: a2>', '<Author: a4>']
  970. )
  971. self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 0)
  972. self.assertEqual(str(q.query).count('INNER JOIN'), 1)
  973. def test_ticket_10790_8(self):
  974. # Querying with combined q-objects should also strip the left outer join
  975. q = Tag.objects.filter(Q(parent__isnull=True) | Q(parent=self.t1))
  976. self.assertQuerysetEqual(
  977. q,
  978. ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>']
  979. )
  980. self.assertEqual(str(q.query).count('LEFT OUTER JOIN'), 0)
  981. self.assertEqual(str(q.query).count('INNER JOIN'), 0)
  982. def test_ticket_10790_combine(self):
  983. # Combining queries should not re-populate the left outer join
  984. q1 = Tag.objects.filter(parent__isnull=True)
  985. q2 = Tag.objects.filter(parent__isnull=False)
  986. q3 = q1 | q2
  987. self.assertQuerysetEqual(
  988. q3,
  989. ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>', '<Tag: t4>', '<Tag: t5>'],
  990. )
  991. self.assertEqual(str(q3.query).count('LEFT OUTER JOIN'), 0)
  992. self.assertEqual(str(q3.query).count('INNER JOIN'), 0)
  993. q3 = q1 & q2
  994. self.assertQuerysetEqual(q3, [])
  995. self.assertEqual(str(q3.query).count('LEFT OUTER JOIN'), 0)
  996. self.assertEqual(str(q3.query).count('INNER JOIN'), 0)
  997. q2 = Tag.objects.filter(parent=self.t1)
  998. q3 = q1 | q2
  999. self.assertQuerysetEqual(
  1000. q3,
  1001. ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>']
  1002. )
  1003. self.assertEqual(str(q3.query).count('LEFT OUTER JOIN'), 0)
  1004. self.assertEqual(str(q3.query).count('INNER JOIN'), 0)
  1005. q3 = q2 | q1
  1006. self.assertQuerysetEqual(
  1007. q3,
  1008. ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>']
  1009. )
  1010. self.assertEqual(str(q3.query).count('LEFT OUTER JOIN'), 0)
  1011. self.assertEqual(str(q3.query).count('INNER JOIN'), 0)
  1012. q1 = Tag.objects.filter(parent__isnull=True)
  1013. q2 = Tag.objects.filter(parent__parent__isnull=True)
  1014. q3 = q1 | q2
  1015. self.assertQuerysetEqual(
  1016. q3,
  1017. ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>']
  1018. )
  1019. self.assertEqual(str(q3.query).count('LEFT OUTER JOIN'), 1)
  1020. self.assertEqual(str(q3.query).count('INNER JOIN'), 0)
  1021. q3 = q2 | q1
  1022. self.assertQuerysetEqual(
  1023. q3,
  1024. ['<Tag: t1>', '<Tag: t2>', '<Tag: t3>']
  1025. )
  1026. self.assertEqual(str(q3.query).count('LEFT OUTER JOIN'), 1)
  1027. self.assertEqual(str(q3.query).count('INNER JOIN'), 0)
  1028. def test_ticket19672(self):
  1029. self.assertQuerysetEqual(
  1030. Report.objects.filter(Q(creator__isnull=False) & ~Q(creator__extra__value=41)),
  1031. ['<Report: r1>']
  1032. )
  1033. def test_ticket_20250(self):
  1034. # A negated Q along with an annotated queryset failed in Django 1.4
  1035. qs = Author.objects.annotate(Count('item'))
  1036. qs = qs.filter(~Q(extra__value=0))
  1037. self.assertIn('SELECT', str(qs.query))
  1038. self.assertQuerysetEqual(
  1039. qs,
  1040. ['<Author: a1>', '<Author: a2>', '<Author: a3>', '<Author: a4>']
  1041. )
  1042. def test_lookup_constraint_fielderror(self):
  1043. msg = (
  1044. "Cannot resolve keyword 'unknown_field' into field. Choices are: "
  1045. "annotation, category, category_id, children, id, item, "
  1046. "managedmodel, name, note, parent, parent_id"
  1047. )
  1048. with self.assertRaisesMessage(FieldError, msg):
  1049. Tag.objects.filter(unknown_field__name='generic')
  1050. class Queries2Tests(TestCase):
  1051. @classmethod
  1052. def setUpTestData(cls):
  1053. Number.objects.create(num=4)
  1054. Number.objects.create(num=8)
  1055. Number.objects.create(num=12)
  1056. def test_ticket4289(self):
  1057. # A slight variation on the restricting the filtering choices by the
  1058. # lookup constraints.
  1059. self.assertQuerysetEqual(Number.objects.filter(num__lt=4), [])
  1060. self.assertQuerysetEqual(Number.objects.filter(num__gt=8, num__lt=12), [])
  1061. self.assertQuerysetEqual(
  1062. Number.objects.filter(num__gt=8, num__lt=13),
  1063. ['<Number: 12>']
  1064. )
  1065. self.assertQuerysetEqual(
  1066. Number.objects.filter(Q(num__lt=4) | Q(num__gt=8, num__lt=12)),
  1067. []
  1068. )
  1069. self.assertQuerysetEqual(
  1070. Number.objects.filter(Q(num__gt=8, num__lt=12) | Q(num__lt=4)),
  1071. []
  1072. )
  1073. self.assertQuerysetEqual(
  1074. Number.objects.filter(Q(num__gt=8) & Q(num__lt=12) | Q(num__lt=4)),
  1075. []
  1076. )
  1077. self.assertQuerysetEqual(
  1078. Number.objects.filter(Q(num__gt=7) & Q(num__lt=12) | Q(num__lt=4)),
  1079. ['<Number: 8>']
  1080. )
  1081. def test_ticket12239(self):
  1082. # Custom lookups are registered to round float values correctly on gte
  1083. # and lt IntegerField queries.
  1084. self.assertQuerysetEqual(
  1085. Number.objects.filter(num__gt=11.9),
  1086. ['<Number: 12>']
  1087. )
  1088. self.assertQuerysetEqual(Number.objects.filter(num__gt=12), [])
  1089. self.assertQuerysetEqual(Number.objects.filter(num__gt=12.0), [])
  1090. self.assertQuerysetEqual(Number.objects.filter(num__gt=12.1), [])
  1091. self.assertQuerysetEqual(
  1092. Number.objects.filter(num__lt=12),
  1093. ['<Number: 4>', '<Number: 8>'],
  1094. ordered=False
  1095. )
  1096. self.assertQuerysetEqual(
  1097. Number.objects.filter(num__lt=12.0),
  1098. ['<Number: 4>', '<Number: 8>'],
  1099. ordered=False
  1100. )
  1101. self.assertQuerysetEqual(
  1102. Number.objects.filter(num__lt=12.1),
  1103. ['<Number: 4>', '<Number: 8>', '<Number: 12>'],
  1104. ordered=False
  1105. )
  1106. self.assertQuerysetEqual(
  1107. Number.objects.filter(num__gte=11.9),
  1108. ['<Number: 12>']
  1109. )
  1110. self.assertQuerysetEqual(
  1111. Number.objects.filter(num__gte=12),
  1112. ['<Number: 12>']
  1113. )
  1114. self.assertQuerysetEqual(
  1115. Number.objects.filter(num__gte=12.0),
  1116. ['<Number: 12>']
  1117. )
  1118. self.assertQuerysetEqual(Number.objects.filter(num__gte=12.1), [])
  1119. self.assertQuerysetEqual(Number.objects.filter(num__gte=12.9), [])
  1120. self.assertQuerysetEqual(
  1121. Number.objects.filter(num__lte=11.9),
  1122. ['<Number: 4>', '<Number: 8>'],
  1123. ordered=False
  1124. )
  1125. self.assertQuerysetEqual(
  1126. Number.objects.filter(num__lte=12),
  1127. ['<Number: 4>', '<Number: 8>', '<Number: 12>'],
  1128. ordered=False
  1129. )
  1130. self.assertQuerysetEqual(
  1131. Number.objects.filter(num__lte=12.0),
  1132. ['<Number: 4>', '<Number: 8>', '<Number: 12>'],
  1133. ordered=False
  1134. )
  1135. self.assertQuerysetEqual(
  1136. Number.objects.filter(num__lte=12.1),
  1137. ['<Number: 4>', '<Number: 8>', '<Number: 12>'],
  1138. ordered=False
  1139. )
  1140. self.assertQuerysetEqual(
  1141. Number.objects.filter(num__lte=12.9),
  1142. ['<Number: 4>', '<Number: 8>', '<Number: 12>'],
  1143. ordered=False
  1144. )
  1145. def test_ticket7759(self):
  1146. # Count should work with a partially read result set.
  1147. count = Number.objects.count()
  1148. qs = Number.objects.all()
  1149. def run():
  1150. for obj in qs:
  1151. return qs.count() == count
  1152. self.assertTrue(run())
  1153. class Queries3Tests(BaseQuerysetTest):
  1154. def test_ticket7107(self):
  1155. # This shouldn't create an infinite loop.
  1156. self.assertQuerysetEqual(Valid.objects.all(), [])
  1157. def test_ticket8683(self):
  1158. # An error should be raised when QuerySet.datetimes() is passed the
  1159. # wrong type of field.
  1160. with self.assertRaisesMessage(AssertionError, "'name' isn't a DateField or DateTimeField."):
  1161. Item.objects.datetimes('name', 'month')
  1162. def test_ticket22023(self):
  1163. with self.assertRaisesMessage(TypeError, "Cannot call only() after .values() or .values_list()"):
  1164. Valid.objects.values().only()
  1165. with self.assertRaisesMessage(TypeError, "Cannot call defer() after .values() or .values_list()"):
  1166. Valid.objects.values().defer()
  1167. class Queries4Tests(BaseQuerysetTest):
  1168. @classmethod
  1169. def setUpTestData(cls):
  1170. generic = NamedCategory.objects.create(name="Generic")
  1171. cls.t1 = Tag.objects.create(name='t1', category=generic)
  1172. n1 = Note.objects.create(note='n1', misc='foo')
  1173. n2 = Note.objects.create(note='n2', misc='bar')
  1174. e1 = ExtraInfo.objects.create(info='e1', note=n1)
  1175. e2 = ExtraInfo.objects.create(info='e2', note=n2)
  1176. cls.a1 = Author.objects.create(name='a1', num=1001, extra=e1)
  1177. cls.a3 = Author.objects.create(name='a3', num=3003, extra=e2)
  1178. cls.r1 = Report.objects.create(name='r1', creator=cls.a1)
  1179. cls.r2 = Report.objects.create(name='r2', creator=cls.a3)
  1180. cls.r3 = Report.objects.create(name='r3')
  1181. Item.objects.create(name='i1', created=datetime.datetime.now(), note=n1, creator=cls.a1)
  1182. Item.objects.create(name='i2', created=datetime.datetime.now(), note=n1, creator=cls.a3)
  1183. def test_ticket24525(self):
  1184. tag = Tag.objects.create()
  1185. anth100 = tag.note_set.create(note='ANTH', misc='100')
  1186. math101 = tag.note_set.create(note='MATH', misc='101')
  1187. s1 = tag.annotation_set.create(name='1')
  1188. s2 = tag.annotation_set.create(name='2')
  1189. s1.notes.set([math101, anth100])
  1190. s2.notes.set([math101])
  1191. result = math101.annotation_set.all() & tag.annotation_set.exclude(notes__in=[anth100])
  1192. self.assertEqual(list(result), [s2])
  1193. def test_ticket11811(self):
  1194. unsaved_category = NamedCategory(name="Other")
  1195. msg = 'Unsaved model instance <NamedCategory: Other> cannot be used in an ORM query.'
  1196. with self.assertRaisesMessage(ValueError, msg):
  1197. Tag.objects.filter(pk=self.t1.pk).update(category=unsaved_category)
  1198. def test_ticket14876(self):
  1199. # Note: when combining the query we need to have information available
  1200. # about the join type of the trimmed "creator__isnull" join. If we
  1201. # don't have that information, then the join is created as INNER JOIN
  1202. # and results will be incorrect.
  1203. q1 = Report.objects.filter(Q(creator__isnull=True) | Q(creator__extra__info='e1'))
  1204. q2 = Report.objects.filter(Q(creator__isnull=True)) | Report.objects.filter(Q(creator__extra__info='e1'))
  1205. self.assertQuerysetEqual(q1, ["<Report: r1>", "<Report: r3>"], ordered=False)
  1206. self.assertEqual(str(q1.query), str(q2.query))
  1207. q1 = Report.objects.filter(Q(creator__extra__info='e1') | Q(creator__isnull=True))
  1208. q2 = Report.objects.filter(Q(creator__extra__info='e1')) | Report.objects.filter(Q(creator__isnull=True))
  1209. self.assertQuerysetEqual(q1, ["<Report: r1>", "<Report: r3>"], ordered=False)
  1210. self.assertEqual(str(q1.query), str(q2.query))
  1211. q1 = Item.objects.filter(Q(creator=self.a1) | Q(creator__report__name='r1')).order_by()
  1212. q2 = (
  1213. Item.objects
  1214. .filter(Q(creator=self.a1)).order_by() | Item.objects.filter(Q(creator__report__name='r1'))
  1215. .order_by()
  1216. )
  1217. self.assertQuerysetEqual(q1, ["<Item: i1>"])
  1218. self.assertEqual(str(q1.query), str(q2.query))
  1219. q1 = Item.objects.filter(Q(creator__report__name='e1') | Q(creator=self.a1)).order_by()
  1220. q2 = (
  1221. Item.objects.filter(Q(creator__report__name='e1')).order_by() |
  1222. Item.objects.filter(Q(creator=self.a1)).order_by()
  1223. )
  1224. self.assertQuerysetEqual(q1, ["<Item: i1>"])
  1225. self.assertEqual(str(q1.query), str(q2.query))
  1226. def test_combine_join_reuse(self):
  1227. # Test that we correctly recreate joins having identical connections
  1228. # in the rhs query, in case the query is ORed together. Related to
  1229. # ticket #18748
  1230. Report.objects.create(name='r4', creator=self.a1)
  1231. q1 = Author.objects.filter(report__name='r5')
  1232. q2 = Author.objects.filter(report__name='r4').filter(report__name='r1')
  1233. combined = q1 | q2
  1234. self.assertEqual(str(combined.query).count('JOIN'), 2)
  1235. self.assertEqual(len(combined), 1)
  1236. self.assertEqual(combined[0].name, 'a1')
  1237. def test_ticket7095(self):
  1238. # Updates that are filtered on the model being updated are somewhat
  1239. # tricky in MySQL. This exercises that case.
  1240. ManagedModel.objects.create(data='mm1', tag=self.t1, public=True)
  1241. self.assertEqual(ManagedModel.objects.update(data='mm'), 1)
  1242. # A values() or values_list() query across joined models must use outer
  1243. # joins appropriately.
  1244. # Note: In Oracle, we expect a null CharField to return '' instead of
  1245. # None.
  1246. if connection.features.interprets_empty_strings_as_nulls:
  1247. expected_null_charfield_repr = ''
  1248. else:
  1249. expected_null_charfield_repr = None
  1250. self.assertValueQuerysetEqual(
  1251. Report.objects.values_list("creator__extra__info", flat=True).order_by("name"),
  1252. ['e1', 'e2', expected_null_charfield_repr],
  1253. )
  1254. # Similarly for select_related(), joins beyond an initial nullable join
  1255. # must use outer joins so that all results are included.
  1256. self.assertQuerysetEqual(
  1257. Report.objects.select_related("creator", "creator__extra").order_by("name"),
  1258. ['<Report: r1>', '<Report: r2>', '<Report: r3>']
  1259. )
  1260. # When there are multiple paths to a table from another table, we have
  1261. # to be careful not to accidentally reuse an inappropriate join when
  1262. # using select_related(). We used to return the parent's Detail record
  1263. # here by mistake.
  1264. d1 = Detail.objects.create(data="d1")
  1265. d2 = Detail.objects.create(data="d2")
  1266. m1 = Member.objects.create(name="m1", details=d1)
  1267. m2 = Member.objects.create(name="m2", details=d2)
  1268. Child.objects.create(person=m2, parent=m1)
  1269. obj = m1.children.select_related("person__details")[0]
  1270. self.assertEqual(obj.person.details.data, 'd2')
  1271. def test_order_by_resetting(self):
  1272. # Calling order_by() with no parameters removes any existing ordering on the
  1273. # model. But it should still be possible to add new ordering after that.
  1274. qs = Author.objects.order_by().order_by('name')
  1275. self.assertIn('ORDER BY', qs.query.get_compiler(qs.db).as_sql()[0])
  1276. def test_order_by_reverse_fk(self):
  1277. # It is possible to order by reverse of foreign key, although that can lead
  1278. # to duplicate results.
  1279. c1 = SimpleCategory.objects.create(name="category1")
  1280. c2 = SimpleCategory.objects.create(name="category2")
  1281. CategoryItem.objects.create(category=c1)
  1282. CategoryItem.objects.create(category=c2)
  1283. CategoryItem.objects.create(category=c1)
  1284. self.assertQuerysetEqual(
  1285. SimpleCategory.objects.order_by('categoryitem', 'pk'),
  1286. [c1, c2, c1], lambda x: x)
  1287. def test_ticket10181(self):
  1288. # Avoid raising an EmptyResultSet if an inner query is probably
  1289. # empty (and hence, not executed).
  1290. self.assertQuerysetEqual(
  1291. Tag.objects.filter(id__in=Tag.objects.filter(id__in=[])),
  1292. []
  1293. )
  1294. def test_ticket15316_filter_false(self):
  1295. c1 = SimpleCategory.objects.create(name="category1")
  1296. c2 = SpecialCategory.objects.create(name="named category1", special_name="special1")
  1297. c3 = SpecialCategory.objects.create(name="named category2", special_name="special2")
  1298. CategoryItem.objects.create(category=c1)
  1299. ci2 = CategoryItem.objects.create(category=c2)
  1300. ci3 = CategoryItem.objects.create(category=c3)
  1301. qs = CategoryItem.objects.filter(category__specialcategory__isnull=False)
  1302. self.assertEqual(qs.count(), 2)
  1303. self.assertQuerysetEqual(qs, [ci2.pk, ci3.pk], lambda x: x.pk, False)
  1304. def test_ticket15316_exclude_false(self):
  1305. c1 = SimpleCategory.objects.create(name="category1")
  1306. c2 = SpecialCategory.objects.create(name="named category1", special_name="special1")
  1307. c3 = SpecialCategory.objects.create(name="named category2", special_name="special2")
  1308. ci1 = CategoryItem.objects.create(category=c1)
  1309. CategoryItem.objects.create(category=c2)
  1310. CategoryItem.objects.create(category=c3)
  1311. qs = CategoryItem.objects.exclude(category__specialcategory__isnull=False)
  1312. self.assertEqual(qs.count(), 1)
  1313. self.assertQuerysetEqual(qs, [ci1.pk], lambda x: x.pk)
  1314. def test_ticket15316_filter_true(self):
  1315. c1 = SimpleCategory.objects.create(name="category1")
  1316. c2 = SpecialCategory.objects.create(name="named category1", special_name="special1")
  1317. c3 = SpecialCategory.objects.create(name="named category2", special_name="special2")
  1318. ci1 = CategoryItem.objects.create(category=c1)
  1319. CategoryItem.objects.create(category=c2)
  1320. CategoryItem.objects.create(category=c3)
  1321. qs = CategoryItem.objects.filter(category__specialcategory__isnull=True)
  1322. self.assertEqual(qs.count(), 1)
  1323. self.assertQuerysetEqual(qs, [ci1.pk], lambda x: x.pk)
  1324. def test_ticket15316_exclude_true(self):
  1325. c1 = SimpleCategory.objects.create(name="category1")
  1326. c2 = SpecialCategory.objects.create(name="named category1", special_name="special1")
  1327. c3 = SpecialCategory.objects.create(name="named category2", special_name="special2")
  1328. CategoryItem.objects.create(category=c1)
  1329. ci2 = CategoryItem.objects.create(category=c2)
  1330. ci3 = CategoryItem.objects.create(category=c3)
  1331. qs = CategoryItem.objects.exclude(category__specialcategory__isnull=True)
  1332. self.assertEqual(qs.count(), 2)
  1333. self.assertQuerysetEqual(qs, [ci2.pk, ci3.pk], lambda x: x.pk, False)
  1334. def test_ticket15316_one2one_filter_false(self):
  1335. c = SimpleCategory.objects.create(name="cat")
  1336. c0 = SimpleCategory.objects.create(name="cat0")
  1337. c1 = SimpleCategory.objects.create(name="category1")
  1338. OneToOneCategory.objects.create(category=c1, new_name="new1")
  1339. OneToOneCategory.objects.create(category=c0, new_name="new2")
  1340. CategoryItem.objects.create(category=c)
  1341. ci2 = CategoryItem.objects.create(category=c0)
  1342. ci3 = CategoryItem.objects.create(category=c1)
  1343. qs = CategoryItem.objects.filter(category__onetoonecategory__isnull=False)
  1344. self.assertEqual(qs.count(), 2)
  1345. self.assertQuerysetEqual(qs, [ci2.pk, ci3.pk], lambda x: x.pk, False)
  1346. def test_ticket15316_one2one_exclude_false(self):
  1347. c = SimpleCategory.objects.create(name="cat")
  1348. c0 = SimpleCategory.objects.create(name="cat0")
  1349. c1 = SimpleCategory.objects.create(name="category1")
  1350. OneToOneCategory.objects.create(category=c1, new_name="new1")
  1351. OneToOneCategory.objects.create(category=c0, new_name="new2")
  1352. ci1 = CategoryItem.objects.create(category=c)
  1353. CategoryItem.objects.create(category=c0)
  1354. CategoryItem.objects.create(category=c1)
  1355. qs = CategoryItem.objects.exclude(category__onetoonecategory__isnull=False)
  1356. self.assertEqual(qs.count(), 1)
  1357. self.assertQuerysetEqual(qs, [ci1.pk], lambda x: x.pk)
  1358. def test_ticket15316_one2one_filter_true(self):
  1359. c = SimpleCategory.objects.create(name="cat")
  1360. c0 = SimpleCategory.objects.create(name="cat0")
  1361. c1 = SimpleCategory.objects.create(name="category1")
  1362. OneToOneCategory.objects.create(category=c1, new_name="new1")
  1363. OneToOneCategory.objects.create(category=c0, new_name="new2")
  1364. ci1 = CategoryItem.objects.create(category=c)
  1365. CategoryItem.objects.create(category=c0)
  1366. CategoryItem.objects.create(category=c1)
  1367. qs = CategoryItem.objects.filter(category__onetoonecategory__isnull=True)
  1368. self.assertEqual(qs.count(), 1)
  1369. self.assertQuerysetEqual(qs, [ci1.pk], lambda x: x.pk)
  1370. def test_ticket15316_one2one_exclude_true(self):
  1371. c = SimpleCategory.objects.create(name="cat")
  1372. c0 = SimpleCategory.objects.create(name="cat0")
  1373. c1 = SimpleCategory.objects.create(name="category1")
  1374. OneToOneCategory.objects.create(category=c1, new_name="new1")
  1375. OneToOneCategory.objects.create(category=c0, new_name="new2")
  1376. CategoryItem.objects.create(category=c)
  1377. ci2 = CategoryItem.objects.create(category=c0)
  1378. ci3 = CategoryItem.objects.create(category=c1)
  1379. qs = CategoryItem.objects.exclude(category__onetoonecategory__isnull=True)
  1380. self.assertEqual(qs.count(), 2)
  1381. self.assertQuerysetEqual(qs, [ci2.pk, ci3.pk], lambda x: x.pk, False)
  1382. class Queries5Tests(TestCase):
  1383. @classmethod
  1384. def setUpTestData(cls):
  1385. # Ordering by 'rank' gives us rank2, rank1, rank3. Ordering by the
  1386. # Meta.ordering will be rank3, rank2, rank1.
  1387. n1 = Note.objects.create(note='n1', misc='foo', id=1)
  1388. n2 = Note.objects.create(note='n2', misc='bar', id=2)
  1389. e1 = ExtraInfo.objects.create(info='e1', note=n1)
  1390. e2 = ExtraInfo.objects.create(info='e2', note=n2)
  1391. a1 = Author.objects.create(name='a1', num=1001, extra=e1)
  1392. a2 = Author.objects.create(name='a2', num=2002, extra=e1)
  1393. a3 = Author.objects.create(name='a3', num=3003, extra=e2)
  1394. cls.rank1 = Ranking.objects.create(rank=2, author=a2)
  1395. Ranking.objects.create(rank=1, author=a3)
  1396. Ranking.objects.create(rank=3, author=a1)
  1397. def test_ordering(self):
  1398. # Cross model ordering is possible in Meta, too.
  1399. self.assertQuerysetEqual(
  1400. Ranking.objects.all(),
  1401. ['<Ranking: 3: a1>', '<Ranking: 2: a2>', '<Ranking: 1: a3>']
  1402. )
  1403. self.assertQuerysetEqual(
  1404. Ranking.objects.all().order_by('rank'),
  1405. ['<Ranking: 1: a3>', '<Ranking: 2: a2>', '<Ranking: 3: a1>']
  1406. )
  1407. # Ordering of extra() pieces is possible, too and you can mix extra
  1408. # fields and model fields in the ordering.
  1409. self.assertQuerysetEqual(
  1410. Ranking.objects.extra(tables=['django_site'], order_by=['-django_site.id', 'rank']),
  1411. ['<Ranking: 1: a3>', '<Ranking: 2: a2>', '<Ranking: 3: a1>']
  1412. )
  1413. qs = Ranking.objects.extra(select={'good': 'case when rank > 2 then 1 else 0 end'})
  1414. self.assertEqual(
  1415. [o.good for o in qs.extra(order_by=('-good',))],
  1416. [True, False, False]
  1417. )
  1418. self.assertQuerysetEqual(
  1419. qs.extra(order_by=('-good', 'id')),
  1420. ['<Ranking: 3: a1>', '<Ranking: 2: a2>', '<Ranking: 1: a3>']
  1421. )
  1422. # Despite having some extra aliases in the query, we can still omit
  1423. # them in a values() query.
  1424. dicts = qs.values('id', 'rank').order_by('id')
  1425. self.assertEqual(
  1426. [d['rank'] for d in dicts],
  1427. [2, 1, 3]
  1428. )
  1429. def test_ticket7256(self):
  1430. # An empty values() call includes all aliases, including those from an
  1431. # extra()
  1432. qs = Ranking.objects.extra(select={'good': 'case when rank > 2 then 1 else 0 end'})
  1433. dicts = qs.values().order_by('id')
  1434. for d in dicts:
  1435. del d['id']
  1436. del d['author_id']
  1437. self.assertEqual(
  1438. [sorted(d.items()) for d in dicts],
  1439. [[('good', 0), ('rank', 2)], [('good', 0), ('rank', 1)], [('good', 1), ('rank', 3)]]
  1440. )
  1441. def test_ticket7045(self):
  1442. # Extra tables used to crash SQL construction on the second use.
  1443. qs = Ranking.objects.extra(tables=['django_site'])
  1444. qs.query.get_compiler(qs.db).as_sql()
  1445. # test passes if this doesn't raise an exception.
  1446. qs.query.get_compiler(qs.db).as_sql()
  1447. def test_ticket9848(self):
  1448. # Make sure that updates which only filter on sub-tables don't
  1449. # inadvertently update the wrong records (bug #9848).
  1450. # Make sure that the IDs from different tables don't happen to match.
  1451. self.assertQuerysetEqual(
  1452. Ranking.objects.filter(author__name='a1'),
  1453. ['<Ranking: 3: a1>']
  1454. )
  1455. self.assertEqual(
  1456. Ranking.objects.filter(author__name='a1').update(rank='4'),
  1457. 1
  1458. )
  1459. r = Ranking.objects.filter(author__name='a1')[0]
  1460. self.assertNotEqual(r.id, r.author.id)
  1461. self.assertEqual(r.rank, 4)
  1462. r.rank = 3
  1463. r.save()
  1464. self.assertQuerysetEqual(
  1465. Ranking.objects.all(),
  1466. ['<Ranking: 3: a1>', '<Ranking: 2: a2>', '<Ranking: 1: a3>']
  1467. )
  1468. def test_ticket5261(self):
  1469. # Test different empty excludes.
  1470. self.assertQuerysetEqual(
  1471. Note.objects.exclude(Q()),
  1472. ['<Note: n1>', '<Note: n2>']
  1473. )
  1474. self.assertQuerysetEqual(
  1475. Note.objects.filter(~Q()),
  1476. ['<Note: n1>', '<Note: n2>']
  1477. )
  1478. self.assertQuerysetEqual(
  1479. Note.objects.filter(~Q() | ~Q()),
  1480. ['<Note: n1>', '<Note: n2>']
  1481. )
  1482. self.assertQuerysetEqual(
  1483. Note.objects.exclude(~Q() & ~Q()),
  1484. ['<Note: n1>', '<Note: n2>']
  1485. )
  1486. def test_extra_select_literal_percent_s(self):
  1487. # Allow %%s to escape select clauses
  1488. self.assertEqual(
  1489. Note.objects.extra(select={'foo': "'%%s'"})[0].foo,
  1490. '%s'
  1491. )
  1492. self.assertEqual(
  1493. Note.objects.extra(select={'foo': "'%%s bar %%s'"})[0].foo,
  1494. '%s bar %s'
  1495. )
  1496. self.assertEqual(
  1497. Note.objects.extra(select={'foo': "'bar %%s'"})[0].foo,
  1498. 'bar %s'
  1499. )
  1500. class SelectRelatedTests(TestCase):
  1501. def test_tickets_3045_3288(self):
  1502. # Once upon a time, select_related() with circular relations would loop
  1503. # infinitely if you forgot to specify "depth". Now we set an arbitrary
  1504. # default upper bound.
  1505. self.assertQuerysetEqual(X.objects.all(), [])
  1506. self.assertQuerysetEqual(X.objects.select_related(), [])
  1507. class SubclassFKTests(TestCase):
  1508. def test_ticket7778(self):
  1509. # Model subclasses could not be deleted if a nullable foreign key
  1510. # relates to a model that relates back.
  1511. num_celebs = Celebrity.objects.count()
  1512. tvc = TvChef.objects.create(name="Huey")
  1513. self.assertEqual(Celebrity.objects.count(), num_celebs + 1)
  1514. Fan.objects.create(fan_of=tvc)
  1515. Fan.objects.create(fan_of=tvc)
  1516. tvc.delete()
  1517. # The parent object should have been deleted as well.
  1518. self.assertEqual(Celebrity.objects.count(), num_celebs)
  1519. class CustomPkTests(TestCase):
  1520. def test_ticket7371(self):
  1521. self.assertQuerysetEqual(Related.objects.order_by('custom'), [])
  1522. class NullableRelOrderingTests(TestCase):
  1523. def test_ticket10028(self):
  1524. # Ordering by model related to nullable relations(!) should use outer
  1525. # joins, so that all results are included.
  1526. Plaything.objects.create(name="p1")
  1527. self.assertQuerysetEqual(
  1528. Plaything.objects.all(),
  1529. ['<Plaything: p1>']
  1530. )
  1531. def test_join_already_in_query(self):
  1532. # Ordering by model related to nullable relations should not change
  1533. # the join type of already existing joins.
  1534. Plaything.objects.create(name="p1")
  1535. s = SingleObject.objects.create(name='s')
  1536. r = RelatedObject.objects.create(single=s, f=1)
  1537. Plaything.objects.create(name="p2", others=r)
  1538. qs = Plaything.objects.all().filter(others__isnull=False).order_by('pk')
  1539. self.assertNotIn('JOIN', str(qs.query))
  1540. qs = Plaything.objects.all().filter(others__f__isnull=False).order_by('pk')
  1541. self.assertIn('INNER', str(qs.query))
  1542. qs = qs.order_by('others__single__name')
  1543. # The ordering by others__single__pk will add one new join (to single)
  1544. # and that join must be LEFT join. The already existing join to related
  1545. # objects must be kept INNER. So, we have both an INNER and a LEFT join
  1546. # in the query.
  1547. self.assertEqual(str(qs.query).count('LEFT'), 1)
  1548. self.assertEqual(str(qs.query).count('INNER'), 1)
  1549. self.assertQuerysetEqual(
  1550. qs,
  1551. ['<Plaything: p2>']
  1552. )
  1553. class DisjunctiveFilterTests(TestCase):
  1554. @classmethod
  1555. def setUpTestData(cls):
  1556. cls.n1 = Note.objects.create(note='n1', misc='foo', id=1)
  1557. ExtraInfo.objects.create(info='e1', note=cls.n1)
  1558. def test_ticket7872(self):
  1559. # Another variation on the disjunctive filtering theme.
  1560. # For the purposes of this regression test, it's important that there is no
  1561. # Join object related to the LeafA we create.
  1562. LeafA.objects.create(data='first')
  1563. self.assertQuerysetEqual(LeafA.objects.all(), ['<LeafA: first>'])
  1564. self.assertQuerysetEqual(
  1565. LeafA.objects.filter(Q(data='first') | Q(join__b__data='second')),
  1566. ['<LeafA: first>']
  1567. )
  1568. def test_ticket8283(self):
  1569. # Checking that applying filters after a disjunction works correctly.
  1570. self.assertQuerysetEqual(
  1571. (ExtraInfo.objects.filter(note=self.n1) | ExtraInfo.objects.filter(info='e2')).filter(note=self.n1),
  1572. ['<ExtraInfo: e1>']
  1573. )
  1574. self.assertQuerysetEqual(
  1575. (ExtraInfo.objects.filter(info='e2') | ExtraInfo.objects.filter(note=self.n1)).filter(note=self.n1),
  1576. ['<ExtraInfo: e1>']
  1577. )
  1578. class Queries6Tests(TestCase):
  1579. @classmethod
  1580. def setUpTestData(cls):
  1581. generic = NamedCategory.objects.create(name="Generic")
  1582. t1 = Tag.objects.create(name='t1', category=generic)
  1583. Tag.objects.create(name='t2', parent=t1, category=generic)
  1584. t3 = Tag.objects.create(name='t3', parent=t1)
  1585. t4 = Tag.objects.create(name='t4', parent=t3)
  1586. Tag.objects.create(name='t5', parent=t3)
  1587. n1 = Note.objects.create(note='n1', misc='foo', id=1)
  1588. ann1 = Annotation.objects.create(name='a1', tag=t1)
  1589. ann1.notes.add(n1)
  1590. Annotation.objects.create(name='a2', tag=t4)
  1591. def test_parallel_iterators(self):
  1592. # Test that parallel iterators work.
  1593. qs = Tag.objects.all()
  1594. i1, i2 = iter(qs), iter(qs)
  1595. self.assertEqual(repr(next(i1)), '<Tag: t1>')
  1596. self.assertEqual(repr(next(i1)), '<Tag: t2>')
  1597. self.assertEqual(repr(next(i2)), '<Tag: t1>')
  1598. self.assertEqual(repr(next(i2)), '<Tag: t2>')
  1599. self.assertEqual(repr(next(i2)), '<Tag: t3>')
  1600. self.assertEqual(repr(next(i1)), '<Tag: t3>')
  1601. qs = X.objects.all()
  1602. self.assertFalse(qs)
  1603. self.assertFalse(qs)
  1604. def test_nested_queries_sql(self):
  1605. # Nested queries should not evaluate the inner query as part of constructing the
  1606. # SQL (so we should see a nested query here, indicated by two "SELECT" calls).
  1607. qs = Annotation.objects.filter(notes__in=Note.objects.filter(note="xyzzy"))
  1608. self.assertEqual(
  1609. qs.query.get_compiler(qs.db).as_sql()[0].count('SELECT'),
  1610. 2
  1611. )
  1612. def test_tickets_8921_9188(self):
  1613. # Incorrect SQL was being generated for certain types of exclude()
  1614. # queries that crossed multi-valued relations (#8921, #9188 and some
  1615. # preemptively discovered cases).
  1616. self.assertQuerysetEqual(
  1617. PointerA.objects.filter(connection__pointerb__id=1),
  1618. []
  1619. )
  1620. self.assertQuerysetEqual(
  1621. PointerA.objects.exclude(connection__pointerb__id=1),
  1622. []
  1623. )
  1624. self.assertQuerysetEqual(
  1625. Tag.objects.exclude(children=None),
  1626. ['<Tag: t1>', '<Tag: t3>']
  1627. )
  1628. # This example is tricky because the parent could be NULL, so only checking
  1629. # parents with annotations omits some results (tag t1, in this case).
  1630. self.assertQuerysetEqual(
  1631. Tag.objects.exclude(parent__annotation__name="a1"),
  1632. ['<Tag: t1>', '<Tag: t4>', '<Tag: t5>']
  1633. )
  1634. # The annotation->tag link is single values and tag->children links is
  1635. # multi-valued. So we have to split the exclude filter in the middle
  1636. # and then optimize the inner query without losing results.
  1637. self.assertQuerysetEqual(
  1638. Annotation.objects.exclude(tag__children__name="t2"),
  1639. ['<Annotation: a2>']
  1640. )
  1641. # Nested queries are possible (although should be used with care, since
  1642. # they have performance problems on backends like MySQL.
  1643. self.assertQuerysetEqual(
  1644. Annotation.objects.filter(notes__in=Note.objects.filter(note="n1")),
  1645. ['<Annotation: a1>']
  1646. )
  1647. def test_ticket3739(self):
  1648. # The all() method on querysets returns a copy of the queryset.
  1649. q1 = Tag.objects.order_by('name')
  1650. self.assertIsNot(q1, q1.all())
  1651. def test_ticket_11320(self):
  1652. qs = Tag.objects.exclude(category=None).exclude(category__name='foo')
  1653. self.assertEqual(str(qs.query).count(' INNER JOIN '), 1)
  1654. class RawQueriesTests(TestCase):
  1655. def setUp(self):
  1656. Note.objects.create(note='n1', misc='foo', id=1)
  1657. def test_ticket14729(self):
  1658. # Test representation of raw query with one or few parameters passed as list
  1659. query = "SELECT * FROM queries_note WHERE note = %s"
  1660. params = ['n1']
  1661. qs = Note.objects.raw(query, params=params)
  1662. self.assertEqual(repr(qs), "<RawQuerySet: SELECT * FROM queries_note WHERE note = n1>")
  1663. query = "SELECT * FROM queries_note WHERE note = %s and misc = %s"
  1664. params = ['n1', 'foo']
  1665. qs = Note.objects.raw(query, params=params)
  1666. self.assertEqual(repr(qs), "<RawQuerySet: SELECT * FROM queries_note WHERE note = n1 and misc = foo>")
  1667. class GeneratorExpressionTests(TestCase):
  1668. def test_ticket10432(self):
  1669. # Using an empty generator expression as the rvalue for an "__in"
  1670. # lookup is legal.
  1671. self.assertQuerysetEqual(
  1672. Note.objects.filter(pk__in=(x for x in ())),
  1673. []
  1674. )
  1675. class ComparisonTests(TestCase):
  1676. def setUp(self):
  1677. self.n1 = Note.objects.create(note='n1', misc='foo', id=1)
  1678. e1 = ExtraInfo.objects.create(info='e1', note=self.n1)
  1679. self.a2 = Author.objects.create(name='a2', num=2002, extra=e1)
  1680. def test_ticket8597(self):
  1681. # Regression tests for case-insensitive comparisons
  1682. Item.objects.create(name="a_b", created=datetime.datetime.now(), creator=self.a2, note=self.n1)
  1683. Item.objects.create(name="x%y", created=datetime.datetime.now(), creator=self.a2, note=self.n1)
  1684. self.assertQuerysetEqual(
  1685. Item.objects.filter(name__iexact="A_b"),
  1686. ['<Item: a_b>']
  1687. )
  1688. self.assertQuerysetEqual(
  1689. Item.objects.filter(name__iexact="x%Y"),
  1690. ['<Item: x%y>']
  1691. )
  1692. self.assertQuerysetEqual(
  1693. Item.objects.filter(name__istartswith="A_b"),
  1694. ['<Item: a_b>']
  1695. )
  1696. self.assertQuerysetEqual(
  1697. Item.objects.filter(name__iendswith="A_b"),
  1698. ['<Item: a_b>']
  1699. )
  1700. class ExistsSql(TestCase):
  1701. def test_exists(self):
  1702. with CaptureQueriesContext(connection) as captured_queries:
  1703. self.assertFalse(Tag.objects.exists())
  1704. # Ok - so the exist query worked - but did it include too many columns?
  1705. self.assertEqual(len(captured_queries), 1)
  1706. qstr = captured_queries[0]['sql']
  1707. id, name = connection.ops.quote_name('id'), connection.ops.quote_name('name')
  1708. self.assertNotIn(id, qstr)
  1709. self.assertNotIn(name, qstr)
  1710. def test_ticket_18414(self):
  1711. Article.objects.create(name='one', created=datetime.datetime.now())
  1712. Article.objects.create(name='one', created=datetime.datetime.now())
  1713. Article.objects.create(name='two', created=datetime.datetime.now())
  1714. self.assertTrue(Article.objects.exists())
  1715. self.assertTrue(Article.objects.distinct().exists())
  1716. self.assertTrue(Article.objects.distinct()[1:3].exists())
  1717. self.assertFalse(Article.objects.distinct()[1:1].exists())
  1718. @skipUnlessDBFeature('can_distinct_on_fields')
  1719. def test_ticket_18414_distinct_on(self):
  1720. Article.objects.create(name='one', created=datetime.datetime.now())
  1721. Article.objects.create(name='one', created=datetime.datetime.now())
  1722. Article.objects.create(name='two', created=datetime.datetime.now())
  1723. self.assertTrue(Article.objects.distinct('name').exists())
  1724. self.assertTrue(Article.objects.distinct('name')[1:2].exists())
  1725. self.assertFalse(Article.objects.distinct('name')[2:3].exists())
  1726. class QuerysetOrderedTests(unittest.TestCase):
  1727. """
  1728. Tests for the Queryset.ordered attribute.
  1729. """
  1730. def test_no_default_or_explicit_ordering(self):
  1731. self.assertIs(Annotation.objects.all().ordered, False)
  1732. def test_cleared_default_ordering(self):
  1733. self.assertIs(Tag.objects.all().ordered, True)
  1734. self.assertIs(Tag.objects.all().order_by().ordered, False)
  1735. def test_explicit_ordering(self):
  1736. self.assertIs(Annotation.objects.all().order_by('id').ordered, True)
  1737. def test_order_by_extra(self):
  1738. self.assertIs(Annotation.objects.all().extra(order_by=['id']).ordered, True)
  1739. def test_annotated_ordering(self):
  1740. qs = Annotation.objects.annotate(num_notes=Count('notes'))
  1741. self.assertIs(qs.ordered, False)
  1742. self.assertIs(qs.order_by('num_notes').ordered, True)
  1743. @skipUnlessDBFeature('allow_sliced_subqueries')
  1744. class SubqueryTests(TestCase):
  1745. @classmethod
  1746. def setUpTestData(cls):
  1747. DumbCategory.objects.create(id=1)
  1748. DumbCategory.objects.create(id=2)
  1749. DumbCategory.objects.create(id=3)
  1750. DumbCategory.objects.create(id=4)
  1751. def test_ordered_subselect(self):
  1752. "Subselects honor any manual ordering"
  1753. query = DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[0:2])
  1754. self.assertEqual(set(query.values_list('id', flat=True)), {3, 4})
  1755. query = DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[:2])
  1756. self.assertEqual(set(query.values_list('id', flat=True)), {3, 4})
  1757. query = DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[1:2])
  1758. self.assertEqual(set(query.values_list('id', flat=True)), {3})
  1759. query = DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[2:])
  1760. self.assertEqual(set(query.values_list('id', flat=True)), {1, 2})
  1761. def test_slice_subquery_and_query(self):
  1762. """
  1763. Slice a query that has a sliced subquery
  1764. """
  1765. query = DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[0:2])[0:2]
  1766. self.assertEqual({x.id for x in query}, {3, 4})
  1767. query = DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[1:3])[1:3]
  1768. self.assertEqual({x.id for x in query}, {3})
  1769. query = DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[2:])[1:]
  1770. self.assertEqual({x.id for x in query}, {2})
  1771. def test_related_sliced_subquery(self):
  1772. """
  1773. Related objects constraints can safely contain sliced subqueries.
  1774. refs #22434
  1775. """
  1776. generic = NamedCategory.objects.create(id=5, name="Generic")
  1777. t1 = Tag.objects.create(name='t1', category=generic)
  1778. t2 = Tag.objects.create(name='t2', category=generic)
  1779. ManagedModel.objects.create(data='mm1', tag=t1, public=True)
  1780. mm2 = ManagedModel.objects.create(data='mm2', tag=t2, public=True)
  1781. query = ManagedModel.normal_manager.filter(
  1782. tag__in=Tag.objects.order_by('-id')[:1]
  1783. )
  1784. self.assertEqual({x.id for x in query}, {mm2.id})
  1785. def test_sliced_delete(self):
  1786. "Delete queries can safely contain sliced subqueries"
  1787. DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[0:1]).delete()
  1788. self.assertEqual(set(DumbCategory.objects.values_list('id', flat=True)), {1, 2, 3})
  1789. DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[1:2]).delete()
  1790. self.assertEqual(set(DumbCategory.objects.values_list('id', flat=True)), {1, 3})
  1791. DumbCategory.objects.filter(id__in=DumbCategory.objects.order_by('-id')[1:]).delete()
  1792. self.assertEqual(set(DumbCategory.objects.values_list('id', flat=True)), {3})
  1793. class CloneTests(TestCase):
  1794. def test_evaluated_queryset_as_argument(self):
  1795. "#13227 -- If a queryset is already evaluated, it can still be used as a query arg"
  1796. n = Note(note='Test1', misc='misc')
  1797. n.save()
  1798. e = ExtraInfo(info='good', note=n)
  1799. e.save()
  1800. n_list = Note.objects.all()
  1801. # Evaluate the Note queryset, populating the query cache
  1802. list(n_list)
  1803. # Use the note queryset in a query, and evaluate
  1804. # that query in a way that involves cloning.
  1805. self.assertEqual(ExtraInfo.objects.filter(note__in=n_list)[0].info, 'good')
  1806. def test_no_model_options_cloning(self):
  1807. """
  1808. Test that cloning a queryset does not get out of hand. While complete
  1809. testing is impossible, this is a sanity check against invalid use of
  1810. deepcopy. refs #16759.
  1811. """
  1812. opts_class = type(Note._meta)
  1813. note_deepcopy = getattr(opts_class, "__deepcopy__", None)
  1814. opts_class.__deepcopy__ = lambda obj, memo: self.fail("Model options shouldn't be cloned.")
  1815. try:
  1816. Note.objects.filter(pk__lte=F('pk') + 1).all()
  1817. finally:
  1818. if note_deepcopy is None:
  1819. delattr(opts_class, "__deepcopy__")
  1820. else:
  1821. opts_class.__deepcopy__ = note_deepcopy
  1822. def test_no_fields_cloning(self):
  1823. """
  1824. Test that cloning a queryset does not get out of hand. While complete
  1825. testing is impossible, this is a sanity check against invalid use of
  1826. deepcopy. refs #16759.
  1827. """
  1828. opts_class = type(Note._meta.get_field("misc"))
  1829. note_deepcopy = getattr(opts_class, "__deepcopy__", None)
  1830. opts_class.__deepcopy__ = lambda obj, memo: self.fail("Model fields shouldn't be cloned")
  1831. try:
  1832. Note.objects.filter(note=F('misc')).all()
  1833. finally:
  1834. if note_deepcopy is None:
  1835. delattr(opts_class, "__deepcopy__")
  1836. else:
  1837. opts_class.__deepcopy__ = note_deepcopy
  1838. class EmptyQuerySetTests(TestCase):
  1839. def test_emptyqueryset_values(self):
  1840. # #14366 -- Calling .values() on an empty QuerySet and then cloning
  1841. # that should not cause an error
  1842. self.assertQuerysetEqual(
  1843. Number.objects.none().values('num').order_by('num'), []
  1844. )
  1845. def test_values_subquery(self):
  1846. self.assertQuerysetEqual(
  1847. Number.objects.filter(pk__in=Number.objects.none().values("pk")),
  1848. []
  1849. )
  1850. self.assertQuerysetEqual(
  1851. Number.objects.filter(pk__in=Number.objects.none().values_list("pk")),
  1852. []
  1853. )
  1854. def test_ticket_19151(self):
  1855. # #19151 -- Calling .values() or .values_list() on an empty QuerySet
  1856. # should return an empty QuerySet and not cause an error.
  1857. q = Author.objects.none()
  1858. self.assertQuerysetEqual(q.values(), [])
  1859. self.assertQuerysetEqual(q.values_list(), [])
  1860. class ValuesQuerysetTests(BaseQuerysetTest):
  1861. @classmethod
  1862. def setUpTestData(cls):
  1863. Number.objects.create(num=72)
  1864. cls.identity = staticmethod(lambda x: x)
  1865. def test_flat_values_list(self):
  1866. qs = Number.objects.values_list("num")
  1867. qs = qs.values_list("num", flat=True)
  1868. self.assertValueQuerysetEqual(qs, [72])
  1869. def test_extra_values(self):
  1870. # testing for ticket 14930 issues
  1871. qs = Number.objects.extra(select=OrderedDict([('value_plus_x', 'num+%s'),
  1872. ('value_minus_x', 'num-%s')]),
  1873. select_params=(1, 2))
  1874. qs = qs.order_by('value_minus_x')
  1875. qs = qs.values('num')
  1876. self.assertQuerysetEqual(qs, [{'num': 72}], self.identity)
  1877. def test_extra_values_order_twice(self):
  1878. # testing for ticket 14930 issues
  1879. qs = Number.objects.extra(select={'value_plus_one': 'num+1', 'value_minus_one': 'num-1'})
  1880. qs = qs.order_by('value_minus_one').order_by('value_plus_one')
  1881. qs = qs.values('num')
  1882. self.assertQuerysetEqual(qs, [{'num': 72}], self.identity)
  1883. def test_extra_values_order_multiple(self):
  1884. # Postgres doesn't allow constants in order by, so check for that.
  1885. qs = Number.objects.extra(select={
  1886. 'value_plus_one': 'num+1',
  1887. 'value_minus_one': 'num-1',
  1888. 'constant_value': '1'
  1889. })
  1890. qs = qs.order_by('value_plus_one', 'value_minus_one', 'constant_value')
  1891. qs = qs.values('num')
  1892. self.assertQuerysetEqual(qs, [{'num': 72}], self.identity)
  1893. def test_extra_values_order_in_extra(self):
  1894. # testing for ticket 14930 issues
  1895. qs = Number.objects.extra(
  1896. select={'value_plus_one': 'num+1', 'value_minus_one': 'num-1'},
  1897. order_by=['value_minus_one'])
  1898. qs = qs.values('num')
  1899. def test_extra_select_params_values_order_in_extra(self):
  1900. # testing for 23259 issue
  1901. qs = Number.objects.extra(
  1902. select={'value_plus_x': 'num+%s'},
  1903. select_params=[1],
  1904. order_by=['value_plus_x'])
  1905. qs = qs.filter(num=72)
  1906. qs = qs.values('num')
  1907. self.assertQuerysetEqual(qs, [{'num': 72}], self.identity)
  1908. def test_extra_multiple_select_params_values_order_by(self):
  1909. # testing for 23259 issue
  1910. qs = Number.objects.extra(select=OrderedDict([('value_plus_x', 'num+%s'),
  1911. ('value_minus_x', 'num-%s')]),
  1912. select_params=(72, 72))
  1913. qs = qs.order_by('value_minus_x')
  1914. qs = qs.filter(num=1)
  1915. qs = qs.values('num')
  1916. self.assertQuerysetEqual(qs, [], self.identity)
  1917. def test_extra_values_list(self):
  1918. # testing for ticket 14930 issues
  1919. qs = Number.objects.extra(select={'value_plus_one': 'num+1'})
  1920. qs = qs.order_by('value_plus_one')
  1921. qs = qs.values_list('num')
  1922. self.assertQuerysetEqual(qs, [(72,)], self.identity)
  1923. def test_flat_extra_values_list(self):
  1924. # testing for ticket 14930 issues
  1925. qs = Number.objects.extra(select={'value_plus_one': 'num+1'})
  1926. qs = qs.order_by('value_plus_one')
  1927. qs = qs.values_list('num', flat=True)
  1928. self.assertQuerysetEqual(qs, [72], self.identity)
  1929. def test_field_error_values_list(self):
  1930. # see #23443
  1931. msg = "Cannot resolve keyword %r into field. Join on 'name' not permitted." % 'foo'
  1932. with self.assertRaisesMessage(FieldError, msg):
  1933. Tag.objects.values_list('name__foo')
  1934. class QuerySetSupportsPythonIdioms(TestCase):
  1935. @classmethod
  1936. def setUpTestData(cls):
  1937. some_date = datetime.datetime(2014, 5, 16, 12, 1)
  1938. for i in range(1, 8):
  1939. Article.objects.create(
  1940. name="Article {}".format(i), created=some_date)
  1941. def get_ordered_articles(self):
  1942. return Article.objects.all().order_by('name')
  1943. def test_can_get_items_using_index_and_slice_notation(self):
  1944. self.assertEqual(self.get_ordered_articles()[0].name, 'Article 1')
  1945. self.assertQuerysetEqual(
  1946. self.get_ordered_articles()[1:3],
  1947. ["<Article: Article 2>", "<Article: Article 3>"]
  1948. )
  1949. def test_slicing_with_steps_can_be_used(self):
  1950. self.assertQuerysetEqual(
  1951. self.get_ordered_articles()[::2], [
  1952. "<Article: Article 1>",
  1953. "<Article: Article 3>",
  1954. "<Article: Article 5>",
  1955. "<Article: Article 7>"
  1956. ]
  1957. )
  1958. @unittest.skipUnless(six.PY2, "Python 2 only -- Python 3 doesn't have longs.")
  1959. def test_slicing_works_with_longs(self):
  1960. # NOQA: long undefined on PY3
  1961. self.assertEqual(self.get_ordered_articles()[long(0)].name, 'Article 1') # NOQA
  1962. self.assertQuerysetEqual(self.get_ordered_articles()[long(1):long(3)], # NOQA
  1963. ["<Article: Article 2>", "<Article: Article 3>"])
  1964. self.assertQuerysetEqual(
  1965. self.get_ordered_articles()[::long(2)], [ # NOQA
  1966. "<Article: Article 1>",
  1967. "<Article: Article 3>",
  1968. "<Article: Article 5>",
  1969. "<Article: Article 7>"
  1970. ]
  1971. )
  1972. # And can be mixed with ints.
  1973. self.assertQuerysetEqual(self.get_ordered_articles()[1:long(3)], # NOQA
  1974. ["<Article: Article 2>", "<Article: Article 3>"])
  1975. def test_slicing_without_step_is_lazy(self):
  1976. with self.assertNumQueries(0):
  1977. self.get_ordered_articles()[0:5]
  1978. def test_slicing_with_tests_is_not_lazy(self):
  1979. with self.assertNumQueries(1):
  1980. self.get_ordered_articles()[0:5:3]
  1981. def test_slicing_can_slice_again_after_slicing(self):
  1982. self.assertQuerysetEqual(
  1983. self.get_ordered_articles()[0:5][0:2],
  1984. ["<Article: Article 1>", "<Article: Article 2>"]
  1985. )
  1986. self.assertQuerysetEqual(self.get_ordered_articles()[0:5][4:], ["<Article: Article 5>"])
  1987. self.assertQuerysetEqual(self.get_ordered_articles()[0:5][5:], [])
  1988. # Some more tests!
  1989. self.assertQuerysetEqual(
  1990. self.get_ordered_articles()[2:][0:2],
  1991. ["<Article: Article 3>", "<Article: Article 4>"]
  1992. )
  1993. self.assertQuerysetEqual(
  1994. self.get_ordered_articles()[2:][:2],
  1995. ["<Article: Article 3>", "<Article: Article 4>"]
  1996. )
  1997. self.assertQuerysetEqual(self.get_ordered_articles()[2:][2:3], ["<Article: Article 5>"])
  1998. # Using an offset without a limit is also possible.
  1999. self.assertQuerysetEqual(
  2000. self.get_ordered_articles()[5:],
  2001. ["<Article: Article 6>", "<Article: Article 7>"]
  2002. )
  2003. def test_slicing_cannot_filter_queryset_once_sliced(self):
  2004. with self.assertRaisesMessage(AssertionError, "Cannot filter a query once a slice has been taken."):
  2005. Article.objects.all()[0:5].filter(id=1, )
  2006. def test_slicing_cannot_reorder_queryset_once_sliced(self):
  2007. with self.assertRaisesMessage(AssertionError, "Cannot reorder a query once a slice has been taken."):
  2008. Article.objects.all()[0:5].order_by('id', )
  2009. def test_slicing_cannot_combine_queries_once_sliced(self):
  2010. with self.assertRaisesMessage(AssertionError, "Cannot combine queries once a slice has been taken."):
  2011. Article.objects.all()[0:1] & Article.objects.all()[4:5]
  2012. def test_slicing_negative_indexing_not_supported_for_single_element(self):
  2013. """hint: inverting your ordering might do what you need"""
  2014. with self.assertRaisesMessage(AssertionError, "Negative indexing is not supported."):
  2015. Article.objects.all()[-1]
  2016. def test_slicing_negative_indexing_not_supported_for_range(self):
  2017. """hint: inverting your ordering might do what you need"""
  2018. with self.assertRaisesMessage(AssertionError, "Negative indexing is not supported."):
  2019. Article.objects.all()[0:-5]
  2020. def test_can_get_number_of_items_in_queryset_using_standard_len(self):
  2021. self.assertEqual(len(Article.objects.filter(name__exact='Article 1')), 1)
  2022. def test_can_combine_queries_using_and_and_or_operators(self):
  2023. s1 = Article.objects.filter(name__exact='Article 1')
  2024. s2 = Article.objects.filter(name__exact='Article 2')
  2025. self.assertQuerysetEqual(
  2026. (s1 | s2).order_by('name'),
  2027. ["<Article: Article 1>", "<Article: Article 2>"]
  2028. )
  2029. self.assertQuerysetEqual(s1 & s2, [])
  2030. class WeirdQuerysetSlicingTests(BaseQuerysetTest):
  2031. @classmethod
  2032. def setUpTestData(cls):
  2033. Number.objects.create(num=1)
  2034. Number.objects.create(num=2)
  2035. Article.objects.create(name='one', created=datetime.datetime.now())
  2036. Article.objects.create(name='two', created=datetime.datetime.now())
  2037. Article.objects.create(name='three', created=datetime.datetime.now())
  2038. Article.objects.create(name='four', created=datetime.datetime.now())
  2039. food = Food.objects.create(name='spam')
  2040. Eaten.objects.create(meal='spam with eggs', food=food)
  2041. def test_tickets_7698_10202(self):
  2042. # People like to slice with '0' as the high-water mark.
  2043. self.assertQuerysetEqual(Article.objects.all()[0:0], [])
  2044. self.assertQuerysetEqual(Article.objects.all()[0:0][:10], [])
  2045. self.assertEqual(Article.objects.all()[:0].count(), 0)
  2046. with self.assertRaisesMessage(AssertionError, 'Cannot change a query once a slice has been taken.'):
  2047. Article.objects.all()[:0].latest('created')
  2048. def test_empty_resultset_sql(self):
  2049. # ticket #12192
  2050. self.assertNumQueries(0, lambda: list(Number.objects.all()[1:1]))
  2051. def test_empty_sliced_subquery(self):
  2052. self.assertEqual(Eaten.objects.filter(food__in=Food.objects.all()[0:0]).count(), 0)
  2053. def test_empty_sliced_subquery_exclude(self):
  2054. self.assertEqual(Eaten.objects.exclude(food__in=Food.objects.all()[0:0]).count(), 1)
  2055. def test_zero_length_values_slicing(self):
  2056. n = 42
  2057. with self.assertNumQueries(0):
  2058. self.assertQuerysetEqual(Article.objects.values()[n:n], [])
  2059. self.assertQuerysetEqual(Article.objects.values_list()[n:n], [])
  2060. class EscapingTests(TestCase):
  2061. def test_ticket_7302(self):
  2062. # Reserved names are appropriately escaped
  2063. ReservedName.objects.create(name='a', order=42)
  2064. ReservedName.objects.create(name='b', order=37)
  2065. self.assertQuerysetEqual(
  2066. ReservedName.objects.all().order_by('order'),
  2067. ['<ReservedName: b>', '<ReservedName: a>']
  2068. )
  2069. self.assertQuerysetEqual(
  2070. ReservedName.objects.extra(select={'stuff': 'name'}, order_by=('order', 'stuff')),
  2071. ['<ReservedName: b>', '<ReservedName: a>']
  2072. )
  2073. class ToFieldTests(TestCase):
  2074. def test_in_query(self):
  2075. apple = Food.objects.create(name="apple")
  2076. pear = Food.objects.create(name="pear")
  2077. lunch = Eaten.objects.create(food=apple, meal="lunch")
  2078. dinner = Eaten.objects.create(food=pear, meal="dinner")
  2079. self.assertEqual(
  2080. set(Eaten.objects.filter(food__in=[apple, pear])),
  2081. {lunch, dinner},
  2082. )
  2083. def test_in_subquery(self):
  2084. apple = Food.objects.create(name="apple")
  2085. lunch = Eaten.objects.create(food=apple, meal="lunch")
  2086. self.assertEqual(
  2087. set(Eaten.objects.filter(food__in=Food.objects.filter(name='apple'))),
  2088. {lunch}
  2089. )
  2090. self.assertEqual(
  2091. set(Eaten.objects.filter(food__in=Food.objects.filter(name='apple').values('eaten__meal'))),
  2092. set()
  2093. )
  2094. self.assertEqual(
  2095. set(Food.objects.filter(eaten__in=Eaten.objects.filter(meal='lunch'))),
  2096. {apple}
  2097. )
  2098. def test_reverse_in(self):
  2099. apple = Food.objects.create(name="apple")
  2100. pear = Food.objects.create(name="pear")
  2101. lunch_apple = Eaten.objects.create(food=apple, meal="lunch")
  2102. lunch_pear = Eaten.objects.create(food=pear, meal="dinner")
  2103. self.assertEqual(
  2104. set(Food.objects.filter(eaten__in=[lunch_apple, lunch_pear])),
  2105. {apple, pear}
  2106. )
  2107. def test_single_object(self):
  2108. apple = Food.objects.create(name="apple")
  2109. lunch = Eaten.objects.create(food=apple, meal="lunch")
  2110. dinner = Eaten.objects.create(food=apple, meal="dinner")
  2111. self.assertEqual(
  2112. set(Eaten.objects.filter(food=apple)),
  2113. {lunch, dinner}
  2114. )
  2115. def test_single_object_reverse(self):
  2116. apple = Food.objects.create(name="apple")
  2117. lunch = Eaten.objects.create(food=apple, meal="lunch")
  2118. self.assertEqual(
  2119. set(Food.objects.filter(eaten=lunch)),
  2120. {apple}
  2121. )
  2122. def test_recursive_fk(self):
  2123. node1 = Node.objects.create(num=42)
  2124. node2 = Node.objects.create(num=1, parent=node1)
  2125. self.assertEqual(
  2126. list(Node.objects.filter(parent=node1)),
  2127. [node2]
  2128. )
  2129. def test_recursive_fk_reverse(self):
  2130. node1 = Node.objects.create(num=42)
  2131. node2 = Node.objects.create(num=1, parent=node1)
  2132. self.assertEqual(
  2133. list(Node.objects.filter(node=node2)),
  2134. [node1]
  2135. )
  2136. class ConditionalTests(BaseQuerysetTest):
  2137. """Tests whose execution depend on different environment conditions like
  2138. Python version or DB backend features"""
  2139. @classmethod
  2140. def setUpTestData(cls):
  2141. generic = NamedCategory.objects.create(name="Generic")
  2142. t1 = Tag.objects.create(name='t1', category=generic)
  2143. Tag.objects.create(name='t2', parent=t1, category=generic)
  2144. t3 = Tag.objects.create(name='t3', parent=t1)
  2145. Tag.objects.create(name='t4', parent=t3)
  2146. Tag.objects.create(name='t5', parent=t3)
  2147. def test_infinite_loop(self):
  2148. # If you're not careful, it's possible to introduce infinite loops via
  2149. # default ordering on foreign keys in a cycle. We detect that.
  2150. with self.assertRaisesMessage(FieldError, 'Infinite loop caused by ordering.'):
  2151. list(LoopX.objects.all()) # Force queryset evaluation with list()
  2152. with self.assertRaisesMessage(FieldError, 'Infinite loop caused by ordering.'):
  2153. list(LoopZ.objects.all()) # Force queryset evaluation with list()
  2154. # Note that this doesn't cause an infinite loop, since the default
  2155. # ordering on the Tag model is empty (and thus defaults to using "id"
  2156. # for the related field).
  2157. self.assertEqual(len(Tag.objects.order_by('parent')), 5)
  2158. # ... but you can still order in a non-recursive fashion among linked
  2159. # fields (the previous test failed because the default ordering was
  2160. # recursive).
  2161. self.assertQuerysetEqual(
  2162. LoopX.objects.all().order_by('y__x__y__x__id'),
  2163. []
  2164. )
  2165. # When grouping without specifying ordering, we add an explicit "ORDER BY NULL"
  2166. # portion in MySQL to prevent unnecessary sorting.
  2167. @skipUnlessDBFeature('requires_explicit_null_ordering_when_grouping')
  2168. def test_null_ordering_added(self):
  2169. query = Tag.objects.values_list('parent_id', flat=True).order_by().query
  2170. query.group_by = ['parent_id']
  2171. sql = query.get_compiler(DEFAULT_DB_ALIAS).as_sql()[0]
  2172. fragment = "ORDER BY "
  2173. pos = sql.find(fragment)
  2174. self.assertEqual(sql.find(fragment, pos + 1), -1)
  2175. self.assertEqual(sql.find("NULL", pos + len(fragment)), pos + len(fragment))
  2176. # Sqlite 3 does not support passing in more than 1000 parameters except by
  2177. # changing a parameter at compilation time.
  2178. @skipUnlessDBFeature('supports_1000_query_parameters')
  2179. def test_ticket14244(self):
  2180. # Test that the "in" lookup works with lists of 1000 items or more.
  2181. # The numbers amount is picked to force three different IN batches
  2182. # for Oracle, yet to be less than 2100 parameter limit for MSSQL.
  2183. numbers = list(range(2050))
  2184. Number.objects.all().delete()
  2185. Number.objects.bulk_create(Number(num=num) for num in numbers)
  2186. self.assertEqual(
  2187. Number.objects.filter(num__in=numbers[:1000]).count(),
  2188. 1000
  2189. )
  2190. self.assertEqual(
  2191. Number.objects.filter(num__in=numbers[:1001]).count(),
  2192. 1001
  2193. )
  2194. self.assertEqual(
  2195. Number.objects.filter(num__in=numbers[:2000]).count(),
  2196. 2000
  2197. )
  2198. self.assertEqual(
  2199. Number.objects.filter(num__in=numbers).count(),
  2200. len(numbers)
  2201. )
  2202. class UnionTests(unittest.TestCase):
  2203. """
  2204. Tests for the union of two querysets. Bug #12252.
  2205. """
  2206. @classmethod
  2207. def setUpTestData(cls):
  2208. objectas = []
  2209. objectbs = []
  2210. objectcs = []
  2211. a_info = ['one', 'two', 'three']
  2212. for name in a_info:
  2213. o = ObjectA(name=name)
  2214. o.save()
  2215. objectas.append(o)
  2216. b_info = [('un', 1, objectas[0]), ('deux', 2, objectas[0]), ('trois', 3, objectas[2])]
  2217. for name, number, objecta in b_info:
  2218. o = ObjectB(name=name, num=number, objecta=objecta)
  2219. o.save()
  2220. objectbs.append(o)
  2221. c_info = [('ein', objectas[2], objectbs[2]), ('zwei', objectas[1], objectbs[1])]
  2222. for name, objecta, objectb in c_info:
  2223. o = ObjectC(name=name, objecta=objecta, objectb=objectb)
  2224. o.save()
  2225. objectcs.append(o)
  2226. def check_union(self, model, Q1, Q2):
  2227. filter = model.objects.filter
  2228. self.assertEqual(set(filter(Q1) | filter(Q2)), set(filter(Q1 | Q2)))
  2229. self.assertEqual(set(filter(Q2) | filter(Q1)), set(filter(Q1 | Q2)))
  2230. def test_A_AB(self):
  2231. Q1 = Q(name='two')
  2232. Q2 = Q(objectb__name='deux')
  2233. self.check_union(ObjectA, Q1, Q2)
  2234. def test_A_AB2(self):
  2235. Q1 = Q(name='two')
  2236. Q2 = Q(objectb__name='deux', objectb__num=2)
  2237. self.check_union(ObjectA, Q1, Q2)
  2238. def test_AB_ACB(self):
  2239. Q1 = Q(objectb__name='deux')
  2240. Q2 = Q(objectc__objectb__name='deux')
  2241. self.check_union(ObjectA, Q1, Q2)
  2242. def test_BAB_BAC(self):
  2243. Q1 = Q(objecta__objectb__name='deux')
  2244. Q2 = Q(objecta__objectc__name='ein')
  2245. self.check_union(ObjectB, Q1, Q2)
  2246. def test_BAB_BACB(self):
  2247. Q1 = Q(objecta__objectb__name='deux')
  2248. Q2 = Q(objecta__objectc__objectb__name='trois')
  2249. self.check_union(ObjectB, Q1, Q2)
  2250. def test_BA_BCA__BAB_BAC_BCA(self):
  2251. Q1 = Q(objecta__name='one', objectc__objecta__name='two')
  2252. Q2 = Q(objecta__objectc__name='ein', objectc__objecta__name='three', objecta__objectb__name='trois')
  2253. self.check_union(ObjectB, Q1, Q2)
  2254. class DefaultValuesInsertTest(TestCase):
  2255. def test_no_extra_params(self):
  2256. # Ticket #17056 -- affects Oracle
  2257. try:
  2258. DumbCategory.objects.create()
  2259. except TypeError:
  2260. self.fail(
  2261. "Creation of an instance of a model with only the PK field "
  2262. "shouldn't error out after bulk insert refactoring (#17056)"
  2263. )
  2264. class ExcludeTests(TestCase):
  2265. @classmethod
  2266. def setUpTestData(cls):
  2267. f1 = Food.objects.create(name='apples')
  2268. Food.objects.create(name='oranges')
  2269. Eaten.objects.create(food=f1, meal='dinner')
  2270. j1 = Job.objects.create(name='Manager')
  2271. r1 = Responsibility.objects.create(description='Playing golf')
  2272. j2 = Job.objects.create(name='Programmer')
  2273. r2 = Responsibility.objects.create(description='Programming')
  2274. JobResponsibilities.objects.create(job=j1, responsibility=r1)
  2275. JobResponsibilities.objects.create(job=j2, responsibility=r2)
  2276. def test_to_field(self):
  2277. self.assertQuerysetEqual(
  2278. Food.objects.exclude(eaten__meal='dinner'),
  2279. ['<Food: oranges>'])
  2280. self.assertQuerysetEqual(
  2281. Job.objects.exclude(responsibilities__description='Playing golf'),
  2282. ['<Job: Programmer>'])
  2283. self.assertQuerysetEqual(
  2284. Responsibility.objects.exclude(jobs__name='Manager'),
  2285. ['<Responsibility: Programming>'])
  2286. def test_ticket14511(self):
  2287. alex = Person.objects.get_or_create(name='Alex')[0]
  2288. jane = Person.objects.get_or_create(name='Jane')[0]
  2289. oracle = Company.objects.get_or_create(name='Oracle')[0]
  2290. google = Company.objects.get_or_create(name='Google')[0]
  2291. microsoft = Company.objects.get_or_create(name='Microsoft')[0]
  2292. intel = Company.objects.get_or_create(name='Intel')[0]
  2293. def employ(employer, employee, title):
  2294. Employment.objects.get_or_create(employee=employee, employer=employer, title=title)
  2295. employ(oracle, alex, 'Engineer')
  2296. employ(oracle, alex, 'Developer')
  2297. employ(google, alex, 'Engineer')
  2298. employ(google, alex, 'Manager')
  2299. employ(microsoft, alex, 'Manager')
  2300. employ(intel, alex, 'Manager')
  2301. employ(microsoft, jane, 'Developer')
  2302. employ(intel, jane, 'Manager')
  2303. alex_tech_employers = alex.employers.filter(
  2304. employment__title__in=('Engineer', 'Developer')).distinct().order_by('name')
  2305. self.assertQuerysetEqual(alex_tech_employers, [google, oracle], lambda x: x)
  2306. alex_nontech_employers = alex.employers.exclude(
  2307. employment__title__in=('Engineer', 'Developer')).distinct().order_by('name')
  2308. self.assertQuerysetEqual(alex_nontech_employers, [google, intel, microsoft], lambda x: x)
  2309. class ExcludeTest17600(TestCase):
  2310. """
  2311. Some regressiontests for ticket #17600. Some of these likely duplicate
  2312. other existing tests.
  2313. """
  2314. @classmethod
  2315. def setUpTestData(cls):
  2316. # Create a few Orders.
  2317. cls.o1 = Order.objects.create(pk=1)
  2318. cls.o2 = Order.objects.create(pk=2)
  2319. cls.o3 = Order.objects.create(pk=3)
  2320. # Create some OrderItems for the first order with homogeneous
  2321. # status_id values
  2322. cls.oi1 = OrderItem.objects.create(order=cls.o1, status=1)
  2323. cls.oi2 = OrderItem.objects.create(order=cls.o1, status=1)
  2324. cls.oi3 = OrderItem.objects.create(order=cls.o1, status=1)
  2325. # Create some OrderItems for the second order with heterogeneous
  2326. # status_id values
  2327. cls.oi4 = OrderItem.objects.create(order=cls.o2, status=1)
  2328. cls.oi5 = OrderItem.objects.create(order=cls.o2, status=2)
  2329. cls.oi6 = OrderItem.objects.create(order=cls.o2, status=3)
  2330. # Create some OrderItems for the second order with heterogeneous
  2331. # status_id values
  2332. cls.oi7 = OrderItem.objects.create(order=cls.o3, status=2)
  2333. cls.oi8 = OrderItem.objects.create(order=cls.o3, status=3)
  2334. cls.oi9 = OrderItem.objects.create(order=cls.o3, status=4)
  2335. def test_exclude_plain(self):
  2336. """
  2337. This should exclude Orders which have some items with status 1
  2338. """
  2339. self.assertQuerysetEqual(
  2340. Order.objects.exclude(items__status=1),
  2341. ['<Order: 3>'])
  2342. def test_exclude_plain_distinct(self):
  2343. """
  2344. This should exclude Orders which have some items with status 1
  2345. """
  2346. self.assertQuerysetEqual(
  2347. Order.objects.exclude(items__status=1).distinct(),
  2348. ['<Order: 3>'])
  2349. def test_exclude_with_q_object_distinct(self):
  2350. """
  2351. This should exclude Orders which have some items with status 1
  2352. """
  2353. self.assertQuerysetEqual(
  2354. Order.objects.exclude(Q(items__status=1)).distinct(),
  2355. ['<Order: 3>'])
  2356. def test_exclude_with_q_object_no_distinct(self):
  2357. """
  2358. This should exclude Orders which have some items with status 1
  2359. """
  2360. self.assertQuerysetEqual(
  2361. Order.objects.exclude(Q(items__status=1)),
  2362. ['<Order: 3>'])
  2363. def test_exclude_with_q_is_equal_to_plain_exclude(self):
  2364. """
  2365. Using exclude(condition) and exclude(Q(condition)) should
  2366. yield the same QuerySet
  2367. """
  2368. self.assertEqual(
  2369. list(Order.objects.exclude(items__status=1).distinct()),
  2370. list(Order.objects.exclude(Q(items__status=1)).distinct()))
  2371. def test_exclude_with_q_is_equal_to_plain_exclude_variation(self):
  2372. """
  2373. Using exclude(condition) and exclude(Q(condition)) should
  2374. yield the same QuerySet
  2375. """
  2376. self.assertEqual(
  2377. list(Order.objects.exclude(items__status=1)),
  2378. list(Order.objects.exclude(Q(items__status=1)).distinct()))
  2379. @unittest.expectedFailure
  2380. def test_only_orders_with_all_items_having_status_1(self):
  2381. """
  2382. This should only return orders having ALL items set to status 1, or
  2383. those items not having any orders at all. The correct way to write
  2384. this query in SQL seems to be using two nested subqueries.
  2385. """
  2386. self.assertQuerysetEqual(
  2387. Order.objects.exclude(~Q(items__status=1)).distinct(),
  2388. ['<Order: 1>'])
  2389. class Exclude15786(TestCase):
  2390. """Regression test for #15786"""
  2391. def test_ticket15786(self):
  2392. c1 = SimpleCategory.objects.create(name='c1')
  2393. c2 = SimpleCategory.objects.create(name='c2')
  2394. OneToOneCategory.objects.create(category=c1)
  2395. OneToOneCategory.objects.create(category=c2)
  2396. rel = CategoryRelationship.objects.create(first=c1, second=c2)
  2397. self.assertEqual(
  2398. CategoryRelationship.objects.exclude(
  2399. first__onetoonecategory=F('second__onetoonecategory')
  2400. ).get(), rel
  2401. )
  2402. class NullInExcludeTest(TestCase):
  2403. @classmethod
  2404. def setUpTestData(cls):
  2405. NullableName.objects.create(name='i1')
  2406. NullableName.objects.create()
  2407. def test_null_in_exclude_qs(self):
  2408. none_val = '' if connection.features.interprets_empty_strings_as_nulls else None
  2409. self.assertQuerysetEqual(
  2410. NullableName.objects.exclude(name__in=[]),
  2411. ['i1', none_val], attrgetter('name'))
  2412. self.assertQuerysetEqual(
  2413. NullableName.objects.exclude(name__in=['i1']),
  2414. [none_val], attrgetter('name'))
  2415. self.assertQuerysetEqual(
  2416. NullableName.objects.exclude(name__in=['i3']),
  2417. ['i1', none_val], attrgetter('name'))
  2418. inner_qs = NullableName.objects.filter(name='i1').values_list('name')
  2419. self.assertQuerysetEqual(
  2420. NullableName.objects.exclude(name__in=inner_qs),
  2421. [none_val], attrgetter('name'))
  2422. # Check that the inner queryset wasn't executed - it should be turned
  2423. # into subquery above
  2424. self.assertIs(inner_qs._result_cache, None)
  2425. @unittest.expectedFailure
  2426. def test_col_not_in_list_containing_null(self):
  2427. """
  2428. The following case is not handled properly because
  2429. SQL's COL NOT IN (list containing null) handling is too weird to
  2430. abstract away.
  2431. """
  2432. self.assertQuerysetEqual(
  2433. NullableName.objects.exclude(name__in=[None]),
  2434. ['i1'], attrgetter('name'))
  2435. def test_double_exclude(self):
  2436. self.assertEqual(
  2437. list(NullableName.objects.filter(~~Q(name='i1'))),
  2438. list(NullableName.objects.filter(Q(name='i1'))))
  2439. self.assertNotIn(
  2440. 'IS NOT NULL',
  2441. str(NullableName.objects.filter(~~Q(name='i1')).query))
  2442. class EmptyStringsAsNullTest(TestCase):
  2443. """
  2444. Test that filtering on non-null character fields works as expected.
  2445. The reason for these tests is that Oracle treats '' as NULL, and this
  2446. can cause problems in query construction. Refs #17957.
  2447. """
  2448. @classmethod
  2449. def setUpTestData(cls):
  2450. cls.nc = NamedCategory.objects.create(name='')
  2451. def test_direct_exclude(self):
  2452. self.assertQuerysetEqual(
  2453. NamedCategory.objects.exclude(name__in=['nonexisting']),
  2454. [self.nc.pk], attrgetter('pk')
  2455. )
  2456. def test_joined_exclude(self):
  2457. self.assertQuerysetEqual(
  2458. DumbCategory.objects.exclude(namedcategory__name__in=['nonexisting']),
  2459. [self.nc.pk], attrgetter('pk')
  2460. )
  2461. def test_21001(self):
  2462. foo = NamedCategory.objects.create(name='foo')
  2463. self.assertQuerysetEqual(
  2464. NamedCategory.objects.exclude(name=''),
  2465. [foo.pk], attrgetter('pk')
  2466. )
  2467. class ProxyQueryCleanupTest(TestCase):
  2468. def test_evaluated_proxy_count(self):
  2469. """
  2470. Test that generating the query string doesn't alter the query's state
  2471. in irreversible ways. Refs #18248.
  2472. """
  2473. ProxyCategory.objects.create()
  2474. qs = ProxyCategory.objects.all()
  2475. self.assertEqual(qs.count(), 1)
  2476. str(qs.query)
  2477. self.assertEqual(qs.count(), 1)
  2478. class WhereNodeTest(TestCase):
  2479. class DummyNode(object):
  2480. def as_sql(self, compiler, connection):
  2481. return 'dummy', []
  2482. class MockCompiler(object):
  2483. def compile(self, node):
  2484. return node.as_sql(self, connection)
  2485. def __call__(self, name):
  2486. return connection.ops.quote_name(name)
  2487. def test_empty_full_handling_conjunction(self):
  2488. compiler = WhereNodeTest.MockCompiler()
  2489. w = WhereNode(children=[NothingNode()])
  2490. with self.assertRaises(EmptyResultSet):
  2491. w.as_sql(compiler, connection)
  2492. w.negate()
  2493. self.assertEqual(w.as_sql(compiler, connection), ('', []))
  2494. w = WhereNode(children=[self.DummyNode(), self.DummyNode()])
  2495. self.assertEqual(w.as_sql(compiler, connection), ('(dummy AND dummy)', []))
  2496. w.negate()
  2497. self.assertEqual(w.as_sql(compiler, connection), ('NOT (dummy AND dummy)', []))
  2498. w = WhereNode(children=[NothingNode(), self.DummyNode()])
  2499. with self.assertRaises(EmptyResultSet):
  2500. w.as_sql(compiler, connection)
  2501. w.negate()
  2502. self.assertEqual(w.as_sql(compiler, connection), ('', []))
  2503. def test_empty_full_handling_disjunction(self):
  2504. compiler = WhereNodeTest.MockCompiler()
  2505. w = WhereNode(children=[NothingNode()], connector='OR')
  2506. with self.assertRaises(EmptyResultSet):
  2507. w.as_sql(compiler, connection)
  2508. w.negate()
  2509. self.assertEqual(w.as_sql(compiler, connection), ('', []))
  2510. w = WhereNode(children=[self.DummyNode(), self.DummyNode()], connector='OR')
  2511. self.assertEqual(w.as_sql(compiler, connection), ('(dummy OR dummy)', []))
  2512. w.negate()
  2513. self.assertEqual(w.as_sql(compiler, connection), ('NOT (dummy OR dummy)', []))
  2514. w = WhereNode(children=[NothingNode(), self.DummyNode()], connector='OR')
  2515. self.assertEqual(w.as_sql(compiler, connection), ('dummy', []))
  2516. w.negate()
  2517. self.assertEqual(w.as_sql(compiler, connection), ('NOT (dummy)', []))
  2518. def test_empty_nodes(self):
  2519. compiler = WhereNodeTest.MockCompiler()
  2520. empty_w = WhereNode()
  2521. w = WhereNode(children=[empty_w, empty_w])
  2522. self.assertEqual(w.as_sql(compiler, connection), ('', []))
  2523. w.negate()
  2524. with self.assertRaises(EmptyResultSet):
  2525. w.as_sql(compiler, connection)
  2526. w.connector = 'OR'
  2527. with self.assertRaises(EmptyResultSet):
  2528. w.as_sql(compiler, connection)
  2529. w.negate()
  2530. self.assertEqual(w.as_sql(compiler, connection), ('', []))
  2531. w = WhereNode(children=[empty_w, NothingNode()], connector='OR')
  2532. self.assertEqual(w.as_sql(compiler, connection), ('', []))
  2533. w = WhereNode(children=[empty_w, NothingNode()], connector='AND')
  2534. with self.assertRaises(EmptyResultSet):
  2535. w.as_sql(compiler, connection)
  2536. class IteratorExceptionsTest(TestCase):
  2537. def test_iter_exceptions(self):
  2538. qs = ExtraInfo.objects.only('author')
  2539. with self.assertRaises(AttributeError):
  2540. list(qs)
  2541. def test_invalid_qs_list(self):
  2542. # Test for #19895 - second iteration over invalid queryset
  2543. # raises errors.
  2544. qs = Article.objects.order_by('invalid_column')
  2545. with self.assertRaises(FieldError):
  2546. list(qs)
  2547. with self.assertRaises(FieldError):
  2548. list(qs)
  2549. class NullJoinPromotionOrTest(TestCase):
  2550. @classmethod
  2551. def setUpTestData(cls):
  2552. cls.d1 = ModelD.objects.create(name='foo')
  2553. d2 = ModelD.objects.create(name='bar')
  2554. cls.a1 = ModelA.objects.create(name='a1', d=cls.d1)
  2555. c = ModelC.objects.create(name='c')
  2556. b = ModelB.objects.create(name='b', c=c)
  2557. cls.a2 = ModelA.objects.create(name='a2', b=b, d=d2)
  2558. def test_ticket_17886(self):
  2559. # The first Q-object is generating the match, the rest of the filters
  2560. # should not remove the match even if they do not match anything. The
  2561. # problem here was that b__name generates a LOUTER JOIN, then
  2562. # b__c__name generates join to c, which the ORM tried to promote but
  2563. # failed as that join isn't nullable.
  2564. q_obj = (
  2565. Q(d__name='foo') |
  2566. Q(b__name='foo') |
  2567. Q(b__c__name='foo')
  2568. )
  2569. qset = ModelA.objects.filter(q_obj)
  2570. self.assertEqual(list(qset), [self.a1])
  2571. # We generate one INNER JOIN to D. The join is direct and not nullable
  2572. # so we can use INNER JOIN for it. However, we can NOT use INNER JOIN
  2573. # for the b->c join, as a->b is nullable.
  2574. self.assertEqual(str(qset.query).count('INNER JOIN'), 1)
  2575. def test_isnull_filter_promotion(self):
  2576. qs = ModelA.objects.filter(Q(b__name__isnull=True))
  2577. self.assertEqual(str(qs.query).count('LEFT OUTER'), 1)
  2578. self.assertEqual(list(qs), [self.a1])
  2579. qs = ModelA.objects.filter(~Q(b__name__isnull=True))
  2580. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2581. self.assertEqual(list(qs), [self.a2])
  2582. qs = ModelA.objects.filter(~~Q(b__name__isnull=True))
  2583. self.assertEqual(str(qs.query).count('LEFT OUTER'), 1)
  2584. self.assertEqual(list(qs), [self.a1])
  2585. qs = ModelA.objects.filter(Q(b__name__isnull=False))
  2586. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2587. self.assertEqual(list(qs), [self.a2])
  2588. qs = ModelA.objects.filter(~Q(b__name__isnull=False))
  2589. self.assertEqual(str(qs.query).count('LEFT OUTER'), 1)
  2590. self.assertEqual(list(qs), [self.a1])
  2591. qs = ModelA.objects.filter(~~Q(b__name__isnull=False))
  2592. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2593. self.assertEqual(list(qs), [self.a2])
  2594. def test_null_join_demotion(self):
  2595. qs = ModelA.objects.filter(Q(b__name__isnull=False) & Q(b__name__isnull=True))
  2596. self.assertIn(' INNER JOIN ', str(qs.query))
  2597. qs = ModelA.objects.filter(Q(b__name__isnull=True) & Q(b__name__isnull=False))
  2598. self.assertIn(' INNER JOIN ', str(qs.query))
  2599. qs = ModelA.objects.filter(Q(b__name__isnull=False) | Q(b__name__isnull=True))
  2600. self.assertIn(' LEFT OUTER JOIN ', str(qs.query))
  2601. qs = ModelA.objects.filter(Q(b__name__isnull=True) | Q(b__name__isnull=False))
  2602. self.assertIn(' LEFT OUTER JOIN ', str(qs.query))
  2603. def test_ticket_21366(self):
  2604. n = Note.objects.create(note='n', misc='m')
  2605. e = ExtraInfo.objects.create(info='info', note=n)
  2606. a = Author.objects.create(name='Author1', num=1, extra=e)
  2607. Ranking.objects.create(rank=1, author=a)
  2608. r1 = Report.objects.create(name='Foo', creator=a)
  2609. r2 = Report.objects.create(name='Bar')
  2610. Report.objects.create(name='Bar', creator=a)
  2611. qs = Report.objects.filter(
  2612. Q(creator__ranking__isnull=True) |
  2613. Q(creator__ranking__rank=1, name='Foo')
  2614. )
  2615. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 2)
  2616. self.assertEqual(str(qs.query).count(' JOIN '), 2)
  2617. self.assertQuerysetEqual(
  2618. qs.order_by('name'), [r2, r1], lambda x: x)
  2619. def test_ticket_21748(self):
  2620. i1 = Identifier.objects.create(name='i1')
  2621. i2 = Identifier.objects.create(name='i2')
  2622. i3 = Identifier.objects.create(name='i3')
  2623. Program.objects.create(identifier=i1)
  2624. Channel.objects.create(identifier=i1)
  2625. Program.objects.create(identifier=i2)
  2626. self.assertQuerysetEqual(
  2627. Identifier.objects.filter(program=None, channel=None),
  2628. [i3], lambda x: x)
  2629. self.assertQuerysetEqual(
  2630. Identifier.objects.exclude(program=None, channel=None).order_by('name'),
  2631. [i1, i2], lambda x: x)
  2632. def test_ticket_21748_double_negated_and(self):
  2633. i1 = Identifier.objects.create(name='i1')
  2634. i2 = Identifier.objects.create(name='i2')
  2635. Identifier.objects.create(name='i3')
  2636. p1 = Program.objects.create(identifier=i1)
  2637. c1 = Channel.objects.create(identifier=i1)
  2638. Program.objects.create(identifier=i2)
  2639. # Check the ~~Q() (or equivalently .exclude(~Q)) works like Q() for
  2640. # join promotion.
  2641. qs1_doubleneg = Identifier.objects.exclude(~Q(program__id=p1.id, channel__id=c1.id)).order_by('pk')
  2642. qs1_filter = Identifier.objects.filter(program__id=p1.id, channel__id=c1.id).order_by('pk')
  2643. self.assertQuerysetEqual(qs1_doubleneg, qs1_filter, lambda x: x)
  2644. self.assertEqual(str(qs1_filter.query).count('JOIN'),
  2645. str(qs1_doubleneg.query).count('JOIN'))
  2646. self.assertEqual(2, str(qs1_doubleneg.query).count('INNER JOIN'))
  2647. self.assertEqual(str(qs1_filter.query).count('INNER JOIN'),
  2648. str(qs1_doubleneg.query).count('INNER JOIN'))
  2649. def test_ticket_21748_double_negated_or(self):
  2650. i1 = Identifier.objects.create(name='i1')
  2651. i2 = Identifier.objects.create(name='i2')
  2652. Identifier.objects.create(name='i3')
  2653. p1 = Program.objects.create(identifier=i1)
  2654. c1 = Channel.objects.create(identifier=i1)
  2655. p2 = Program.objects.create(identifier=i2)
  2656. # Test OR + doubleneg. The expected result is that channel is LOUTER
  2657. # joined, program INNER joined
  2658. qs1_filter = Identifier.objects.filter(
  2659. Q(program__id=p2.id, channel__id=c1.id) | Q(program__id=p1.id)
  2660. ).order_by('pk')
  2661. qs1_doubleneg = Identifier.objects.exclude(
  2662. ~Q(Q(program__id=p2.id, channel__id=c1.id) | Q(program__id=p1.id))
  2663. ).order_by('pk')
  2664. self.assertQuerysetEqual(qs1_doubleneg, qs1_filter, lambda x: x)
  2665. self.assertEqual(str(qs1_filter.query).count('JOIN'),
  2666. str(qs1_doubleneg.query).count('JOIN'))
  2667. self.assertEqual(1, str(qs1_doubleneg.query).count('INNER JOIN'))
  2668. self.assertEqual(str(qs1_filter.query).count('INNER JOIN'),
  2669. str(qs1_doubleneg.query).count('INNER JOIN'))
  2670. def test_ticket_21748_complex_filter(self):
  2671. i1 = Identifier.objects.create(name='i1')
  2672. i2 = Identifier.objects.create(name='i2')
  2673. Identifier.objects.create(name='i3')
  2674. p1 = Program.objects.create(identifier=i1)
  2675. c1 = Channel.objects.create(identifier=i1)
  2676. p2 = Program.objects.create(identifier=i2)
  2677. # Finally, a more complex case, one time in a way where each
  2678. # NOT is pushed to lowest level in the boolean tree, and
  2679. # another query where this isn't done.
  2680. qs1 = Identifier.objects.filter(
  2681. ~Q(~Q(program__id=p2.id, channel__id=c1.id) & Q(program__id=p1.id))
  2682. ).order_by('pk')
  2683. qs2 = Identifier.objects.filter(
  2684. Q(Q(program__id=p2.id, channel__id=c1.id) | ~Q(program__id=p1.id))
  2685. ).order_by('pk')
  2686. self.assertQuerysetEqual(qs1, qs2, lambda x: x)
  2687. self.assertEqual(str(qs1.query).count('JOIN'),
  2688. str(qs2.query).count('JOIN'))
  2689. self.assertEqual(0, str(qs1.query).count('INNER JOIN'))
  2690. self.assertEqual(str(qs1.query).count('INNER JOIN'),
  2691. str(qs2.query).count('INNER JOIN'))
  2692. class ReverseJoinTrimmingTest(TestCase):
  2693. def test_reverse_trimming(self):
  2694. # Check that we don't accidentally trim reverse joins - we can't know
  2695. # if there is anything on the other side of the join, so trimming
  2696. # reverse joins can't be done, ever.
  2697. t = Tag.objects.create()
  2698. qs = Tag.objects.filter(annotation__tag=t.pk)
  2699. self.assertIn('INNER JOIN', str(qs.query))
  2700. self.assertEqual(list(qs), [])
  2701. class JoinReuseTest(TestCase):
  2702. """
  2703. Test that the queries reuse joins sensibly (for example, direct joins
  2704. are always reused).
  2705. """
  2706. def test_fk_reuse(self):
  2707. qs = Annotation.objects.filter(tag__name='foo').filter(tag__name='bar')
  2708. self.assertEqual(str(qs.query).count('JOIN'), 1)
  2709. def test_fk_reuse_select_related(self):
  2710. qs = Annotation.objects.filter(tag__name='foo').select_related('tag')
  2711. self.assertEqual(str(qs.query).count('JOIN'), 1)
  2712. def test_fk_reuse_annotation(self):
  2713. qs = Annotation.objects.filter(tag__name='foo').annotate(cnt=Count('tag__name'))
  2714. self.assertEqual(str(qs.query).count('JOIN'), 1)
  2715. def test_fk_reuse_disjunction(self):
  2716. qs = Annotation.objects.filter(Q(tag__name='foo') | Q(tag__name='bar'))
  2717. self.assertEqual(str(qs.query).count('JOIN'), 1)
  2718. def test_fk_reuse_order_by(self):
  2719. qs = Annotation.objects.filter(tag__name='foo').order_by('tag__name')
  2720. self.assertEqual(str(qs.query).count('JOIN'), 1)
  2721. def test_revo2o_reuse(self):
  2722. qs = Detail.objects.filter(member__name='foo').filter(member__name='foo')
  2723. self.assertEqual(str(qs.query).count('JOIN'), 1)
  2724. def test_revfk_noreuse(self):
  2725. qs = Author.objects.filter(report__name='r4').filter(report__name='r1')
  2726. self.assertEqual(str(qs.query).count('JOIN'), 2)
  2727. class DisjunctionPromotionTests(TestCase):
  2728. def test_disjunction_promotion_select_related(self):
  2729. fk1 = FK1.objects.create(f1='f1', f2='f2')
  2730. basea = BaseA.objects.create(a=fk1)
  2731. qs = BaseA.objects.filter(Q(a=fk1) | Q(b=2))
  2732. self.assertEqual(str(qs.query).count(' JOIN '), 0)
  2733. qs = qs.select_related('a', 'b')
  2734. self.assertEqual(str(qs.query).count(' INNER JOIN '), 0)
  2735. self.assertEqual(str(qs.query).count(' LEFT OUTER JOIN '), 2)
  2736. with self.assertNumQueries(1):
  2737. self.assertQuerysetEqual(qs, [basea], lambda x: x)
  2738. self.assertEqual(qs[0].a, fk1)
  2739. self.assertIs(qs[0].b, None)
  2740. def test_disjunction_promotion1(self):
  2741. # Pre-existing join, add two ORed filters to the same join,
  2742. # all joins can be INNER JOINS.
  2743. qs = BaseA.objects.filter(a__f1='foo')
  2744. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2745. qs = qs.filter(Q(b__f1='foo') | Q(b__f2='foo'))
  2746. self.assertEqual(str(qs.query).count('INNER JOIN'), 2)
  2747. # Reverse the order of AND and OR filters.
  2748. qs = BaseA.objects.filter(Q(b__f1='foo') | Q(b__f2='foo'))
  2749. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2750. qs = qs.filter(a__f1='foo')
  2751. self.assertEqual(str(qs.query).count('INNER JOIN'), 2)
  2752. def test_disjunction_promotion2(self):
  2753. qs = BaseA.objects.filter(a__f1='foo')
  2754. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2755. # Now we have two different joins in an ORed condition, these
  2756. # must be OUTER joins. The pre-existing join should remain INNER.
  2757. qs = qs.filter(Q(b__f1='foo') | Q(c__f2='foo'))
  2758. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2759. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 2)
  2760. # Reverse case.
  2761. qs = BaseA.objects.filter(Q(b__f1='foo') | Q(c__f2='foo'))
  2762. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 2)
  2763. qs = qs.filter(a__f1='foo')
  2764. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2765. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 2)
  2766. def test_disjunction_promotion3(self):
  2767. qs = BaseA.objects.filter(a__f2='bar')
  2768. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2769. # The ANDed a__f2 filter allows us to use keep using INNER JOIN
  2770. # even inside the ORed case. If the join to a__ returns nothing,
  2771. # the ANDed filter for a__f2 can't be true.
  2772. qs = qs.filter(Q(a__f1='foo') | Q(b__f2='foo'))
  2773. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2774. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 1)
  2775. def test_disjunction_promotion3_demote(self):
  2776. # This one needs demotion logic: the first filter causes a to be
  2777. # outer joined, the second filter makes it inner join again.
  2778. qs = BaseA.objects.filter(
  2779. Q(a__f1='foo') | Q(b__f2='foo')).filter(a__f2='bar')
  2780. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2781. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 1)
  2782. def test_disjunction_promotion4_demote(self):
  2783. qs = BaseA.objects.filter(Q(a=1) | Q(a=2))
  2784. self.assertEqual(str(qs.query).count('JOIN'), 0)
  2785. # Demote needed for the "a" join. It is marked as outer join by
  2786. # above filter (even if it is trimmed away).
  2787. qs = qs.filter(a__f1='foo')
  2788. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2789. def test_disjunction_promotion4(self):
  2790. qs = BaseA.objects.filter(a__f1='foo')
  2791. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2792. qs = qs.filter(Q(a=1) | Q(a=2))
  2793. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2794. def test_disjunction_promotion5_demote(self):
  2795. qs = BaseA.objects.filter(Q(a=1) | Q(a=2))
  2796. # Note that the above filters on a force the join to an
  2797. # inner join even if it is trimmed.
  2798. self.assertEqual(str(qs.query).count('JOIN'), 0)
  2799. qs = qs.filter(Q(a__f1='foo') | Q(b__f1='foo'))
  2800. # So, now the a__f1 join doesn't need promotion.
  2801. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2802. # But b__f1 does.
  2803. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 1)
  2804. qs = BaseA.objects.filter(Q(a__f1='foo') | Q(b__f1='foo'))
  2805. # Now the join to a is created as LOUTER
  2806. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 2)
  2807. qs = qs.filter(Q(a=1) | Q(a=2))
  2808. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2809. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 1)
  2810. def test_disjunction_promotion6(self):
  2811. qs = BaseA.objects.filter(Q(a=1) | Q(a=2))
  2812. self.assertEqual(str(qs.query).count('JOIN'), 0)
  2813. qs = BaseA.objects.filter(Q(a__f1='foo') & Q(b__f1='foo'))
  2814. self.assertEqual(str(qs.query).count('INNER JOIN'), 2)
  2815. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 0)
  2816. qs = BaseA.objects.filter(Q(a__f1='foo') & Q(b__f1='foo'))
  2817. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 0)
  2818. self.assertEqual(str(qs.query).count('INNER JOIN'), 2)
  2819. qs = qs.filter(Q(a=1) | Q(a=2))
  2820. self.assertEqual(str(qs.query).count('INNER JOIN'), 2)
  2821. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 0)
  2822. def test_disjunction_promotion7(self):
  2823. qs = BaseA.objects.filter(Q(a=1) | Q(a=2))
  2824. self.assertEqual(str(qs.query).count('JOIN'), 0)
  2825. qs = BaseA.objects.filter(Q(a__f1='foo') | (Q(b__f1='foo') & Q(a__f1='bar')))
  2826. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2827. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 1)
  2828. qs = BaseA.objects.filter(
  2829. (Q(a__f1='foo') | Q(b__f1='foo')) & (Q(a__f1='bar') | Q(c__f1='foo'))
  2830. )
  2831. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 3)
  2832. self.assertEqual(str(qs.query).count('INNER JOIN'), 0)
  2833. qs = BaseA.objects.filter(
  2834. (Q(a__f1='foo') | (Q(a__f1='bar')) & (Q(b__f1='bar') | Q(c__f1='foo')))
  2835. )
  2836. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 2)
  2837. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2838. def test_disjunction_promotion_fexpression(self):
  2839. qs = BaseA.objects.filter(Q(a__f1=F('b__f1')) | Q(b__f1='foo'))
  2840. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 1)
  2841. self.assertEqual(str(qs.query).count('INNER JOIN'), 1)
  2842. qs = BaseA.objects.filter(Q(a__f1=F('c__f1')) | Q(b__f1='foo'))
  2843. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 3)
  2844. qs = BaseA.objects.filter(Q(a__f1=F('b__f1')) | Q(a__f2=F('b__f2')) | Q(c__f1='foo'))
  2845. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 3)
  2846. qs = BaseA.objects.filter(Q(a__f1=F('c__f1')) | (Q(pk=1) & Q(pk=2)))
  2847. self.assertEqual(str(qs.query).count('LEFT OUTER JOIN'), 2)
  2848. self.assertEqual(str(qs.query).count('INNER JOIN'), 0)
  2849. class ManyToManyExcludeTest(TestCase):
  2850. def test_exclude_many_to_many(self):
  2851. Identifier.objects.create(name='extra')
  2852. program = Program.objects.create(identifier=Identifier.objects.create(name='program'))
  2853. channel = Channel.objects.create(identifier=Identifier.objects.create(name='channel'))
  2854. channel.programs.add(program)
  2855. # channel contains 'program1', so all Identifiers except that one
  2856. # should be returned
  2857. self.assertQuerysetEqual(
  2858. Identifier.objects.exclude(program__channel=channel).order_by('name'),
  2859. ['<Identifier: channel>', '<Identifier: extra>']
  2860. )
  2861. self.assertQuerysetEqual(
  2862. Identifier.objects.exclude(program__channel=None).order_by('name'),
  2863. ['<Identifier: program>']
  2864. )
  2865. def test_ticket_12823(self):
  2866. pg3 = Page.objects.create(text='pg3')
  2867. pg2 = Page.objects.create(text='pg2')
  2868. pg1 = Page.objects.create(text='pg1')
  2869. pa1 = Paragraph.objects.create(text='pa1')
  2870. pa1.page.set([pg1, pg2])
  2871. pa2 = Paragraph.objects.create(text='pa2')
  2872. pa2.page.set([pg2, pg3])
  2873. pa3 = Paragraph.objects.create(text='pa3')
  2874. ch1 = Chapter.objects.create(title='ch1', paragraph=pa1)
  2875. ch2 = Chapter.objects.create(title='ch2', paragraph=pa2)
  2876. ch3 = Chapter.objects.create(title='ch3', paragraph=pa3)
  2877. b1 = Book.objects.create(title='b1', chapter=ch1)
  2878. b2 = Book.objects.create(title='b2', chapter=ch2)
  2879. b3 = Book.objects.create(title='b3', chapter=ch3)
  2880. q = Book.objects.exclude(chapter__paragraph__page__text='pg1')
  2881. self.assertNotIn('IS NOT NULL', str(q.query))
  2882. self.assertEqual(len(q), 2)
  2883. self.assertNotIn(b1, q)
  2884. self.assertIn(b2, q)
  2885. self.assertIn(b3, q)
  2886. class RelabelCloneTest(TestCase):
  2887. def test_ticket_19964(self):
  2888. my1 = MyObject.objects.create(data='foo')
  2889. my1.parent = my1
  2890. my1.save()
  2891. my2 = MyObject.objects.create(data='bar', parent=my1)
  2892. parents = MyObject.objects.filter(parent=F('id'))
  2893. children = MyObject.objects.filter(parent__in=parents).exclude(parent=F('id'))
  2894. self.assertEqual(list(parents), [my1])
  2895. # Evaluating the children query (which has parents as part of it) does
  2896. # not change results for the parents query.
  2897. self.assertEqual(list(children), [my2])
  2898. self.assertEqual(list(parents), [my1])
  2899. class Ticket20101Tests(TestCase):
  2900. def test_ticket_20101(self):
  2901. """
  2902. Tests QuerySet ORed combining in exclude subquery case.
  2903. """
  2904. t = Tag.objects.create(name='foo')
  2905. a1 = Annotation.objects.create(tag=t, name='a1')
  2906. a2 = Annotation.objects.create(tag=t, name='a2')
  2907. a3 = Annotation.objects.create(tag=t, name='a3')
  2908. n = Note.objects.create(note='foo', misc='bar')
  2909. qs1 = Note.objects.exclude(annotation__in=[a1, a2])
  2910. qs2 = Note.objects.filter(annotation__in=[a3])
  2911. self.assertIn(n, qs1)
  2912. self.assertNotIn(n, qs2)
  2913. self.assertIn(n, (qs1 | qs2))
  2914. class EmptyStringPromotionTests(TestCase):
  2915. def test_empty_string_promotion(self):
  2916. qs = RelatedObject.objects.filter(single__name='')
  2917. if connection.features.interprets_empty_strings_as_nulls:
  2918. self.assertIn('LEFT OUTER JOIN', str(qs.query))
  2919. else:
  2920. self.assertNotIn('LEFT OUTER JOIN', str(qs.query))
  2921. class ValuesSubqueryTests(TestCase):
  2922. def test_values_in_subquery(self):
  2923. # Check that if a values() queryset is used, then the given values
  2924. # will be used instead of forcing use of the relation's field.
  2925. o1 = Order.objects.create(id=-2)
  2926. o2 = Order.objects.create(id=-1)
  2927. oi1 = OrderItem.objects.create(order=o1, status=0)
  2928. oi1.status = oi1.pk
  2929. oi1.save()
  2930. OrderItem.objects.create(order=o2, status=0)
  2931. # The query below should match o1 as it has related order_item
  2932. # with id == status.
  2933. self.assertQuerysetEqual(
  2934. Order.objects.filter(items__in=OrderItem.objects.values_list('status')),
  2935. [o1.pk], lambda x: x.pk)
  2936. class DoubleInSubqueryTests(TestCase):
  2937. def test_double_subquery_in(self):
  2938. lfa1 = LeafA.objects.create(data='foo')
  2939. lfa2 = LeafA.objects.create(data='bar')
  2940. lfb1 = LeafB.objects.create(data='lfb1')
  2941. lfb2 = LeafB.objects.create(data='lfb2')
  2942. Join.objects.create(a=lfa1, b=lfb1)
  2943. Join.objects.create(a=lfa2, b=lfb2)
  2944. leaf_as = LeafA.objects.filter(data='foo').values_list('pk', flat=True)
  2945. joins = Join.objects.filter(a__in=leaf_as).values_list('b__id', flat=True)
  2946. qs = LeafB.objects.filter(pk__in=joins)
  2947. self.assertQuerysetEqual(
  2948. qs, [lfb1], lambda x: x)
  2949. class Ticket18785Tests(TestCase):
  2950. def test_ticket_18785(self):
  2951. # Test join trimming from ticket18785
  2952. qs = Item.objects.exclude(
  2953. note__isnull=False
  2954. ).filter(
  2955. name='something', creator__extra__isnull=True
  2956. ).order_by()
  2957. self.assertEqual(1, str(qs.query).count('INNER JOIN'))
  2958. self.assertEqual(0, str(qs.query).count('OUTER JOIN'))
  2959. class Ticket20788Tests(TestCase):
  2960. def test_ticket_20788(self):
  2961. Paragraph.objects.create()
  2962. paragraph = Paragraph.objects.create()
  2963. page = paragraph.page.create()
  2964. chapter = Chapter.objects.create(paragraph=paragraph)
  2965. Book.objects.create(chapter=chapter)
  2966. paragraph2 = Paragraph.objects.create()
  2967. Page.objects.create()
  2968. chapter2 = Chapter.objects.create(paragraph=paragraph2)
  2969. book2 = Book.objects.create(chapter=chapter2)
  2970. sentences_not_in_pub = Book.objects.exclude(
  2971. chapter__paragraph__page=page)
  2972. self.assertQuerysetEqual(
  2973. sentences_not_in_pub, [book2], lambda x: x)
  2974. class Ticket12807Tests(TestCase):
  2975. def test_ticket_12807(self):
  2976. p1 = Paragraph.objects.create()
  2977. p2 = Paragraph.objects.create()
  2978. # The ORed condition below should have no effect on the query - the
  2979. # ~Q(pk__in=[]) will always be True.
  2980. qs = Paragraph.objects.filter((Q(pk=p2.pk) | ~Q(pk__in=[])) & Q(pk=p1.pk))
  2981. self.assertQuerysetEqual(qs, [p1], lambda x: x)
  2982. class RelatedLookupTypeTests(TestCase):
  2983. error = 'Cannot query "%s": Must be "%s" instance.'
  2984. @classmethod
  2985. def setUpTestData(cls):
  2986. cls.oa = ObjectA.objects.create(name="oa")
  2987. cls.poa = ProxyObjectA.objects.get(name="oa")
  2988. cls.coa = ChildObjectA.objects.create(name="coa")
  2989. cls.wrong_type = Order.objects.create(id=cls.oa.pk)
  2990. cls.ob = ObjectB.objects.create(name="ob", objecta=cls.oa, num=1)
  2991. ProxyObjectB.objects.create(name="pob", objecta=cls.oa, num=2)
  2992. cls.pob = ProxyObjectB.objects.all()
  2993. ObjectC.objects.create(childobjecta=cls.coa)
  2994. def test_wrong_type_lookup(self):
  2995. """
  2996. A ValueError is raised when the incorrect object type is passed to a
  2997. query lookup.
  2998. """
  2999. # Passing incorrect object type
  3000. with self.assertRaisesMessage(ValueError, self.error % (self.wrong_type, ObjectA._meta.object_name)):
  3001. ObjectB.objects.get(objecta=self.wrong_type)
  3002. with self.assertRaisesMessage(ValueError, self.error % (self.wrong_type, ObjectA._meta.object_name)):
  3003. ObjectB.objects.filter(objecta__in=[self.wrong_type])
  3004. with self.assertRaisesMessage(ValueError, self.error % (self.wrong_type, ObjectA._meta.object_name)):
  3005. ObjectB.objects.filter(objecta=self.wrong_type)
  3006. with self.assertRaisesMessage(ValueError, self.error % (self.wrong_type, ObjectB._meta.object_name)):
  3007. ObjectA.objects.filter(objectb__in=[self.wrong_type, self.ob])
  3008. # Passing an object of the class on which query is done.
  3009. with self.assertRaisesMessage(ValueError, self.error % (self.ob, ObjectA._meta.object_name)):
  3010. ObjectB.objects.filter(objecta__in=[self.poa, self.ob])
  3011. with self.assertRaisesMessage(ValueError, self.error % (self.ob, ChildObjectA._meta.object_name)):
  3012. ObjectC.objects.exclude(childobjecta__in=[self.coa, self.ob])
  3013. def test_wrong_backward_lookup(self):
  3014. """
  3015. A ValueError is raised when the incorrect object type is passed to a
  3016. query lookup for backward relations.
  3017. """
  3018. with self.assertRaisesMessage(ValueError, self.error % (self.oa, ObjectB._meta.object_name)):
  3019. ObjectA.objects.filter(objectb__in=[self.oa, self.ob])
  3020. with self.assertRaisesMessage(ValueError, self.error % (self.oa, ObjectB._meta.object_name)):
  3021. ObjectA.objects.exclude(objectb=self.oa)
  3022. with self.assertRaisesMessage(ValueError, self.error % (self.wrong_type, ObjectB._meta.object_name)):
  3023. ObjectA.objects.get(objectb=self.wrong_type)
  3024. def test_correct_lookup(self):
  3025. """
  3026. When passing proxy model objects, child objects, or parent objects,
  3027. lookups work fine.
  3028. """
  3029. out_a = ['<ObjectA: oa>', ]
  3030. out_b = ['<ObjectB: ob>', '<ObjectB: pob>']
  3031. out_c = ['<ObjectC: >']
  3032. # proxy model objects
  3033. self.assertQuerysetEqual(ObjectB.objects.filter(objecta=self.poa).order_by('name'), out_b)
  3034. self.assertQuerysetEqual(ObjectA.objects.filter(objectb__in=self.pob).order_by('pk'), out_a * 2)
  3035. # child objects
  3036. self.assertQuerysetEqual(ObjectB.objects.filter(objecta__in=[self.coa]), [])
  3037. self.assertQuerysetEqual(ObjectB.objects.filter(objecta__in=[self.poa, self.coa]).order_by('name'), out_b)
  3038. self.assertQuerysetEqual(
  3039. ObjectB.objects.filter(objecta__in=iter([self.poa, self.coa])).order_by('name'),
  3040. out_b
  3041. )
  3042. # parent objects
  3043. self.assertQuerysetEqual(ObjectC.objects.exclude(childobjecta=self.oa), out_c)
  3044. # QuerySet related object type checking shouldn't issue queries
  3045. # (the querysets aren't evaluated here, hence zero queries) (#23266).
  3046. with self.assertNumQueries(0):
  3047. ObjectB.objects.filter(objecta__in=ObjectA.objects.all())
  3048. def test_values_queryset_lookup(self):
  3049. """
  3050. #23396 - Ensure ValueQuerySets are not checked for compatibility with the lookup field
  3051. """
  3052. # Make sure the num and objecta field values match.
  3053. ob = ObjectB.objects.get(name='ob')
  3054. ob.num = ob.objecta.pk
  3055. ob.save()
  3056. pob = ObjectB.objects.get(name='pob')
  3057. pob.num = pob.objecta.pk
  3058. pob.save()
  3059. self.assertQuerysetEqual(ObjectB.objects.filter(
  3060. objecta__in=ObjectB.objects.all().values_list('num')
  3061. ).order_by('pk'), ['<ObjectB: ob>', '<ObjectB: pob>'])
  3062. class Ticket14056Tests(TestCase):
  3063. def test_ticket_14056(self):
  3064. s1 = SharedConnection.objects.create(data='s1')
  3065. s2 = SharedConnection.objects.create(data='s2')
  3066. s3 = SharedConnection.objects.create(data='s3')
  3067. PointerA.objects.create(connection=s2)
  3068. expected_ordering = (
  3069. [s1, s3, s2] if connection.features.nulls_order_largest
  3070. else [s2, s1, s3]
  3071. )
  3072. self.assertQuerysetEqual(
  3073. SharedConnection.objects.order_by('-pointera__connection', 'pk'),
  3074. expected_ordering, lambda x: x
  3075. )
  3076. class Ticket20955Tests(TestCase):
  3077. def test_ticket_20955(self):
  3078. jack = Staff.objects.create(name='jackstaff')
  3079. jackstaff = StaffUser.objects.create(staff=jack)
  3080. jill = Staff.objects.create(name='jillstaff')
  3081. jillstaff = StaffUser.objects.create(staff=jill)
  3082. task = Task.objects.create(creator=jackstaff, owner=jillstaff, title="task")
  3083. task_get = Task.objects.get(pk=task.pk)
  3084. # Load data so that assertNumQueries doesn't complain about the get
  3085. # version's queries.
  3086. task_get.creator.staffuser.staff
  3087. task_get.owner.staffuser.staff
  3088. qs = Task.objects.select_related(
  3089. 'creator__staffuser__staff', 'owner__staffuser__staff')
  3090. self.assertEqual(str(qs.query).count(' JOIN '), 6)
  3091. task_select_related = qs.get(pk=task.pk)
  3092. with self.assertNumQueries(0):
  3093. self.assertEqual(task_select_related.creator.staffuser.staff,
  3094. task_get.creator.staffuser.staff)
  3095. self.assertEqual(task_select_related.owner.staffuser.staff,
  3096. task_get.owner.staffuser.staff)
  3097. class Ticket21203Tests(TestCase):
  3098. def test_ticket_21203(self):
  3099. p = Ticket21203Parent.objects.create(parent_bool=True)
  3100. c = Ticket21203Child.objects.create(parent=p)
  3101. qs = Ticket21203Child.objects.select_related('parent').defer('parent__created')
  3102. self.assertQuerysetEqual(qs, [c], lambda x: x)
  3103. self.assertIs(qs[0].parent.parent_bool, True)
  3104. class ValuesJoinPromotionTests(TestCase):
  3105. def test_values_no_promotion_for_existing(self):
  3106. qs = Node.objects.filter(parent__parent__isnull=False)
  3107. self.assertIn(' INNER JOIN ', str(qs.query))
  3108. qs = qs.values('parent__parent__id')
  3109. self.assertIn(' INNER JOIN ', str(qs.query))
  3110. # Make sure there is a left outer join without the filter.
  3111. qs = Node.objects.values('parent__parent__id')
  3112. self.assertIn(' LEFT OUTER JOIN ', str(qs.query))
  3113. def test_non_nullable_fk_not_promoted(self):
  3114. qs = ObjectB.objects.values('objecta__name')
  3115. self.assertIn(' INNER JOIN ', str(qs.query))
  3116. def test_ticket_21376(self):
  3117. a = ObjectA.objects.create()
  3118. ObjectC.objects.create(objecta=a)
  3119. qs = ObjectC.objects.filter(
  3120. Q(objecta=a) | Q(objectb__objecta=a),
  3121. )
  3122. qs = qs.filter(
  3123. Q(objectb=1) | Q(objecta=a),
  3124. )
  3125. self.assertEqual(qs.count(), 1)
  3126. tblname = connection.ops.quote_name(ObjectB._meta.db_table)
  3127. self.assertIn(' LEFT OUTER JOIN %s' % tblname, str(qs.query))
  3128. class ForeignKeyToBaseExcludeTests(TestCase):
  3129. def test_ticket_21787(self):
  3130. sc1 = SpecialCategory.objects.create(special_name='sc1', name='sc1')
  3131. sc2 = SpecialCategory.objects.create(special_name='sc2', name='sc2')
  3132. sc3 = SpecialCategory.objects.create(special_name='sc3', name='sc3')
  3133. c1 = CategoryItem.objects.create(category=sc1)
  3134. CategoryItem.objects.create(category=sc2)
  3135. self.assertQuerysetEqual(
  3136. SpecialCategory.objects.exclude(
  3137. categoryitem__id=c1.pk).order_by('name'),
  3138. [sc2, sc3], lambda x: x
  3139. )
  3140. self.assertQuerysetEqual(
  3141. SpecialCategory.objects.filter(categoryitem__id=c1.pk),
  3142. [sc1], lambda x: x
  3143. )
  3144. class ReverseM2MCustomPkTests(TestCase):
  3145. def test_ticket_21879(self):
  3146. cpt1 = CustomPkTag.objects.create(id='cpt1', tag='cpt1')
  3147. cp1 = CustomPk.objects.create(name='cp1', extra='extra')
  3148. cp1.custompktag_set.add(cpt1)
  3149. self.assertQuerysetEqual(
  3150. CustomPk.objects.filter(custompktag=cpt1), [cp1],
  3151. lambda x: x)
  3152. self.assertQuerysetEqual(
  3153. CustomPkTag.objects.filter(custom_pk=cp1), [cpt1],
  3154. lambda x: x)
  3155. class Ticket22429Tests(TestCase):
  3156. def test_ticket_22429(self):
  3157. sc1 = School.objects.create()
  3158. st1 = Student.objects.create(school=sc1)
  3159. sc2 = School.objects.create()
  3160. st2 = Student.objects.create(school=sc2)
  3161. cr = Classroom.objects.create(school=sc1)
  3162. cr.students.add(st1)
  3163. queryset = Student.objects.filter(~Q(classroom__school=F('school')))
  3164. self.assertQuerysetEqual(queryset, [st2], lambda x: x)
  3165. class Ticket23605Tests(TestCase):
  3166. def test_ticket_23605(self):
  3167. # Test filtering on a complicated q-object from ticket's report.
  3168. # The query structure is such that we have multiple nested subqueries.
  3169. # The original problem was that the inner queries weren't relabeled
  3170. # correctly.
  3171. # See also #24090.
  3172. a1 = Ticket23605A.objects.create()
  3173. a2 = Ticket23605A.objects.create()
  3174. c1 = Ticket23605C.objects.create(field_c0=10000.0)
  3175. Ticket23605B.objects.create(
  3176. field_b0=10000.0, field_b1=True,
  3177. modelc_fk=c1, modela_fk=a1)
  3178. complex_q = Q(pk__in=Ticket23605A.objects.filter(
  3179. Q(
  3180. # True for a1 as field_b0 = 10000, field_c0=10000
  3181. # False for a2 as no ticket23605b found
  3182. ticket23605b__field_b0__gte=1000000 /
  3183. F("ticket23605b__modelc_fk__field_c0")
  3184. ) &
  3185. # True for a1 (field_b1=True)
  3186. Q(ticket23605b__field_b1=True) & ~Q(ticket23605b__pk__in=Ticket23605B.objects.filter(
  3187. ~(
  3188. # Same filters as above commented filters, but
  3189. # double-negated (one for Q() above, one for
  3190. # parentheses). So, again a1 match, a2 not.
  3191. Q(field_b1=True) &
  3192. Q(field_b0__gte=1000000 / F("modelc_fk__field_c0"))
  3193. )
  3194. ))).filter(ticket23605b__field_b1=True))
  3195. qs1 = Ticket23605A.objects.filter(complex_q)
  3196. self.assertQuerysetEqual(qs1, [a1], lambda x: x)
  3197. qs2 = Ticket23605A.objects.exclude(complex_q)
  3198. self.assertQuerysetEqual(qs2, [a2], lambda x: x)
  3199. class TestTicket24279(TestCase):
  3200. def test_ticket_24278(self):
  3201. School.objects.create()
  3202. qs = School.objects.filter(Q(pk__in=()) | Q())
  3203. self.assertQuerysetEqual(qs, [])
  3204. class TestInvalidValuesRelation(TestCase):
  3205. def test_invalid_values(self):
  3206. with self.assertRaises(ValueError):
  3207. Annotation.objects.filter(tag='abc')
  3208. with self.assertRaises(ValueError):
  3209. Annotation.objects.filter(tag__in=[123, 'abc'])
  3210. class TestTicket24605(TestCase):
  3211. def test_ticket_24605(self):
  3212. """
  3213. Subquery table names should be quoted.
  3214. """
  3215. i1 = Individual.objects.create(alive=True)
  3216. RelatedIndividual.objects.create(related=i1)
  3217. i2 = Individual.objects.create(alive=False)
  3218. RelatedIndividual.objects.create(related=i2)
  3219. i3 = Individual.objects.create(alive=True)
  3220. i4 = Individual.objects.create(alive=False)
  3221. self.assertQuerysetEqual(
  3222. Individual.objects.filter(Q(alive=False), Q(related_individual__isnull=True)),
  3223. [i4], lambda x: x
  3224. )
  3225. self.assertQuerysetEqual(
  3226. Individual.objects.exclude(
  3227. Q(alive=False), Q(related_individual__isnull=True)
  3228. ).order_by('pk'),
  3229. [i1, i2, i3], lambda x: x
  3230. )
  3231. class Ticket23622Tests(TestCase):
  3232. @skipUnlessDBFeature('can_distinct_on_fields')
  3233. def test_ticket_23622(self):
  3234. """
  3235. Make sure __pk__in and __in work the same for related fields when
  3236. using a distinct on subquery.
  3237. """
  3238. a1 = Ticket23605A.objects.create()
  3239. a2 = Ticket23605A.objects.create()
  3240. c1 = Ticket23605C.objects.create(field_c0=0.0)
  3241. Ticket23605B.objects.create(
  3242. modela_fk=a1, field_b0=123,
  3243. field_b1=True,
  3244. modelc_fk=c1,
  3245. )
  3246. Ticket23605B.objects.create(
  3247. modela_fk=a1, field_b0=23,
  3248. field_b1=True,
  3249. modelc_fk=c1,
  3250. )
  3251. Ticket23605B.objects.create(
  3252. modela_fk=a1, field_b0=234,
  3253. field_b1=True,
  3254. modelc_fk=c1,
  3255. )
  3256. Ticket23605B.objects.create(
  3257. modela_fk=a1, field_b0=12,
  3258. field_b1=True,
  3259. modelc_fk=c1,
  3260. )
  3261. Ticket23605B.objects.create(
  3262. modela_fk=a2, field_b0=567,
  3263. field_b1=True,
  3264. modelc_fk=c1,
  3265. )
  3266. Ticket23605B.objects.create(
  3267. modela_fk=a2, field_b0=76,
  3268. field_b1=True,
  3269. modelc_fk=c1,
  3270. )
  3271. Ticket23605B.objects.create(
  3272. modela_fk=a2, field_b0=7,
  3273. field_b1=True,
  3274. modelc_fk=c1,
  3275. )
  3276. Ticket23605B.objects.create(
  3277. modela_fk=a2, field_b0=56,
  3278. field_b1=True,
  3279. modelc_fk=c1,
  3280. )
  3281. qx = (
  3282. Q(ticket23605b__pk__in=Ticket23605B.objects.order_by('modela_fk', '-field_b1').distinct('modela_fk')) &
  3283. Q(ticket23605b__field_b0__gte=300)
  3284. )
  3285. qy = (
  3286. Q(ticket23605b__in=Ticket23605B.objects.order_by('modela_fk', '-field_b1').distinct('modela_fk')) &
  3287. Q(ticket23605b__field_b0__gte=300)
  3288. )
  3289. self.assertEqual(
  3290. set(Ticket23605A.objects.filter(qx).values_list('pk', flat=True)),
  3291. set(Ticket23605A.objects.filter(qy).values_list('pk', flat=True))
  3292. )
  3293. self.assertQuerysetEqual(
  3294. Ticket23605A.objects.filter(qx),
  3295. [a2], lambda x: x
  3296. )