def test_header_encoder(empty_config):
context_data = {
'foo': 'FOO',
'bar': 'BAR',
'baz': 'BAZ',
# unserialisable, shouldn't be included in the processed headers
'none': None,
}
encoder = HeaderEncoder()
with patch.object(encoder, 'header_prefix', new="testprefix"):
worker_ctx_cls = worker_context_factory('foo', 'bar', 'xxx')
worker_ctx = worker_ctx_cls(data=context_data)
worker_ctx.call_id_stack = ['x']
res = encoder.get_message_headers(worker_ctx)
assert res == {'testprefix.foo': 'FOO', 'testprefix.bar': 'BAR',
'testprefix.call_id_stack': ['x']}
def test_header_encoder(empty_config):
context_data = {
'foo': 'FOO',
'bar': 'BAR',
'baz': 'BAZ',
# unserialisable, shouldn't be included in the processed headers
'none': None,
}
encoder = HeaderEncoder()
with patch.object(encoder, 'header_prefix', new="testprefix"):
worker_ctx = Mock(context_data=context_data)
res = encoder.get_message_headers(worker_ctx)
assert res == {
'testprefix.foo': 'FOO',
'testprefix.bar': 'BAR',
'testprefix.baz': 'BAZ',
}
