def test_register_origin_keys(monkeypatch): class Fake1: pass class Fake2: pass key_maker = pretend.stub() key_maker_factory = pretend.call_recorder(lambda **kw: key_maker) monkeypatch.setattr(origin, "key_maker_factory", key_maker_factory) config = pretend.stub(registry={}) origin.register_origin_cache_keys(config, Fake1, cache_keys=["one", "two/{obj.attr}"]) origin.register_origin_cache_keys( config, Fake2, cache_keys=["three"], purge_keys=["lol"], ) assert key_maker_factory.calls == [ pretend.call(cache_keys=["one", "two/{obj.attr}"], purge_keys=None), pretend.call(cache_keys=["three"], purge_keys=["lol"]), ] assert config.registry == { "cache_keys": { Fake1: key_maker, Fake2: key_maker, }, }
def test_register_origin_keys(monkeypatch): class Fake1: pass class Fake2: pass key_maker = pretend.stub() key_maker_factory = pretend.call_recorder(lambda **kw: key_maker) monkeypatch.setattr(origin, "key_maker_factory", key_maker_factory) config = pretend.stub(registry={}) origin.register_origin_cache_keys( config, Fake1, cache_keys=["one", "two/{obj.attr}"]) origin.register_origin_cache_keys( config, Fake2, cache_keys=["three"], purge_keys=["lol"], ) assert key_maker_factory.calls == [ pretend.call(cache_keys=["one", "two/{obj.attr}"], purge_keys=None), pretend.call(cache_keys=["three"], purge_keys=["lol"]), ] assert config.registry == { "cache_keys": { Fake1: key_maker, Fake2: key_maker, }, }