def test_get_versions(self, fetch_grouped_history):
fetch_grouped_history.return_value = [
{'timeline': 'future', 'version': 'a'},
{'timeline': 'current', 'version': 'b'}]
current_ver, next_ver = reg_landing.get_versions('204')
self.assertEqual({'timeline': 'current', 'version': 'b'}, current_ver)
self.assertEqual({'timeline': 'future', 'version': 'a'}, next_ver)
def test_get_versions_no_next(self, fetch_grouped_history):
fetch_grouped_history.return_value = [
{'timeline': Timeline.present, 'version': 'b'}]
current_ver, next_ver = reg_landing.get_versions('204')
self.assertEqual({'timeline': Timeline.present, 'version': 'b'},
current_ver)
self.assertEqual(None, next_ver)
def test_get_versions_no_next(self, fetch_grouped_history):
fetch_grouped_history.return_value = [{
'timeline': 'current',
'version': 'b'
}]
current_ver, next_ver = reg_landing.get_versions('204')
self.assertEqual({'timeline': 'current', 'version': 'b'}, current_ver)
self.assertEqual(None, next_ver)
def fill_kwargs(self, kwargs):
"""Add the version and replace the label_id for the chrome context"""
reg_part = kwargs['label_id']
if not regulation_exists(reg_part):
raise error_handling.MissingContentException()
current, _ = get_versions(kwargs['label_id'])
kwargs['version'] = current['version']
kwargs['label_id'] = utils.first_section(reg_part, current['version'])
return kwargs
def fill_kwargs(self, kwargs):
"""Get the version for the chrome context"""
kwargs['version'] = self.request.GET.get('version', '')
kwargs['skip_count'] = True
if not kwargs['version']:
current, _ = get_versions(kwargs['label_id'])
kwargs['version'] = current['version']
kwargs['label_id'] = utils.first_section(kwargs['label_id'],
kwargs['version'])
return kwargs
def fill_kwargs(self, kwargs):
"""Add the version and replace the label_id for the chrome context"""
reg_part = kwargs['label_id']
if not regulation_exists(reg_part):
raise error_handling.MissingContentException()
current, _ = get_versions(kwargs['label_id'])
kwargs['version'] = current['version']
kwargs['label_id'] = utils.first_section(reg_part, current['version'])
return kwargs
def fill_kwargs(self, kwargs):
"""Get the version for the chrome context"""
kwargs['version'] = self.request.GET.get('version', '')
kwargs['skip_count'] = True
if not kwargs['version']:
current, _ = get_versions(kwargs['label_id'])
kwargs['version'] = current['version']
kwargs['label_id'] = utils.first_section(kwargs['label_id'],
kwargs['version'])
return kwargs
def get_context_data(self, **kwargs):
"""Add the version and replace the label_id for the chrome context"""
reg_part = kwargs['label_id']
if not regulation_exists(reg_part):
raise error_handling.MissingContentException()
current, _ = get_versions(kwargs['label_id'])
kwargs['version'] = current['version']
kwargs['label_id'] = utils.first_section(reg_part, current['version'])
return super(ChromeLandingView, self).get_context_data(**kwargs)
def get_context_data(self, **kwargs):
"""Add the version and replace the label_id for the chrome context"""
reg_part = kwargs['label_id']
if not regulation_exists(reg_part):
raise error_handling.MissingContentException()
current, _ = get_versions(kwargs['label_id'])
kwargs['version'] = current['version']
kwargs['label_id'] = utils.first_section(reg_part, current['version'])
return super(ChromeLandingView, self).get_context_data(**kwargs)
def test_get_versions(self, fetch_grouped_history):
fetch_grouped_history.return_value = [{
'timeline': Timeline.future,
'version': 'a'
}, {
'timeline': Timeline.present,
'version': 'b'
}]
current_ver, next_ver = reg_landing.get_versions('204')
self.assertEqual({
'timeline': Timeline.present,
'version': 'b'
}, current_ver)
self.assertEqual({
'timeline': Timeline.future,
'version': 'a'
}, next_ver)
