celery/celery/tests/test_serialization.py

from __future__ import with_statement
import sys
import unittest


class TestAAPickle(unittest.TestCase):

    def test_no_cpickle(self):
        from celery.tests.utils import mask_modules
        prev = sys.modules.pop("billiard.serialization")
        with mask_modules("cPickle"):
            from billiard.serialization import pickle
            import pickle as orig_pickle
            self.assertTrue(pickle.dumps is orig_pickle.dumps)
        sys.modules["billiard.serialization"] = prev