def test_get_land_compensation_payment(self):
"""should respond with a 200 and render the expected template"""
charge = LocalLandChargeItem()
charge.land_compensation_paid = '150000'
charge.land_compensation_amount_type = 'Agreed amount'
self.mock_session.return_value.add_charge_state = charge
self.client.set_cookie('localhost', Session.session_cookie_name, 'cookie_value')
self.mock_session.return_value.user.permissions = [Permissions.vary_llc]
response = self.client.get(url_for('modify_land_charge.get_land_compensation_payment'))
self.status = self.assert_status(response, 200)
self.assert_template_used(HTML_PAYMENT)
def test_post_land_compensation_payment_success_change_amount(
self, mock_land_compensation_payment_validator,
mock_review_router):
self.client.set_cookie('localhost', Session.session_cookie_name,
'cookie_value')
mock_review_router.get_redirect_url.return_value = url_for(
'add_land_charge.get_land_compensation_owned')
state = LocalLandChargeItem()
state.land_compensation_paid = '50000'
state.land_compensation_amount_type = 'Estimated amount'
state.land_capacity_description = 'Freehold'
self.mock_session.return_value.add_charge_state = state
self.mock_session.return_value.user.permissions = [Permissions.add_llc]
mock_review_router.get_redirect_url.return_value = url_for(
'add_land_charge.get_review')
mock_validation_errors = MagicMock()
mock_validation_errors.errors = {}
mock_land_compensation_payment_validator.validate.return_value = mock_validation_errors
response = self.client.post(
url_for('add_land_charge.post_land_compensation_payment'),
data={
'land-compensation-paid': '100000',
'amount-of-compensation': '900000',
'land-compensation-amount-type': 'Estimated amount'
})
calls = [
call('land_compensation_paid', '100000.00'),
call('amount_of_compensation', '900000.00'),
call('land_compensation_amount_type', 'Estimated amount')
]
mock_review_router.update_edited_field.assert_has_calls(calls)
self.assert_status(response, 302)
self.assertRedirects(response, url_for('add_land_charge.get_review'))
def test_get_land_compensation_payment_success_with_values(self):
self.client.set_cookie('localhost', Session.session_cookie_name,
'cookie_value')
state = LocalLandChargeItem()
state.land_compensation_paid = '50000.10'
state.land_compensation_amount_type = 'Agreed amount'
self.mock_session.return_value.add_charge_state = state
self.mock_session.return_value.user.permissions = [Permissions.add_llc]
response = self.client.get(
url_for('add_land_charge.get_land_compensation_payment'))
self.assert_status(response, 200)
self.assert_template_used('land_compensation_payment.html')
self.assertEqual(
self.get_context_variable('request_body')
['land-compensation-amount-type'], 'Agreed amount')
self.assertEqual(
self.get_context_variable('request_body')
['land-compensation-paid'], '50000.10')
