diff --git a/opentech/apply/funds/tests/test_models.py b/opentech/apply/funds/tests/test_models.py index 431676d67ef2af9934ae104923530950c794edf3..e0237520c6ca736db669c54102bb29150cc3b037 100644 --- a/opentech/apply/funds/tests/test_models.py +++ b/opentech/apply/funds/tests/test_models.py @@ -13,44 +13,40 @@ def days_from_today(days): class TestFundModel(TestCase): + def setUp(self): + self.fund = FundTypeFactory(parent=None) + def test_can_access_workflow_class(self): - fund = FundTypeFactory(parent=None) - self.assertEqual(fund.workflow, 'single') - self.assertEqual(fund.workflow_class, SingleStage) + self.assertEqual(self.fund.workflow, 'single') + self.assertEqual(self.fund.workflow_class, SingleStage) def test_no_open_rounds(self): - fund = FundTypeFactory(parent=None) - self.assertIsNone(fund.open_round) + self.assertIsNone(self.fund.open_round) def test_open_ended_round(self): - fund = FundTypeFactory(parent=None) - open_round = RoundFactory(end_date=None, parent=fund) - self.assertEqual(fund.open_round, open_round) + open_round = RoundFactory(end_date=None, parent=self.fund) + self.assertEqual(self.fund.open_round, open_round) def test_normal_round(self): - fund = FundTypeFactory(parent=None) - open_round = RoundFactory(parent=fund) - self.assertEqual(fund.open_round, open_round) + open_round = RoundFactory(parent=self.fund) + self.assertEqual(self.fund.open_round, open_round) def test_closed_round(self): - fund = FundTypeFactory(parent=None) yesterday = days_from_today(-1) last_week = days_from_today(-7) - RoundFactory(start_date=last_week, end_date=yesterday, parent=fund) - self.assertIsNone(fund.open_round) + RoundFactory(start_date=last_week, end_date=yesterday, parent=self.fund) + self.assertIsNone(self.fund.open_round) def test_round_not_open(self): - fund = FundTypeFactory(parent=None) tomorrow = days_from_today(1) - RoundFactory(start_date=tomorrow, parent=fund) - self.assertIsNone(fund.open_round) + RoundFactory(start_date=tomorrow, parent=self.fund) + self.assertIsNone(self.fund.open_round) def test_multiple_open_rounds(self): - fund = FundTypeFactory(parent=None) - open_round = RoundFactory(parent=fund) + open_round = RoundFactory(parent=self.fund) next_round_start = open_round.end_date + timedelta(days=1) - RoundFactory(start_date=next_round_start, end_date=None, parent=fund) - self.assertEqual(fund.open_round, open_round) + RoundFactory(start_date=next_round_start, end_date=None, parent=self.fund) + self.assertEqual(self.fund.open_round, open_round) class TestRoundModel(TestCase):