Prodream/ormar
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add connectiong to db in tests

Browse Source
This commit is contained in:
collerek
2021-01-12 14:47:55 +01:00
parent 4209d37364
commit 8d619d5fc1
1 changed files with 62 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
tests/test_forward_refs.py
View File

@ -123,11 +123,15 @@ def test_proper_field_init():
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_self_relation(): async def test_self_relation():
async with db:
async with db.transaction(force_rollback=True):
sam = await Person.objects.create(name="Sam") sam = await Person.objects.create(name="Sam")
joe = await Person(name="Joe", supervisor=sam).save() joe = await Person(name="Joe", supervisor=sam).save()
assert joe.supervisor.name == "Sam" assert joe.supervisor.name == "Sam"
joe_check = await Person.objects.select_related("supervisor").get(name="Joe") joe_check = await Person.objects.select_related("supervisor").get(
name="Joe"
)
assert joe_check.supervisor.name == "Sam" assert joe_check.supervisor.name == "Sam"
sam_check = await Person.objects.select_related("employees").get(name="Sam") sam_check = await Person.objects.select_related("employees").get(name="Sam")
@ -137,11 +141,17 @@ async def test_self_relation():
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_other_forwardref_relation(cleanup): async def test_other_forwardref_relation(cleanup):
async with db:
async with db.transaction(force_rollback=True):
checkers = await Game.objects.create(name="checkers") checkers = await Game.objects.create(name="checkers")
uno = await Game(name="Uno").save() uno = await Game(name="Uno").save()
await Child(name="Billy", favourite_game=uno, least_favourite_game=checkers).save() await Child(
await Child(name="Kate", favourite_game=checkers, least_favourite_game=uno).save() name="Billy", favourite_game=uno, least_favourite_game=checkers
).save()
await Child(
name="Kate", favourite_game=checkers, least_favourite_game=uno
).save()
billy_check = await Child.objects.select_related( billy_check = await Child.objects.select_related(
["favourite_game", "least_favourite_game"] ["favourite_game", "least_favourite_game"]
@ -149,15 +159,17 @@ async def test_other_forwardref_relation(cleanup):
assert billy_check.favourite_game == uno assert billy_check.favourite_game == uno
assert billy_check.least_favourite_game == checkers assert billy_check.least_favourite_game == checkers
uno_check = await Game.objects.select_related(["liked_by", "not_liked_by"]).get( uno_check = await Game.objects.select_related(
name="Uno" ["liked_by", "not_liked_by"]
) ).get(name="Uno")
assert uno_check.liked_by[0].name == "Billy" assert uno_check.liked_by[0].name == "Billy"
assert uno_check.not_liked_by[0].name == "Kate" assert uno_check.not_liked_by[0].name == "Kate"
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_m2m_self_forwardref_relation(cleanup): async def test_m2m_self_forwardref_relation(cleanup):
async with db:
async with db.transaction(force_rollback=True):
checkers = await Game.objects.create(name="checkers") checkers = await Game.objects.create(name="checkers")
uno = await Game(name="Uno").save() uno = await Game(name="Uno").save()
jenga = await Game(name="Jenga").save() jenga = await Game(name="Jenga").save()
@ -186,6 +198,8 @@ async def test_m2m_self_forwardref_relation(cleanup):
assert billy_check.friends[1].name == "Steve" assert billy_check.friends[1].name == "Steve"
assert billy_check.favourite_game.name == "Uno" assert billy_check.favourite_game.name == "Uno"
kate_check = await Child.objects.select_related(["also_friends",]).get(name="Kate") kate_check = await Child.objects.select_related(["also_friends",]).get(
name="Kate"
)
assert len(kate_check.also_friends) == 1 assert len(kate_check.also_friends) == 1
assert kate_check.also_friends[0].name == "Billy" assert kate_check.also_friends[0].name == "Billy"