113 lines
3.9 KiB
Python
113 lines
3.9 KiB
Python
from django.test import TestCase
|
|
from events.models import Player, Event, Availability, Team, Positioning
|
|
import datetime
|
|
import pytz
|
|
# Create your tests here.
|
|
|
|
class BenchcoachTestCase(TestCase):
|
|
def setUp(self):
|
|
for first_name, last_name, jersey_number in [
|
|
("Rush", "Valenzuela", 1),
|
|
("Baby", "Triumphant", 2)
|
|
]:
|
|
Player.objects.create(first_name=first_name, last_name=last_name, jersey_number=jersey_number)
|
|
|
|
for name in ["Firefighters", "Garages"]:
|
|
Team.objects.create(name=name)
|
|
|
|
for start, home_team_id, away_team_id in [
|
|
(datetime.datetime(2020, 1, 2, 12, 0), 1, 2), # id = 1
|
|
(datetime.datetime(2020, 1, 3, 12, 0), 2, 1), # id = 2
|
|
(datetime.datetime(2020, 1, 4, 12, 0), 2, 1) # id = 3
|
|
]:
|
|
Event.objects.create(start=start, home_team_id=home_team_id, away_team_id=away_team_id)
|
|
|
|
for event_id in [
|
|
1
|
|
]:
|
|
for player_id, available in [
|
|
(1,"Yes"),
|
|
(2,"No")
|
|
]:
|
|
Availability.objects.create(event_id=event_id, player_id=player_id, available=available)
|
|
|
|
for event_id in [
|
|
1
|
|
]:
|
|
for player_id, position in [
|
|
(1, "C"),
|
|
(2, "1B")
|
|
]:
|
|
Positioning.objects.create(event_id=event_id, player_id=player_id, position=position)
|
|
|
|
for event_id in [
|
|
2
|
|
]:
|
|
for player_id, available in [
|
|
(1,"Yes"),
|
|
(2,"Yes")
|
|
]:
|
|
Availability.objects.create(event_id=event_id, player_id=player_id, available=available)
|
|
|
|
for event_id in [
|
|
3
|
|
]:
|
|
for player_id, available in [
|
|
(1,"No"),
|
|
(2,"No")
|
|
]:
|
|
Availability.objects.create(event_id=event_id, player_id=player_id, available=available)
|
|
pass
|
|
|
|
|
|
def test_player(self):
|
|
"""Test that player works"""
|
|
player_1 = Player.objects.get(first_name="Rush")
|
|
player_2 = Player.objects.get(first_name="Baby")
|
|
self.assertEqual(str(player_1), f"Valenzuela, Rush")
|
|
self.assertEqual(str(player_2), f"Triumphant, Baby")
|
|
pass
|
|
|
|
def test_event(self):
|
|
event = Event.objects.get(pk=1)
|
|
self.assertEqual(event.start.year, 2020)
|
|
self.assertEqual(event.start.month, 1)
|
|
self.assertEqual(event.start.day, 2)
|
|
self.assertEqual(event.start.hour, 12)
|
|
self.assertEqual(event.start.minute, 0)
|
|
self.assertEqual(event.home_team.name, "Firefighters")
|
|
self.assertEqual(event.away_team.name, "Garages")
|
|
|
|
def test_availability(self):
|
|
availability_1 = Availability.objects.get(event_id=1, player_id=1)
|
|
availability_2 = Availability.objects.get(event_id=1, player_id=2)
|
|
|
|
self.assertEqual(availability_1.available, "Yes")
|
|
self.assertEqual(availability_2.available, "No")
|
|
pass
|
|
|
|
def test_positioning(self):
|
|
positioning_1 = Positioning.objects.get(event_id=1, player_id=1)
|
|
positioning_2 = Positioning.objects.get(event_id=1, player_id=2)
|
|
# positing_1
|
|
pass
|
|
|
|
def test_combine_info(self):
|
|
event = Event.objects.get(pk=1)
|
|
event_id = 1
|
|
player_query = Player.objects.all()
|
|
positioning_query = Positioning.objects.filter(event=event).select_related('player')
|
|
availability_query = Availability.objects.filter(event=event).select_related('player')
|
|
player_event_info = []
|
|
|
|
for player in player_query:
|
|
player_event_info.append(
|
|
{
|
|
"player":player,
|
|
"positioning":positioning_query.get(player_id=player.id),
|
|
"availability":availability_query.get(player_id=player.id)
|
|
}
|
|
)
|
|
|
|
pass
|