def test_always_rollback(self):
connection = transaction.get_connection()
with patch.object(connection, 'rollback', wraps=connection.rollback) as mock_rollback:
with pytest.raises(TestException):
confirm_email_change(self.request, self.key)
mock_rollback.assert_called_with()
def check_confirm_email_change(self, expected_template, expected_context):
"""
Call `confirm_email_change` and assert that the content was generated as expected
`expected_template`: The name of the template that should have been used
to generate the content
`expected_context`: The context dictionary that should have been used to
generate the content
"""
response = confirm_email_change(self.request, self.key)
assert response.status_code == 200
assert mock_render_to_response(expected_template, expected_context).content.decode('utf-8') \
== response.content.decode('utf-8')
