expose querysetproxy on reverse of foreignkey (virtual fk), add additional methods from queryset to querysetproxy
This commit is contained in:
collerek
@ -9,7 +9,7 @@ import pytest
|
||||
import sqlalchemy
|
||||
|
||||
import ormar
|
||||
from ormar.exceptions import QueryDefinitionError, NoMatch
|
||||
from ormar.exceptions import QueryDefinitionError, NoMatch, ModelError
|
||||
from tests.settings import DATABASE_URL
|
||||
|
||||
database = databases.Database(DATABASE_URL, force_rollback=True)
|
||||
@ -117,6 +117,11 @@ def test_model_class():
|
||||
assert isinstance(User.Meta.table, sqlalchemy.Table)
|
||||
|
||||
|
||||
def test_wrong_field_name():
|
||||
with pytest.raises(ModelError):
|
||||
User(non_existing_pk=1)
|
||||
|
||||
|
||||
def test_model_pk():
|
||||
user = User(pk=1)
|
||||
assert user.pk == 1
|
||||
|
||||
182
tests/test_queryproxy_on_m2m_models.py
Normal file
182
tests/test_queryproxy_on_m2m_models.py
Normal file
@ -0,0 +1,182 @@
|
||||
import asyncio
|
||||
from typing import List, Optional, Union
|
||||
|
||||
import databases
|
||||
import pytest
|
||||
import sqlalchemy
|
||||
|
||||
import ormar
|
||||
from tests.settings import DATABASE_URL
|
||||
|
||||
database = databases.Database(DATABASE_URL, force_rollback=True)
|
||||
metadata = sqlalchemy.MetaData()
|
||||
|
||||
|
||||
class Subject(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "subjects"
|
||||
database = database
|
||||
metadata = metadata
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
name: str = ormar.String(max_length=80)
|
||||
|
||||
|
||||
class Author(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "authors"
|
||||
database = database
|
||||
metadata = metadata
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
first_name: str = ormar.String(max_length=80)
|
||||
last_name: str = ormar.String(max_length=80)
|
||||
|
||||
|
||||
class Category(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "categories"
|
||||
database = database
|
||||
metadata = metadata
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
name: str = ormar.String(max_length=40)
|
||||
sort_order: int = ormar.Integer(nullable=True)
|
||||
subject: Optional[Subject] = ormar.ForeignKey(Subject)
|
||||
|
||||
|
||||
class PostCategory(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "posts_categories"
|
||||
database = database
|
||||
metadata = metadata
|
||||
|
||||
|
||||
class Post(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "posts"
|
||||
database = database
|
||||
metadata = metadata
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
title: str = ormar.String(max_length=200)
|
||||
categories: Optional[Union[Category, List[Category]]] = ormar.ManyToMany(
|
||||
Category, through=PostCategory
|
||||
)
|
||||
author: Optional[Author] = ormar.ForeignKey(Author)
|
||||
|
||||
|
||||
@pytest.fixture(scope="module")
|
||||
def event_loop():
|
||||
loop = asyncio.get_event_loop()
|
||||
yield loop
|
||||
loop.close()
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True, scope="module")
|
||||
async def create_test_database():
|
||||
engine = sqlalchemy.create_engine(DATABASE_URL)
|
||||
metadata.create_all(engine)
|
||||
yield
|
||||
metadata.drop_all(engine)
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_queryset_methods():
|
||||
async with database:
|
||||
async with database.transaction(force_rollback=True):
|
||||
guido = await Author.objects.create(
|
||||
first_name="Guido", last_name="Van Rossum"
|
||||
)
|
||||
subject = await Subject(name="Random").save()
|
||||
post = await Post.objects.create(title="Hello, M2M", author=guido)
|
||||
news = await Category.objects.create(
|
||||
name="News", sort_order=1, subject=subject
|
||||
)
|
||||
breaking = await Category.objects.create(
|
||||
name="Breaking", sort_order=3, subject=subject
|
||||
)
|
||||
|
||||
# Add a category to a post.
|
||||
await post.categories.add(news)
|
||||
await post.categories.add(breaking)
|
||||
|
||||
category = await post.categories.get_or_create(name="News")
|
||||
assert category == news
|
||||
assert len(post.categories) == 1
|
||||
|
||||
category = await post.categories.get_or_create(name="Breaking News")
|
||||
assert category != breaking
|
||||
assert category.pk is not None
|
||||
assert len(post.categories) == 2
|
||||
|
||||
await post.categories.update_or_create(pk=category.pk, name="Urgent News")
|
||||
assert len(post.categories) == 2
|
||||
cat = await post.categories.get_or_create(name="Urgent News")
|
||||
assert cat.pk == category.pk
|
||||
assert len(post.categories) == 1
|
||||
|
||||
await post.categories.remove(cat)
|
||||
await cat.delete()
|
||||
|
||||
assert len(post.categories) == 0
|
||||
|
||||
category = await post.categories.update_or_create(
|
||||
name="Weather News", sort_order=2, subject=subject
|
||||
)
|
||||
assert category.pk is not None
|
||||
assert category.posts[0] == post
|
||||
|
||||
assert len(post.categories) == 1
|
||||
|
||||
categories = await post.categories.all()
|
||||
assert len(categories) == 3 == len(post.categories)
|
||||
|
||||
assert await post.categories.exists()
|
||||
assert 3 == await post.categories.count()
|
||||
|
||||
categories = await post.categories.limit(2).all()
|
||||
assert len(categories) == 2 == len(post.categories)
|
||||
|
||||
categories2 = await post.categories.limit(2).offset(1).all()
|
||||
assert len(categories2) == 2 == len(post.categories)
|
||||
assert categories != categories2
|
||||
|
||||
categories = await post.categories.order_by("-sort_order").all()
|
||||
assert len(categories) == 3 == len(post.categories)
|
||||
assert post.categories[2].name == "News"
|
||||
assert post.categories[0].name == "Breaking"
|
||||
|
||||
categories = await post.categories.exclude(name__icontains="news").all()
|
||||
assert len(categories) == 1 == len(post.categories)
|
||||
assert post.categories[0].name == "Breaking"
|
||||
|
||||
categories = (
|
||||
await post.categories.filter(name__icontains="news")
|
||||
.order_by("-name")
|
||||
.all()
|
||||
)
|
||||
assert len(categories) == 2 == len(post.categories)
|
||||
assert post.categories[0].name == "Weather News"
|
||||
assert post.categories[1].name == "News"
|
||||
|
||||
categories = await post.categories.fields("name").all()
|
||||
assert len(categories) == 3 == len(post.categories)
|
||||
for cat in post.categories:
|
||||
assert cat.sort_order is None
|
||||
|
||||
categories = await post.categories.exclude_fields("sort_order").all()
|
||||
assert len(categories) == 3 == len(post.categories)
|
||||
for cat in post.categories:
|
||||
assert cat.sort_order is None
|
||||
assert cat.subject.name is None
|
||||
|
||||
categories = await post.categories.select_related("subject").all()
|
||||
assert len(categories) == 3 == len(post.categories)
|
||||
for cat in post.categories:
|
||||
assert cat.subject.name is not None
|
||||
|
||||
categories = await post.categories.prefetch_related("subject").all()
|
||||
assert len(categories) == 3 == len(post.categories)
|
||||
for cat in post.categories:
|
||||
assert cat.subject.name is not None
|
||||
233
tests/test_reverse_fk_queryset.py
Normal file
233
tests/test_reverse_fk_queryset.py
Normal file
@ -0,0 +1,233 @@
|
||||
from typing import Optional
|
||||
|
||||
import databases
|
||||
import pytest
|
||||
import sqlalchemy
|
||||
|
||||
import ormar
|
||||
from tests.settings import DATABASE_URL
|
||||
|
||||
database = databases.Database(DATABASE_URL, force_rollback=True)
|
||||
metadata = sqlalchemy.MetaData()
|
||||
|
||||
|
||||
class Album(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "albums"
|
||||
metadata = metadata
|
||||
database = database
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
name: str = ormar.String(max_length=100)
|
||||
is_best_seller: bool = ormar.Boolean(default=False)
|
||||
|
||||
|
||||
class Writer(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "writers"
|
||||
metadata = metadata
|
||||
database = database
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
name: str = ormar.String(max_length=100)
|
||||
|
||||
|
||||
class Track(ormar.Model):
|
||||
class Meta:
|
||||
tablename = "tracks"
|
||||
metadata = metadata
|
||||
database = database
|
||||
|
||||
id: int = ormar.Integer(primary_key=True)
|
||||
album: Optional[Album] = ormar.ForeignKey(Album)
|
||||
title: str = ormar.String(max_length=100)
|
||||
position: int = ormar.Integer()
|
||||
play_count: int = ormar.Integer(nullable=True)
|
||||
written_by: Optional[Writer] = ormar.ForeignKey(Writer)
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True)
|
||||
@pytest.mark.asyncio
|
||||
async def sample_data():
|
||||
album = await Album(name="Malibu").save()
|
||||
writer1 = await Writer.objects.create(name="John")
|
||||
writer2 = await Writer.objects.create(name="Sue")
|
||||
track1 = await Track(
|
||||
album=album, title="The Bird", position=1, play_count=30, written_by=writer1
|
||||
).save()
|
||||
track2 = await Track(
|
||||
album=album,
|
||||
title="Heart don't stand a chance",
|
||||
position=2,
|
||||
play_count=20,
|
||||
written_by=writer2,
|
||||
).save()
|
||||
tracks3 = await Track(
|
||||
album=album, title="The Waters", position=3, play_count=10, written_by=writer1
|
||||
).save()
|
||||
return album, [track1, track2, tracks3]
|
||||
|
||||
|
||||
@pytest.fixture(autouse=True, scope="module")
|
||||
def create_test_database():
|
||||
engine = sqlalchemy.create_engine(DATABASE_URL)
|
||||
metadata.drop_all(engine)
|
||||
metadata.create_all(engine)
|
||||
yield
|
||||
metadata.drop_all(engine)
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_quering_by_reverse_fk(sample_data):
|
||||
async with database:
|
||||
async with database.transaction(force_rollback=True):
|
||||
track1 = sample_data[1][0]
|
||||
album = await Album.objects.first()
|
||||
|
||||
assert await album.tracks.exists()
|
||||
assert await album.tracks.count() == 3
|
||||
|
||||
track = await album.tracks.get_or_create(
|
||||
title="The Bird", position=1, play_count=30
|
||||
)
|
||||
assert track == track1
|
||||
assert len(album.tracks) == 1
|
||||
|
||||
track = await album.tracks.get_or_create(
|
||||
title="The Bird2", position=4, play_count=5
|
||||
)
|
||||
assert track != track1
|
||||
assert track.pk is not None
|
||||
assert len(album.tracks) == 2
|
||||
|
||||
await album.tracks.update_or_create(pk=track.pk, play_count=50)
|
||||
assert len(album.tracks) == 2
|
||||
track = await album.tracks.get_or_create(title="The Bird2")
|
||||
assert track.play_count == 50
|
||||
assert len(album.tracks) == 1
|
||||
|
||||
await album.tracks.remove(track)
|
||||
assert track.album is None
|
||||
await track.delete()
|
||||
|
||||
assert len(album.tracks) == 0
|
||||
|
||||
track6 = await album.tracks.update_or_create(
|
||||
title="The Bird3", position=4, play_count=5
|
||||
)
|
||||
assert track6.pk is not None
|
||||
assert track6.play_count == 5
|
||||
|
||||
assert len(album.tracks) == 1
|
||||
|
||||
await album.tracks.remove(track6)
|
||||
assert track6.album is None
|
||||
await track6.delete()
|
||||
|
||||
assert len(album.tracks) == 0
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_getting(sample_data):
|
||||
async with database:
|
||||
async with database.transaction(force_rollback=True):
|
||||
album = sample_data[0]
|
||||
track1 = await album.tracks.fields(["album", "title", "position"]).get(
|
||||
title="The Bird"
|
||||
)
|
||||
track2 = await album.tracks.exclude_fields("play_count").get(
|
||||
title="The Bird"
|
||||
)
|
||||
for track in [track1, track2]:
|
||||
assert track.title == "The Bird"
|
||||
assert track.album == album
|
||||
assert track.play_count is None
|
||||
|
||||
assert len(album.tracks) == 1
|
||||
|
||||
tracks = await album.tracks.all()
|
||||
assert len(tracks) == 3
|
||||
|
||||
assert len(album.tracks) == 3
|
||||
|
||||
tracks = await album.tracks.order_by("play_count").all()
|
||||
assert len(tracks) == 3
|
||||
assert tracks[0].title == "The Waters"
|
||||
assert tracks[2].title == "The Bird"
|
||||
|
||||
assert len(album.tracks) == 3
|
||||
|
||||
track = await album.tracks.create(
|
||||
title="The Bird Fly Away", position=4, play_count=10
|
||||
)
|
||||
assert track.title == "The Bird Fly Away"
|
||||
assert track.position == 4
|
||||
assert track.album == album
|
||||
|
||||
assert len(album.tracks) == 4
|
||||
|
||||
tracks = await album.tracks.all()
|
||||
assert len(tracks) == 4
|
||||
|
||||
tracks = await album.tracks.limit(2).all()
|
||||
assert len(tracks) == 2
|
||||
|
||||
tracks2 = await album.tracks.limit(2).offset(2).all()
|
||||
assert len(tracks2) == 2
|
||||
assert tracks != tracks2
|
||||
|
||||
tracks3 = await album.tracks.filter(play_count__lt=15).all()
|
||||
assert len(tracks3) == 2
|
||||
|
||||
tracks4 = await album.tracks.exclude(play_count__lt=15).all()
|
||||
assert len(tracks4) == 2
|
||||
assert tracks3 != tracks4
|
||||
|
||||
assert len(album.tracks) == 2
|
||||
|
||||
await album.tracks.clear()
|
||||
tracks = await album.tracks.all()
|
||||
assert len(tracks) == 0
|
||||
assert len(album.tracks) == 0
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_loading_related(sample_data):
|
||||
async with database:
|
||||
async with database.transaction(force_rollback=True):
|
||||
album = sample_data[0]
|
||||
tracks = await album.tracks.select_related("written_by").all()
|
||||
assert len(tracks) == 3
|
||||
assert len(album.tracks) == 3
|
||||
for track in tracks:
|
||||
assert track.written_by is not None
|
||||
|
||||
tracks = await album.tracks.prefetch_related("written_by").all()
|
||||
assert len(tracks) == 3
|
||||
assert len(album.tracks) == 3
|
||||
for track in tracks:
|
||||
assert track.written_by is not None
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_adding_removing(sample_data):
|
||||
async with database:
|
||||
async with database.transaction(force_rollback=True):
|
||||
album = sample_data[0]
|
||||
track_new = await Track(title="Rainbow", position=5, play_count=300).save()
|
||||
await album.tracks.add(track_new)
|
||||
assert track_new.album == album
|
||||
assert len(album.tracks) == 4
|
||||
|
||||
track_check = await Track.objects.get(title="Rainbow")
|
||||
assert track_check.album == album
|
||||
|
||||
track_test = await Track.objects.get(title="Rainbow")
|
||||
assert track_test.album == album
|
||||
|
||||
await album.tracks.remove(track_new)
|
||||
assert track_new.album is None
|
||||
assert len(album.tracks) == 3
|
||||
|
||||
track_test = await Track.objects.get(title="Rainbow")
|
||||
assert track_test.album is None
|
||||
Reference in New Issue
Block a user