def test_team_list(self):
TeamModel('test team', 'test', 'test', 'test', 'test', 0, 0).save_to_db()
TeamModel('test team2', 'test2', 'test2', 'test2', 'test2', 0, 0).save_to_db()
response = self.client.get('/teams')
self.assertEqual(200, response.status_code)
expected = {
'teams':
[
{
'id': 1,
'name': 'test team',
'base': 'test',
'chief': 'test',
'car': 'test',
'power_unit': 'test',
'since': '0',
'championships': 0,
'drivers': []
},
{
'id': 2,
'name': 'test team2',
'base': 'test2',
'chief': 'test2',
'car': 'test2',
'power_unit': 'test2',
'since': '0',
'championships': 0,
'drivers': []
}
]
}
self.assertDictEqual(expected, json.loads(response.data))
def test_driver_relationship(self):
TeamModel('test team', 'test', 'test', 'test', 'test', 0, 0).save_to_db()
DriverModel('test fn', 'test ln', 0, 1, 'test', 0, 0, 0, date(1995, 8, 25)).save_to_db()
team = TeamModel.find_by_name('test team')
self.assertEqual(team.drivers.count(), 1)
self.assertEqual(team.drivers.first().first_name, 'test fn')
def test_delete_team(self):
TeamModel('test team', 'test', 'test', 'test', 'test', 0, 0).save_to_db()
headers = {'Authorization': self.token}
response = self.client.delete('/teams/1',
headers=headers)
self.assertEqual(response.status_code, 202)
self.assertDictEqual({'message': 'Team id:1 deleted.'}, json.loads(response.data))
def test_team_list_with_drivers(self):
TeamModel('test team', 'test', 'test', 'test', 'test', 0, 0).save_to_db()
DriverModel('test fn', 'test ln', 0, 1, 'test', 0, 0, 0, date(1995, 8, 25)).save_to_db()
response = self.client.get('/teams')
self.assertEqual(200, response.status_code)
expected = {
'teams':
[
{
'id': 1,
'name': 'test team',
'base': 'test',
'chief': 'test',
'car': 'test',
'power_unit': 'test',
'since': '0',
'championships': 0,
'drivers':
[
{'id': 1, 'first_name': 'test fn', 'last_name': 'test ln', 'number': 0, 'points': 0}
]
}
]
}
self.assertDictEqual(expected, json.loads(response.data))
def test_team_relationship(self):
TeamModel('test team', 'test', 'test', 'test', 'test', 0,
0).save_to_db()
DriverModel('test fn', 'test ln', 0, 1, 'test', 0, 0, 0,
date(1995, 8, 25)).save_to_db()
driver = DriverModel.find_by_id(1)
self.assertEqual('test team', driver.team.name)
def test_team_crud(self):
team = TeamModel('test team', 'test', 'test', 'test', 'test', 0, 0)
self.assertIsNone(team.find_by_name('test team'),
"Found a team with name '{}', but expected not to.".format(team.name))
team.save_to_db()
self.assertIsNotNone(team.find_by_name('test team'),
"Did not find a team with name '{}', but expected to.".format(team.name))
team.delete_from_db()
self.assertIsNone(team.find_by_name('test team'),
"Found a team with name '{}', but expected not to.".format(team.name))
def test_create_team(self):
team = TeamModel('test', 'test', 'test', 'test', 'test', 0, 0)
self.assertIsNotNone(team)
self.assertEqual(team.name, 'test')
self.assertEqual(team.base, 'test')
self.assertEqual(team.chief, 'test')
self.assertEqual(team.car, 'test')
self.assertEqual(team.power_unit, 'test')
self.assertEqual(team.since, 0)
self.assertEqual(team.championships, 0)
def test_team_json_with_driver(self):
team = TeamModel('test team', 'test', 'test', 'test', 'test', 0, 0)
driver = DriverModel('test fn', 'test ln', 0, 1, 'test', 0, 0, 0, date(1995, 8, 25))
team.save_to_db()
driver.save_to_db()
expected = {
'id': 1,
'name': 'test team',
'base': 'test',
'chief': 'test',
'car': 'test',
'power_unit': 'test',
'founded': 0,
'championships': 0,
'drivers': [driver.json()],
}
self.assertEqual(team.json(), expected)
def test_driver_crud(self):
TeamModel('test', 'test', 'test', 'test', 'test', 0, 0).save_to_db()
driver = DriverModel('test fn', 'test ln', 0, 1, 'test', 0, 0, 0,
date(1995, 8, 25))
self.assertIsNone(DriverModel.find_by_id(1),
"Found an driver with id: 1, but expected not to.")
driver.save_to_db()
self.assertIsNotNone(
DriverModel.find_by_id(1),
"Did not find an item with id: 1, but expected to.")
driver.delete_from_db()
self.assertIsNone(DriverModel.find_by_id(1),
"Found an driver with id: 1, but expected not to.")
def test_create_duplicate_team(self):
TeamModel('test team', 'test', 'test', 'test', 'test', 0, 0).save_to_db()
headers = {'Authorization': self.token,
'Content-Type': 'application/json'}
data = {
'name': 'test',
'base': 'test',
'chief': 'test',
'car': 'test',
'power_unit': 'test'
}
response = self.client.post('/teams/1',
data=json.dumps(data),
headers=headers)
self.assertEqual(response.status_code, 400)
expected = {"message": "Team with id:1 already exists"}
self.assertDictEqual(expected, json.loads(response.data))
def test_create_team_drivers_empty(self):
TeamModel('test team', 'test', 'test', 'test', 'test', 0, 0).save_to_db()
team = TeamModel.find_by_name('test team')
self.assertListEqual(team.drivers.all(), [])