def test_get_export_models(self):
"""
Test that `get_export_models` returns the expected list of installed models.
"""
from django.contrib.auth.models import User, Group, Permission
from django.contrib.sites.models import Site
from django.contrib.sessions.models import Session
from django.contrib.admin.models import LogEntry
from django.contrib.contenttypes.models import ContentType
export_models = set(get_export_models(admin_only=False))
expected_export_models = set([
User, Group, Permission, Site, Session, LogEntry, ContentType,
Author, Book, Category
])
self.assertEqual(export_models, expected_export_models)
export_models_admin = set(get_export_models(admin_only=True))
expected_export_models_admin = set([
User, Group, Site,
Author, Book, Category
])
self.assertEqual(export_models_admin, expected_export_models_admin)
def test_kwargs_forwarded(self, resource_for_model, mocked_export):
"""
Test that kwargs passed to `export` are forwarded to the ModelResource
for export.
"""
from exportdb.tasks import export
book = Databook()
book.add_sheet(Dataset())
mocked_export.return_value = book
kwargs = {'foo': 'bar', 'baz': 10}
export(Exporter, **kwargs)
export_models = get_export_models()
for i, export_model in enumerate(export_models):
self.assertEqual(
resource_for_model.call_args_list[i],
mock.call(export_model, foo='bar', baz=10)
)
