def test_to_source_empty_mapping(self): mapping = {} model = Mock() converter = DatabaseConverter(mapping, self.Schema, self.Model) result = converter.to_source(model) self.Schema.assert_called_once_with() assert_that(result, same_instance(self.db_row))
def test_to_source(self): value = 'value' mapping = {'db_field': 'model_field'} model = Mock() model.model_field = value converter = DatabaseConverter(mapping, self.Schema, self.Model) result = converter.to_source(model) self.Schema.assert_called_once_with(db_field=value) assert_that(result, same_instance(self.db_row))
def test_update_source(self): value = 'value' mapping = {'db_field': 'model_field'} db_row = Mock() model = Mock() model.model_field = value converter = DatabaseConverter(mapping, self.Schema, self.Model) converter.update_source(db_row, model) assert_that(db_row.db_field, equal_to(value))
def to_source(self, model): context_row = DatabaseConverter.to_source(self, model) if context_row.description is None: context_row.description = '' if context_row.commented is None: context_row.commented = 0 return context_row
def to_source(self, model): source = DatabaseConverter.to_source(self, model) if not source.main_line: source.main_line = True if not source.main_user: source.main_user = True return source
def test_update_source(self): value = 'value' mapping = { 'db_field': 'model_field' } db_row = Mock() model = Mock() model.model_field = value converter = DatabaseConverter(mapping, self.Schema, self.Model) converter.update_source(db_row, model) assert_that(db_row.db_field, equal_to(value))
def test_to_source_with_nonexistent_column(self): value = 'value' mapping = { 'db_field': 'model_field', 'does_not_exist': 'does_not_exist' } model = Mock() model.model_field = value del model.does_not_exist converter = DatabaseConverter(mapping, self.Schema, self.Model) self.assertRaises(ValueError, converter.to_source, model)
def test_to_model_with_nonexistent_column(self): value = 'value' mapping = { 'db_field': 'model_field', 'does_not_exist': 'does_not_exist', } db_row = Mock() db_row.db_field = value del db_row.does_not_exist converter = DatabaseConverter(mapping, self.Schema, self.Model) self.assertRaises(ValueError, converter.to_model, db_row)
def __init__(self): DatabaseConverter.__init__(self, DB_TO_MODEL_MAPPING, UserLineSchema, LineExtension)
def __init__(self): DatabaseConverter.__init__(self, DB_TO_MODEL_MAPPING, None, UserVoicemail)
def __init__(self): DatabaseConverter.__init__(self, self.DB_TO_MODEL_MAPPING, UserSchema, User)
def to_model(self, db_row): model = DatabaseConverter.to_model(self, db_row) if model.description == '': model.description = None return model
def to_model(self, source): model = DatabaseConverter.to_model(self, source) self._convert_model_fields(source, model) return model
def update_source(self, source, model): DatabaseConverter.update_source(self, source, model) self._convert_source_fields(source, model)
def __init__(self): DatabaseConverter.__init__(self, self.DB_TO_MODEL_MAPPING, FuncKeySchema, FuncKey)
def __init__(self): DatabaseConverter.__init__(self, DB_TO_MODEL_MAPPING, QueueMember, QueueMemberAgent)
] _RELATION = {} def __init__(self, *args, **kwargs): NewModel.__init__(self, *args, **kwargs) self._related_cels = [] def get_related_cels(self): return self._related_cels def add_related_cels(self, cel_ids): self._related_cels.extend(cel_ids) DB_TO_MODEL_MAPPING = { 'id': 'id', 'date': 'date', 'source_name': 'source_name', 'source_exten': 'source_exten', 'destination_name': 'destination_name', 'destination_exten': 'destination_exten', 'user_field': 'user_field', 'answered': 'answered', 'duration': 'duration', 'source_line_identity': 'source_line_identity', 'destination_line_identity': 'destination_line_identity', } db_converter = DatabaseConverter(DB_TO_MODEL_MAPPING, CallLogSchema, CallLog)
def __init__(self): DatabaseConverter.__init__(self, DB_TO_MODEL_MAPPING, None, UserCtiProfile)
def __init__(self): DatabaseConverter.__init__(self, self.DB_TO_MODEL_MAPPING, IncallSchema, Incall)
def to_source(self, model): source = DatabaseConverter.to_source(self, model) source.enablevoicemail = int(model.enabled) return source
def __init__(self): DatabaseConverter.__init__(self, self.DB_TO_MODEL_MAPPING, VoicemailSchema, Voicemail)
def to_source(self, model): source = DatabaseConverter.to_source(self, model) self._convert_source_fields(source, model) return source
def to_model(self, source): model = DatabaseConverter.to_model(self, source) model.enabled = bool(source.enablevoicemail) return model
def __init__(self): DatabaseConverter.__init__(self, self.DB_TO_MODEL_MAPPING, CtiProfileSchema, CtiProfile)
def update_source(self, source, model): DatabaseConverter.update_source(self, source, model) self._update_source_fields(source, model)
def __init__(self): DatabaseConverter.__init__(self, self.DB_TO_MODEL_MAPPING, ContextRangeSchema, ContextRange)
def to_model(self, source): context_range = DatabaseConverter.to_model(self, source) if context_range.end == '': context_range.end = None return context_range
def __init__(self): DatabaseConverter.__init__(self, DB_TO_MODEL_MAPPING, ExtensionSchema, Extension)