def test_find_approval_rules_error_means_rollback(
self, patched_get_worksheet, patched_find_approval_rules,
patched_rollback):
eie.import_upload_excel_file_for_approval(None, "remco",
"norealfile.xls", [])
self.assertTrue(patched_rollback.called)
def test_find_approval_rules_error_means_rollback(
self, patched_get_worksheet,
patched_find_approval_rules, patched_rollback):
eie.import_upload_excel_file_for_approval(
None, "remco", "norealfile.xls", [])
self.assertTrue(patched_rollback.called)
def test_find_approval_rules_works_empty_worksheet_no_errors(
self, patched_find_approval_rules, patched_commit):
worksheet = MockWorksheet()
with mock.patch('flooding_lib.excel_import_export.get_worksheet',
return_value=worksheet):
eie.import_upload_excel_file_for_approval(None, "remco",
"norealfile.xls", [])
self.assertTrue(patched_commit.called)
def test_find_approval_rules_works_empty_worksheet_no_errors(
self, patched_find_approval_rules, patched_commit):
worksheet = MockWorksheet()
with mock.patch('flooding_lib.excel_import_export.get_worksheet',
return_value=worksheet):
eie.import_upload_excel_file_for_approval(
None, "remco", "norealfile.xls", [])
self.assertTrue(patched_commit.called)
def test_put_row_in_worksheet_row_returns_errors(self, patched_rollback):
scenario = ScenarioF.create()
project = ProjectF.create()
scenario.set_project(project)
worksheet = MockWorksheet()
worksheet.write(eie.HEADER_ROWS, 1, scenario.id)
rule_dict = dict()
allowed_scenario_ids = [] # This will give an error
with mock.patch("flooding_lib.excel_import_export.get_worksheet", return_value=worksheet):
with mock.patch("flooding_lib.excel_import_export.find_approval_rules", return_value=([], rule_dict)):
eie.import_upload_excel_file_for_approval(project, "remco", "somepath.xls", allowed_scenario_ids)
self.assertTrue(patched_rollback.called)
def post_excel(request, project, province):
form = forms.ExcelImportForm(request.POST, request.FILES)
errors = []
if form.is_valid():
filename = request.FILES['excel_file'].name
if not filename.startswith(province.name):
errors += [
_('Filename does not start with the name of the province. '
'Wrong file?')
]
else:
dest_path = os.path.join('/tmp', filename)
with open(dest_path, 'wb') as dest:
for chunk in request.FILES['excel_file'].chunks():
dest.write(chunk)
allowed_scenario_ids = [
scenario.id for scenario in scenario_list(project, province)
]
errors = (
excel_import_export.import_upload_excel_file_for_approval(
project, request.user.username, dest_path,
allowed_scenario_ids))
if errors:
request.session['messages'] = errors
else:
request.session['messages'] = ['Opslaan is geslaagd.']
return HttpResponseRedirect(reverse('sharedproject_dashboard'))
def test_put_row_in_worksheet_row_returns_errors(self, patched_rollback):
scenario = ScenarioF.create()
project = ProjectF.create()
scenario.set_project(project)
worksheet = MockWorksheet()
worksheet.write(eie.HEADER_ROWS, 1, scenario.id)
rule_dict = dict()
allowed_scenario_ids = [] # This will give an error
with mock.patch('flooding_lib.excel_import_export.get_worksheet',
return_value=worksheet):
with mock.patch(
'flooding_lib.excel_import_export.find_approval_rules',
return_value=([], rule_dict)):
eie.import_upload_excel_file_for_approval(
project, 'remco', 'somepath.xls', allowed_scenario_ids)
self.assertTrue(patched_rollback.called)
