123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487 |
- import logging
- import os
- import unittest
- import warnings
- from io import StringIO
- from unittest import mock
- from django.conf import settings
- from django.contrib.staticfiles.finders import get_finder, get_finders
- from django.contrib.staticfiles.storage import staticfiles_storage
- from django.core.exceptions import ImproperlyConfigured
- from django.core.files.storage import default_storage
- from django.db import (
- IntegrityError,
- connection,
- connections,
- models,
- router,
- transaction,
- )
- from django.forms import (
- CharField,
- EmailField,
- Form,
- IntegerField,
- ValidationError,
- formset_factory,
- )
- from django.http import HttpResponse
- from django.template.loader import render_to_string
- from django.test import (
- SimpleTestCase,
- TestCase,
- TransactionTestCase,
- skipIfDBFeature,
- skipUnlessDBFeature,
- )
- from django.test.html import HTMLParseError, parse_html
- from django.test.testcases import DatabaseOperationForbidden
- from django.test.utils import (
- CaptureQueriesContext,
- TestContextDecorator,
- ignore_warnings,
- isolate_apps,
- override_settings,
- setup_test_environment,
- )
- from django.urls import NoReverseMatch, path, reverse, reverse_lazy
- from django.utils.deprecation import RemovedInDjango50Warning
- from django.utils.log import DEFAULT_LOGGING
- from django.utils.version import PY311
- from .models import Car, Person, PossessedCar
- from .views import empty_response
- class SkippingTestCase(SimpleTestCase):
- def _assert_skipping(self, func, expected_exc, msg=None):
- try:
- if msg is not None:
- with self.assertRaisesMessage(expected_exc, msg):
- func()
- else:
- with self.assertRaises(expected_exc):
- func()
- except unittest.SkipTest:
- self.fail("%s should not result in a skipped test." % func.__name__)
- def test_skip_unless_db_feature(self):
- """
- Testing the django.test.skipUnlessDBFeature decorator.
- """
- # Total hack, but it works, just want an attribute that's always true.
- @skipUnlessDBFeature("__class__")
- def test_func():
- raise ValueError
- @skipUnlessDBFeature("notprovided")
- def test_func2():
- raise ValueError
- @skipUnlessDBFeature("__class__", "__class__")
- def test_func3():
- raise ValueError
- @skipUnlessDBFeature("__class__", "notprovided")
- def test_func4():
- raise ValueError
- self._assert_skipping(test_func, ValueError)
- self._assert_skipping(test_func2, unittest.SkipTest)
- self._assert_skipping(test_func3, ValueError)
- self._assert_skipping(test_func4, unittest.SkipTest)
- class SkipTestCase(SimpleTestCase):
- @skipUnlessDBFeature("missing")
- def test_foo(self):
- pass
- self._assert_skipping(
- SkipTestCase("test_foo").test_foo,
- ValueError,
- "skipUnlessDBFeature cannot be used on test_foo (test_utils.tests."
- "SkippingTestCase.test_skip_unless_db_feature.<locals>.SkipTestCase%s) "
- "as SkippingTestCase.test_skip_unless_db_feature.<locals>.SkipTestCase "
- "doesn't allow queries against the 'default' database."
- # Python 3.11 uses fully qualified test name in the output.
- % (".test_foo" if PY311 else ""),
- )
- def test_skip_if_db_feature(self):
- """
- Testing the django.test.skipIfDBFeature decorator.
- """
- @skipIfDBFeature("__class__")
- def test_func():
- raise ValueError
- @skipIfDBFeature("notprovided")
- def test_func2():
- raise ValueError
- @skipIfDBFeature("__class__", "__class__")
- def test_func3():
- raise ValueError
- @skipIfDBFeature("__class__", "notprovided")
- def test_func4():
- raise ValueError
- @skipIfDBFeature("notprovided", "notprovided")
- def test_func5():
- raise ValueError
- self._assert_skipping(test_func, unittest.SkipTest)
- self._assert_skipping(test_func2, ValueError)
- self._assert_skipping(test_func3, unittest.SkipTest)
- self._assert_skipping(test_func4, unittest.SkipTest)
- self._assert_skipping(test_func5, ValueError)
- class SkipTestCase(SimpleTestCase):
- @skipIfDBFeature("missing")
- def test_foo(self):
- pass
- self._assert_skipping(
- SkipTestCase("test_foo").test_foo,
- ValueError,
- "skipIfDBFeature cannot be used on test_foo (test_utils.tests."
- "SkippingTestCase.test_skip_if_db_feature.<locals>.SkipTestCase%s) "
- "as SkippingTestCase.test_skip_if_db_feature.<locals>.SkipTestCase "
- "doesn't allow queries against the 'default' database."
- # Python 3.11 uses fully qualified test name in the output.
- % (".test_foo" if PY311 else ""),
- )
- class SkippingClassTestCase(TestCase):
- def test_skip_class_unless_db_feature(self):
- @skipUnlessDBFeature("__class__")
- class NotSkippedTests(TestCase):
- def test_dummy(self):
- return
- @skipUnlessDBFeature("missing")
- @skipIfDBFeature("__class__")
- class SkippedTests(TestCase):
- def test_will_be_skipped(self):
- self.fail("We should never arrive here.")
- @skipIfDBFeature("__dict__")
- class SkippedTestsSubclass(SkippedTests):
- pass
- test_suite = unittest.TestSuite()
- test_suite.addTest(NotSkippedTests("test_dummy"))
- try:
- test_suite.addTest(SkippedTests("test_will_be_skipped"))
- test_suite.addTest(SkippedTestsSubclass("test_will_be_skipped"))
- except unittest.SkipTest:
- self.fail("SkipTest should not be raised here.")
- result = unittest.TextTestRunner(stream=StringIO()).run(test_suite)
- self.assertEqual(result.testsRun, 3)
- self.assertEqual(len(result.skipped), 2)
- self.assertEqual(result.skipped[0][1], "Database has feature(s) __class__")
- self.assertEqual(result.skipped[1][1], "Database has feature(s) __class__")
- def test_missing_default_databases(self):
- @skipIfDBFeature("missing")
- class MissingDatabases(SimpleTestCase):
- def test_assertion_error(self):
- pass
- suite = unittest.TestSuite()
- try:
- suite.addTest(MissingDatabases("test_assertion_error"))
- except unittest.SkipTest:
- self.fail("SkipTest should not be raised at this stage")
- runner = unittest.TextTestRunner(stream=StringIO())
- msg = (
- "skipIfDBFeature cannot be used on <class 'test_utils.tests."
- "SkippingClassTestCase.test_missing_default_databases.<locals>."
- "MissingDatabases'> as it doesn't allow queries against the "
- "'default' database."
- )
- with self.assertRaisesMessage(ValueError, msg):
- runner.run(suite)
- @override_settings(ROOT_URLCONF="test_utils.urls")
- class AssertNumQueriesTests(TestCase):
- def test_assert_num_queries(self):
- def test_func():
- raise ValueError
- with self.assertRaises(ValueError):
- self.assertNumQueries(2, test_func)
- def test_assert_num_queries_with_client(self):
- person = Person.objects.create(name="test")
- self.assertNumQueries(
- 1, self.client.get, "/test_utils/get_person/%s/" % person.pk
- )
- self.assertNumQueries(
- 1, self.client.get, "/test_utils/get_person/%s/" % person.pk
- )
- def test_func():
- self.client.get("/test_utils/get_person/%s/" % person.pk)
- self.client.get("/test_utils/get_person/%s/" % person.pk)
- self.assertNumQueries(2, test_func)
- @unittest.skipUnless(
- connection.vendor != "sqlite" or not connection.is_in_memory_db(),
- "For SQLite in-memory tests, closing the connection destroys the database.",
- )
- class AssertNumQueriesUponConnectionTests(TransactionTestCase):
- available_apps = []
- def test_ignores_connection_configuration_queries(self):
- real_ensure_connection = connection.ensure_connection
- connection.close()
- def make_configuration_query():
- is_opening_connection = connection.connection is None
- real_ensure_connection()
- if is_opening_connection:
- # Avoid infinite recursion. Creating a cursor calls
- # ensure_connection() which is currently mocked by this method.
- with connection.cursor() as cursor:
- cursor.execute("SELECT 1" + connection.features.bare_select_suffix)
- ensure_connection = (
- "django.db.backends.base.base.BaseDatabaseWrapper.ensure_connection"
- )
- with mock.patch(ensure_connection, side_effect=make_configuration_query):
- with self.assertNumQueries(1):
- list(Car.objects.all())
- class AssertQuerysetEqualTests(TestCase):
- @classmethod
- def setUpTestData(cls):
- cls.p1 = Person.objects.create(name="p1")
- cls.p2 = Person.objects.create(name="p2")
- def test_empty(self):
- self.assertQuerysetEqual(Person.objects.filter(name="p3"), [])
- def test_ordered(self):
- self.assertQuerysetEqual(
- Person.objects.order_by("name"),
- [self.p1, self.p2],
- )
- def test_unordered(self):
- self.assertQuerysetEqual(
- Person.objects.order_by("name"), [self.p2, self.p1], ordered=False
- )
- def test_queryset(self):
- self.assertQuerysetEqual(
- Person.objects.order_by("name"),
- Person.objects.order_by("name"),
- )
- def test_flat_values_list(self):
- self.assertQuerysetEqual(
- Person.objects.order_by("name").values_list("name", flat=True),
- ["p1", "p2"],
- )
- def test_transform(self):
- self.assertQuerysetEqual(
- Person.objects.order_by("name"),
- [self.p1.pk, self.p2.pk],
- transform=lambda x: x.pk,
- )
- def test_repr_transform(self):
- self.assertQuerysetEqual(
- Person.objects.order_by("name"),
- [repr(self.p1), repr(self.p2)],
- transform=repr,
- )
- def test_undefined_order(self):
- # Using an unordered queryset with more than one ordered value
- # is an error.
- msg = (
- "Trying to compare non-ordered queryset against more than one "
- "ordered value."
- )
- with self.assertRaisesMessage(ValueError, msg):
- self.assertQuerysetEqual(
- Person.objects.all(),
- [self.p1, self.p2],
- )
- # No error for one value.
- self.assertQuerysetEqual(Person.objects.filter(name="p1"), [self.p1])
- def test_repeated_values(self):
- """
- assertQuerysetEqual checks the number of appearance of each item
- when used with option ordered=False.
- """
- batmobile = Car.objects.create(name="Batmobile")
- k2000 = Car.objects.create(name="K 2000")
- PossessedCar.objects.bulk_create(
- [
- PossessedCar(car=batmobile, belongs_to=self.p1),
- PossessedCar(car=batmobile, belongs_to=self.p1),
- PossessedCar(car=k2000, belongs_to=self.p1),
- PossessedCar(car=k2000, belongs_to=self.p1),
- PossessedCar(car=k2000, belongs_to=self.p1),
- PossessedCar(car=k2000, belongs_to=self.p1),
- ]
- )
- with self.assertRaises(AssertionError):
- self.assertQuerysetEqual(
- self.p1.cars.all(), [batmobile, k2000], ordered=False
- )
- self.assertQuerysetEqual(
- self.p1.cars.all(), [batmobile] * 2 + [k2000] * 4, ordered=False
- )
- def test_maxdiff(self):
- names = ["Joe Smith %s" % i for i in range(20)]
- Person.objects.bulk_create([Person(name=name) for name in names])
- names.append("Extra Person")
- with self.assertRaises(AssertionError) as ctx:
- self.assertQuerysetEqual(
- Person.objects.filter(name__startswith="Joe"),
- names,
- ordered=False,
- transform=lambda p: p.name,
- )
- self.assertIn("Set self.maxDiff to None to see it.", str(ctx.exception))
- original = self.maxDiff
- self.maxDiff = None
- try:
- with self.assertRaises(AssertionError) as ctx:
- self.assertQuerysetEqual(
- Person.objects.filter(name__startswith="Joe"),
- names,
- ordered=False,
- transform=lambda p: p.name,
- )
- finally:
- self.maxDiff = original
- exception_msg = str(ctx.exception)
- self.assertNotIn("Set self.maxDiff to None to see it.", exception_msg)
- for name in names:
- self.assertIn(name, exception_msg)
- @override_settings(ROOT_URLCONF="test_utils.urls")
- class CaptureQueriesContextManagerTests(TestCase):
- @classmethod
- def setUpTestData(cls):
- cls.person_pk = str(Person.objects.create(name="test").pk)
- def test_simple(self):
- with CaptureQueriesContext(connection) as captured_queries:
- Person.objects.get(pk=self.person_pk)
- self.assertEqual(len(captured_queries), 1)
- self.assertIn(self.person_pk, captured_queries[0]["sql"])
- with CaptureQueriesContext(connection) as captured_queries:
- pass
- self.assertEqual(0, len(captured_queries))
- def test_within(self):
- with CaptureQueriesContext(connection) as captured_queries:
- Person.objects.get(pk=self.person_pk)
- self.assertEqual(len(captured_queries), 1)
- self.assertIn(self.person_pk, captured_queries[0]["sql"])
- def test_nested(self):
- with CaptureQueriesContext(connection) as captured_queries:
- Person.objects.count()
- with CaptureQueriesContext(connection) as nested_captured_queries:
- Person.objects.count()
- self.assertEqual(1, len(nested_captured_queries))
- self.assertEqual(2, len(captured_queries))
- def test_failure(self):
- with self.assertRaises(TypeError):
- with CaptureQueriesContext(connection):
- raise TypeError
- def test_with_client(self):
- with CaptureQueriesContext(connection) as captured_queries:
- self.client.get("/test_utils/get_person/%s/" % self.person_pk)
- self.assertEqual(len(captured_queries), 1)
- self.assertIn(self.person_pk, captured_queries[0]["sql"])
- with CaptureQueriesContext(connection) as captured_queries:
- self.client.get("/test_utils/get_person/%s/" % self.person_pk)
- self.assertEqual(len(captured_queries), 1)
- self.assertIn(self.person_pk, captured_queries[0]["sql"])
- with CaptureQueriesContext(connection) as captured_queries:
- self.client.get("/test_utils/get_person/%s/" % self.person_pk)
- self.client.get("/test_utils/get_person/%s/" % self.person_pk)
- self.assertEqual(len(captured_queries), 2)
- self.assertIn(self.person_pk, captured_queries[0]["sql"])
- self.assertIn(self.person_pk, captured_queries[1]["sql"])
- @override_settings(ROOT_URLCONF="test_utils.urls")
- class AssertNumQueriesContextManagerTests(TestCase):
- def test_simple(self):
- with self.assertNumQueries(0):
- pass
- with self.assertNumQueries(1):
- Person.objects.count()
- with self.assertNumQueries(2):
- Person.objects.count()
- Person.objects.count()
- def test_failure(self):
- msg = "1 != 2 : 1 queries executed, 2 expected\nCaptured queries were:\n1."
- with self.assertRaisesMessage(AssertionError, msg):
- with self.assertNumQueries(2):
- Person.objects.count()
- with self.assertRaises(TypeError):
- with self.assertNumQueries(4000):
- raise TypeError
- def test_with_client(self):
- person = Person.objects.create(name="test")
- with self.assertNumQueries(1):
- self.client.get("/test_utils/get_person/%s/" % person.pk)
- with self.assertNumQueries(1):
- self.client.get("/test_utils/get_person/%s/" % person.pk)
- with self.assertNumQueries(2):
- self.client.get("/test_utils/get_person/%s/" % person.pk)
- self.client.get("/test_utils/get_person/%s/" % person.pk)
- @override_settings(ROOT_URLCONF="test_utils.urls")
- class AssertTemplateUsedContextManagerTests(SimpleTestCase):
- def test_usage(self):
- with self.assertTemplateUsed("template_used/base.html"):
- render_to_string("template_used/base.html")
- with self.assertTemplateUsed(template_name="template_used/base.html"):
- render_to_string("template_used/base.html")
- with self.assertTemplateUsed("template_used/base.html"):
- render_to_string("template_used/include.html")
- with self.assertTemplateUsed("template_used/base.html"):
- render_to_string("template_used/extends.html")
- with self.assertTemplateUsed("template_used/base.html"):
- render_to_string("template_used/base.html")
- render_to_string("template_used/base.html")
- def test_nested_usage(self):
- with self.assertTemplateUsed("template_used/base.html"):
- with self.assertTemplateUsed("template_used/include.html"):
- render_to_string("template_used/include.html")
- with self.assertTemplateUsed("template_used/extends.html"):
- with self.assertTemplateUsed("template_used/base.html"):
- render_to_string("template_used/extends.html")
- with self.assertTemplateUsed("template_used/base.html"):
- with self.assertTemplateUsed("template_used/alternative.html"):
- render_to_string("template_used/alternative.html")
- render_to_string("template_used/base.html")
- with self.assertTemplateUsed("template_used/base.html"):
- render_to_string("template_used/extends.html")
- with self.assertTemplateNotUsed("template_used/base.html"):
- render_to_string("template_used/alternative.html")
- render_to_string("template_used/base.html")
- def test_not_used(self):
- with self.assertTemplateNotUsed("template_used/base.html"):
- pass
- with self.assertTemplateNotUsed("template_used/alternative.html"):
- pass
- def test_error_message(self):
- msg = "No templates used to render the response"
- with self.assertRaisesMessage(AssertionError, msg):
- with self.assertTemplateUsed("template_used/base.html"):
- pass
- with self.assertRaisesMessage(AssertionError, msg):
- with self.assertTemplateUsed(template_name="template_used/base.html"):
- pass
- msg2 = (
- "Template 'template_used/base.html' was not a template used to render "
- "the response. Actual template(s) used: template_used/alternative.html"
- )
- with self.assertRaisesMessage(AssertionError, msg2):
- with self.assertTemplateUsed("template_used/base.html"):
- render_to_string("template_used/alternative.html")
- with self.assertRaisesMessage(
- AssertionError, "No templates used to render the response"
- ):
- response = self.client.get("/test_utils/no_template_used/")
- self.assertTemplateUsed(response, "template_used/base.html")
- def test_msg_prefix(self):
- msg_prefix = "Prefix"
- msg = f"{msg_prefix}: No templates used to render the response"
- with self.assertRaisesMessage(AssertionError, msg):
- with self.assertTemplateUsed(
- "template_used/base.html", msg_prefix=msg_prefix
- ):
- pass
- with self.assertRaisesMessage(AssertionError, msg):
- with self.assertTemplateUsed(
- template_name="template_used/base.html",
- msg_prefix=msg_prefix,
- ):
- pass
- msg = (
- f"{msg_prefix}: Template 'template_used/base.html' was not a "
- f"template used to render the response. Actual template(s) used: "
- f"template_used/alternative.html"
- )
- with self.assertRaisesMessage(AssertionError, msg):
- with self.assertTemplateUsed(
- "template_used/base.html", msg_prefix=msg_prefix
- ):
- render_to_string("template_used/alternative.html")
- def test_count(self):
- with self.assertTemplateUsed("template_used/base.html", count=2):
- render_to_string("template_used/base.html")
- render_to_string("template_used/base.html")
- msg = (
- "Template 'template_used/base.html' was expected to be rendered "
- "3 time(s) but was actually rendered 2 time(s)."
- )
- with self.assertRaisesMessage(AssertionError, msg):
- with self.assertTemplateUsed("template_used/base.html", count=3):
- render_to_string("template_used/base.html")
- render_to_string("template_used/base.html")
- def test_failure(self):
- msg = "response and/or template_name argument must be provided"
- with self.assertRaisesMessage(TypeError, msg):
- with self.assertTemplateUsed():
- pass
- msg = "No templates used to render the response"
- with self.assertRaisesMessage(AssertionError, msg):
- with self.assertTemplateUsed(""):
- pass
- with self.assertRaisesMessage(AssertionError, msg):
- with self.assertTemplateUsed(""):
- render_to_string("template_used/base.html")
- with self.assertRaisesMessage(AssertionError, msg):
- with self.assertTemplateUsed(template_name=""):
- pass
- msg = (
- "Template 'template_used/base.html' was not a template used to "
- "render the response. Actual template(s) used: "
- "template_used/alternative.html"
- )
- with self.assertRaisesMessage(AssertionError, msg):
- with self.assertTemplateUsed("template_used/base.html"):
- render_to_string("template_used/alternative.html")
- def test_assert_used_on_http_response(self):
- response = HttpResponse()
- msg = "%s() is only usable on responses fetched using the Django test Client."
- with self.assertRaisesMessage(ValueError, msg % "assertTemplateUsed"):
- self.assertTemplateUsed(response, "template.html")
- with self.assertRaisesMessage(ValueError, msg % "assertTemplateNotUsed"):
- self.assertTemplateNotUsed(response, "template.html")
- class HTMLEqualTests(SimpleTestCase):
- def test_html_parser(self):
- element = parse_html("<div><p>Hello</p></div>")
- self.assertEqual(len(element.children), 1)
- self.assertEqual(element.children[0].name, "p")
- self.assertEqual(element.children[0].children[0], "Hello")
- parse_html("<p>")
- parse_html("<p attr>")
- dom = parse_html("<p>foo")
- self.assertEqual(len(dom.children), 1)
- self.assertEqual(dom.name, "p")
- self.assertEqual(dom[0], "foo")
- def test_parse_html_in_script(self):
- parse_html('<script>var a = "<p" + ">";</script>')
- parse_html(
- """
- <script>
- var js_sha_link='<p>***</p>';
- </script>
- """
- )
- # script content will be parsed to text
- dom = parse_html(
- """
- <script><p>foo</p> '</scr'+'ipt>' <span>bar</span></script>
- """
- )
- self.assertEqual(len(dom.children), 1)
- self.assertEqual(dom.children[0], "<p>foo</p> '</scr'+'ipt>' <span>bar</span>")
- def test_self_closing_tags(self):
- self_closing_tags = [
- "area",
- "base",
- "br",
- "col",
- "embed",
- "hr",
- "img",
- "input",
- "link",
- "meta",
- "param",
- "source",
- "track",
- "wbr",
- # Deprecated tags
- "frame",
- "spacer",
- ]
- for tag in self_closing_tags:
- with self.subTest(tag):
- dom = parse_html("<p>Hello <%s> world</p>" % tag)
- self.assertEqual(len(dom.children), 3)
- self.assertEqual(dom[0], "Hello")
- self.assertEqual(dom[1].name, tag)
- self.assertEqual(dom[2], "world")
- dom = parse_html("<p>Hello <%s /> world</p>" % tag)
- self.assertEqual(len(dom.children), 3)
- self.assertEqual(dom[0], "Hello")
- self.assertEqual(dom[1].name, tag)
- self.assertEqual(dom[2], "world")
- def test_simple_equal_html(self):
- self.assertHTMLEqual("", "")
- self.assertHTMLEqual("<p></p>", "<p></p>")
- self.assertHTMLEqual("<p></p>", " <p> </p> ")
- self.assertHTMLEqual("<div><p>Hello</p></div>", "<div><p>Hello</p></div>")
- self.assertHTMLEqual("<div><p>Hello</p></div>", "<div> <p>Hello</p> </div>")
- self.assertHTMLEqual("<div>\n<p>Hello</p></div>", "<div><p>Hello</p></div>\n")
- self.assertHTMLEqual(
- "<div><p>Hello\nWorld !</p></div>", "<div><p>Hello World\n!</p></div>"
- )
- self.assertHTMLEqual(
- "<div><p>Hello\nWorld !</p></div>", "<div><p>Hello World\n!</p></div>"
- )
- self.assertHTMLEqual("<p>Hello World !</p>", "<p>Hello World\n\n!</p>")
- self.assertHTMLEqual("<p> </p>", "<p></p>")
- self.assertHTMLEqual("<p/>", "<p></p>")
- self.assertHTMLEqual("<p />", "<p></p>")
- self.assertHTMLEqual("<input checked>", '<input checked="checked">')
- self.assertHTMLEqual("<p>Hello", "<p> Hello")
- self.assertHTMLEqual("<p>Hello</p>World", "<p>Hello</p> World")
- def test_ignore_comments(self):
- self.assertHTMLEqual(
- "<div>Hello<!-- this is a comment --> World!</div>",
- "<div>Hello World!</div>",
- )
- def test_unequal_html(self):
- self.assertHTMLNotEqual("<p>Hello</p>", "<p>Hello!</p>")
- self.assertHTMLNotEqual("<p>foobar</p>", "<p>foo bar</p>")
- self.assertHTMLNotEqual("<p>foo bar</p>", "<p>foo bar</p>")
- self.assertHTMLNotEqual("<p>foo nbsp</p>", "<p>foo </p>")
- self.assertHTMLNotEqual("<p>foo #20</p>", "<p>foo </p>")
- self.assertHTMLNotEqual(
- "<p><span>Hello</span><span>World</span></p>",
- "<p><span>Hello</span>World</p>",
- )
- self.assertHTMLNotEqual(
- "<p><span>Hello</span>World</p>",
- "<p><span>Hello</span><span>World</span></p>",
- )
- def test_attributes(self):
- self.assertHTMLEqual(
- '<input type="text" id="id_name" />', '<input id="id_name" type="text" />'
- )
- self.assertHTMLEqual(
- """<input type='text' id="id_name" />""",
- '<input id="id_name" type="text" />',
- )
- self.assertHTMLNotEqual(
- '<input type="text" id="id_name" />',
- '<input type="password" id="id_name" />',
- )
- def test_class_attribute(self):
- pairs = [
- ('<p class="foo bar"></p>', '<p class="bar foo"></p>'),
- ('<p class=" foo bar "></p>', '<p class="bar foo"></p>'),
- ('<p class=" foo bar "></p>', '<p class="bar foo"></p>'),
- ('<p class="foo\tbar"></p>', '<p class="bar foo"></p>'),
- ('<p class="\tfoo\tbar\t"></p>', '<p class="bar foo"></p>'),
- ('<p class="\t\t\tfoo\t\t\tbar\t\t\t"></p>', '<p class="bar foo"></p>'),
- ('<p class="\t \nfoo \t\nbar\n\t "></p>', '<p class="bar foo"></p>'),
- ]
- for html1, html2 in pairs:
- with self.subTest(html1):
- self.assertHTMLEqual(html1, html2)
- def test_boolean_attribute(self):
- html1 = "<input checked>"
- html2 = '<input checked="">'
- html3 = '<input checked="checked">'
- self.assertHTMLEqual(html1, html2)
- self.assertHTMLEqual(html1, html3)
- self.assertHTMLEqual(html2, html3)
- self.assertHTMLNotEqual(html1, '<input checked="invalid">')
- self.assertEqual(str(parse_html(html1)), "<input checked>")
- self.assertEqual(str(parse_html(html2)), "<input checked>")
- self.assertEqual(str(parse_html(html3)), "<input checked>")
- def test_non_boolean_attibutes(self):
- html1 = "<input value>"
- html2 = '<input value="">'
- html3 = '<input value="value">'
- self.assertHTMLEqual(html1, html2)
- self.assertHTMLNotEqual(html1, html3)
- self.assertEqual(str(parse_html(html1)), '<input value="">')
- self.assertEqual(str(parse_html(html2)), '<input value="">')
- def test_normalize_refs(self):
- pairs = [
- ("'", "'"),
- ("'", "'"),
- ("'", "'"),
- ("'", "'"),
- ("'", "'"),
- ("'", "'"),
- ("&", "&"),
- ("&", "&"),
- ("&", "&"),
- ("&", "&"),
- ("&", "&"),
- ("&", "&"),
- ("&", "&"),
- ("&", "&"),
- ("&", "&"),
- ("&", "&"),
- ("&", "&"),
- ("&", "&"),
- ]
- for pair in pairs:
- with self.subTest(repr(pair)):
- self.assertHTMLEqual(*pair)
- def test_complex_examples(self):
- self.assertHTMLEqual(
- """<tr><th><label for="id_first_name">First name:</label></th>
- <td><input type="text" name="first_name" value="John" id="id_first_name" /></td></tr>
- <tr><th><label for="id_last_name">Last name:</label></th>
- <td><input type="text" id="id_last_name" name="last_name" value="Lennon" /></td></tr>
- <tr><th><label for="id_birthday">Birthday:</label></th>
- <td><input type="text" value="1940-10-9" name="birthday" id="id_birthday" /></td></tr>""", # NOQA
- """
- <tr><th>
- <label for="id_first_name">First name:</label></th><td>
- <input type="text" name="first_name" value="John" id="id_first_name" />
- </td></tr>
- <tr><th>
- <label for="id_last_name">Last name:</label></th><td>
- <input type="text" name="last_name" value="Lennon" id="id_last_name" />
- </td></tr>
- <tr><th>
- <label for="id_birthday">Birthday:</label></th><td>
- <input type="text" name="birthday" value="1940-10-9" id="id_birthday" />
- </td></tr>
- """,
- )
- self.assertHTMLEqual(
- """<!DOCTYPE html>
- <html>
- <head>
- <link rel="stylesheet">
- <title>Document</title>
- <meta attribute="value">
- </head>
- <body>
- <p>
- This is a valid paragraph
- <div> this is a div AFTER the p</div>
- </body>
- </html>""",
- """
- <html>
- <head>
- <link rel="stylesheet">
- <title>Document</title>
- <meta attribute="value">
- </head>
- <body>
- <p> This is a valid paragraph
- <!-- browsers would close the p tag here -->
- <div> this is a div AFTER the p</div>
- </p> <!-- this is invalid HTML parsing, but it should make no
- difference in most cases -->
- </body>
- </html>""",
- )
- def test_html_contain(self):
- # equal html contains each other
- dom1 = parse_html("<p>foo")
- dom2 = parse_html("<p>foo</p>")
- self.assertIn(dom1, dom2)
- self.assertIn(dom2, dom1)
- dom2 = parse_html("<div><p>foo</p></div>")
- self.assertIn(dom1, dom2)
- self.assertNotIn(dom2, dom1)
- self.assertNotIn("<p>foo</p>", dom2)
- self.assertIn("foo", dom2)
- # when a root element is used ...
- dom1 = parse_html("<p>foo</p><p>bar</p>")
- dom2 = parse_html("<p>foo</p><p>bar</p>")
- self.assertIn(dom1, dom2)
- dom1 = parse_html("<p>foo</p>")
- self.assertIn(dom1, dom2)
- dom1 = parse_html("<p>bar</p>")
- self.assertIn(dom1, dom2)
- dom1 = parse_html("<div><p>foo</p><p>bar</p></div>")
- self.assertIn(dom2, dom1)
- def test_count(self):
- # equal html contains each other one time
- dom1 = parse_html("<p>foo")
- dom2 = parse_html("<p>foo</p>")
- self.assertEqual(dom1.count(dom2), 1)
- self.assertEqual(dom2.count(dom1), 1)
- dom2 = parse_html("<p>foo</p><p>bar</p>")
- self.assertEqual(dom2.count(dom1), 1)
- dom2 = parse_html("<p>foo foo</p><p>foo</p>")
- self.assertEqual(dom2.count("foo"), 3)
- dom2 = parse_html('<p class="bar">foo</p>')
- self.assertEqual(dom2.count("bar"), 0)
- self.assertEqual(dom2.count("class"), 0)
- self.assertEqual(dom2.count("p"), 0)
- self.assertEqual(dom2.count("o"), 2)
- dom2 = parse_html("<p>foo</p><p>foo</p>")
- self.assertEqual(dom2.count(dom1), 2)
- dom2 = parse_html('<div><p>foo<input type=""></p><p>foo</p></div>')
- self.assertEqual(dom2.count(dom1), 1)
- dom2 = parse_html("<div><div><p>foo</p></div></div>")
- self.assertEqual(dom2.count(dom1), 1)
- dom2 = parse_html("<p>foo<p>foo</p></p>")
- self.assertEqual(dom2.count(dom1), 1)
- dom2 = parse_html("<p>foo<p>bar</p></p>")
- self.assertEqual(dom2.count(dom1), 0)
- # HTML with a root element contains the same HTML with no root element.
- dom1 = parse_html("<p>foo</p><p>bar</p>")
- dom2 = parse_html("<div><p>foo</p><p>bar</p></div>")
- self.assertEqual(dom2.count(dom1), 1)
- # Target of search is a sequence of child elements and appears more
- # than once.
- dom2 = parse_html("<div><p>foo</p><p>bar</p><p>foo</p><p>bar</p></div>")
- self.assertEqual(dom2.count(dom1), 2)
- # Searched HTML has additional children.
- dom1 = parse_html("<a/><b/>")
- dom2 = parse_html("<a/><b/><c/>")
- self.assertEqual(dom2.count(dom1), 1)
- # No match found in children.
- dom1 = parse_html("<b/><a/>")
- self.assertEqual(dom2.count(dom1), 0)
- # Target of search found among children and grandchildren.
- dom1 = parse_html("<b/><b/>")
- dom2 = parse_html("<a><b/><b/></a><b/><b/>")
- self.assertEqual(dom2.count(dom1), 2)
- def test_root_element_escaped_html(self):
- html = "<br>"
- parsed = parse_html(html)
- self.assertEqual(str(parsed), html)
- def test_parsing_errors(self):
- with self.assertRaises(AssertionError):
- self.assertHTMLEqual("<p>", "")
- with self.assertRaises(AssertionError):
- self.assertHTMLEqual("", "<p>")
- error_msg = (
- "First argument is not valid HTML:\n"
- "('Unexpected end tag `div` (Line 1, Column 6)', (1, 6))"
- )
- with self.assertRaisesMessage(AssertionError, error_msg):
- self.assertHTMLEqual("< div></ div>", "<div></div>")
- with self.assertRaises(HTMLParseError):
- parse_html("</p>")
- def test_escaped_html_errors(self):
- msg = "<p>\n<foo>\n</p> != <p>\n<foo>\n</p>\n"
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertHTMLEqual("<p><foo></p>", "<p><foo></p>")
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertHTMLEqual("<p><foo></p>", "<p><foo></p>")
- def test_contains_html(self):
- response = HttpResponse(
- """<body>
- This is a form: <form method="get">
- <input type="text" name="Hello" />
- </form></body>"""
- )
- self.assertNotContains(response, "<input name='Hello' type='text'>")
- self.assertContains(response, '<form method="get">')
- self.assertContains(response, "<input name='Hello' type='text'>", html=True)
- self.assertNotContains(response, '<form method="get">', html=True)
- invalid_response = HttpResponse("""<body <bad>>""")
- with self.assertRaises(AssertionError):
- self.assertContains(invalid_response, "<p></p>")
- with self.assertRaises(AssertionError):
- self.assertContains(response, '<p "whats" that>')
- def test_unicode_handling(self):
- response = HttpResponse(
- '<p class="help">Some help text for the title (with Unicode ŠĐĆŽćžšđ)</p>'
- )
- self.assertContains(
- response,
- '<p class="help">Some help text for the title (with Unicode ŠĐĆŽćžšđ)</p>',
- html=True,
- )
- class JSONEqualTests(SimpleTestCase):
- def test_simple_equal(self):
- json1 = '{"attr1": "foo", "attr2":"baz"}'
- json2 = '{"attr1": "foo", "attr2":"baz"}'
- self.assertJSONEqual(json1, json2)
- def test_simple_equal_unordered(self):
- json1 = '{"attr1": "foo", "attr2":"baz"}'
- json2 = '{"attr2":"baz", "attr1": "foo"}'
- self.assertJSONEqual(json1, json2)
- def test_simple_equal_raise(self):
- json1 = '{"attr1": "foo", "attr2":"baz"}'
- json2 = '{"attr2":"baz"}'
- with self.assertRaises(AssertionError):
- self.assertJSONEqual(json1, json2)
- def test_equal_parsing_errors(self):
- invalid_json = '{"attr1": "foo, "attr2":"baz"}'
- valid_json = '{"attr1": "foo", "attr2":"baz"}'
- with self.assertRaises(AssertionError):
- self.assertJSONEqual(invalid_json, valid_json)
- with self.assertRaises(AssertionError):
- self.assertJSONEqual(valid_json, invalid_json)
- def test_simple_not_equal(self):
- json1 = '{"attr1": "foo", "attr2":"baz"}'
- json2 = '{"attr2":"baz"}'
- self.assertJSONNotEqual(json1, json2)
- def test_simple_not_equal_raise(self):
- json1 = '{"attr1": "foo", "attr2":"baz"}'
- json2 = '{"attr1": "foo", "attr2":"baz"}'
- with self.assertRaises(AssertionError):
- self.assertJSONNotEqual(json1, json2)
- def test_not_equal_parsing_errors(self):
- invalid_json = '{"attr1": "foo, "attr2":"baz"}'
- valid_json = '{"attr1": "foo", "attr2":"baz"}'
- with self.assertRaises(AssertionError):
- self.assertJSONNotEqual(invalid_json, valid_json)
- with self.assertRaises(AssertionError):
- self.assertJSONNotEqual(valid_json, invalid_json)
- class XMLEqualTests(SimpleTestCase):
- def test_simple_equal(self):
- xml1 = "<elem attr1='a' attr2='b' />"
- xml2 = "<elem attr1='a' attr2='b' />"
- self.assertXMLEqual(xml1, xml2)
- def test_simple_equal_unordered(self):
- xml1 = "<elem attr1='a' attr2='b' />"
- xml2 = "<elem attr2='b' attr1='a' />"
- self.assertXMLEqual(xml1, xml2)
- def test_simple_equal_raise(self):
- xml1 = "<elem attr1='a' />"
- xml2 = "<elem attr2='b' attr1='a' />"
- with self.assertRaises(AssertionError):
- self.assertXMLEqual(xml1, xml2)
- def test_simple_equal_raises_message(self):
- xml1 = "<elem attr1='a' />"
- xml2 = "<elem attr2='b' attr1='a' />"
- msg = """{xml1} != {xml2}
- - <elem attr1='a' />
- + <elem attr2='b' attr1='a' />
- ? ++++++++++
- """.format(
- xml1=repr(xml1), xml2=repr(xml2)
- )
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertXMLEqual(xml1, xml2)
- def test_simple_not_equal(self):
- xml1 = "<elem attr1='a' attr2='c' />"
- xml2 = "<elem attr1='a' attr2='b' />"
- self.assertXMLNotEqual(xml1, xml2)
- def test_simple_not_equal_raise(self):
- xml1 = "<elem attr1='a' attr2='b' />"
- xml2 = "<elem attr2='b' attr1='a' />"
- with self.assertRaises(AssertionError):
- self.assertXMLNotEqual(xml1, xml2)
- def test_parsing_errors(self):
- xml_unvalid = "<elem attr1='a attr2='b' />"
- xml2 = "<elem attr2='b' attr1='a' />"
- with self.assertRaises(AssertionError):
- self.assertXMLNotEqual(xml_unvalid, xml2)
- def test_comment_root(self):
- xml1 = "<?xml version='1.0'?><!-- comment1 --><elem attr1='a' attr2='b' />"
- xml2 = "<?xml version='1.0'?><!-- comment2 --><elem attr2='b' attr1='a' />"
- self.assertXMLEqual(xml1, xml2)
- def test_simple_equal_with_leading_or_trailing_whitespace(self):
- xml1 = "<elem>foo</elem> \t\n"
- xml2 = " \t\n<elem>foo</elem>"
- self.assertXMLEqual(xml1, xml2)
- def test_simple_not_equal_with_whitespace_in_the_middle(self):
- xml1 = "<elem>foo</elem><elem>bar</elem>"
- xml2 = "<elem>foo</elem> <elem>bar</elem>"
- self.assertXMLNotEqual(xml1, xml2)
- def test_doctype_root(self):
- xml1 = '<?xml version="1.0"?><!DOCTYPE root SYSTEM "example1.dtd"><root />'
- xml2 = '<?xml version="1.0"?><!DOCTYPE root SYSTEM "example2.dtd"><root />'
- self.assertXMLEqual(xml1, xml2)
- def test_processing_instruction(self):
- xml1 = (
- '<?xml version="1.0"?>'
- '<?xml-model href="http://www.example1.com"?><root />'
- )
- xml2 = (
- '<?xml version="1.0"?>'
- '<?xml-model href="http://www.example2.com"?><root />'
- )
- self.assertXMLEqual(xml1, xml2)
- self.assertXMLEqual(
- '<?xml-stylesheet href="style1.xslt" type="text/xsl"?><root />',
- '<?xml-stylesheet href="style2.xslt" type="text/xsl"?><root />',
- )
- class SkippingExtraTests(TestCase):
- fixtures = ["should_not_be_loaded.json"]
- # HACK: This depends on internals of our TestCase subclasses
- def __call__(self, result=None):
- # Detect fixture loading by counting SQL queries, should be zero
- with self.assertNumQueries(0):
- super().__call__(result)
- @unittest.skip("Fixture loading should not be performed for skipped tests.")
- def test_fixtures_are_skipped(self):
- pass
- class AssertRaisesMsgTest(SimpleTestCase):
- def test_assert_raises_message(self):
- msg = "'Expected message' not found in 'Unexpected message'"
- # context manager form of assertRaisesMessage()
- with self.assertRaisesMessage(AssertionError, msg):
- with self.assertRaisesMessage(ValueError, "Expected message"):
- raise ValueError("Unexpected message")
- # callable form
- def func():
- raise ValueError("Unexpected message")
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertRaisesMessage(ValueError, "Expected message", func)
- def test_special_re_chars(self):
- """assertRaisesMessage shouldn't interpret RE special chars."""
- def func1():
- raise ValueError("[.*x+]y?")
- with self.assertRaisesMessage(ValueError, "[.*x+]y?"):
- func1()
- class AssertWarnsMessageTests(SimpleTestCase):
- def test_context_manager(self):
- with self.assertWarnsMessage(UserWarning, "Expected message"):
- warnings.warn("Expected message", UserWarning)
- def test_context_manager_failure(self):
- msg = "Expected message' not found in 'Unexpected message'"
- with self.assertRaisesMessage(AssertionError, msg):
- with self.assertWarnsMessage(UserWarning, "Expected message"):
- warnings.warn("Unexpected message", UserWarning)
- def test_callable(self):
- def func():
- warnings.warn("Expected message", UserWarning)
- self.assertWarnsMessage(UserWarning, "Expected message", func)
- def test_special_re_chars(self):
- def func1():
- warnings.warn("[.*x+]y?", UserWarning)
- with self.assertWarnsMessage(UserWarning, "[.*x+]y?"):
- func1()
- # TODO: Remove when dropping support for PY39.
- class AssertNoLogsTest(SimpleTestCase):
- @classmethod
- def setUpClass(cls):
- super().setUpClass()
- logging.config.dictConfig(DEFAULT_LOGGING)
- cls.addClassCleanup(logging.config.dictConfig, settings.LOGGING)
- def setUp(self):
- self.logger = logging.getLogger("django")
- @override_settings(DEBUG=True)
- def test_fails_when_log_emitted(self):
- msg = "Unexpected logs found: ['INFO:django:FAIL!']"
- with self.assertRaisesMessage(AssertionError, msg):
- with self.assertNoLogs("django", "INFO"):
- self.logger.info("FAIL!")
- @override_settings(DEBUG=True)
- def test_text_level(self):
- with self.assertNoLogs("django", "INFO"):
- self.logger.debug("DEBUG logs are ignored.")
- @override_settings(DEBUG=True)
- def test_int_level(self):
- with self.assertNoLogs("django", logging.INFO):
- self.logger.debug("DEBUG logs are ignored.")
- @override_settings(DEBUG=True)
- def test_default_level(self):
- with self.assertNoLogs("django"):
- self.logger.debug("DEBUG logs are ignored.")
- @override_settings(DEBUG=True)
- def test_does_not_hide_other_failures(self):
- msg = "1 != 2"
- with self.assertRaisesMessage(AssertionError, msg):
- with self.assertNoLogs("django"):
- self.assertEqual(1, 2)
- class AssertFieldOutputTests(SimpleTestCase):
- def test_assert_field_output(self):
- error_invalid = ["Enter a valid email address."]
- self.assertFieldOutput(
- EmailField, {"a@a.com": "a@a.com"}, {"aaa": error_invalid}
- )
- with self.assertRaises(AssertionError):
- self.assertFieldOutput(
- EmailField,
- {"a@a.com": "a@a.com"},
- {"aaa": error_invalid + ["Another error"]},
- )
- with self.assertRaises(AssertionError):
- self.assertFieldOutput(
- EmailField, {"a@a.com": "Wrong output"}, {"aaa": error_invalid}
- )
- with self.assertRaises(AssertionError):
- self.assertFieldOutput(
- EmailField,
- {"a@a.com": "a@a.com"},
- {"aaa": ["Come on, gimme some well formatted data, dude."]},
- )
- def test_custom_required_message(self):
- class MyCustomField(IntegerField):
- default_error_messages = {
- "required": "This is really required.",
- }
- self.assertFieldOutput(MyCustomField, {}, {}, empty_value=None)
- @override_settings(ROOT_URLCONF="test_utils.urls")
- class AssertURLEqualTests(SimpleTestCase):
- def test_equal(self):
- valid_tests = (
- ("http://example.com/?", "http://example.com/"),
- ("http://example.com/?x=1&", "http://example.com/?x=1"),
- ("http://example.com/?x=1&y=2", "http://example.com/?y=2&x=1"),
- ("http://example.com/?x=1&y=2", "http://example.com/?y=2&x=1"),
- (
- "http://example.com/?x=1&y=2&a=1&a=2",
- "http://example.com/?a=1&a=2&y=2&x=1",
- ),
- ("/path/to/?x=1&y=2&z=3", "/path/to/?z=3&y=2&x=1"),
- ("?x=1&y=2&z=3", "?z=3&y=2&x=1"),
- ("/test_utils/no_template_used/", reverse_lazy("no_template_used")),
- )
- for url1, url2 in valid_tests:
- with self.subTest(url=url1):
- self.assertURLEqual(url1, url2)
- def test_not_equal(self):
- invalid_tests = (
- # Protocol must be the same.
- ("http://example.com/", "https://example.com/"),
- ("http://example.com/?x=1&x=2", "https://example.com/?x=2&x=1"),
- ("http://example.com/?x=1&y=bar&x=2", "https://example.com/?y=bar&x=2&x=1"),
- # Parameters of the same name must be in the same order.
- ("/path/to?a=1&a=2", "/path/to/?a=2&a=1"),
- )
- for url1, url2 in invalid_tests:
- with self.subTest(url=url1), self.assertRaises(AssertionError):
- self.assertURLEqual(url1, url2)
- def test_message(self):
- msg = (
- "Expected 'http://example.com/?x=1&x=2' to equal "
- "'https://example.com/?x=2&x=1'"
- )
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertURLEqual(
- "http://example.com/?x=1&x=2", "https://example.com/?x=2&x=1"
- )
- def test_msg_prefix(self):
- msg = (
- "Prefix: Expected 'http://example.com/?x=1&x=2' to equal "
- "'https://example.com/?x=2&x=1'"
- )
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertURLEqual(
- "http://example.com/?x=1&x=2",
- "https://example.com/?x=2&x=1",
- msg_prefix="Prefix: ",
- )
- class TestForm(Form):
- field = CharField()
- def clean_field(self):
- value = self.cleaned_data.get("field", "")
- if value == "invalid":
- raise ValidationError("invalid value")
- return value
- def clean(self):
- if self.cleaned_data.get("field") == "invalid_non_field":
- raise ValidationError("non-field error")
- return self.cleaned_data
- @classmethod
- def _get_cleaned_form(cls, field_value):
- form = cls({"field": field_value})
- form.full_clean()
- return form
- @classmethod
- def valid(cls):
- return cls._get_cleaned_form("valid")
- @classmethod
- def invalid(cls, nonfield=False):
- return cls._get_cleaned_form("invalid_non_field" if nonfield else "invalid")
- class TestFormset(formset_factory(TestForm)):
- @classmethod
- def _get_cleaned_formset(cls, field_value):
- formset = cls(
- {
- "form-TOTAL_FORMS": "1",
- "form-INITIAL_FORMS": "0",
- "form-0-field": field_value,
- }
- )
- formset.full_clean()
- return formset
- @classmethod
- def valid(cls):
- return cls._get_cleaned_formset("valid")
- @classmethod
- def invalid(cls, nonfield=False, nonform=False):
- if nonform:
- formset = cls({}, error_messages={"missing_management_form": "error"})
- formset.full_clean()
- return formset
- return cls._get_cleaned_formset("invalid_non_field" if nonfield else "invalid")
- class AssertFormErrorTests(SimpleTestCase):
- @ignore_warnings(category=RemovedInDjango50Warning)
- def test_non_client_response(self):
- msg = (
- "assertFormError() is only usable on responses fetched using the "
- "Django test Client."
- )
- response = HttpResponse()
- with self.assertRaisesMessage(ValueError, msg):
- self.assertFormError(response, "form", "field", "invalid value")
- @ignore_warnings(category=RemovedInDjango50Warning)
- def test_response_with_no_context(self):
- msg = "Response did not use any contexts to render the response"
- response = mock.Mock(context=[])
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertFormError(response, "form", "field", "invalid value")
- msg_prefix = "Custom prefix"
- with self.assertRaisesMessage(AssertionError, f"{msg_prefix}: {msg}"):
- self.assertFormError(
- response,
- "form",
- "field",
- "invalid value",
- msg_prefix=msg_prefix,
- )
- @ignore_warnings(category=RemovedInDjango50Warning)
- def test_form_not_in_context(self):
- msg = "The form 'form' was not used to render the response"
- response = mock.Mock(context=[{}])
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertFormError(response, "form", "field", "invalid value")
- msg_prefix = "Custom prefix"
- with self.assertRaisesMessage(AssertionError, f"{msg_prefix}: {msg}"):
- self.assertFormError(
- response, "form", "field", "invalid value", msg_prefix=msg_prefix
- )
- def test_single_error(self):
- self.assertFormError(TestForm.invalid(), "field", "invalid value")
- def test_error_list(self):
- self.assertFormError(TestForm.invalid(), "field", ["invalid value"])
- def test_empty_errors_valid_form(self):
- self.assertFormError(TestForm.valid(), "field", [])
- def test_empty_errors_valid_form_non_field_errors(self):
- self.assertFormError(TestForm.valid(), None, [])
- def test_field_not_in_form(self):
- msg = (
- "The form <TestForm bound=True, valid=False, fields=(field)> does not "
- "contain the field 'other_field'."
- )
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertFormError(TestForm.invalid(), "other_field", "invalid value")
- msg_prefix = "Custom prefix"
- with self.assertRaisesMessage(AssertionError, f"{msg_prefix}: {msg}"):
- self.assertFormError(
- TestForm.invalid(),
- "other_field",
- "invalid value",
- msg_prefix=msg_prefix,
- )
- def test_field_with_no_errors(self):
- msg = (
- "The errors of field 'field' on form <TestForm bound=True, valid=True, "
- "fields=(field)> don't match."
- )
- with self.assertRaisesMessage(AssertionError, msg) as ctx:
- self.assertFormError(TestForm.valid(), "field", "invalid value")
- self.assertIn("[] != ['invalid value']", str(ctx.exception))
- msg_prefix = "Custom prefix"
- with self.assertRaisesMessage(AssertionError, f"{msg_prefix}: {msg}"):
- self.assertFormError(
- TestForm.valid(), "field", "invalid value", msg_prefix=msg_prefix
- )
- def test_field_with_different_error(self):
- msg = (
- "The errors of field 'field' on form <TestForm bound=True, valid=False, "
- "fields=(field)> don't match."
- )
- with self.assertRaisesMessage(AssertionError, msg) as ctx:
- self.assertFormError(TestForm.invalid(), "field", "other error")
- self.assertIn("['invalid value'] != ['other error']", str(ctx.exception))
- msg_prefix = "Custom prefix"
- with self.assertRaisesMessage(AssertionError, f"{msg_prefix}: {msg}"):
- self.assertFormError(
- TestForm.invalid(), "field", "other error", msg_prefix=msg_prefix
- )
- def test_unbound_form(self):
- msg = (
- "The form <TestForm bound=False, valid=Unknown, fields=(field)> is not "
- "bound, it will never have any errors."
- )
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertFormError(TestForm(), "field", [])
- msg_prefix = "Custom prefix"
- with self.assertRaisesMessage(AssertionError, f"{msg_prefix}: {msg}"):
- self.assertFormError(TestForm(), "field", [], msg_prefix=msg_prefix)
- def test_empty_errors_invalid_form(self):
- msg = (
- "The errors of field 'field' on form <TestForm bound=True, valid=False, "
- "fields=(field)> don't match."
- )
- with self.assertRaisesMessage(AssertionError, msg) as ctx:
- self.assertFormError(TestForm.invalid(), "field", [])
- self.assertIn("['invalid value'] != []", str(ctx.exception))
- def test_non_field_errors(self):
- self.assertFormError(TestForm.invalid(nonfield=True), None, "non-field error")
- def test_different_non_field_errors(self):
- msg = (
- "The non-field errors of form <TestForm bound=True, valid=False, "
- "fields=(field)> don't match."
- )
- with self.assertRaisesMessage(AssertionError, msg) as ctx:
- self.assertFormError(
- TestForm.invalid(nonfield=True), None, "other non-field error"
- )
- self.assertIn(
- "['non-field error'] != ['other non-field error']", str(ctx.exception)
- )
- msg_prefix = "Custom prefix"
- with self.assertRaisesMessage(AssertionError, f"{msg_prefix}: {msg}"):
- self.assertFormError(
- TestForm.invalid(nonfield=True),
- None,
- "other non-field error",
- msg_prefix=msg_prefix,
- )
- class AssertFormsetErrorTests(SimpleTestCase):
- @ignore_warnings(category=RemovedInDjango50Warning)
- def test_non_client_response(self):
- msg = (
- "assertFormsetError() is only usable on responses fetched using "
- "the Django test Client."
- )
- response = HttpResponse()
- with self.assertRaisesMessage(ValueError, msg):
- self.assertFormsetError(response, "formset", 0, "field", "invalid value")
- @ignore_warnings(category=RemovedInDjango50Warning)
- def test_response_with_no_context(self):
- msg = "Response did not use any contexts to render the response"
- response = mock.Mock(context=[])
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertFormsetError(response, "formset", 0, "field", "invalid value")
- @ignore_warnings(category=RemovedInDjango50Warning)
- def test_formset_not_in_context(self):
- msg = "The formset 'formset' was not used to render the response"
- response = mock.Mock(context=[{}])
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertFormsetError(response, "formset", 0, "field", "invalid value")
- msg_prefix = "Custom prefix"
- with self.assertRaisesMessage(AssertionError, f"{msg_prefix}: {msg}"):
- self.assertFormsetError(
- response, "formset", 0, "field", "invalid value", msg_prefix=msg_prefix
- )
- def test_single_error(self):
- self.assertFormsetError(TestFormset.invalid(), 0, "field", "invalid value")
- def test_error_list(self):
- self.assertFormsetError(TestFormset.invalid(), 0, "field", ["invalid value"])
- def test_empty_errors_valid_formset(self):
- self.assertFormsetError(TestFormset.valid(), 0, "field", [])
- def test_multiple_forms(self):
- formset = TestFormset(
- {
- "form-TOTAL_FORMS": "2",
- "form-INITIAL_FORMS": "0",
- "form-0-field": "valid",
- "form-1-field": "invalid",
- }
- )
- formset.full_clean()
- self.assertFormsetError(formset, 0, "field", [])
- self.assertFormsetError(formset, 1, "field", ["invalid value"])
- def test_field_not_in_form(self):
- msg = (
- "The form 0 of formset <TestFormset: bound=True valid=False total_forms=1> "
- "does not contain the field 'other_field'."
- )
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertFormsetError(
- TestFormset.invalid(), 0, "other_field", "invalid value"
- )
- msg_prefix = "Custom prefix"
- with self.assertRaisesMessage(AssertionError, f"{msg_prefix}: {msg}"):
- self.assertFormsetError(
- TestFormset.invalid(),
- 0,
- "other_field",
- "invalid value",
- msg_prefix=msg_prefix,
- )
- def test_field_with_no_errors(self):
- msg = (
- "The errors of field 'field' on form 0 of formset <TestFormset: bound=True "
- "valid=True total_forms=1> don't match."
- )
- with self.assertRaisesMessage(AssertionError, msg) as ctx:
- self.assertFormsetError(TestFormset.valid(), 0, "field", "invalid value")
- self.assertIn("[] != ['invalid value']", str(ctx.exception))
- msg_prefix = "Custom prefix"
- with self.assertRaisesMessage(AssertionError, f"{msg_prefix}: {msg}"):
- self.assertFormsetError(
- TestFormset.valid(), 0, "field", "invalid value", msg_prefix=msg_prefix
- )
- def test_field_with_different_error(self):
- msg = (
- "The errors of field 'field' on form 0 of formset <TestFormset: bound=True "
- "valid=False total_forms=1> don't match."
- )
- with self.assertRaisesMessage(AssertionError, msg) as ctx:
- self.assertFormsetError(TestFormset.invalid(), 0, "field", "other error")
- self.assertIn("['invalid value'] != ['other error']", str(ctx.exception))
- msg_prefix = "Custom prefix"
- with self.assertRaisesMessage(AssertionError, f"{msg_prefix}: {msg}"):
- self.assertFormsetError(
- TestFormset.invalid(), 0, "field", "other error", msg_prefix=msg_prefix
- )
- def test_unbound_formset(self):
- msg = (
- "The formset <TestFormset: bound=False valid=Unknown total_forms=1> is not "
- "bound, it will never have any errors."
- )
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertFormsetError(TestFormset(), 0, "field", [])
- def test_empty_errors_invalid_formset(self):
- msg = (
- "The errors of field 'field' on form 0 of formset <TestFormset: bound=True "
- "valid=False total_forms=1> don't match."
- )
- with self.assertRaisesMessage(AssertionError, msg) as ctx:
- self.assertFormsetError(TestFormset.invalid(), 0, "field", [])
- self.assertIn("['invalid value'] != []", str(ctx.exception))
- def test_non_field_errors(self):
- self.assertFormsetError(
- TestFormset.invalid(nonfield=True), 0, None, "non-field error"
- )
- def test_different_non_field_errors(self):
- msg = (
- "The non-field errors of form 0 of formset <TestFormset: bound=True "
- "valid=False total_forms=1> don't match."
- )
- with self.assertRaisesMessage(AssertionError, msg) as ctx:
- self.assertFormsetError(
- TestFormset.invalid(nonfield=True), 0, None, "other non-field error"
- )
- self.assertIn(
- "['non-field error'] != ['other non-field error']", str(ctx.exception)
- )
- msg_prefix = "Custom prefix"
- with self.assertRaisesMessage(AssertionError, f"{msg_prefix}: {msg}"):
- self.assertFormsetError(
- TestFormset.invalid(nonfield=True),
- 0,
- None,
- "other non-field error",
- msg_prefix=msg_prefix,
- )
- def test_no_non_field_errors(self):
- msg = (
- "The non-field errors of form 0 of formset <TestFormset: bound=True "
- "valid=False total_forms=1> don't match."
- )
- with self.assertRaisesMessage(AssertionError, msg) as ctx:
- self.assertFormsetError(TestFormset.invalid(), 0, None, "non-field error")
- self.assertIn("[] != ['non-field error']", str(ctx.exception))
- msg_prefix = "Custom prefix"
- with self.assertRaisesMessage(AssertionError, f"{msg_prefix}: {msg}"):
- self.assertFormsetError(
- TestFormset.invalid(), 0, None, "non-field error", msg_prefix=msg_prefix
- )
- def test_non_form_errors(self):
- self.assertFormsetError(TestFormset.invalid(nonform=True), None, None, "error")
- def test_different_non_form_errors(self):
- msg = (
- "The non-form errors of formset <TestFormset: bound=True valid=False "
- "total_forms=0> don't match."
- )
- with self.assertRaisesMessage(AssertionError, msg) as ctx:
- self.assertFormsetError(
- TestFormset.invalid(nonform=True), None, None, "other error"
- )
- self.assertIn("['error'] != ['other error']", str(ctx.exception))
- msg_prefix = "Custom prefix"
- with self.assertRaisesMessage(AssertionError, f"{msg_prefix}: {msg}"):
- self.assertFormsetError(
- TestFormset.invalid(nonform=True),
- None,
- None,
- "other error",
- msg_prefix=msg_prefix,
- )
- def test_no_non_form_errors(self):
- msg = (
- "The non-form errors of formset <TestFormset: bound=True valid=False "
- "total_forms=1> don't match."
- )
- with self.assertRaisesMessage(AssertionError, msg) as ctx:
- self.assertFormsetError(TestFormset.invalid(), None, None, "error")
- self.assertIn("[] != ['error']", str(ctx.exception))
- msg_prefix = "Custom prefix"
- with self.assertRaisesMessage(AssertionError, f"{msg_prefix}: {msg}"):
- self.assertFormsetError(
- TestFormset.invalid(),
- None,
- None,
- "error",
- msg_prefix=msg_prefix,
- )
- def test_non_form_errors_with_field(self):
- msg = "You must use field=None with form_index=None."
- with self.assertRaisesMessage(ValueError, msg):
- self.assertFormsetError(
- TestFormset.invalid(nonform=True), None, "field", "error"
- )
- def test_form_index_too_big(self):
- msg = (
- "The formset <TestFormset: bound=True valid=False total_forms=1> only has "
- "1 form."
- )
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertFormsetError(TestFormset.invalid(), 2, "field", "error")
- def test_form_index_too_big_plural(self):
- formset = TestFormset(
- {
- "form-TOTAL_FORMS": "2",
- "form-INITIAL_FORMS": "0",
- "form-0-field": "valid",
- "form-1-field": "valid",
- }
- )
- formset.full_clean()
- msg = (
- "The formset <TestFormset: bound=True valid=True total_forms=2> only has 2 "
- "forms."
- )
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertFormsetError(formset, 2, "field", "error")
- # RemovedInDjango50Warning
- class AssertFormErrorDeprecationTests(SimpleTestCase):
- """
- Exhaustively test all possible combinations of args/kwargs for the old
- signature.
- """
- @ignore_warnings(category=RemovedInDjango50Warning)
- def test_assert_form_error_errors_none(self):
- msg = (
- "The errors of field 'field' on form <TestForm bound=True, valid=False, "
- "fields=(field)> don't match."
- )
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertFormError(TestForm.invalid(), "field", None)
- def test_assert_form_error_errors_none_warning(self):
- msg = (
- "Passing errors=None to assertFormError() is deprecated, use "
- "errors=[] instead."
- )
- with self.assertWarnsMessage(RemovedInDjango50Warning, msg):
- self.assertFormError(TestForm.valid(), "field", None)
- def _assert_form_error_old_api_cases(self, form, field, errors, msg_prefix):
- response = mock.Mock(context=[{"form": TestForm.invalid()}])
- return (
- ((response, form, field, errors), {}),
- ((response, form, field, errors, msg_prefix), {}),
- ((response, form, field, errors), {"msg_prefix": msg_prefix}),
- ((response, form, field), {"errors": errors}),
- ((response, form, field), {"errors": errors, "msg_prefix": msg_prefix}),
- ((response, form), {"field": field, "errors": errors}),
- (
- (response, form),
- {"field": field, "errors": errors, "msg_prefix": msg_prefix},
- ),
- ((response,), {"form": form, "field": field, "errors": errors}),
- (
- (response,),
- {
- "form": form,
- "field": field,
- "errors": errors,
- "msg_prefix": msg_prefix,
- },
- ),
- (
- (),
- {"response": response, "form": form, "field": field, "errors": errors},
- ),
- (
- (),
- {
- "response": response,
- "form": form,
- "field": field,
- "errors": errors,
- "msg_prefix": msg_prefix,
- },
- ),
- )
- def test_assert_form_error_old_api(self):
- deprecation_msg = (
- "Passing response to assertFormError() is deprecated. Use the form object "
- "directly: assertFormError(response.context['form'], 'field', ...)"
- )
- for args, kwargs in self._assert_form_error_old_api_cases(
- form="form",
- field="field",
- errors=["invalid value"],
- msg_prefix="Custom prefix",
- ):
- with self.subTest(args=args, kwargs=kwargs):
- with self.assertWarnsMessage(RemovedInDjango50Warning, deprecation_msg):
- self.assertFormError(*args, **kwargs)
- @ignore_warnings(category=RemovedInDjango50Warning)
- def test_assert_form_error_old_api_assertion_error(self):
- for args, kwargs in self._assert_form_error_old_api_cases(
- form="form",
- field="field",
- errors=["other error"],
- msg_prefix="Custom prefix",
- ):
- with self.subTest(args=args, kwargs=kwargs):
- with self.assertRaises(AssertionError):
- self.assertFormError(*args, **kwargs)
- @ignore_warnings(category=RemovedInDjango50Warning)
- def test_assert_formset_error_errors_none(self):
- msg = (
- "The errors of field 'field' on form 0 of formset <TestFormset: bound=True "
- "valid=False total_forms=1> don't match."
- )
- with self.assertRaisesMessage(AssertionError, msg):
- self.assertFormsetError(TestFormset.invalid(), 0, "field", None)
- def test_assert_formset_error_errors_none_warning(self):
- msg = (
- "Passing errors=None to assertFormsetError() is deprecated, use "
- "errors=[] instead."
- )
- with self.assertWarnsMessage(RemovedInDjango50Warning, msg):
- self.assertFormsetError(TestFormset.valid(), 0, "field", None)
- def _assert_formset_error_old_api_cases(
- self, formset, form_index, field, errors, msg_prefix
- ):
- response = mock.Mock(context=[{"formset": TestFormset.invalid()}])
- return (
- ((response, formset, form_index, field, errors), {}),
- ((response, formset, form_index, field, errors, msg_prefix), {}),
- (
- (response, formset, form_index, field, errors),
- {"msg_prefix": msg_prefix},
- ),
- ((response, formset, form_index, field), {"errors": errors}),
- (
- (response, formset, form_index, field),
- {"errors": errors, "msg_prefix": msg_prefix},
- ),
- ((response, formset, form_index), {"field": field, "errors": errors}),
- (
- (response, formset, form_index),
- {"field": field, "errors": errors, "msg_prefix": msg_prefix},
- ),
- (
- (response, formset),
- {"form_index": form_index, "field": field, "errors": errors},
- ),
- (
- (response, formset),
- {
- "form_index": form_index,
- "field": field,
- "errors": errors,
- "msg_prefix": msg_prefix,
- },
- ),
- (
- (response,),
- {
- "formset": formset,
- "form_index": form_index,
- "field": field,
- "errors": errors,
- },
- ),
- (
- (response,),
- {
- "formset": formset,
- "form_index": form_index,
- "field": field,
- "errors": errors,
- "msg_prefix": msg_prefix,
- },
- ),
- (
- (),
- {
- "response": response,
- "formset": formset,
- "form_index": form_index,
- "field": field,
- "errors": errors,
- },
- ),
- (
- (),
- {
- "response": response,
- "formset": formset,
- "form_index": form_index,
- "field": field,
- "errors": errors,
- "msg_prefix": msg_prefix,
- },
- ),
- )
- def test_assert_formset_error_old_api(self):
- deprecation_msg = (
- "Passing response to assertFormsetError() is deprecated. Use the formset "
- "object directly: assertFormsetError(response.context['formset'], 0, ...)"
- )
- for args, kwargs in self._assert_formset_error_old_api_cases(
- formset="formset",
- form_index=0,
- field="field",
- errors=["invalid value"],
- msg_prefix="Custom prefix",
- ):
- with self.subTest(args=args, kwargs=kwargs):
- with self.assertWarnsMessage(RemovedInDjango50Warning, deprecation_msg):
- self.assertFormsetError(*args, **kwargs)
- @ignore_warnings(category=RemovedInDjango50Warning)
- def test_assert_formset_error_old_api_assertion_error(self):
- for args, kwargs in self._assert_formset_error_old_api_cases(
- formset="formset",
- form_index=0,
- field="field",
- errors=["other error"],
- msg_prefix="Custom prefix",
- ):
- with self.subTest(args=args, kwargs=kwargs):
- with self.assertRaises(AssertionError):
- self.assertFormsetError(*args, **kwargs)
- class FirstUrls:
- urlpatterns = [path("first/", empty_response, name="first")]
- class SecondUrls:
- urlpatterns = [path("second/", empty_response, name="second")]
- class SetupTestEnvironmentTests(SimpleTestCase):
- def test_setup_test_environment_calling_more_than_once(self):
- with self.assertRaisesMessage(
- RuntimeError, "setup_test_environment() was already called"
- ):
- setup_test_environment()
- def test_allowed_hosts(self):
- for type_ in (list, tuple):
- with self.subTest(type_=type_):
- allowed_hosts = type_("*")
- with mock.patch("django.test.utils._TestState") as x:
- del x.saved_data
- with self.settings(ALLOWED_HOSTS=allowed_hosts):
- setup_test_environment()
- self.assertEqual(settings.ALLOWED_HOSTS, ["*", "testserver"])
- class OverrideSettingsTests(SimpleTestCase):
- # #21518 -- If neither override_settings nor a setting_changed receiver
- # clears the URL cache between tests, then one of test_first or
- # test_second will fail.
- @override_settings(ROOT_URLCONF=FirstUrls)
- def test_urlconf_first(self):
- reverse("first")
- @override_settings(ROOT_URLCONF=SecondUrls)
- def test_urlconf_second(self):
- reverse("second")
- def test_urlconf_cache(self):
- with self.assertRaises(NoReverseMatch):
- reverse("first")
- with self.assertRaises(NoReverseMatch):
- reverse("second")
- with override_settings(ROOT_URLCONF=FirstUrls):
- self.client.get(reverse("first"))
- with self.assertRaises(NoReverseMatch):
- reverse("second")
- with override_settings(ROOT_URLCONF=SecondUrls):
- with self.assertRaises(NoReverseMatch):
- reverse("first")
- self.client.get(reverse("second"))
- self.client.get(reverse("first"))
- with self.assertRaises(NoReverseMatch):
- reverse("second")
- with self.assertRaises(NoReverseMatch):
- reverse("first")
- with self.assertRaises(NoReverseMatch):
- reverse("second")
- def test_override_media_root(self):
- """
- Overriding the MEDIA_ROOT setting should be reflected in the
- base_location attribute of django.core.files.storage.default_storage.
- """
- self.assertEqual(default_storage.base_location, "")
- with self.settings(MEDIA_ROOT="test_value"):
- self.assertEqual(default_storage.base_location, "test_value")
- def test_override_media_url(self):
- """
- Overriding the MEDIA_URL setting should be reflected in the
- base_url attribute of django.core.files.storage.default_storage.
- """
- self.assertEqual(default_storage.base_location, "")
- with self.settings(MEDIA_URL="/test_value/"):
- self.assertEqual(default_storage.base_url, "/test_value/")
- def test_override_file_upload_permissions(self):
- """
- Overriding the FILE_UPLOAD_PERMISSIONS setting should be reflected in
- the file_permissions_mode attribute of
- django.core.files.storage.default_storage.
- """
- self.assertEqual(default_storage.file_permissions_mode, 0o644)
- with self.settings(FILE_UPLOAD_PERMISSIONS=0o777):
- self.assertEqual(default_storage.file_permissions_mode, 0o777)
- def test_override_file_upload_directory_permissions(self):
- """
- Overriding the FILE_UPLOAD_DIRECTORY_PERMISSIONS setting should be
- reflected in the directory_permissions_mode attribute of
- django.core.files.storage.default_storage.
- """
- self.assertIsNone(default_storage.directory_permissions_mode)
- with self.settings(FILE_UPLOAD_DIRECTORY_PERMISSIONS=0o777):
- self.assertEqual(default_storage.directory_permissions_mode, 0o777)
- def test_override_database_routers(self):
- """
- Overriding DATABASE_ROUTERS should update the base router.
- """
- test_routers = [object()]
- with self.settings(DATABASE_ROUTERS=test_routers):
- self.assertEqual(router.routers, test_routers)
- def test_override_static_url(self):
- """
- Overriding the STATIC_URL setting should be reflected in the
- base_url attribute of
- django.contrib.staticfiles.storage.staticfiles_storage.
- """
- with self.settings(STATIC_URL="/test/"):
- self.assertEqual(staticfiles_storage.base_url, "/test/")
- def test_override_static_root(self):
- """
- Overriding the STATIC_ROOT setting should be reflected in the
- location attribute of
- django.contrib.staticfiles.storage.staticfiles_storage.
- """
- with self.settings(STATIC_ROOT="/tmp/test"):
- self.assertEqual(staticfiles_storage.location, os.path.abspath("/tmp/test"))
- def test_override_staticfiles_storage(self):
- """
- Overriding the STATICFILES_STORAGE setting should be reflected in
- the value of django.contrib.staticfiles.storage.staticfiles_storage.
- """
- new_class = "ManifestStaticFilesStorage"
- new_storage = "django.contrib.staticfiles.storage." + new_class
- with self.settings(STATICFILES_STORAGE=new_storage):
- self.assertEqual(staticfiles_storage.__class__.__name__, new_class)
- def test_override_staticfiles_finders(self):
- """
- Overriding the STATICFILES_FINDERS setting should be reflected in
- the return value of django.contrib.staticfiles.finders.get_finders.
- """
- current = get_finders()
- self.assertGreater(len(list(current)), 1)
- finders = ["django.contrib.staticfiles.finders.FileSystemFinder"]
- with self.settings(STATICFILES_FINDERS=finders):
- self.assertEqual(len(list(get_finders())), len(finders))
- def test_override_staticfiles_dirs(self):
- """
- Overriding the STATICFILES_DIRS setting should be reflected in
- the locations attribute of the
- django.contrib.staticfiles.finders.FileSystemFinder instance.
- """
- finder = get_finder("django.contrib.staticfiles.finders.FileSystemFinder")
- test_path = "/tmp/test"
- expected_location = ("", test_path)
- self.assertNotIn(expected_location, finder.locations)
- with self.settings(STATICFILES_DIRS=[test_path]):
- finder = get_finder("django.contrib.staticfiles.finders.FileSystemFinder")
- self.assertIn(expected_location, finder.locations)
- @skipUnlessDBFeature("supports_transactions")
- class TestBadSetUpTestData(TestCase):
- """
- An exception in setUpTestData() shouldn't leak a transaction which would
- cascade across the rest of the test suite.
- """
- class MyException(Exception):
- pass
- @classmethod
- def setUpClass(cls):
- try:
- super().setUpClass()
- except cls.MyException:
- cls._in_atomic_block = connection.in_atomic_block
- @classmethod
- def tearDownClass(Cls):
- # override to avoid a second cls._rollback_atomics() which would fail.
- # Normal setUpClass() methods won't have exception handling so this
- # method wouldn't typically be run.
- pass
- @classmethod
- def setUpTestData(cls):
- # Simulate a broken setUpTestData() method.
- raise cls.MyException()
- def test_failure_in_setUpTestData_should_rollback_transaction(self):
- # setUpTestData() should call _rollback_atomics() so that the
- # transaction doesn't leak.
- self.assertFalse(self._in_atomic_block)
- @skipUnlessDBFeature("supports_transactions")
- class CaptureOnCommitCallbacksTests(TestCase):
- databases = {"default", "other"}
- callback_called = False
- def enqueue_callback(self, using="default"):
- def hook():
- self.callback_called = True
- transaction.on_commit(hook, using=using)
- def test_no_arguments(self):
- with self.captureOnCommitCallbacks() as callbacks:
- self.enqueue_callback()
- self.assertEqual(len(callbacks), 1)
- self.assertIs(self.callback_called, False)
- callbacks[0]()
- self.assertIs(self.callback_called, True)
- def test_using(self):
- with self.captureOnCommitCallbacks(using="other") as callbacks:
- self.enqueue_callback(using="other")
- self.assertEqual(len(callbacks), 1)
- self.assertIs(self.callback_called, False)
- callbacks[0]()
- self.assertIs(self.callback_called, True)
- def test_different_using(self):
- with self.captureOnCommitCallbacks(using="default") as callbacks:
- self.enqueue_callback(using="other")
- self.assertEqual(callbacks, [])
- def test_execute(self):
- with self.captureOnCommitCallbacks(execute=True) as callbacks:
- self.enqueue_callback()
- self.assertEqual(len(callbacks), 1)
- self.assertIs(self.callback_called, True)
- def test_pre_callback(self):
- def pre_hook():
- pass
- transaction.on_commit(pre_hook, using="default")
- with self.captureOnCommitCallbacks() as callbacks:
- self.enqueue_callback()
- self.assertEqual(len(callbacks), 1)
- self.assertNotEqual(callbacks[0], pre_hook)
- def test_with_rolled_back_savepoint(self):
- with self.captureOnCommitCallbacks() as callbacks:
- try:
- with transaction.atomic():
- self.enqueue_callback()
- raise IntegrityError
- except IntegrityError:
- # Inner transaction.atomic() has been rolled back.
- pass
- self.assertEqual(callbacks, [])
- def test_execute_recursive(self):
- with self.captureOnCommitCallbacks(execute=True) as callbacks:
- transaction.on_commit(self.enqueue_callback)
- self.assertEqual(len(callbacks), 2)
- self.assertIs(self.callback_called, True)
- def test_execute_tree(self):
- """
- A visualisation of the callback tree tested. Each node is expected to
- be visited only once:
- └─branch_1
- ├─branch_2
- │ ├─leaf_1
- │ └─leaf_2
- └─leaf_3
- """
- branch_1_call_counter = 0
- branch_2_call_counter = 0
- leaf_1_call_counter = 0
- leaf_2_call_counter = 0
- leaf_3_call_counter = 0
- def leaf_1():
- nonlocal leaf_1_call_counter
- leaf_1_call_counter += 1
- def leaf_2():
- nonlocal leaf_2_call_counter
- leaf_2_call_counter += 1
- def leaf_3():
- nonlocal leaf_3_call_counter
- leaf_3_call_counter += 1
- def branch_1():
- nonlocal branch_1_call_counter
- branch_1_call_counter += 1
- transaction.on_commit(branch_2)
- transaction.on_commit(leaf_3)
- def branch_2():
- nonlocal branch_2_call_counter
- branch_2_call_counter += 1
- transaction.on_commit(leaf_1)
- transaction.on_commit(leaf_2)
- with self.captureOnCommitCallbacks(execute=True) as callbacks:
- transaction.on_commit(branch_1)
- self.assertEqual(branch_1_call_counter, 1)
- self.assertEqual(branch_2_call_counter, 1)
- self.assertEqual(leaf_1_call_counter, 1)
- self.assertEqual(leaf_2_call_counter, 1)
- self.assertEqual(leaf_3_call_counter, 1)
- self.assertEqual(callbacks, [branch_1, branch_2, leaf_3, leaf_1, leaf_2])
- def test_execute_robust(self):
- class MyException(Exception):
- pass
- def hook():
- self.callback_called = True
- raise MyException("robust callback")
- with self.assertLogs("django.test", "ERROR") as cm:
- with self.captureOnCommitCallbacks(execute=True) as callbacks:
- transaction.on_commit(hook, robust=True)
- self.assertEqual(len(callbacks), 1)
- self.assertIs(self.callback_called, True)
- log_record = cm.records[0]
- self.assertEqual(
- log_record.getMessage(),
- "Error calling CaptureOnCommitCallbacksTests.test_execute_robust.<locals>."
- "hook in on_commit() (robust callback).",
- )
- self.assertIsNotNone(log_record.exc_info)
- raised_exception = log_record.exc_info[1]
- self.assertIsInstance(raised_exception, MyException)
- self.assertEqual(str(raised_exception), "robust callback")
- class DisallowedDatabaseQueriesTests(SimpleTestCase):
- def test_disallowed_database_connections(self):
- expected_message = (
- "Database connections to 'default' are not allowed in SimpleTestCase "
- "subclasses. Either subclass TestCase or TransactionTestCase to "
- "ensure proper test isolation or add 'default' to "
- "test_utils.tests.DisallowedDatabaseQueriesTests.databases to "
- "silence this failure."
- )
- with self.assertRaisesMessage(DatabaseOperationForbidden, expected_message):
- connection.connect()
- with self.assertRaisesMessage(DatabaseOperationForbidden, expected_message):
- connection.temporary_connection()
- def test_disallowed_database_queries(self):
- expected_message = (
- "Database queries to 'default' are not allowed in SimpleTestCase "
- "subclasses. Either subclass TestCase or TransactionTestCase to "
- "ensure proper test isolation or add 'default' to "
- "test_utils.tests.DisallowedDatabaseQueriesTests.databases to "
- "silence this failure."
- )
- with self.assertRaisesMessage(DatabaseOperationForbidden, expected_message):
- Car.objects.first()
- def test_disallowed_database_chunked_cursor_queries(self):
- expected_message = (
- "Database queries to 'default' are not allowed in SimpleTestCase "
- "subclasses. Either subclass TestCase or TransactionTestCase to "
- "ensure proper test isolation or add 'default' to "
- "test_utils.tests.DisallowedDatabaseQueriesTests.databases to "
- "silence this failure."
- )
- with self.assertRaisesMessage(DatabaseOperationForbidden, expected_message):
- next(Car.objects.iterator())
- class AllowedDatabaseQueriesTests(SimpleTestCase):
- databases = {"default"}
- def test_allowed_database_queries(self):
- Car.objects.first()
- def test_allowed_database_chunked_cursor_queries(self):
- next(Car.objects.iterator(), None)
- class DatabaseAliasTests(SimpleTestCase):
- def setUp(self):
- self.addCleanup(setattr, self.__class__, "databases", self.databases)
- def test_no_close_match(self):
- self.__class__.databases = {"void"}
- message = (
- "test_utils.tests.DatabaseAliasTests.databases refers to 'void' which is "
- "not defined in settings.DATABASES."
- )
- with self.assertRaisesMessage(ImproperlyConfigured, message):
- self._validate_databases()
- def test_close_match(self):
- self.__class__.databases = {"defualt"}
- message = (
- "test_utils.tests.DatabaseAliasTests.databases refers to 'defualt' which "
- "is not defined in settings.DATABASES. Did you mean 'default'?"
- )
- with self.assertRaisesMessage(ImproperlyConfigured, message):
- self._validate_databases()
- def test_match(self):
- self.__class__.databases = {"default", "other"}
- self.assertEqual(self._validate_databases(), frozenset({"default", "other"}))
- def test_all(self):
- self.__class__.databases = "__all__"
- self.assertEqual(self._validate_databases(), frozenset(connections))
- @isolate_apps("test_utils", attr_name="class_apps")
- class IsolatedAppsTests(SimpleTestCase):
- def test_installed_apps(self):
- self.assertEqual(
- [app_config.label for app_config in self.class_apps.get_app_configs()],
- ["test_utils"],
- )
- def test_class_decoration(self):
- class ClassDecoration(models.Model):
- pass
- self.assertEqual(ClassDecoration._meta.apps, self.class_apps)
- @isolate_apps("test_utils", kwarg_name="method_apps")
- def test_method_decoration(self, method_apps):
- class MethodDecoration(models.Model):
- pass
- self.assertEqual(MethodDecoration._meta.apps, method_apps)
- def test_context_manager(self):
- with isolate_apps("test_utils") as context_apps:
- class ContextManager(models.Model):
- pass
- self.assertEqual(ContextManager._meta.apps, context_apps)
- @isolate_apps("test_utils", kwarg_name="method_apps")
- def test_nested(self, method_apps):
- class MethodDecoration(models.Model):
- pass
- with isolate_apps("test_utils") as context_apps:
- class ContextManager(models.Model):
- pass
- with isolate_apps("test_utils") as nested_context_apps:
- class NestedContextManager(models.Model):
- pass
- self.assertEqual(MethodDecoration._meta.apps, method_apps)
- self.assertEqual(ContextManager._meta.apps, context_apps)
- self.assertEqual(NestedContextManager._meta.apps, nested_context_apps)
- class DoNothingDecorator(TestContextDecorator):
- def enable(self):
- pass
- def disable(self):
- pass
- class TestContextDecoratorTests(SimpleTestCase):
- @mock.patch.object(DoNothingDecorator, "disable")
- def test_exception_in_setup(self, mock_disable):
- """An exception is setUp() is reraised after disable() is called."""
- class ExceptionInSetUp(unittest.TestCase):
- def setUp(self):
- raise NotImplementedError("reraised")
- decorator = DoNothingDecorator()
- decorated_test_class = decorator.__call__(ExceptionInSetUp)()
- self.assertFalse(mock_disable.called)
- with self.assertRaisesMessage(NotImplementedError, "reraised"):
- decorated_test_class.setUp()
- decorated_test_class.doCleanups()
- self.assertTrue(mock_disable.called)
- def test_cleanups_run_after_tearDown(self):
- calls = []
- class SaveCallsDecorator(TestContextDecorator):
- def enable(self):
- calls.append("enable")
- def disable(self):
- calls.append("disable")
- class AddCleanupInSetUp(unittest.TestCase):
- def setUp(self):
- calls.append("setUp")
- self.addCleanup(lambda: calls.append("cleanup"))
- decorator = SaveCallsDecorator()
- decorated_test_class = decorator.__call__(AddCleanupInSetUp)()
- decorated_test_class.setUp()
- decorated_test_class.tearDown()
- decorated_test_class.doCleanups()
- self.assertEqual(calls, ["enable", "setUp", "cleanup", "disable"])
|