def test_configure_alembic(monkeypatch): config_obj = pretend.stub( set_main_option=pretend.call_recorder(lambda *a: None)) def config_cls(): return config_obj monkeypatch.setattr(alembic.config, "Config", config_cls) config = pretend.stub(registry=pretend.stub( settings={"database.url": pretend.stub()})) alembic_config = _configure_alembic(config) assert alembic_config is config_obj assert alembic_config.set_main_option.calls == [ pretend.call("script_location", "warehouse:migrations"), pretend.call("url", config.registry.settings["database.url"]), ]
def test_configure_alembic(monkeypatch): config_obj = pretend.stub(set_main_option=pretend.call_recorder(lambda *a: None)) def config_cls(): return config_obj monkeypatch.setattr(alembic.config, "Config", config_cls) config = pretend.stub( registry=pretend.stub(settings={"database.url": pretend.stub()}) ) alembic_config = _configure_alembic(config) assert alembic_config is config_obj assert alembic_config.set_main_option.calls == [ pretend.call("script_location", "warehouse:migrations"), pretend.call("url", config.registry.settings["database.url"]), ]