def test_parse_yaml_example():
file_example = StringIO(YAML_EXAMPLE)
config = Config(None, 'test')
constructor = YamlParser(config)
constructor.load_from_buffer(file_example)
migration = constructor.parse()
assert len(migration.versions) == 4
def parser(filename, allow_serie=True, mode=None):
migration_file = os.path.join(request.fspath.dirname,
'examples', filename)
config = Config(migration_file,
'test',
allow_serie=allow_serie,
mode=mode)
migration_parser = YamlParser.parse_from_file(config.migration_file)
return migration_parser, config
def runner(filename, allow_serie=True, mode=None, db_versions=None):
migration_file = os.path.join(request.fspath.dirname,
'examples', filename)
config = Config(migration_file,
'test',
allow_serie=allow_serie,
mode=mode)
migration_parser = YamlParser.parse_from_file(config.migration_file)
migration = migration_parser.parse()
table = mock.MagicMock(spec=MigrationTable)
table.versions.return_value = db_versions or []
database = mock.MagicMock(spec=Database)
return Runner(config, migration, database, table)
def placeholders_config():
database = 'a_db'
db_user = '******'
db_password = '******'
db_port = 'a_port'
db_host = 'a_host'
return Config(
'migration.yml',
database,
db_user=db_user,
db_password=db_password,
db_port=db_port,
db_host=db_host,
)