def test_import_headers_returns_errors(self, patched_import_header):
cell = mock.MagicMock(value=15)
cells = [cell]
sheetattrs = {"row.return_value": cells}
worksheet = mock.MagicMock(**sheetattrs)
with mock.patch("flooding_lib.excel_import_export.get_worksheet", return_value=worksheet):
self.assertEquals(["error!"], eie.import_uploaded_excel_file("", set()))
def test_get_scenario_row_returns_errors(self, patched_import_header,
patched_import_scenario_row):
worksheet = mock.MagicMock(row=lambda i: [], nrows=5)
with mock.patch('flooding_lib.excel_import_export.get_worksheet',
return_value=worksheet):
errors = eie.import_uploaded_excel_file('', set())
self.assertEquals(errors, ["error!"])
def test_get_scenario_row_returns_errors(
self, patched_import_header, patched_import_scenario_row):
worksheet = mock.MagicMock(row=lambda i: [], nrows=5)
with mock.patch(
'flooding_lib.excel_import_export.get_worksheet',
return_value=worksheet):
errors = eie.import_uploaded_excel_file('', set())
self.assertEquals(errors, ["error!"])
def test_import_headers_returns_errors(self, patched_import_header):
cell = mock.MagicMock(value=15)
cells = [cell]
sheetattrs = {'row.return_value': cells}
worksheet = mock.MagicMock(**sheetattrs)
with mock.patch('flooding_lib.excel_import_export.get_worksheet',
return_value=worksheet):
self.assertEquals(["error!"],
eie.import_uploaded_excel_file('', set()))
def test_rows_calls_get_scenario_row(self, patched_import_header, patched_import_scenario_row):
cell = object()
onecellrow = [cell]
def row(i):
if i < 4:
return []
return onecellrow
worksheet = mock.MagicMock(row=row, nrows=5)
allowed_ids = set()
with mock.patch("flooding_lib.excel_import_export.get_worksheet", return_value=worksheet):
eie.import_uploaded_excel_file("", allowed_ids)
patched_import_scenario_row.assertCalledWith(
(patched_import_header.return_value, 4, onecellrow, allowed_ids)
)
def test_rows_calls_get_scenario_row(self, patched_import_header,
patched_import_scenario_row):
cell = object()
onecellrow = [cell]
def row(i):
if i < 4:
return []
return onecellrow
worksheet = mock.MagicMock(row=row, nrows=5)
allowed_ids = set()
with mock.patch('flooding_lib.excel_import_export.get_worksheet',
return_value=worksheet):
eie.import_uploaded_excel_file('', allowed_ids)
patched_import_scenario_row.assertCalledWith(
(patched_import_header.return_value, 4, onecellrow,
allowed_ids))
def test_calls_import_header(self, patched_import_header):
cell = mock.MagicMock(value=15)
cells = [cell]
sheetattrs = {'row.return_value': cells}
worksheet = mock.MagicMock(**sheetattrs)
with mock.patch(
'flooding_lib.excel_import_export.get_worksheet',
return_value=worksheet):
self.assertEquals([], eie.import_uploaded_excel_file('', set()))
patched_import_header.assertCalledWith(cells)
