def test_passes_args_to_serializer_init(self): setup_encrypted_serializer(self.key) self.assertEqual(1, self.register_mock.call_count) self.encrypted_serializer_mock.assert_called_once_with( key=self.key, serializer='pickle', )
def test_kombu_serializer_registration(self): setup_encrypted_serializer(key=self.key, serializer='pickle') self.register_mock.assert_called_once_with( 'encrypted_pickle', self.encrypted_serializer_mock_instance.serialize, self.encrypted_serializer_mock_instance.deserialize, content_type='application/x-encrypted-pickle', content_encoding='utf-8', )
def test_name_reflects_serializer_by_default(self): name = setup_encrypted_serializer(self.key, serializer='json') self.assertEqual(name, 'encrypted_json') self.register_mock.assert_called_once_with( 'encrypted_json', self.encrypted_serializer_mock_instance.serialize, self.encrypted_serializer_mock_instance.deserialize, content_type='application/x-encrypted-json', content_encoding='utf-8', )
import os from celery import Celery from kombu_encrypted_serializer import setup_encrypted_serializer TEST_KEY = 'WgFNqB8eokKER0aFxEmfnK7qyZmGhGmxxOqccW3oZoM=' os.environ['KOMBU_ENCRYPTED_SERIALIZER_KEY'] = TEST_KEY serializer_name = setup_encrypted_serializer(serializer='pickle') app = Celery('tasks', broker='redis://', backend='redis://') app.conf.update( CELERY_TASK_SERIALIZER=serializer_name, CELERY_RESULT_SERIALIZER=serializer_name, CELERY_ACCEPT_CONTENT=[serializer_name], ) @app.task def add(x, y): return x + y @app.task def give_it_back(data): return data
def test_calls_kombu_registry_register(self): setup_encrypted_serializer(self.key) self.assertEqual(1, self.register_mock.call_count)