def test_get_context_adds_units_and_errors(self, get_mock):
view = ActivityView()
get_mock.return_value = dict(super=sentinel.super)
# When getting the context data
context = view.get_context_data()
# Then the val_errors and units fields are included
assert context['val_errors'] is not None
assert context['units'] is not None
assert context['super'] == sentinel.super
def test_get_object_returns_activity_after_verify(self, get_mock,
verify_mock):
# Given a mock parent that returns a mock activity for current user
view = ActivityView()
view.request = sentinel.req
get_mock.return_value = sentinel.activity
# When getting the object
activity = view.get_object()
# Then the mock activity is returned and helper called
assert activity == sentinel.activity
verify_mock.assert_called_with(sentinel.activity, sentinel.req)