def test_migrate(export_file, tmpdir):
runner = CliRunner()
memory_db = peewee.SqliteDatabase(':memory:')
migrate_mock = mock.MagicMock(spec=migrate.migrate, side_effect=migrate.migrate)
with mock.patch('tracboat.cli.peewee.SqliteDatabase', lambda uri: memory_db), \
mock.patch('tracboat.migrate.migrate', migrate_mock):
result = runner.invoke(
cli.migrate, obj={}, catch_exceptions=False,
args=['--from-export-file', export_file, '--mock', '--mock-path', str(tmpdir)]
)
migrate_mock.assert_called()
assert result.exit_code == 0
评论列表
文章目录