charasheet/src/character/tests/test_access.py

46 lines
1.4 KiB
Python

import pytest
from model_bakery import baker
from character.models import Character
from common.models import User
from party.models import Party
@pytest.mark.django_db
def test_can_access_own_character(client):
# Create a user
player = User.objects.create_user("username", password="password")
character = baker.make(Character, player=player)
client.force_login(player)
res = client.get(character.get_absolute_url())
assert res.status_code == 200
@pytest.mark.django_db
def test_cant_access_random_character(client):
# Create a user
player = User.objects.create_user("user", password="password")
other = User.objects.create_user("other", password="password")
character = baker.make(Character, player=other)
client.force_login(player)
res = client.get(character.get_absolute_url())
assert res.status_code == 404
@pytest.mark.django_db
def test_can_access_character_in_party(client):
# Create a user
player = User.objects.create_user("user", password="password")
friend = User.objects.create_user("friend", password="password")
character = baker.make(Character, player=player)
friend_character = baker.make(Character, player=friend)
party = baker.make(Party)
party.characters.add(character)
party.characters.add(friend_character)
client.force_login(player)
res = client.get(character.get_absolute_url())
assert res.status_code == 200