do_title_change test/fix
This commit is contained in:
parent
30db77781a
commit
68b369c5bd
@ -103,9 +103,9 @@ class Match(models.Model):
|
|||||||
if self.title_at_stake:
|
if self.title_at_stake:
|
||||||
victors = list(self.teams.get(victorious=True).members.all())
|
victors = list(self.teams.get(victorious=True).members.all())
|
||||||
if len(victors) == 1:
|
if len(victors) == 1:
|
||||||
victor[0].win_title(self.title_at_stake)
|
victors[0].win_title(self.title_at_stake)
|
||||||
elif len(victors) == 2 and self.title_at_stake == 'tag':
|
elif len(victors) == 2 and self.title_at_stake == 'tag':
|
||||||
victor[0].win_title(self.title_at_stake, victor[1])
|
victors[0].win_title(self.title_at_stake, victors[1])
|
||||||
else:
|
else:
|
||||||
raise ValueError('invalid number of victors for title change')
|
raise ValueError('invalid number of victors for title change')
|
||||||
|
|
||||||
|
@ -51,6 +51,40 @@ class MatchTest(TestCase):
|
|||||||
self.assertEqual(unicode(match), 'CM Punk (c) (v) vs. Rey Mysterio')
|
self.assertEqual(unicode(match), 'CM Punk (c) (v) vs. Rey Mysterio')
|
||||||
|
|
||||||
|
|
||||||
|
def test_do_title_change(self):
|
||||||
|
# title to punk
|
||||||
|
match = self.event.add_match('cmpunk', 'reymysterio', winner='cmpunk',
|
||||||
|
outcome='normal', title_at_stake='wwe')
|
||||||
|
match.do_title_change()
|
||||||
|
self.assertEqual(Star.objects.get(pk='cmpunk').title, 'wwe')
|
||||||
|
|
||||||
|
# title to mysterio
|
||||||
|
match = self.event.add_match('cmpunk', 'reymysterio',
|
||||||
|
winner='reymysterio', outcome='normal',
|
||||||
|
title_at_stake='wwe')
|
||||||
|
match.do_title_change()
|
||||||
|
self.assertEqual(Star.objects.get(pk='reymysterio').title, 'wwe')
|
||||||
|
self.assertEqual(Star.objects.get(pk='cmpunk').title, None)
|
||||||
|
|
||||||
|
# tag title
|
||||||
|
match = self.event.add_match(['kofikingston', 'rtruth'],
|
||||||
|
['jackswagger', 'dolphziggler'],
|
||||||
|
winner='dolphziggler', outcome='normal',
|
||||||
|
title_at_stake='tag')
|
||||||
|
match.do_title_change()
|
||||||
|
self.assertEqual(Star.objects.get(pk='kofikingston').title, None)
|
||||||
|
self.assertEqual(Star.objects.get(pk='rtruth').title, None)
|
||||||
|
self.assertEqual(Star.objects.get(pk='dolphziggler').title, 'tag')
|
||||||
|
self.assertEqual(Star.objects.get(pk='jackswagger').title, 'tag')
|
||||||
|
|
||||||
|
match = self.event.add_match(['kofikingston', 'rtruth'],
|
||||||
|
['jackswagger', 'dolphziggler'],
|
||||||
|
winner='dolphziggler', outcome='normal',
|
||||||
|
title_at_stake='diva')
|
||||||
|
# diva title on a tag match
|
||||||
|
with self.assertRaises(ValueError):
|
||||||
|
match.do_title_change()
|
||||||
|
|
||||||
def test_scoring(self):
|
def test_scoring(self):
|
||||||
# one on one : 2 points
|
# one on one : 2 points
|
||||||
match = self.event.add_match('tripleh', 'undertaker',
|
match = self.event.add_match('tripleh', 'undertaker',
|
||||||
|
Loading…
Reference in New Issue
Block a user