def test_enable_insecure_serializers(self):
with patch('kombu.serialization.registry') as registry:
enable_insecure_serializers()
registry.assert_has_calls([
call.enable('pickle'), call.enable('yaml'),
call.enable('msgpack'),
])
registry.enable.side_effect = KeyError()
enable_insecure_serializers()
with patch('kombu.serialization.registry') as registry:
enable_insecure_serializers(['msgpack'])
registry.assert_has_calls([call.enable('msgpack')])
def test_enable(self):
agent = AgentMock(enabled=False)
controller = make_bamboo_agent_controller(agent=agent)
controller.set_enabled(True)
self.assertEqual(agent.method_calls, [call.enabled(), call.enable()])