def test_sheet_save_to_django_model(self): model=FakeExceptionDjangoModel() writer = DjangoModelWriter([model, self.data[0], None, None]) writer.write_array(self.data[1:]) writer.close() # now raise excpetion model=FakeExceptionDjangoModel(raiseException=True) writer = DjangoModelWriter([model, self.data[0], None, None]) writer.write_array(self.data[1:]) writer.close()
def test_mapping_array(self): data2 = [ ["A", "B", "C"], [1, 2, 3], [4, 5, 6] ] mapdict = ["X", "Y", "Z"] model=FakeDjangoModel() writer = DjangoModelWriter([model, data2[0], mapdict, None]) writer.write_array(data2[1:]) writer.close() assert model.objects.objs == self.result
def test_sheet_save_to_django_model_with_empty_array(self): model=FakeDjangoModel() data = [ ["X", "Y", "Z"], ['', '', ''], [1, 2, 3], [4, 5, 6] ] writer = DjangoModelWriter([model, data[0], None, None]) writer.write_array(data[1:]) writer.close() assert model.objects.objs == self.result
def test_sheet_save_to_django_model_3(self): model=FakeDjangoModel() def wrapper(row): row[0] = row[0] + 1 return row writer = DjangoModelWriter([model, self.data[0], None, wrapper]) writer.write_array(self.data[1:]) writer.close() assert model.objects.objs == [ {'Y': 2, 'X': 2, 'Z': 3}, {'Y': 5, 'X': 5, 'Z': 6} ]
def test_sheet_save_to_django_model(self): model=FakeDjangoModel() writer = DjangoModelWriter([model, self.data[0], None, None]) writer.write_array(self.data[1:]) writer.close() assert model.objects.objs == self.result