1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
from django.db import models
from psqlextra.util import postgres_manager
from .fake_model import get_fake_model
def test_manager_context():
"""Tests whether the :see:postgres_manager context manager can be used to
get access to :see:PostgresManager on a model that does not use it directly
or inherits from :see:PostgresModel."""
model = get_fake_model(
{"myfield": models.CharField(max_length=255, unique=True)}, models.Model
)
with postgres_manager(model) as manager:
manager.upsert(conflict_target=["myfield"], fields=dict(myfield="beer"))
assert manager.first().myfield == "beer"
|
