def test_add_schemas_worksheet(self):
# given
sheets = ['A', 'B', 'C', 'X']
workbook = create_test_workbook(*sheets)
wb = IngestWorkbook(workbook)
# when
wb.add_schemas_worksheet(['schema1', 'schema2', 'schema3'])
# then
self.assertEqual(wb.workbook['Schemas'].cell(1, 1).value, 'Schemas')
self.assertEqual(wb.workbook['Schemas'].cell(2, 1).value, 'schema1')
self.assertEqual(wb.workbook['Schemas'].cell(3, 1).value, 'schema2')
self.assertEqual(wb.workbook['Schemas'].cell(4, 1).value, 'schema3')
def test_add_schemas_worksheet_existing_schemas(self):
# given
sheets = ['A', 'B', 'C']
workbook = create_test_workbook(*sheets)
wb = IngestWorkbook(workbook)
wb.workbook.create_sheet('Schemas')
wb.workbook['Schemas'].cell(1, 1).value = 'X'
wb.workbook['Schemas'].cell(2, 1).value = 'x'
wb.workbook['Schemas'].cell(3, 1).value = 'y'
wb.workbook['Schemas'].cell(4, 1).value = 'z'
# when
wb.add_schemas_worksheet(['schema1', 'schema2', 'schema3'])
# then
self.assertEqual(wb.workbook['Schemas'].cell(1, 1).value, 'X')
self.assertEqual(wb.workbook['Schemas'].cell(2, 1).value, 'x')
self.assertEqual(wb.workbook['Schemas'].cell(3, 1).value, 'y')
self.assertEqual(wb.workbook['Schemas'].cell(4, 1).value, 'z')