import gzip import os import sys import tempfile import unittest import warnings from io import StringIO from unittest import mock from django.apps import apps from django.contrib.sites.models import Site from django.core import management from django.core.files.temp import NamedTemporaryFile from django.core.management import CommandError from django.core.management.commands.dumpdata import ProxyModelWarning from django.core.serializers.base import ProgressBar from django.db import IntegrityError, connection from django.test import TestCase, TransactionTestCase, skipUnlessDBFeature from .models import ( Article, Category, CircularA, CircularB, NaturalKeyThing, PrimaryKeyUUIDModel, ProxySpy, Spy, Tag, Visa, ) try: import bz2 # NOQA HAS_BZ2 = True except ImportError: HAS_BZ2 = False try: import lzma # NOQA HAS_LZMA = True except ImportError: HAS_LZMA = False class TestCaseFixtureLoadingTests(TestCase): fixtures = ["fixture1.json", "fixture2.json"] def test_class_fixtures(self): "Test case has installed 3 fixture objects" self.assertSequenceEqual( Article.objects.values_list("headline", flat=True), [ "Django conquers world!", "Copyright is fine the way it is", "Poker has no place on ESPN", ], ) class SubclassTestCaseFixtureLoadingTests(TestCaseFixtureLoadingTests): """ Make sure that subclasses can remove fixtures from parent class (#21089). """ fixtures = [] def test_class_fixtures(self): "There were no fixture objects installed" self.assertEqual(Article.objects.count(), 0) class DumpDataAssertMixin: def _dumpdata_assert( self, args, output, format="json", filename=None, natural_foreign_keys=False, natural_primary_keys=False, use_base_manager=False, exclude_list=[], primary_keys="", ): new_io = StringIO() filename = filename and os.path.join(tempfile.gettempdir(), filename) management.call_command( "dumpdata", *args, format=format, stdout=new_io, stderr=new_io, output=filename, use_natural_foreign_keys=natural_foreign_keys, use_natural_primary_keys=natural_primary_keys, use_base_manager=use_base_manager, exclude=exclude_list, primary_keys=primary_keys, ) if filename: file_root, file_ext = os.path.splitext(filename) compression_formats = { ".bz2": (open, file_root), ".gz": (gzip.open, filename), ".lzma": (open, file_root), ".xz": (open, file_root), ".zip": (open, file_root), } if HAS_BZ2: compression_formats[".bz2"] = (bz2.open, filename) if HAS_LZMA: compression_formats[".lzma"] = (lzma.open, filename) compression_formats[".xz"] = (lzma.open, filename) try: open_method, file_path = compression_formats[file_ext] except KeyError: open_method, file_path = open, filename with open_method(file_path, "rt") as f: command_output = f.read() os.remove(file_path) else: command_output = new_io.getvalue().strip() if format == "json": self.assertJSONEqual(command_output, output) elif format == "xml": self.assertXMLEqual(command_output, output) else: self.assertEqual(command_output, output) class FixtureLoadingTests(DumpDataAssertMixin, TestCase): def test_loading_and_dumping(self): apps.clear_cache() Site.objects.all().delete() # Load fixture 1. Single JSON file, with two objects. management.call_command("loaddata", "fixture1.json", verbosity=0) self.assertSequenceEqual( Article.objects.values_list("headline", flat=True), ["Time to reform copyright", "Poker has no place on ESPN"], ) # Dump the current contents of the database as a JSON fixture self._dumpdata_assert( ["fixtures"], '[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": ' '"News Stories"}}, {"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place ' 'on ESPN", "pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": ' '{"headline": "Time to reform copyright", "pub_date": "2006-06-16T13:00:00"}}]', ) # Try just dumping the contents of fixtures.Category self._dumpdata_assert( ["fixtures.Category"], '[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", ' '"title": "News Stories"}}]', ) # ...and just fixtures.Article self._dumpdata_assert( ["fixtures.Article"], '[{"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place on ESPN", ' '"pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": {"headline": ' '"Time to reform copyright", "pub_date": "2006-06-16T13:00:00"}}]', ) # ...and both self._dumpdata_assert( ["fixtures.Category", "fixtures.Article"], '[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", ' '"title": "News Stories"}}, {"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has ' 'no place on ESPN", "pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", ' '"fields": {"headline": "Time to reform copyright", "pub_date": "2006-06-16T13:00:00"}}]', ) # Specify a specific model twice self._dumpdata_assert( ["fixtures.Article", "fixtures.Article"], ( '[{"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place on ESPN", ' '"pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": {"headline": ' '"Time to reform copyright", "pub_date": "2006-06-16T13:00:00"}}]' ), ) # Specify a dump that specifies Article both explicitly and implicitly self._dumpdata_assert( ["fixtures.Article", "fixtures"], '[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": ' '"News Stories"}}, {"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place ' 'on ESPN", "pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": ' '{"headline": "Time to reform copyright", "pub_date": "2006-06-16T13:00:00"}}]', ) # Specify a dump that specifies Article both explicitly and implicitly, # but lists the app first (#22025). self._dumpdata_assert( ["fixtures", "fixtures.Article"], '[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": ' '"News Stories"}}, {"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place ' 'on ESPN", "pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": ' '{"headline": "Time to reform copyright", "pub_date": "2006-06-16T13:00:00"}}]', ) # Same again, but specify in the reverse order self._dumpdata_assert( ["fixtures"], '[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": ' '"News Stories"}}, {"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no ' 'place on ESPN", "pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields":' ' {"headline": "Time to reform copyright", "pub_date": "2006-06-16T13:00:00"}}]', ) # Specify one model from one application, and an entire other application. self._dumpdata_assert( ["fixtures.Category", "sites"], '[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": ' '"News Stories"}}, {"pk": 1, "model": "sites.site", "fields": {"domain": "example.com", "name": ' '"example.com"}}]', ) # Load fixture 2. JSON file imported by default. Overwrites some existing objects management.call_command("loaddata", "fixture2.json", verbosity=0) self.assertSequenceEqual( Article.objects.values_list("headline", flat=True), [ "Django conquers world!", "Copyright is fine the way it is", "Poker has no place on ESPN", ], ) # Load fixture 3, XML format. management.call_command("loaddata", "fixture3.xml", verbosity=0) self.assertSequenceEqual( Article.objects.values_list("headline", flat=True), [ "XML identified as leading cause of cancer", "Django conquers world!", "Copyright is fine the way it is", "Poker on TV is great!", ], ) # Load fixture 6, JSON file with dynamic ContentType fields. Testing ManyToOne. management.call_command("loaddata", "fixture6.json", verbosity=0) self.assertQuerysetEqual( Tag.objects.all(), [ ' tagged "copyright">', ' tagged "law">', ], transform=repr, ordered=False, ) # Load fixture 7, XML file with dynamic ContentType fields. Testing ManyToOne. management.call_command("loaddata", "fixture7.xml", verbosity=0) self.assertQuerysetEqual( Tag.objects.all(), [ ' tagged "copyright">', ' tagged "legal">', ' tagged "django">', ' tagged "world domination">', ], transform=repr, ordered=False, ) # Load fixture 8, JSON file with dynamic Permission fields. Testing ManyToMany. management.call_command("loaddata", "fixture8.json", verbosity=0) self.assertQuerysetEqual( Visa.objects.all(), [ "", "", "", ], transform=repr, ordered=False, ) # Load fixture 9, XML file with dynamic Permission fields. Testing ManyToMany. management.call_command("loaddata", "fixture9.xml", verbosity=0) self.assertQuerysetEqual( Visa.objects.all(), [ "", "", '', ], transform=repr, ordered=False, ) # object list is unaffected self.assertSequenceEqual( Article.objects.values_list("headline", flat=True), [ "XML identified as leading cause of cancer", "Django conquers world!", "Copyright is fine the way it is", "Poker on TV is great!", ], ) # By default, you get raw keys on dumpdata self._dumpdata_assert( ["fixtures.book"], '[{"pk": 1, "model": "fixtures.book", "fields": {"name": "Music for all ages", "authors": [3, 1]}}]', ) # But you can get natural keys if you ask for them and they are available self._dumpdata_assert( ["fixtures.book"], '[{"pk": 1, "model": "fixtures.book", "fields": {"name": "Music for all ages", "authors": [["Artist ' 'formerly known as \\"Prince\\""], ["Django Reinhardt"]]}}]', natural_foreign_keys=True, ) # You can also omit the primary keys for models that we can get later with natural keys. self._dumpdata_assert( ["fixtures.person"], '[{"fields": {"name": "Django Reinhardt"}, "model": "fixtures.person"}, {"fields": {"name": "Stephane ' 'Grappelli"}, "model": "fixtures.person"}, {"fields": {"name": "Artist formerly known as ' '\\"Prince\\""}, "model": "fixtures.person"}]', natural_primary_keys=True, ) # Dump the current contents of the database as a JSON fixture self._dumpdata_assert( ["fixtures"], '[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": ' '"News Stories"}}, {"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker on TV is ' 'great!", "pub_date": "2006-06-16T11:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": ' '{"headline": "Copyright is fine the way it is", "pub_date": "2006-06-16T14:00:00"}}, {"pk": 4, ' '"model": "fixtures.article", "fields": {"headline": "Django conquers world!", "pub_date": ' '"2006-06-16T15:00:00"}}, {"pk": 5, "model": "fixtures.article", "fields": {"headline": "XML ' 'identified as leading cause of cancer", "pub_date": "2006-06-16T16:00:00"}}, {"pk": 1, "model": ' '"fixtures.tag", "fields": {"tagged_type": ["fixtures", "article"], "name": "copyright", "tagged_id": ' '3}}, {"pk": 2, "model": "fixtures.tag", "fields": {"tagged_type": ["fixtures", "article"], "name": ' '"legal", "tagged_id": 3}}, {"pk": 3, "model": "fixtures.tag", "fields": {"tagged_type": ["fixtures", ' '"article"], "name": "django", "tagged_id": 4}}, {"pk": 4, "model": "fixtures.tag", "fields": ' '{"tagged_type": ["fixtures", "article"], "name": "world domination", "tagged_id": 4}}, {"pk": 1, ' '"model": "fixtures.person", "fields": {"name": "Django Reinhardt"}}, {"pk": 2, "model": ' '"fixtures.person", "fields": {"name": "Stephane Grappelli"}}, {"pk": 3, "model": "fixtures.person", ' '"fields": {"name": "Artist formerly known as \\"Prince\\""}}, {"pk": 1, "model": "fixtures.visa", ' '"fields": {"person": ["Django Reinhardt"], "permissions": [["add_user", "auth", "user"], ' '["change_user", "auth", "user"], ["delete_user", "auth", "user"]]}}, {"pk": 2, "model": ' '"fixtures.visa", "fields": {"person": ["Stephane Grappelli"], "permissions": [["add_user", "auth", ' '"user"], ["delete_user", "auth", "user"]]}}, {"pk": 3, "model": "fixtures.visa", "fields": {"person":' ' ["Artist formerly known as \\"Prince\\""], "permissions": [["change_user", "auth", "user"]]}}, ' '{"pk": 1, "model": "fixtures.book", "fields": {"name": "Music for all ages", "authors": [["Artist ' 'formerly known as \\"Prince\\""], ["Django Reinhardt"]]}}]', natural_foreign_keys=True, ) # Dump the current contents of the database as an XML fixture self._dumpdata_assert( ["fixtures"], 'News StoriesLatest news storiesPoker on TV is great!2006-06-16T11:00:00Copyright is fine the way it ' 'is2006-06-16T14:00:00Django conquers world!' '2006-06-16T15:00:00XML identified as leading ' 'cause of cancer2006-06-16T16:00:00' 'copyright' 'fixtures' 'article3' 'legal' '' 'fixturesarticle3djangofixturesarticle4' 'world dominationfixturesarticle' '4Django Reinhardt' 'Stephane Grappelli' '' 'Artist formerly known as "Prince"Django Reinhardt' 'add_user' "authuserchange_user" "authuserdelete_user" 'authuserStephane' ' Grappelli' "add_userauthuser" "delete_userauthuser" 'Artist formerly known as "Prince"change_user' 'authuserMusic for all agesArtist formerly known as ' '"Prince"Django Reinhardt' "", format="xml", natural_foreign_keys=True, ) def test_dumpdata_with_excludes(self): # Load fixture1 which has a site, two articles, and a category Site.objects.all().delete() management.call_command("loaddata", "fixture1.json", verbosity=0) # Excluding fixtures app should only leave sites self._dumpdata_assert( ["sites", "fixtures"], '[{"pk": 1, "model": "sites.site", "fields": {"domain": "example.com", "name": "example.com"}}]', exclude_list=["fixtures"], ) # Excluding fixtures.Article/Book should leave fixtures.Category self._dumpdata_assert( ["sites", "fixtures"], '[{"pk": 1, "model": "sites.site", "fields": {"domain": "example.com", "name": "example.com"}}, ' '{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": ' '"News Stories"}}]', exclude_list=["fixtures.Article", "fixtures.Book"], ) # Excluding fixtures and fixtures.Article/Book should be a no-op self._dumpdata_assert( ["sites", "fixtures"], '[{"pk": 1, "model": "sites.site", "fields": {"domain": "example.com", "name": "example.com"}}, ' '{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": ' '"News Stories"}}]', exclude_list=["fixtures.Article", "fixtures.Book"], ) # Excluding sites and fixtures.Article/Book should only leave fixtures.Category self._dumpdata_assert( ["sites", "fixtures"], '[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": ' '"News Stories"}}]', exclude_list=["fixtures.Article", "fixtures.Book", "sites"], ) # Excluding a bogus app should throw an error with self.assertRaisesMessage( management.CommandError, "No installed app with label 'foo_app'." ): self._dumpdata_assert(["fixtures", "sites"], "", exclude_list=["foo_app"]) # Excluding a bogus model should throw an error with self.assertRaisesMessage( management.CommandError, "Unknown model: fixtures.FooModel" ): self._dumpdata_assert( ["fixtures", "sites"], "", exclude_list=["fixtures.FooModel"] ) @unittest.skipIf( sys.platform == "win32", "Windows doesn't support '?' in filenames." ) def test_load_fixture_with_special_characters(self): management.call_command("loaddata", "fixture_with[special]chars", verbosity=0) self.assertEqual( Article.objects.get().headline, "How To Deal With Special Characters", ) def test_dumpdata_with_filtering_manager(self): spy1 = Spy.objects.create(name="Paul") spy2 = Spy.objects.create(name="Alex", cover_blown=True) self.assertSequenceEqual(Spy.objects.all(), [spy1]) # Use the default manager self._dumpdata_assert( ["fixtures.Spy"], '[{"pk": %d, "model": "fixtures.spy", "fields": {"cover_blown": false}}]' % spy1.pk, ) # Dump using Django's base manager. Should return all objects, # even those normally filtered by the manager self._dumpdata_assert( ["fixtures.Spy"], '[{"pk": %d, "model": "fixtures.spy", "fields": {"cover_blown": true}}, {"pk": %d, "model": ' '"fixtures.spy", "fields": {"cover_blown": false}}]' % (spy2.pk, spy1.pk), use_base_manager=True, ) def test_dumpdata_with_pks(self): management.call_command("loaddata", "fixture1.json", verbosity=0) management.call_command("loaddata", "fixture2.json", verbosity=0) self._dumpdata_assert( ["fixtures.Article"], '[{"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place on ESPN", ' '"pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": {"headline": ' '"Copyright is fine the way it is", "pub_date": "2006-06-16T14:00:00"}}]', primary_keys="2,3", ) self._dumpdata_assert( ["fixtures.Article"], '[{"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place on ESPN", ' '"pub_date": "2006-06-16T12:00:00"}}]', primary_keys="2", ) with self.assertRaisesMessage( management.CommandError, "You can only use --pks option with one model" ): self._dumpdata_assert( ["fixtures"], '[{"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place on ESPN", ' '"pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": ' '{"headline": "Copyright is fine the way it is", "pub_date": "2006-06-16T14:00:00"}}]', primary_keys="2,3", ) with self.assertRaisesMessage( management.CommandError, "You can only use --pks option with one model" ): self._dumpdata_assert( "", '[{"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place on ESPN", ' '"pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": ' '{"headline": "Copyright is fine the way it is", "pub_date": "2006-06-16T14:00:00"}}]', primary_keys="2,3", ) with self.assertRaisesMessage( management.CommandError, "You can only use --pks option with one model" ): self._dumpdata_assert( ["fixtures.Article", "fixtures.category"], '[{"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place on ESPN", ' '"pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": ' '{"headline": "Copyright is fine the way it is", "pub_date": "2006-06-16T14:00:00"}}]', primary_keys="2,3", ) def test_dumpdata_with_uuid_pks(self): m1 = PrimaryKeyUUIDModel.objects.create() m2 = PrimaryKeyUUIDModel.objects.create() output = StringIO() management.call_command( "dumpdata", "fixtures.PrimaryKeyUUIDModel", "--pks", ", ".join([str(m1.id), str(m2.id)]), stdout=output, ) result = output.getvalue() self.assertIn('"pk": "%s"' % m1.id, result) self.assertIn('"pk": "%s"' % m2.id, result) def test_dumpdata_with_file_output(self): management.call_command("loaddata", "fixture1.json", verbosity=0) self._dumpdata_assert( ["fixtures"], '[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": ' '"News Stories"}}, {"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place ' 'on ESPN", "pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": ' '{"headline": "Time to reform copyright", "pub_date": "2006-06-16T13:00:00"}}]', filename="dumpdata.json", ) def test_dumpdata_with_file_gzip_output(self): management.call_command("loaddata", "fixture1.json", verbosity=0) self._dumpdata_assert( ["fixtures"], '[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": ' '"News Stories"}}, {"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place ' 'on ESPN", "pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": ' '{"headline": "Time to reform copyright", "pub_date": "2006-06-16T13:00:00"}}]', filename="dumpdata.json.gz", ) @unittest.skipUnless(HAS_BZ2, "No bz2 library detected.") def test_dumpdata_with_file_bz2_output(self): management.call_command("loaddata", "fixture1.json", verbosity=0) self._dumpdata_assert( ["fixtures"], '[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": ' '"News Stories"}}, {"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place ' 'on ESPN", "pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": ' '{"headline": "Time to reform copyright", "pub_date": "2006-06-16T13:00:00"}}]', filename="dumpdata.json.bz2", ) @unittest.skipUnless(HAS_LZMA, "No lzma library detected.") def test_dumpdata_with_file_lzma_output(self): management.call_command("loaddata", "fixture1.json", verbosity=0) self._dumpdata_assert( ["fixtures"], '[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": ' '"News Stories"}}, {"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place ' 'on ESPN", "pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": ' '{"headline": "Time to reform copyright", "pub_date": "2006-06-16T13:00:00"}}]', filename="dumpdata.json.lzma", ) @unittest.skipUnless(HAS_LZMA, "No lzma library detected.") def test_dumpdata_with_file_xz_output(self): management.call_command("loaddata", "fixture1.json", verbosity=0) self._dumpdata_assert( ["fixtures"], '[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": ' '"News Stories"}}, {"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place ' 'on ESPN", "pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": ' '{"headline": "Time to reform copyright", "pub_date": "2006-06-16T13:00:00"}}]', filename="dumpdata.json.xz", ) def test_dumpdata_with_file_zip_output(self): management.call_command("loaddata", "fixture1.json", verbosity=0) msg = "Unsupported file extension (.zip). Fixtures saved in 'dumpdata.json'." with self.assertWarnsMessage(RuntimeWarning, msg): self._dumpdata_assert( ["fixtures"], '[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": ' '"News Stories"}}, {"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place ' 'on ESPN", "pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": ' '{"headline": "Time to reform copyright", "pub_date": "2006-06-16T13:00:00"}}]', filename="dumpdata.json.zip", ) def test_dumpdata_progressbar(self): """ Dumpdata shows a progress bar on the command line when --output is set, stdout is a tty, and verbosity > 0. """ management.call_command("loaddata", "fixture1.json", verbosity=0) new_io = StringIO() new_io.isatty = lambda: True with NamedTemporaryFile() as file: options = { "format": "json", "stdout": new_io, "stderr": new_io, "output": file.name, } management.call_command("dumpdata", "fixtures", **options) self.assertTrue( new_io.getvalue().endswith( "[" + "." * ProgressBar.progress_width + "]\n" ) ) # Test no progress bar when verbosity = 0 options["verbosity"] = 0 new_io = StringIO() new_io.isatty = lambda: True options.update({"stdout": new_io, "stderr": new_io}) management.call_command("dumpdata", "fixtures", **options) self.assertEqual(new_io.getvalue(), "") def test_dumpdata_proxy_without_concrete(self): """ A warning is displayed if a proxy model is dumped without its concrete parent. """ ProxySpy.objects.create(name="Paul") msg = "fixtures.ProxySpy is a proxy model and won't be serialized." with self.assertWarnsMessage(ProxyModelWarning, msg): self._dumpdata_assert(["fixtures.ProxySpy"], "[]") def test_dumpdata_proxy_with_concrete(self): """ A warning isn't displayed if a proxy model is dumped with its concrete parent. """ spy = ProxySpy.objects.create(name="Paul") with warnings.catch_warnings(record=True) as warning_list: warnings.simplefilter("always") self._dumpdata_assert( ["fixtures.ProxySpy", "fixtures.Spy"], '[{"pk": %d, "model": "fixtures.spy", "fields": {"cover_blown": false}}]' % spy.pk, ) self.assertEqual(len(warning_list), 0) def test_compress_format_loading(self): # Load fixture 4 (compressed), using format specification management.call_command("loaddata", "fixture4.json", verbosity=0) self.assertEqual(Article.objects.get().headline, "Django pets kitten") def test_compressed_specified_loading(self): # Load fixture 5 (compressed), using format *and* compression specification management.call_command("loaddata", "fixture5.json.zip", verbosity=0) self.assertEqual( Article.objects.get().headline, "WoW subscribers now outnumber readers", ) def test_compressed_loading(self): # Load fixture 5 (compressed), only compression specification management.call_command("loaddata", "fixture5.zip", verbosity=0) self.assertEqual( Article.objects.get().headline, "WoW subscribers now outnumber readers", ) def test_compressed_loading_gzip(self): management.call_command("loaddata", "fixture5.json.gz", verbosity=0) self.assertEqual( Article.objects.get().headline, "WoW subscribers now outnumber readers", ) @unittest.skipUnless(HAS_BZ2, "No bz2 library detected.") def test_compressed_loading_bz2(self): management.call_command("loaddata", "fixture5.json.bz2", verbosity=0) self.assertEqual( Article.objects.get().headline, "WoW subscribers now outnumber readers", ) @unittest.skipUnless(HAS_LZMA, "No lzma library detected.") def test_compressed_loading_lzma(self): management.call_command("loaddata", "fixture5.json.lzma", verbosity=0) self.assertEqual( Article.objects.get().headline, "WoW subscribers now outnumber readers", ) @unittest.skipUnless(HAS_LZMA, "No lzma library detected.") def test_compressed_loading_xz(self): management.call_command("loaddata", "fixture5.json.xz", verbosity=0) self.assertEqual( Article.objects.get().headline, "WoW subscribers now outnumber readers", ) def test_ambiguous_compressed_fixture(self): # The name "fixture5" is ambiguous, so loading raises an error. msg = "Multiple fixtures named 'fixture5'" with self.assertRaisesMessage(management.CommandError, msg): management.call_command("loaddata", "fixture5", verbosity=0) def test_db_loading(self): # Load db fixtures 1 and 2. These will load using the 'default' database identifier implicitly management.call_command("loaddata", "db_fixture_1", verbosity=0) management.call_command("loaddata", "db_fixture_2", verbosity=0) self.assertSequenceEqual( Article.objects.values_list("headline", flat=True), [ "Who needs more than one database?", "Who needs to use compressed data?", ], ) def test_loaddata_error_message(self): """ Loading a fixture which contains an invalid object outputs an error message which contains the pk of the object that triggered the error. """ # MySQL needs a little prodding to reject invalid data. # This won't affect other tests because the database connection # is closed at the end of each test. if connection.vendor == "mysql": with connection.cursor() as cursor: cursor.execute("SET sql_mode = 'TRADITIONAL'") msg = "Could not load fixtures.Article(pk=1):" with self.assertRaisesMessage(IntegrityError, msg): management.call_command("loaddata", "invalid.json", verbosity=0) @unittest.skipUnless( connection.vendor == "postgresql", "psycopg2 prohibits null characters in data." ) def test_loaddata_null_characters_on_postgresql(self): msg = ( "Could not load fixtures.Article(pk=2): " "A string literal cannot contain NUL (0x00) characters." ) with self.assertRaisesMessage(ValueError, msg): management.call_command("loaddata", "null_character_in_field_value.json") def test_loaddata_app_option(self): with self.assertRaisesMessage( CommandError, "No fixture named 'db_fixture_1' found." ): management.call_command( "loaddata", "db_fixture_1", verbosity=0, app_label="someotherapp" ) self.assertQuerysetEqual(Article.objects.all(), []) management.call_command( "loaddata", "db_fixture_1", verbosity=0, app_label="fixtures" ) self.assertEqual( Article.objects.get().headline, "Who needs more than one database?", ) def test_loaddata_verbosity_three(self): output = StringIO() management.call_command( "loaddata", "fixture1.json", verbosity=3, stdout=output, stderr=output ) command_output = output.getvalue() self.assertIn( "\rProcessed 1 object(s).\rProcessed 2 object(s)." "\rProcessed 3 object(s).\rProcessed 4 object(s).\n", command_output, ) def test_loading_using(self): # Load db fixtures 1 and 2. These will load using the 'default' database identifier explicitly management.call_command( "loaddata", "db_fixture_1", verbosity=0, database="default" ) management.call_command( "loaddata", "db_fixture_2", verbosity=0, database="default" ) self.assertSequenceEqual( Article.objects.values_list("headline", flat=True), [ "Who needs more than one database?", "Who needs to use compressed data?", ], ) def test_unmatched_identifier_loading(self): # Try to load db fixture 3. This won't load because the database identifier doesn't match with self.assertRaisesMessage( CommandError, "No fixture named 'db_fixture_3' found." ): management.call_command("loaddata", "db_fixture_3", verbosity=0) with self.assertRaisesMessage( CommandError, "No fixture named 'db_fixture_3' found." ): management.call_command( "loaddata", "db_fixture_3", verbosity=0, database="default" ) self.assertQuerysetEqual(Article.objects.all(), []) def test_output_formats(self): # Load back in fixture 1, we need the articles from it management.call_command("loaddata", "fixture1", verbosity=0) # Try to load fixture 6 using format discovery management.call_command("loaddata", "fixture6", verbosity=0) self.assertQuerysetEqual( Tag.objects.all(), [ ' tagged "copyright">', ' tagged "law">', ], transform=repr, ordered=False, ) # Dump the current contents of the database as a JSON fixture self._dumpdata_assert( ["fixtures"], '[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": ' '"News Stories"}}, {"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place ' 'on ESPN", "pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": ' '{"headline": "Time to reform copyright", "pub_date": "2006-06-16T13:00:00"}}, {"pk": 1, "model": ' '"fixtures.tag", "fields": {"tagged_type": ["fixtures", "article"], "name": "copyright", "tagged_id": ' '3}}, {"pk": 2, "model": "fixtures.tag", "fields": {"tagged_type": ["fixtures", "article"], "name": ' '"law", "tagged_id": 3}}, {"pk": 1, "model": "fixtures.person", "fields": {"name": "Django ' 'Reinhardt"}}, {"pk": 2, "model": "fixtures.person", "fields": {"name": "Stephane Grappelli"}}, ' '{"pk": 3, "model": "fixtures.person", "fields": {"name": "Prince"}}]', natural_foreign_keys=True, ) # Dump the current contents of the database as an XML fixture self._dumpdata_assert( ["fixtures"], 'News StoriesLatest news storiesPoker has no place on ESPN' '2006-06-16T12:00:00Time to reform copyright' '2006-06-16T13:00:00copyrightfixtures' 'article3' 'lawfixtures' 'article3' 'Django Reinhardt' 'Stephane ' 'Grappelli' "Prince", format="xml", natural_foreign_keys=True, ) def test_loading_with_exclude_app(self): Site.objects.all().delete() management.call_command( "loaddata", "fixture1", exclude=["fixtures"], verbosity=0 ) self.assertFalse(Article.objects.exists()) self.assertFalse(Category.objects.exists()) self.assertEqual(Site.objects.get().domain, "example.com") def test_loading_with_exclude_model(self): Site.objects.all().delete() management.call_command( "loaddata", "fixture1", exclude=["fixtures.Article"], verbosity=0 ) self.assertFalse(Article.objects.exists()) self.assertEqual(Category.objects.get().title, "News Stories") self.assertEqual(Site.objects.get().domain, "example.com") def test_exclude_option_errors(self): """Excluding a bogus app or model should raise an error.""" msg = "No installed app with label 'foo_app'." with self.assertRaisesMessage(management.CommandError, msg): management.call_command( "loaddata", "fixture1", exclude=["foo_app"], verbosity=0 ) msg = "Unknown model: fixtures.FooModel" with self.assertRaisesMessage(management.CommandError, msg): management.call_command( "loaddata", "fixture1", exclude=["fixtures.FooModel"], verbosity=0 ) def test_stdin_without_format(self): """Reading from stdin raises an error if format isn't specified.""" msg = "--format must be specified when reading from stdin." with self.assertRaisesMessage(management.CommandError, msg): management.call_command("loaddata", "-", verbosity=0) def test_loading_stdin(self): """Loading fixtures from stdin with json and xml.""" tests_dir = os.path.dirname(__file__) fixture_json = os.path.join(tests_dir, "fixtures", "fixture1.json") fixture_xml = os.path.join(tests_dir, "fixtures", "fixture3.xml") with mock.patch( "django.core.management.commands.loaddata.sys.stdin", open(fixture_json) ): management.call_command("loaddata", "--format=json", "-", verbosity=0) self.assertSequenceEqual( Article.objects.values_list("headline", flat=True), ["Time to reform copyright", "Poker has no place on ESPN"], ) with mock.patch( "django.core.management.commands.loaddata.sys.stdin", open(fixture_xml) ): management.call_command("loaddata", "--format=xml", "-", verbosity=0) self.assertSequenceEqual( Article.objects.values_list("headline", flat=True), [ "XML identified as leading cause of cancer", "Time to reform copyright", "Poker on TV is great!", ], ) class NonexistentFixtureTests(TestCase): """ Custom class to limit fixture dirs. """ def test_loaddata_not_existent_fixture_file(self): stdout_output = StringIO() with self.assertRaisesMessage( CommandError, "No fixture named 'this_fixture_doesnt_exist' found." ): management.call_command( "loaddata", "this_fixture_doesnt_exist", stdout=stdout_output ) @mock.patch("django.db.connection.enable_constraint_checking") @mock.patch("django.db.connection.disable_constraint_checking") def test_nonexistent_fixture_no_constraint_checking( self, disable_constraint_checking, enable_constraint_checking ): """ If no fixtures match the loaddata command, constraints checks on the database shouldn't be disabled. This is performance critical on MSSQL. """ with self.assertRaisesMessage( CommandError, "No fixture named 'this_fixture_doesnt_exist' found." ): management.call_command( "loaddata", "this_fixture_doesnt_exist", verbosity=0 ) disable_constraint_checking.assert_not_called() enable_constraint_checking.assert_not_called() class FixtureTransactionTests(DumpDataAssertMixin, TransactionTestCase): available_apps = [ "fixtures", "django.contrib.sites", ] @skipUnlessDBFeature("supports_forward_references") def test_format_discovery(self): # Load fixture 1 again, using format discovery management.call_command("loaddata", "fixture1", verbosity=0) self.assertSequenceEqual( Article.objects.values_list("headline", flat=True), ["Time to reform copyright", "Poker has no place on ESPN"], ) # Try to load fixture 2 using format discovery; this will fail # because there are two fixture2's in the fixtures directory msg = "Multiple fixtures named 'fixture2'" with self.assertRaisesMessage(management.CommandError, msg): management.call_command("loaddata", "fixture2", verbosity=0) # object list is unaffected self.assertSequenceEqual( Article.objects.values_list("headline", flat=True), ["Time to reform copyright", "Poker has no place on ESPN"], ) # Dump the current contents of the database as a JSON fixture self._dumpdata_assert( ["fixtures"], '[{"pk": 1, "model": "fixtures.category", "fields": {"description": "Latest news stories", "title": ' '"News Stories"}}, {"pk": 2, "model": "fixtures.article", "fields": {"headline": "Poker has no place ' 'on ESPN", "pub_date": "2006-06-16T12:00:00"}}, {"pk": 3, "model": "fixtures.article", "fields": ' '{"headline": "Time to reform copyright", "pub_date": "2006-06-16T13:00:00"}}]', ) # Load fixture 4 (compressed), using format discovery management.call_command("loaddata", "fixture4", verbosity=0) self.assertSequenceEqual( Article.objects.values_list("headline", flat=True), [ "Django pets kitten", "Time to reform copyright", "Poker has no place on ESPN", ], ) class ForwardReferenceTests(DumpDataAssertMixin, TestCase): def test_forward_reference_fk(self): management.call_command("loaddata", "forward_reference_fk.json", verbosity=0) t1, t2 = NaturalKeyThing.objects.all() self.assertEqual(t1.other_thing, t2) self.assertEqual(t2.other_thing, t1) self._dumpdata_assert( ["fixtures"], '[{"model": "fixtures.naturalkeything", "pk": 1, ' '"fields": {"key": "t1", "other_thing": 2, "other_things": []}}, ' '{"model": "fixtures.naturalkeything", "pk": 2, ' '"fields": {"key": "t2", "other_thing": 1, "other_things": []}}]', ) def test_forward_reference_fk_natural_key(self): management.call_command( "loaddata", "forward_reference_fk_natural_key.json", verbosity=0, ) t1, t2 = NaturalKeyThing.objects.all() self.assertEqual(t1.other_thing, t2) self.assertEqual(t2.other_thing, t1) self._dumpdata_assert( ["fixtures"], '[{"model": "fixtures.naturalkeything", ' '"fields": {"key": "t1", "other_thing": ["t2"], "other_things": []}}, ' '{"model": "fixtures.naturalkeything", ' '"fields": {"key": "t2", "other_thing": ["t1"], "other_things": []}}]', natural_primary_keys=True, natural_foreign_keys=True, ) def test_forward_reference_m2m(self): management.call_command("loaddata", "forward_reference_m2m.json", verbosity=0) self.assertEqual(NaturalKeyThing.objects.count(), 3) t1 = NaturalKeyThing.objects.get_by_natural_key("t1") self.assertSequenceEqual( t1.other_things.order_by("key").values_list("key", flat=True), ["t2", "t3"], ) self._dumpdata_assert( ["fixtures"], '[{"model": "fixtures.naturalkeything", "pk": 1, ' '"fields": {"key": "t1", "other_thing": null, "other_things": [2, 3]}}, ' '{"model": "fixtures.naturalkeything", "pk": 2, ' '"fields": {"key": "t2", "other_thing": null, "other_things": []}}, ' '{"model": "fixtures.naturalkeything", "pk": 3, ' '"fields": {"key": "t3", "other_thing": null, "other_things": []}}]', ) def test_forward_reference_m2m_natural_key(self): management.call_command( "loaddata", "forward_reference_m2m_natural_key.json", verbosity=0, ) self.assertEqual(NaturalKeyThing.objects.count(), 3) t1 = NaturalKeyThing.objects.get_by_natural_key("t1") self.assertSequenceEqual( t1.other_things.order_by("key").values_list("key", flat=True), ["t2", "t3"], ) self._dumpdata_assert( ["fixtures"], '[{"model": "fixtures.naturalkeything", ' '"fields": {"key": "t1", "other_thing": null, "other_things": [["t2"], ["t3"]]}}, ' '{"model": "fixtures.naturalkeything", ' '"fields": {"key": "t2", "other_thing": null, "other_things": []}}, ' '{"model": "fixtures.naturalkeything", ' '"fields": {"key": "t3", "other_thing": null, "other_things": []}}]', natural_primary_keys=True, natural_foreign_keys=True, ) class CircularReferenceTests(DumpDataAssertMixin, TestCase): def test_circular_reference(self): management.call_command("loaddata", "circular_reference.json", verbosity=0) obj_a = CircularA.objects.get() obj_b = CircularB.objects.get() self.assertEqual(obj_a.obj, obj_b) self.assertEqual(obj_b.obj, obj_a) self._dumpdata_assert( ["fixtures"], '[{"model": "fixtures.circulara", "pk": 1, ' '"fields": {"key": "x", "obj": 1}}, ' '{"model": "fixtures.circularb", "pk": 1, ' '"fields": {"key": "y", "obj": 1}}]', ) def test_circular_reference_natural_key(self): management.call_command( "loaddata", "circular_reference_natural_key.json", verbosity=0, ) obj_a = CircularA.objects.get() obj_b = CircularB.objects.get() self.assertEqual(obj_a.obj, obj_b) self.assertEqual(obj_b.obj, obj_a) self._dumpdata_assert( ["fixtures"], '[{"model": "fixtures.circulara", ' '"fields": {"key": "x", "obj": ["y"]}}, ' '{"model": "fixtures.circularb", ' '"fields": {"key": "y", "obj": ["x"]}}]', natural_primary_keys=True, natural_foreign_keys=True, )