54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
from models.player import Player
|
|
from models.team import Team
|
|
from models.player_team import PlayerTeam
|
|
from models.team_integration import TeamDiscordIntegration, TeamLogsTfIntegration
|
|
from flask_testing import TestCase
|
|
from app_db import app, db, connect_db_with_app
|
|
|
|
SQLALCHEMY_DATABASE_URI = "sqlite:///:memory:"
|
|
connect_db_with_app(SQLALCHEMY_DATABASE_URI, False)
|
|
|
|
|
|
class BaseTestCase(TestCase):
|
|
TESTING = True
|
|
|
|
def create_app(self):
|
|
return app
|
|
|
|
def setUp(self):
|
|
db.create_all()
|
|
self.populate_db()
|
|
return app
|
|
|
|
def tearDown(self):
|
|
from app_db import db
|
|
db.session.remove()
|
|
db.drop_all()
|
|
|
|
def populate_db(self):
|
|
print(list(map(lambda x: x.username, db.session.query(Player).all())))
|
|
player = Player(steam_id=76561198248436608, username="pyro from csgo")
|
|
team = Team(team_name="Team Pepeja", tz_timezone="America/New_York", minute_offset=30)
|
|
|
|
db.session.add(player)
|
|
db.session.add(team)
|
|
|
|
db.session.flush()
|
|
|
|
|
|
player_team = PlayerTeam(
|
|
player_id=player.steam_id,
|
|
team_id=team.id,
|
|
team_role=PlayerTeam.TeamRole.Player,
|
|
is_team_leader=True,
|
|
)
|
|
logs_tf_integration = TeamLogsTfIntegration(
|
|
team_id=team.id,
|
|
min_team_member_count=2,
|
|
)
|
|
|
|
db.session.add(player_team)
|
|
db.session.add(logs_tf_integration)
|
|
|
|
db.session.commit()
|