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 runner(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_constructor = YamlParser(config)
migration_constructor.load_from_file(config.migration_file)
migration = migration_constructor.parse()
table = mock.MagicMock(spec=MigrationTable)
table.versions.return_value = []
database = mock.MagicMock(spec=Database)
return Runner(config, migration, database, table)