def load_descriptor(self):
"""
Set fields from descriptor file
"""
descriptor = ReadDescriptor(self.Meta.descriptor,
self.Meta.descriptor_model)
self.fields = descriptor.get_fields()
self.exclude_fields()
def test_invalid_model(self):
descriptor = ReadDescriptor(file_name=JSON_FILE,
model_name='TestInvalidModel')
self.assertRaises(InvalidModel, lambda: descriptor.get_model())
def test_invalid_file(self):
self.assertRaises(
InvalidDescriptor,
lambda: ReadDescriptor(file_name='invalid_file.er',
model_name='TestInvalidModel'))
def setUp(self):
self.descriptor = ReadDescriptor(file_name=JSON_FILE,
model_name='Contact')
