def test_special_case_where_only_one_model_used(self):
model1=FakeDjangoModel()
to_store = {
"Sheet1": self.content['Sheet1'][1:],
"Sheet2": self.content['Sheet2'][1:]
}
save_data(DB_DJANGO, to_store, models={
"Sheet1": [model1, self.content['Sheet1'][0], None, None]
})
assert model1.objects.objs == self.result1
model1._meta.model_name = "Sheet1"
model1._meta.update(["X", "Y", "Z"])
reader = DjangoBookReader([model1])
data = reader.sheets()
assert data['Sheet1'] == self.content['Sheet1']
def test_reading_from_more_models(self):
model1=FakeDjangoModel()
model2=FakeDjangoModel()
to_store = {
"Sheet1": self.content['Sheet1'][1:],
"Sheet2": self.content['Sheet2'][1:]
}
save_data(DB_DJANGO, to_store, models={
"Sheet1": [model1, self.content['Sheet1'][0], None, None],
"Sheet2": [model2, self.content['Sheet2'][0], None, None]
})
assert model1.objects.objs == self.result1
assert model2.objects.objs == self.result2
model1._meta.model_name = "Sheet1"
model2._meta.model_name = "Sheet2"
model1._meta.update(["X", "Y", "Z"])
model2._meta.update(["A", "B", "C"])
reader = DjangoBookReader([model1, model2])
data = reader.sheets()
assert data == self.content