• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• django-test-migrations-master
  • .github
    • FUNDING.yml
    • workflows
      • test.yml
      • misspell.yml
  • poetry.lock
  • Makefile
  • LICENSE
  • .editorconfig
  • CHANGELOG.md
  • django_test_migrations
    • sql.py
    • logic
      • __init__.py
      • migrations.py
    • exceptions.py
    • autonames.py
    • py.typed
    • migrator.py
    • __init__.py
    • types.py
    • plan.py
    • contrib
      • pytest_plugin.py
      • django_checks.py
      • __init__.py
      • unittest_case.py
  • setup.cfg
  • README.md
  • tests
    • test_contrib
      • test_pytest_plugin
        • test_pytest_plugin_direct.py
        • test_pytest_plugin_reverse.py
        • test_signals.py
        • __init__.py
        • test_plugin.py
      • test_django_checks
        • test_autonames_check.py
      • test_unittest_case
        • test_unittest_case.py
        • test_signals.py
        • __init__.py
    • test_plan
      • test_truncate_plan.py
      • test_all_migrations.py
      • conftest.py
    • test_migrator
      • test_migrator.py
    • test_autonames
      • test_autonames.py
    • test_logic
      • test_migrations.py
    • test_exceptions
      • test_migration_not_found.py
    • test_sql
      • test_flush_utils.py
      • test_drop_models_table.py
      • test_django_migrations_table_utils.py
    • conftest.py
  • pyproject.toml
  • django_test_app
    • manage.py
    • django_test_app
      • wsgi.py
      • urls.py
      • __init__.py
      • settings.py
    • main_app
      • apps.py
      • logic
        • pure
          • __init__.py
          • migrations.py
        • __init__.py
      • models.py
      • views.py
      • __init__.py
      • migrations
        • 0003_update_is_clean.py
        • 0002_someitem_is_clean.py
        • 0001_initial.py
        • 0004_auto_20191119_2125.py
        • 0005_auto_20200329_1118.py
        • __init__.py
  • .gitignore

"""
This module covers simple direct migrations.

We test both schema and data-migrations here.
"""

import pytest
from django.core.exceptions import FieldError
from django.db.utils import IntegrityError


@pytest.mark.django_db
def test_pytest_plugin_initial(migrator):
    """Ensures that the initial migration works."""
    old_state = migrator.apply_initial_migration(('main_app', None))

    with pytest.raises(LookupError):
        # Models does not yet exist:
        old_state.apps.get_model('main_app', 'SomeItem')

    new_state = migrator.apply_tested_migration(('main_app', '0001_initial'))
    # After the initial migration is done, we can use the model state:
    SomeItem = new_state.apps.get_model('main_app', 'SomeItem')
    assert SomeItem.objects.filter(string_field='').count() == 0


@pytest.mark.django_db
def test_pytest_plugin0001(migrator):
    """Ensures that the first migration works."""
    old_state = migrator.apply_initial_migration(('main_app', '0001_initial'))
    SomeItem = old_state.apps.get_model('main_app', 'SomeItem')

    with pytest.raises(FieldError):
        SomeItem.objects.filter(is_clean=True)

    new_state = migrator.apply_tested_migration(
        ('main_app', '0002_someitem_is_clean'),
    )
    SomeItem = new_state.apps.get_model('main_app', 'SomeItem')

    assert SomeItem.objects.filter(is_clean=True).count() == 0


@pytest.mark.django_db
def test_pytest_plugin0002(migrator):
    """Ensures that the second migration works."""
    old_state = migrator.apply_initial_migration(
        ('main_app', '0002_someitem_is_clean'),
    )
    SomeItem = old_state.apps.get_model('main_app', 'SomeItem')
    SomeItem.objects.create(string_field='a')
    SomeItem.objects.create(string_field='a b')

    assert SomeItem.objects.count() == 2
    assert SomeItem.objects.filter(is_clean=True).count() == 2

    new_state = migrator.apply_tested_migration(
        ('main_app', '0003_update_is_clean'),
    )
    SomeItem = new_state.apps.get_model('main_app', 'SomeItem')

    assert SomeItem.objects.count() == 2
    assert SomeItem.objects.filter(is_clean=True).count() == 1


@pytest.mark.django_db
def test_pytest_plugin0003(migrator):
    """Ensures that the third migration works."""
    old_state = migrator.apply_initial_migration(
        ('main_app', '0003_update_is_clean'),
    )
    SomeItem = old_state.apps.get_model('main_app', 'SomeItem')
    SomeItem.objects.create(string_field='a')  # default is still there

    assert SomeItem.objects.count() == 1
    assert SomeItem.objects.filter(is_clean=True).count() == 1

    new_state = migrator.apply_tested_migration(
        ('main_app', '0004_auto_20191119_2125'),
    )
    SomeItem = new_state.apps.get_model('main_app', 'SomeItem')

    with pytest.raises(IntegrityError):
        SomeItem.objects.create(string_field='b')  # no default anymore