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),
]
