class AdjustedBudgetTestCase(TestCase):
"""Unit tests of adjusted budget data summary for a department"""
def setUp(self):
year = FinancialYear(slug="2030-31")
sphere = Sphere(financial_year=year, name="A sphere")
government = Government(sphere=sphere, name="A government")
self.department = Department(
government=government,
name="Fake",
vote_number=1,
is_vote_primary=True,
intro="",
)
mock_dataset = Mock()
mock_openspending_api = Mock()
self.mock_openspending_api = mock_openspending_api
mock_openspending_api.get_adjustment_kind_ref = Mock
mock_openspending_api.get_phase_ref = Mock
mock_openspending_api.get_programme_name_ref = Mock
mock_openspending_api.get_department_name_ref = Mock(
return_value="department_name_ref")
mock_openspending_api.get_financial_year_ref = Mock(
return_value="financial_year_ref")
mock_openspending_api.aggregate = Mock
mock_openspending_api.filter_dept = Mock(return_value={"cells": []})
mock_openspending_api.filter_by_ref_exclusion = Mock
mock_openspending_api.aggregate_by_refs = Mock
mock_openspending_api.aggregate_url = Mock
mock_dataset.get_openspending_api = Mock(
return_value=mock_openspending_api)
self.department.get_adjusted_estimates_expenditure_dataset = Mock(
return_value=mock_dataset)
self.department.get_financial_year = Mock(return_value=year)
self.department._get_adjustments_by_econ_class = Mock(
return_value=Mock())
self.department._get_adjustments_by_programme = Mock(
return_value=Mock())
self.department._get_adjustments_by_type = Mock(return_value=Mock())
self.department._get_budget_virements = Mock(return_value=Mock())
self.department._get_budget_special_appropriations = Mock(
return_value=Mock())
self.department._get_budget_direct_charges = Mock(return_value=Mock())
models.csv_url = Mock(return_value=Mock())
def test_no_adjustment(self):
self.department._get_total_budget_adjustment = Mock(return_value=(123,
0))
result = self.department.get_adjusted_budget_summary()
self.assertEqual(result["total_change"]["amount"], 0)
self.assertEqual(result["total_change"]["percentage"], 0)
def test_adjustment(self):
self.department._get_total_budget_adjustment = Mock(return_value=(100,
11))
result = self.department.get_adjusted_budget_summary()
self.assertEqual(result["total_change"]["amount"], 11)
self.assertEqual(result["total_change"]["percentage"], 11)
class AdjustedBudgetMissingTestCase(TestCase):
"""Unit tests of adjusted budget data summary for a department"""
def setUp(self):
year = FinancialYear(slug="2030-31")
sphere = Sphere(financial_year=year, name="A sphere")
government = Government(sphere=sphere, name="A government")
self.department = Department(
government=government,
name="Fake",
vote_number=1,
is_vote_primary=True,
intro="",
)
self.department.get_adjusted_estimates_expenditure_dataset = Mock(
return_value=None)
def test_missing_dataset(self):
self.assertEqual(self.department.get_adjusted_budget_summary(), None)