def test_includeme(env, ssl): registry_dict = {} config = pretend.stub( action=pretend.call_recorder(lambda *a, **kw: None), add_directive=pretend.call_recorder(lambda *a, **kw: None), add_request_method=pretend.call_recorder(lambda *a, **kw: None), registry=pretend.stub( __getitem__=registry_dict.__getitem__, __setitem__=registry_dict.__setitem__, settings={ "warehouse.env": env, "celery.broker_url": pretend.stub(), "celery.result_url": pretend.stub(), "celery.scheduler_url": pretend.stub(), }, ), ) tasks.includeme(config) app = config.registry["celery.app"] assert app.Task is tasks.WarehouseTask assert app.pyramid_config is config for key, value in { "broker_url": config.registry.settings["celery.broker_url"], "broker_use_ssl": ssl, "worker_disable_rate_limits": True, "result_backend": config.registry.settings["celery.result_url"], "result_serializer": "json", "task_serializer": "json", "accept_content": ["json", "msgpack"], "result_compression": "gzip", "task_queue_ha_policy": "all", "REDBEAT_REDIS_URL": (config.registry.settings["celery.scheduler_url"]) }.items(): assert app.conf[key] == value assert config.action.calls == [ pretend.call(("celery", "finalize"), app.finalize), ] assert config.add_directive.calls == [ pretend.call( "add_periodic_task", tasks._add_periodic_task, action_wrap=False, ), pretend.call( "make_celery_app", tasks._get_celery_app, action_wrap=False, ), pretend.call("task", tasks._get_task_from_config, action_wrap=False), ] assert config.add_request_method.calls == [ pretend.call(tasks._get_task_from_request, name="task", reify=True), ]
def test_includeme(env, ssl): registry_dict = {} config = pretend.stub( action=pretend.call_recorder(lambda *a, **kw: None), add_directive=pretend.call_recorder(lambda *a, **kw: None), add_request_method=pretend.call_recorder(lambda *a, **kw: None), registry=pretend.stub( __getitem__=registry_dict.__getitem__, __setitem__=registry_dict.__setitem__, settings={ "warehouse.env": env, "celery.broker_url": pretend.stub(), "celery.result_url": pretend.stub(), "celery.scheduler_url": pretend.stub(), }, ), ) tasks.includeme(config) app = config.registry["celery.app"] assert app.Task is tasks.WarehouseTask assert app.pyramid_config is config for key, value in { "broker_url": config.registry.settings["celery.broker_url"], "broker_use_ssl": ssl, "worker_disable_rate_limits": True, "result_backend": config.registry.settings["celery.result_url"], "result_serializer": "json", "task_serializer": "json", "accept_content": ["json", "msgpack"], "result_compression": "gzip", "task_queue_ha_policy": "all", "REDBEAT_REDIS_URL": ( config.registry.settings["celery.scheduler_url"])}.items(): assert app.conf[key] == value assert config.action.calls == [ pretend.call(("celery", "finalize"), app.finalize), ] assert config.add_directive.calls == [ pretend.call( "add_periodic_task", tasks._add_periodic_task, action_wrap=False, ), pretend.call( "make_celery_app", tasks._get_celery_app, action_wrap=False, ), pretend.call("task", tasks._get_task_from_config, action_wrap=False), ] assert config.add_request_method.calls == [ pretend.call(tasks._get_task_from_request, name="task", reify=True), ]