mirror of https://github.com/sgoudham/Enso-Bot.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
55 lines
2.0 KiB
Python
55 lines
2.0 KiB
Python
from django.utils import unittest
|
|
from django.db import IntegrityError
|
|
from voice.models import Feature, Vote
|
|
|
|
class FeatureTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
self.feature = Feature.objects.create(title='Feature',
|
|
description='This is a feature', votes_needed=350)
|
|
|
|
def testString(self):
|
|
self.assertEqual('Feature', str(self.feature))
|
|
|
|
def testDefaultState(self):
|
|
self.assertEqual('V', self.feature.state)
|
|
|
|
def testChangesState(self):
|
|
for i in range(0, self.feature.votes_needed):
|
|
voter = 'voter%d@domain.com' % i
|
|
Vote.objects.create(feature=self.feature, voter=voter)
|
|
self.assertEqual('W', self.feature.state)
|
|
|
|
def testVoting(self):
|
|
Vote.objects.create(feature=self.feature, voter='v@d.com')
|
|
self.assertEqual(self.feature.votes.count(), 1)
|
|
|
|
class VoteTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
self.feature = Feature.objects.create(title='Feature',
|
|
description='This is a feature', votes_needed=350)
|
|
|
|
def testString(self):
|
|
voter = 'v@d.com'
|
|
vote = Vote.objects.create(feature=self.feature, voter=voter)
|
|
self.assertEqual(voter, str(vote))
|
|
|
|
def testDisallowDuplicates(self):
|
|
voter = 'v@d.com'
|
|
vote = Vote.objects.create(feature=self.feature, voter=voter)
|
|
self.assertRaises(IntegrityError, Vote.objects.create,
|
|
feature=self.feature, voter=voter)
|
|
|
|
def testDecreasesVotesNeeded(self):
|
|
for i in range(0, self.feature.votes_needed):
|
|
voter = 'voter%d@domain.com' % i
|
|
vote = Vote.objects.create(feature=self.feature, voter=voter)
|
|
|
|
self.assertEqual(self.feature.votes_left(), 0)
|
|
|
|
for i in range(self.feature.votes_needed + 1,
|
|
self.feature.votes_needed * 2):
|
|
voter = 'voter%d@domain.com' % i
|
|
vote = Vote.objects.create(feature=self.feature, voter=voter)
|
|
|
|
self.assertEqual(self.feature.votes_left(), 0)
|