You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
2.8 KiB
87 lines
2.8 KiB
4 years ago
|
|
||
|
|
import unittest
|
||
|
|
from app import create_app, db
|
||
|
|
from app.models import Player, Game, Role, GamePlayer, Objective, ObjectiveMinimalEncoder, LocationEncoder
|
||
|
|
import app.main.routes as routes
|
||
|
|
from config import Config
|
||
|
|
|
||
|
|
class TestConfig(Config):
|
||
|
|
TESTING = True
|
||
|
|
WTF_CSRF_ENABLED = False
|
||
|
|
DEBUG = False
|
||
|
|
SQLALCHEMY_DATABASE_URI = 'sqlite://'
|
||
|
|
|
||
|
|
class RoutesCase(unittest.TestCase):
|
||
|
|
# implement this: https://stackoverflow.com/questions/47294304/how-to-mock-current-user-in-flask-templates
|
||
|
|
def setUp(self):
|
||
|
|
self.app = create_app(TestConfig)
|
||
|
|
self.app_context = self.app.app_context()
|
||
|
|
self.app_context.push()
|
||
|
|
db.create_all()
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
db.session.remove()
|
||
|
|
db.drop_all()
|
||
|
|
self.app_context.pop()
|
||
|
|
|
||
|
|
def test_is_game_owner(self):
|
||
|
|
g1 = Game(name='TestGame')
|
||
|
|
p1 = Player(name='Henk')
|
||
|
|
p2 = Player(name='Alfred')
|
||
|
|
g1.game_players.append(GamePlayer(player=p1, role=Role.owner))
|
||
|
|
g1.game_players.append(GamePlayer(player=p2, role=Role.bunny))
|
||
|
|
db.session.add(g1)
|
||
|
|
db.session.commit()
|
||
|
|
self.assertTrue(routes.is_game_owner(g1, p1))
|
||
|
|
self.assertFalse(routes.is_game_owner(g1, p2))
|
||
|
|
|
||
|
|
def test_is_player_game_owner(self):
|
||
|
|
g1 = Game(name='TestGame')
|
||
|
|
g2 = Game(name='AnotherGame')
|
||
|
|
|
||
|
|
p1 = Player(name='Henk')
|
||
|
|
p2 = Player(name='Alfred')
|
||
|
|
p3 = Player(name='Sasha')
|
||
|
|
|
||
|
|
g1.game_players.append(GamePlayer(player=p1, role=Role.owner))
|
||
|
|
g1.game_players.append(GamePlayer(player=p2, role=Role.bunny))
|
||
|
|
|
||
|
|
g2.game_players.append(GamePlayer(player=p1, role=Role.hunter))
|
||
|
|
g2.game_players.append(GamePlayer(player=p3, role=Role.bunny))
|
||
|
|
|
||
|
|
|
||
|
|
db.session.add(g1)
|
||
|
|
db.session.add(g2)
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
self.assertTrue(routes.is_player_game_owner(subject_player=p2, owner=p1), "owner owns subject_player's game")
|
||
|
|
self.assertFalse(routes.is_player_game_owner(subject_player=p3, owner=p1), "owner doesn't own subject_player's game")
|
||
|
|
self.assertTrue(routes.is_player_game_owner(subject_player=p1, owner=p1), "owner owns it own's game")
|
||
|
|
|
||
|
|
|
||
|
|
def test_is_objective_owner(self):
|
||
|
|
g1 = Game(name='TestGame')
|
||
|
|
g2 = Game(name='AnotherGame')
|
||
|
|
p1 = Player(name='Henk')
|
||
|
|
|
||
|
|
g1.game_players.append(GamePlayer(player=p1, role=Role.owner))
|
||
|
|
g2.game_players.append(GamePlayer(player=p1, role=Role.bunny))
|
||
|
|
|
||
|
|
o1 = Objective(name='o1')
|
||
|
|
o1.set_hash()
|
||
|
|
o2 = Objective(name='o2')
|
||
|
|
o2.set_hash()
|
||
|
|
|
||
|
|
g1.objectives.append(o1)
|
||
|
|
g2.objectives.append(o2)
|
||
|
|
|
||
|
|
db.session.add(g1)
|
||
|
|
db.session.add(g2)
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
self.assertTrue(routes.is_objective_owner(o1, p1))
|
||
|
|
self.assertFalse(routes.is_objective_owner(o2, p1))
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main(verbosity=2)
|