def test_generate_schema_add_columns_to_schema(schema_adapter):
schema = Schema(schema_adapter)
assert [column.name
for column in schema.tables[0].columns] == ['column_one']
assert [column.name for column in schema.tables[1].columns
] == ['column_1', 'column_2']
def test_generate_schema_adds_triggers_to_table(build_schema_adapter):
mock_adapter = build_schema_adapter({
'tables': {
'table_one': {
'triggers': {
'trigger_one': {
'hook': 'hook_one'
}
}
}
}
})
schema = Schema(mock_adapter)
assert schema.triggers[0].name == 'trigger_one'
assert schema.triggers[0].hook == 'hook_one'
def test_generate_schema_assigns_the_primary_key_to_the_table(
build_schema_adapter):
mock_conn = build_schema_adapter({
'tables': {
'table_primary_key': {
'columns': {
'primary_column': {
'type': 'integer',
'primary_key': True
}
}
}
}
})
schema = Schema(mock_conn)
assert schema.tables[0].primary_key.name == 'primary_column'
def test_generate_schema_assigns_index(build_schema_adapter):
mock_conn = build_schema_adapter({
'tables': {
'table': {
'columns': {
'id': {
'type': 'integer',
'indexes': {
'id_index': {}
}
}
}
}
}
})
schema = Schema(mock_conn)
assert schema.tables[0].columns[0].index == schema.indexes[0]
def report(build_schema_adapter):
validations = validations = {
'table': [fail_validation],
'column': [pass_validation],
'function': [raise_validation]
}
schema = Schema(
build_schema_adapter({
'tables': {
'table_one': {
'columns': {
'column_one': {}
}
},
'ignore': {}
},
'functions': {
'function_one': {}
}
}))
return validate_entities(validations, ['ignore'], schema)
def test_generate_schema_add_constraints_to_column(build_schema_adapter):
tables_info = {
'tables': {
'table_one': {
'columns': {
'column_one': {
'type': 'type_one',
'constraints': {
'constraint_one': {
'type': 'type_one'
}
}
}
}
}
}
}
mock_adapter = build_schema_adapter(tables_info)
schema = Schema(mock_adapter)
assert schema.columns[0].constraints[0].name == 'constraint_one'
assert schema.columns[0].constraints[0].type == 'type_one'
def test_generate_schema_assigns_references_to_foreign_keys_columns(
build_schema_adapter):
mock_conn = build_schema_adapter({
'tables': {
'table': {
'columns': {
'id': {
'primary_key': True
}
}
},
'foreign_key_table': {
'columns': {
'table_id': {
'references': 'table'
}
}
}
}
})
schema = Schema(mock_conn)
assert schema.tables[1].columns[0].references == schema.tables[0]
def _serial_schema(info: dict):
return Schema(SerializedAdapter(info))
def test_generate_schema_assigns_the_column_type_to_column(schema_adapter):
schema = Schema(schema_adapter)
assert [column.type for column in schema.tables[0].columns] == ['text']
def test_generate_schema_adds_tables_to_schema(schema_adapter):
schema = Schema(schema_adapter)
assert [table.name
for table in schema.tables] == ['table_one', 'table_two']
def test_generate_schema_adds_sequences_to_schema(build_schema_adapter):
sequences = ['sequence_one']
mock_adapter = build_schema_adapter({'sequences': {'sequence_one': {}}})
schema = Schema(mock_adapter)
assert [sequence.name for sequence in schema.sequences] == sequences
def test_generate_schema_adds_procedures_to_schema(build_schema_adapter):
procedures = ['procedure_one']
mock_adapter = build_schema_adapter({'procedures': {'procedure_one': {}}})
schema = Schema(mock_adapter)
assert [procedure.name for procedure in schema.procedures] == procedures
def test_generate_schema_adds_functions_to_schema(build_schema_adapter):
functions = ['function_one']
mock_adapter = build_schema_adapter({'functions': {'function_one': {}}})
schema = Schema(mock_adapter)
assert [function.name for function in schema.functions] == functions