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