• Search by Module
• Search by Words
• Search Projects

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

• django-postgres-extra-master
  • .circleci
    • config.yml
  • .prospector.yaml
  • .coveragerc
  • psqlextra
    • apps.py
    • sql.py
    • util.py
    • expressions.py
    • type_assertions.py
    • backend
      • introspection.py
      • schema.py
      • base_impl.py
      • __init__.py
      • migrations
        • operations
          • create_view_model.py
          • apply_state.py
          • delete_view_model.py
          • add_default_partition.py
          • add_range_partition.py
          • delete_materialized_view_model.py
          • add_list_partition.py
          • create_materialized_view_model.py
          • delete_partitioned_model.py
          • delete_partition.py
          • create_partitioned_model.py
          • partition.py
          • __init__.py
          • delete_range_partition.py
          • delete_list_partition.py
          • delete_default_partition.py
        • state
          • materialized_view.py
          • model.py
          • __init__.py
          • view.py
          • partitioning.py
        • patched_project_state.py
        • patched_autodetector.py
        • __init__.py
        • README.md
        • patched_migrations.py
      • side_effects
        • hstore_required.py
        • __init__.py
        • hstore_unique.py
      • base.py
      • operations.py
    • models
      • options.py
      • __init__.py
      • partitioned.py
      • view.py
      • base.py
    • fields
      • hstore_field.py
      • __init__.py
    • query.py
    • __init__.py
    • compiler.py
    • partitioning
      • current_time_strategy.py
      • time_partition_size.py
      • config.py
      • time_strategy.py
      • range_partition.py
      • constants.py
      • manager.py
      • range_strategy.py
      • partition.py
      • __init__.py
      • shorthands.py
      • error.py
      • plan.py
      • strategy.py
      • time_partition.py
    • manager
      • manager.py
      • __init__.py
    • types.py
    • indexes
      • case_insensitive_unique_index.py
      • conditional_unique_index.py
      • __init__.py
    • management
      • commands
        • pgpartition.py
        • pgmakemigrations.py
        • pgrefreshmv.py
        • __init__.py
      • __init__.py
  • LICENSE
  • .readthedocs.yml
  • manage.py
  • CONTRIBUTING.md
  • setup.py
  • setup.cfg
  • README.md
  • tests
    • fake_model.py
    • test_schema_editor_view.py
    • test_on_conflict.py
    • test_partitioned_model.py
    • test_db_backend.py
    • test_manager_context.py
    • test_upsert.py
    • test_insert.py
    • db_introspection.py
    • test_view_models.py
    • snapshots
      • __init__.py
      • snap_test_management_command_partition.py
    • test_partitioning_manager.py
    • test_management_command_partition.py
    • test_query.py
    • test_migration_operations.py
    • test_conditional_unique_index.py
    • test_schema_editor_partitioning.py
    • benchmarks
      • test_upsert.py
      • test_insert_nothing.py
      • __init__.py
      • test_upsert_bulk.py
    • test_partitioning_time.py
    • test_materialized_view_model.py
    • test_on_conflict_update.py
    • test_hstore_required.py
    • __init__.py
    • test_hstore_autodetect.py
    • test_query_values.py
    • test_hstore_unique.py
    • test_partitioned_model_state.py
    • test_hstore_field.py
    • test_make_migrations.py
    • test_case_insensitive_unique_index.py
    • migrations.py
    • test_on_conflict_nothing.py
    • conftest.py
    • test_manager.py
  • pytest.ini
  • settings.py
  • pyproject.toml
  • .gitignore
  • docs
    • Makefile
    • source
      • conflict_handling.rst
      • expressions.rst
      • table_partitioning.rst
      • deletion.rst
      • snippets
        • manager_model_warning.rst
        • postgres_doc_links.rst
        • django_doc_links.rst
      • annotations.rst
      • hstore.rst
      • indexes.rst
      • major_releases.rst
      • api_reference.rst
      • settings.rst
      • index.rst
      • conf.py
      • managers_models.rst
      • installation.rst
    • make.bat
    • .gitignore
  • pytest-benchmark.ini
  • tox.ini

import uuid

import pytest

from django.apps import apps
from django.db import models

from psqlextra.backend.migrations.state import (
    PostgresPartitionedModelState,
    PostgresPartitionState,
)
from psqlextra.manager import PostgresManager
from psqlextra.models import PostgresPartitionedModel
from psqlextra.types import PostgresPartitioningMethod

from .fake_model import define_fake_partitioned_model


@pytest.fixture
def model():
    fields = {"name": models.TextField(), "category": models.TextField()}

    partitioning_options = {
        "method": PostgresPartitioningMethod.LIST,
        "key": ["category"],
    }

    model = define_fake_partitioned_model(fields, partitioning_options)
    return model


def test_partitioned_model_state_copies():
    """Tests whether cloning the model state properly copies all the options.

    If it does not copy them, bad things can happen as the state is
    mutated to build up migration state.
    """

    options = dict(method=PostgresPartitioningMethod.RANGE, key=["timestamp"])

    state = PostgresPartitionedModelState(
        app_label="tests",
        name=str(uuid.uuid4()),
        fields=[],
        options=None,
        partitioning_options=options,
        bases=(PostgresPartitionedModel,),
    )

    assert options is not state.partitioning_options


def test_partitioned_model_state_from_model(model):
    """Tests whether creating state from an existing model works as
    expected."""

    state = PostgresPartitionedModelState.from_model(model)
    assert state.partitions == {}
    assert (
        state.partitioning_options["method"] == model._partitioning_meta.method
    )
    assert state.partitioning_options["key"] == model._partitioning_meta.key


def test_partitioned_model_clone(model):
    """Tests whether cloning the state actually clones the partitioning
    options.

    If its not a copy, but a reference instead, bad things can happen as
    the options are mutated to build up migration state.
    """

    state = PostgresPartitionedModelState.from_model(model)
    state.partitions = {
        "pt1": PostgresPartitionState(
            app_label="tests", model_name="tests", name="pt1"
        )
    }

    state_copy = state.clone()
    assert state.partitions is not state_copy.partitions
    assert state.partitioning_options is not state_copy.partitioning_options


def test_partitioned_model_render(model):
    """Tests whether the state can be rendered into a valid model class."""

    options = dict(method=PostgresPartitioningMethod.RANGE, key=["timestamp"])

    state = PostgresPartitionedModelState(
        app_label="tests",
        name=str(uuid.uuid4()),
        fields=[("name", models.TextField())],
        options=None,
        partitioning_options=options,
        bases=(PostgresPartitionedModel,),
        managers=[("cookie", PostgresManager())],
    )

    rendered_model = state.render(apps)

    assert issubclass(rendered_model, PostgresPartitionedModel)
    assert rendered_model.name
    assert isinstance(rendered_model.objects, PostgresManager)
    assert isinstance(rendered_model.cookie, PostgresManager)
    assert rendered_model.__name__ == state.name
    assert rendered_model._meta.apps == apps
    assert rendered_model._meta.app_label == "tests"
    assert rendered_model._partitioning_meta.method == options["method"]
    assert rendered_model._partitioning_meta.key == options["key"]