def setUp(self):
self.decision = Decision()
self.member = Member()
self.bill = Bill()
self.strategy = BalanceTheBooksStrategy(self.decision, self.member, self.bill)
self.stance = Stance()
self.stance.issue = "Some Issue"
self.stance.side = outcomes.FOR
self.stance1 = Stance()
self.stance1.issue = "Some Other Issue"
self.stance1.side = outcomes.AGN
self.stance2 = Stance()
self.stance2.issue = "Another Issue"
self.stance2.side = outcomes.FOR
self.decision.for_stances = [self.stance]
self.decision.agn_stances = [self.stance1]
class BalanceTheBooksStrategyTest(unittest.TestCase):
""" Test suite for balance_the_books_strategy.py."""
def setUp(self):
self.decision = Decision()
self.member = Member()
self.bill = Bill()
self.strategy = BalanceTheBooksStrategy(self.decision, self.member, self.bill)
self.stance = Stance()
self.stance.issue = "Some Issue"
self.stance.side = outcomes.FOR
self.stance1 = Stance()
self.stance1.issue = "Some Other Issue"
self.stance1.side = outcomes.AGN
self.stance2 = Stance()
self.stance2.issue = "Another Issue"
self.stance2.side = outcomes.FOR
self.decision.for_stances = [self.stance]
self.decision.agn_stances = [self.stance1]
def test_init(self):
"""Tests whether a strategy can be constructed properly"""
result = BalanceTheBooksStrategy(self.decision, self.member, self.bill)
self.assertEqual(result._name, "Balance the Books")
self.assertEqual(result._decision, self.decision)
self.assertEqual(result._member, self.member)
self.assertEqual(result._bill, self.bill)
self.assertEqual(result._success, False)
def test_run_fail_no_majority(self):
""" Verifies the function fails if there is no majority."""
self.decision.for_stances = [self.stance]
self.decision.agn_stances = [self.stance1]
self.decision.split_record = [self.stance, self.stance1]
result = self.strategy.run()
self.assertFalse(result)
self.assertFalse(self.strategy._success)
self.assertEquals(self.decision.result, None)
self.assertEquals(self.decision.reason, None)
def test_run_fail_no_split_record(self):
""" Verifies the function fails if the member's record is not split on the decision."""
self.decision.for_stances = [self.stance, self.stance2]
self.decision.agn_stances = [self.stance1]
result = self.strategy.run()
self.assertFalse(result)
self.assertFalse(self.strategy._success)
self.assertEquals(self.decision.result, None)
self.assertEquals(self.decision.reason, None)
def test_run_success_FOR(self):
""" Verifies that run() successfully makes a FOR decision"""
self.decision.for_stances = [self.stance, self.stance2]
self.decision.agn_stances = [self.stance1]
self.decision.split_record = [self.stance, self.stance1]
result = self.strategy.run()
self.assertTrue(result)
self.assertTrue(self.strategy._success)
self.assertEquals(self.decision.result, outcomes.FOR)
self.assertEquals(self.decision.reason, self.decision.for_stances)
self.assertEquals(self.decision.downside, self.decision.agn_stances)
def test_run_success_AGN(self):
""" Verifies that run() successfully makes a AGN decision"""
self.decision.for_stances = [self.stance]
self.decision.agn_stances = [self.stance1, self.stance2]
self.decision.split_record = [self.stance, self.stance1]
result = self.strategy.run()
self.assertTrue(result)
self.assertTrue(self.strategy._success)
self.assertEquals(self.decision.result, outcomes.AGN)
self.assertEquals(self.decision.reason, self.decision.agn_stances)
self.assertEquals(self.decision.downside, self.decision.for_stances)
def test_explain(self):
""" Verifies explain runs if there is a success [aka _explain is implemented]."""
self.decision.for_stances = [self.stance]
self.decision.agn_stances = [self.stance1, self.stance2]
self.decision.split_record = [self.stance, self.stance1]
result = self.strategy.run()
self.assertTrue(result)
self.strategy.explain()
