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_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',
}
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',
}