tests.py 148 KB

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