• 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

from django.db import models

from psqlextra.fields import HStoreField

from .fake_model import get_fake_model


def test_upsert():
    """Tests whether simple upserts works correctly."""

    model = get_fake_model(
        {
            "title": HStoreField(uniqueness=["key1"]),
            "cookies": models.CharField(max_length=255, null=True),
        }
    )

    obj1 = model.objects.upsert_and_get(
        conflict_target=[("title", "key1")],
        fields=dict(title={"key1": "beer"}, cookies="cheers"),
    )

    obj1.refresh_from_db()
    assert obj1.title["key1"] == "beer"
    assert obj1.cookies == "cheers"

    obj2 = model.objects.upsert_and_get(
        conflict_target=[("title", "key1")],
        fields=dict(title={"key1": "beer"}, cookies="choco"),
    )

    obj1.refresh_from_db()
    obj2.refresh_from_db()

    # assert both objects are the same
    assert obj1.id == obj2.id
    assert obj1.title["key1"] == "beer"
    assert obj1.cookies == "choco"
    assert obj2.title["key1"] == "beer"
    assert obj2.cookies == "choco"


def test_upsert_explicit_pk():
    """Tests whether upserts works when the primary key is explicitly
    specified."""

    model = get_fake_model(
        {
            "name": models.CharField(max_length=255, primary_key=True),
            "cookies": models.CharField(max_length=255, null=True),
        }
    )

    obj1 = model.objects.upsert_and_get(
        conflict_target=[("name")],
        fields=dict(name="the-object", cookies="first-cheers"),
    )

    obj1.refresh_from_db()
    assert obj1.name == "the-object"
    assert obj1.cookies == "first-cheers"

    obj2 = model.objects.upsert_and_get(
        conflict_target=[("name")],
        fields=dict(name="the-object", cookies="second-boo"),
    )

    obj1.refresh_from_db()
    obj2.refresh_from_db()

    # assert both objects are the same
    assert obj1.pk == obj2.pk
    assert obj1.name == "the-object"
    assert obj1.cookies == "second-boo"
    assert obj2.name == "the-object"
    assert obj2.cookies == "second-boo"


def test_upsert_bulk():
    """Tests whether bulk_upsert works properly."""

    model = get_fake_model(
        {
            "first_name": models.CharField(
                max_length=255, null=True, unique=True
            ),
            "last_name": models.CharField(max_length=255, null=True),
        }
    )

    model.objects.bulk_upsert(
        conflict_target=["first_name"],
        rows=[
            dict(first_name="Swen", last_name="Kooij"),
            dict(first_name="Henk", last_name="Test"),
        ],
    )

    row_a = model.objects.get(first_name="Swen")
    row_b = model.objects.get(first_name="Henk")

    model.objects.bulk_upsert(
        conflict_target=["first_name"],
        rows=[
            dict(first_name="Swen", last_name="Test"),
            dict(first_name="Henk", last_name="Kooij"),
        ],
    )

    row_a.refresh_from_db()
    assert row_a.last_name == "Test"

    row_b.refresh_from_db()
    assert row_b.last_name == "Kooij"


def test_upsert_bulk_no_rows():
    """Tests whether bulk_upsert doesn't crash when specifying no rows or a
    falsy value."""

    model = get_fake_model(
        {"name": models.CharField(max_length=255, null=True, unique=True)}
    )

    model.objects.bulk_upsert(conflict_target=["name"], rows=[])

    model.objects.bulk_upsert(conflict_target=["name"], rows=None)


def test_bulk_upsert_return_models():
    """Tests whether models are returned instead of dictionaries when
    specifying the return_model=True argument."""

    model = get_fake_model(
        {
            "id": models.BigAutoField(primary_key=True),
            "name": models.CharField(max_length=255, unique=True),
        }
    )

    rows = [dict(name="John Smith"), dict(name="Jane Doe")]

    objs = model.objects.bulk_upsert(
        conflict_target=["name"], rows=rows, return_model=True
    )

    for index, obj in enumerate(objs, 1):
        assert isinstance(obj, model)
        assert obj.id == index


def test_bulk_upsert_accepts_getitem_iterable():
    """Tests whether an iterable only implementing the __getitem__ method works
    correctly."""

    class GetItemIterable:
        def __init__(self, items):
            self.items = items

        def __getitem__(self, key):
            return self.items[key]

    model = get_fake_model(
        {
            "id": models.BigAutoField(primary_key=True),
            "name": models.CharField(max_length=255, unique=True),
        }
    )

    rows = GetItemIterable([dict(name="John Smith"), dict(name="Jane Doe")])

    objs = model.objects.bulk_upsert(
        conflict_target=["name"], rows=rows, return_model=True
    )

    for index, obj in enumerate(objs, 1):
        assert isinstance(obj, model)
        assert obj.id == index


def test_bulk_upsert_accepts_iter_iterable():
    """Tests whether an iterable only implementing the __iter__ method works
    correctly."""

    class IterIterable:
        def __init__(self, items):
            self.items = items

        def __iter__(self):
            return iter(self.items)

    model = get_fake_model(
        {
            "id": models.BigAutoField(primary_key=True),
            "name": models.CharField(max_length=255, unique=True),
        }
    )

    rows = IterIterable([dict(name="John Smith"), dict(name="Jane Doe")])

    objs = model.objects.bulk_upsert(
        conflict_target=["name"], rows=rows, return_model=True
    )

    for index, obj in enumerate(objs, 1):
        assert isinstance(obj, model)
        assert obj.id == index