def test_image_add_to_collection():
root_collection = wagtail_factories.CollectionFactory(parent=None)
image = wagtail_factories.ImageFactory(
collection__parent=root_collection,
collection__name='new')
assert image.collection.name == 'new'
def test_document_add_to_collection():
root_collection = wagtail_factories.CollectionFactory(parent=None)
document = wagtail_factories.DocumentFactory(
collection__parent=root_collection, collection__name="new")
assert document.collection.name == "new"
def handle(self, *args, **options):
if options['delete']:
Page.objects.filter(slug='home').delete()
try:
HomePage.objects.get(slug='home')
except ObjectDoesNotExist:
Page.objects.filter(slug='home').delete()
# homepage cannot be saved without a parent
home_page = HomePageFactory.build(
description_header="Share and accept documents securely.",
slug="home")
root_page = Page.objects.get(title='Root')
root_page.add_child(instance=home_page)
site = Site.objects.create(site_name='SecureDrop.org (Dev)',
hostname='localhost',
port='8000',
root_page=home_page,
is_default_site=True)
image = CustomImage.objects.filter(title='Sample Image').first()
if not image:
image = CustomImage.objects.create(
title='Sample Image',
file=ImageFile(open(
'common/static/images/logo_solid_white.png', 'rb'),
name='logo'),
attribution='createdevdata')
sssettings = SocialSharingSEOSettings.for_site(site)
sssettings.default_description = 'SecureDrop'
sssettings.default_image = image
sssettings.save()
home_page.save()
site.save()
# IMAGES
icon_collection = wagtail_factories.CollectionFactory(name='Icons')
if options.get('download_images', True):
self.stdout.write('Fetching images')
self.stdout.flush()
image_fail = False
for i in range(15):
if not self.fetch_image(500, 500, icon_collection, 'animals'):
image_fail = True
if image_fail:
self.stdout.write(
self.style.NOTICE('NOTICE: Some images failed to save'))
else:
self.stdout.write(self.style.SUCCESS('OK'))
else:
faker = factory.faker.Faker._get_faker(locale='en-US')
for i in range(20):
CustomImageFactory.create(
file__width=500,
file__height=500,
file__color=faker.safe_color_name(),
collection=icon_collection,
)
management.call_command('createblogdata', '10')
management.call_command('createdirectory', '10')
management.call_command('createnavmenu')
management.call_command('createfootersettings')
management.call_command('createresultgroups')
management.call_command('createsearchmenus')
management.call_command('createmarketing')
# Create superuser
if not User.objects.filter(is_superuser=True).exists():
User.objects.create_superuser(
'test',
'test@securedrop',
'test',
)
self.stdout.write('Superuser created:\n'
'\tname: test\n'
'\temail: test@securedrop\n'
'\tpassword: test')