tests.py 151 KB

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