python source code of test_form_submission_file

Project: wagtailstreamforms (GitHub Link)

wagtailstreamforms-master
- .github
  - CONTRIBUTING.md
  - ISSUE_TEMPLATE.md
  - workflows
    - python-test.yml
    - python-release.yml
  - CODE_OF_CONDUCT.md
- .circleci
  - config.yml
- Makefile
- .coveragerc
- wagtailstreamforms
  - fields.py
  - wagtailstreamforms_fields.py
  - wagtail_hooks.py
  - utils
    - apps.py
    - general.py
    - requests.py
    - version.py
    - __init__.py
    - loading.py
  - views
    - advanced_settings.py
    - copy.py
    - submission_delete.py
    - __init__.py
    - submission_list.py
  - wagtailstreamforms_hooks.py
  - blocks.py
  - models
    - form.py
    - file.py
    - __init__.py
    - abstract.py
    - submission.py
  - hooks.py
  - templates
    - streamforms
      - wagtailadmin
        shared
        datetimepicker_translations.html
      - confirm_delete.html
      - form_block.html
      - confirm_copy.html
      - advanced_settings.html
      - non_existent_form.html
      - list_submissions.html
      - index_submissions.html
      - partials
        form_field.html
        pagination_nav.html
  - templatetags
    - streamforms_tags.py
    - __init__.py
  - serializers.py
  - locale
    - nl
      - LC_MESSAGES
        django.mo
        django.po
    - tr
      - LC_MESSAGES
        django.mo
        django.po
    - es
      - LC_MESSAGES
        django.mo
        django.po
  - urls.py
  - forms.py
  - __init__.py
  - migrations
    - 0002_form_site.py
    - 0001_initial.py
    - __init__.py
  - conf.py
  - streamfield.py
  - management
    - commands
      - prunesubmissions.py
      - __init__.py
    - __init__.py
- readthedocs.yml
- postcss.config.js
- LICENSE
- manage.py
- README.rst
- setup.py
- CHANGELOG.rst
- example
  - wagtailstreamforms_fields.py
  - wsgi.py
  - wagtailstreamforms_hooks.py
  - templates
    - streamforms
      - form_block.html
      - partials
        form_field.html
    - example
      - basic_page.html
  - models.py
  - urls.py
  - migrations
    - 0001_initial.py
    - 0002_advancedformsetting.py
    - __init__.py
  - settings.py
  - scss
    - _variables.scss
    - karma.scss
    - module
      - _form.scss
      - _layout.scss
      - _message.scss
  - static
    - css
      - karma.min.css
      - karma.css.map
      - karma.css
      - karma.min.css.map
- docker-entrypoint.sh
- setup.cfg
- package.json
- tests
  - test_utils.py
  - views
    - test_admin_list.py
    - test_advanced.py
    - test_submission_list.py
    - test_copy.py
    - __init__.py
    - test_delete.py
  - test_urls.py
  - fixtures
    - test.json
  - hooks
    - test_save_form_submission_data.py
    - test_process_form.py
    - test_registering.py
    - __init__.py
  - models
    - test_form_submission_file.py
    - test_form_submission.py
    - test_form_settings.py
    - test_form.py
    - __init__.py
  - templatetags
    - test_form.py
    - __init__.py
    - test_url_replace.py
  - fields
    - test_fields.py
    - test_streamfield.py
    - test_registering.py
    - test_base_field.py
    - __init__.py
    - test_hook_select_field.py
  - urls.py
  - test_case.py
  - forms
    - test_form_builder.py
    - __init__.py
  - __init__.py
  - migrations
    - 0001_initial.py
    - __init__.py
  - settings.py
  - blocks
    - test_form_chooser_block.py
    - test_info_block.py
    - __init__.py
    - test_form_block.py
  - serializers
    - test_form_submission.py
    - __init__.py
  - management
    - test_prunesubmissions.py
    - __init__.py
- requirements.txt
- Dockerfile
- .gitignore
- docs
  - templates.rst
  - hooks.rst
  - Makefile
  - advanced.rst
  - usage.rst
  - housekeeping.rst
  - permissions.rst
  - contributors.rst
  - screenshots.rst
  - fields.rst
  - README.md
  - settings.rst
  - requirements.txt
  - index.rst
  - conf.py
  - changelog.rst
  - submission.rst
  - _static
    - images
  - installation.rst
- .dockerignore
- MANIFEST.in
- docker-compose.yml
- tox.ini

from django.db import models, transaction
from django.test import TransactionTestCase
from wagtailstreamforms.models import Form, FormSubmission, FormSubmissionFile

from ..test_case import AppTestCase


class ModelGenericTests(AppTestCase):
    def test_str(self):
        model = FormSubmissionFile(file=self.get_file())
        self.assertEqual(model.__str__(), model.file.name)

    def test_ordering(self):
        self.assertEqual(FormSubmissionFile._meta.ordering, ["field", "file"])


class ModelFieldTests(AppTestCase):
    def test_submission(self):
        field = self.get_field(FormSubmissionFile, "submission")
        self.assertModelPKField(field, FormSubmission, models.CASCADE)

    def test_field(self):
        field = self.get_field(FormSubmissionFile, "field")
        self.assertModelField(field, models.CharField)

    def test_file(self):
        field = self.get_field(FormSubmissionFile, "file")
        self.assertModelField(field, models.FileField)


class ModelPropertyTests(AppTestCase):
    def test_url(self):
        model = FormSubmissionFile(file=self.get_file())
        self.assertEqual(model.url, model.file.url)


class DeleteTests(TransactionTestCase):
    fixtures = ["test"]

    def test_files_are_deleted_on_commit(self):
        test_file = AppTestCase().get_file()
        name = None
        with transaction.atomic():
            form = Form.objects.get(pk=1)
            submission = FormSubmission.objects.create(form=form, form_data={})
            file = FormSubmissionFile.objects.create(
                submission=submission, field="field", file=test_file
            )
            name = file.file.name
            self.assertTrue(file.file.storage.exists(name))
            file.delete()
            self.assertTrue(file.file.storage.exists(name))
        self.assertFalse(file.file.storage.exists(name))