def _generate_summary_response(self, transitions, uploaded_filename):
filename = uploaded_filename.split('.')[0] + " Summary"
response_file = Dumper(self.domain).dump(transitions)
response = HttpResponse(response_file,
content_type="text/html; charset=utf-8")
response['Content-Disposition'] = safe_filename_header(
filename, 'xlsx')
return response
def test_dump(self, mock_active_site_codes_search,
mock_inactive_site_codes_search, mock_case_count,
mock_location_ids):
mock_case_count.return_value = []
transitions = {
'awc': [
Transition(domain=self.domain,
location_type_code='awc',
operation=MOVE_OPERATION,
old_site_codes=['111'],
new_site_codes=['112']),
Transition(domain=self.domain,
location_type_code='awc',
operation=MERGE_OPERATION,
old_site_codes=['113', '114'],
new_site_codes=['115']),
Transition(domain=self.domain,
location_type_code='awc',
operation=SPLIT_OPERATION,
old_site_codes=['116'],
new_site_codes=['117', '118']),
Transition(domain=self.domain,
location_type_code='awc',
operation=EXTRACT_OPERATION,
old_site_codes=['119'],
new_site_codes=['120'])
],
'supervisor': [
Transition(domain=self.domain,
location_type_code='awc',
operation=MOVE_OPERATION,
old_site_codes=['12'],
new_site_codes=['13']),
],
'state': {}
}
mock_location_ids.return_value = {
'111': 'locationid111',
'113': 'locationid113',
'114': 'locationid114',
'116': 'locationid116',
'119': 'locationid119',
'12': 'locationid12'
}
mock_active_site_codes_search.return_value.values_list.return_value = [
'112', '115', '117'
]
mock_inactive_site_codes_search.return_value.values_list.return_value = [
'120'
]
filestream = Dumper(self.domain).dump(transitions)
workbook = get_workbook(filestream)
awc_worksheet_rows = list(workbook.get_worksheet('awc'))
supervisor_worksheet_rows = list(workbook.get_worksheet('supervisor'))
self.assertEqual(supervisor_worksheet_rows, [{
OLD_LOCATION_CODE_COLUMN: '12',
TRANSITION_COLUMN: MOVE_OPERATION,
NEW_LOCATION_CODE_COLUMN: '13',
MISSING_COLUMN: True,
ARCHIVED_COLUMN: False,
CASE_COUNT_COLUMN: 0
}])
self.assertEqual(awc_worksheet_rows, [{
OLD_LOCATION_CODE_COLUMN: '111',
TRANSITION_COLUMN: MOVE_OPERATION,
NEW_LOCATION_CODE_COLUMN: '112',
MISSING_COLUMN: False,
ARCHIVED_COLUMN: False,
CASE_COUNT_COLUMN: 0
}, {
OLD_LOCATION_CODE_COLUMN: '113',
TRANSITION_COLUMN: MERGE_OPERATION,
NEW_LOCATION_CODE_COLUMN: '115',
MISSING_COLUMN: False,
ARCHIVED_COLUMN: False,
CASE_COUNT_COLUMN: 0
}, {
OLD_LOCATION_CODE_COLUMN: '114',
TRANSITION_COLUMN: MERGE_OPERATION,
NEW_LOCATION_CODE_COLUMN: '115',
MISSING_COLUMN: False,
ARCHIVED_COLUMN: False,
CASE_COUNT_COLUMN: 0
}, {
OLD_LOCATION_CODE_COLUMN: '116',
TRANSITION_COLUMN: SPLIT_OPERATION,
NEW_LOCATION_CODE_COLUMN: '117',
MISSING_COLUMN: False,
ARCHIVED_COLUMN: False,
CASE_COUNT_COLUMN: 0
}, {
OLD_LOCATION_CODE_COLUMN: '116',
TRANSITION_COLUMN: SPLIT_OPERATION,
NEW_LOCATION_CODE_COLUMN: '118',
MISSING_COLUMN: True,
ARCHIVED_COLUMN: False,
CASE_COUNT_COLUMN: 0
}, {
OLD_LOCATION_CODE_COLUMN: '119',
TRANSITION_COLUMN: EXTRACT_OPERATION,
NEW_LOCATION_CODE_COLUMN: '120',
MISSING_COLUMN: True,
ARCHIVED_COLUMN: True,
CASE_COUNT_COLUMN: 0
}])