celery/tests/testunits/test_worker_revoke.py

import unittest

from celery.worker import revoke


class TestRevokeRegistry(unittest.TestCase):

    def test_is_working(self):
        revoke.revoked.add("foo")
        self.assertTrue("foo" in revoke.revoked)
        revoke.revoked.pop_value("foo")
        self.assertTrue("foo" not in revoke.revoked)