tests.py 177 KB

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