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)