Python Client 예제들

예제 #1

파일: test_locmem_backend.py 프로젝트: scarsman/alligator

    def setUp(self):
        super(LocmemTestCase, self).setUp()
        self.backend = LocmemClient(CONN_STRING)

        # Just reach in & clear things out.
        LocmemClient.queues = {}
        LocmemClient.task_data = {}

예제 #2

파일: test_locmem_backend.py 프로젝트: philippeowagner/alligator

    def setUp(self):
        super(LocmemTestCase, self).setUp()
        self.backend = LocmemClient(CONN_STRING)

        # Just reach in & clear things out.
        LocmemClient.queues = {}
        LocmemClient.task_data = {}

예제 #3

파일: test_locmem_backend.py 프로젝트: scarsman/alligator

class LocmemTestCase(unittest.TestCase):
    def setUp(self):
        super(LocmemTestCase, self).setUp()
        self.backend = LocmemClient(CONN_STRING)

        # Just reach in & clear things out.
        LocmemClient.queues = {}
        LocmemClient.task_data = {}

    def test_init(self):
        self.assertEqual(LocmemClient.queues, {})
        self.assertEqual(LocmemClient.task_data, {})

    def test_len(self):
        LocmemClient.queues = {'all': ['a', 'b', 'c']}
        self.assertEqual(self.backend.len('all'), 3)
        self.assertEqual(self.backend.len('something'), 0)

    def test_drop_all(self):
        LocmemClient.queues = {'all': ['a', 'b', 'c']}
        LocmemClient.task_data = {
            'a': {
                'whatev': True
            },
            'b': 'grump',
            'd': 'another',
        }

        self.backend.drop_all('all')
        self.assertEqual(LocmemClient.queues, {'all': []})
        self.assertEqual(LocmemClient.task_data, {'d': 'another'})

    def test_push(self):
        self.assertEqual(LocmemClient.queues, {})
        self.assertEqual(LocmemClient.task_data, {})

        self.backend.push('all', 'hello', {'whee': 1})
        self.assertEqual(LocmemClient.queues, {'all': ['hello']})
        self.assertEqual(LocmemClient.task_data, {
            'hello': {
                'whee': 1
            },
        })

    def test_pop(self):
        self.backend.push('all', 'hello', {'whee': 1})

        data = self.backend.pop('all')
        self.assertEqual(data, {'whee': 1})
        self.assertEqual(LocmemClient.queues, {'all': []})
        self.assertEqual(LocmemClient.task_data, {})

    def test_get(self):
        self.backend.push('all', 'hello', {'whee': 1})
        self.backend.push('all', 'world', {'whee': 2})

        data = self.backend.get('all', 'world')
        self.assertEqual(data, {'whee': 2})
        self.assertEqual(LocmemClient.queues, {'all': ['hello']})
        self.assertEqual(LocmemClient.task_data, {'hello': {'whee': 1}})

        # Try a non-existent one.
        data = self.backend.get('all', 'nopenopenope')
        self.assertEqual(data, None)

예제 #4

파일: test_locmem_backend.py 프로젝트: philippeowagner/alligator

class LocmemTestCase(unittest.TestCase):
    def setUp(self):
        super(LocmemTestCase, self).setUp()
        self.backend = LocmemClient(CONN_STRING)

        # Just reach in & clear things out.
        LocmemClient.queues = {}
        LocmemClient.task_data = {}

    def test_init(self):
        self.assertEqual(LocmemClient.queues, {})
        self.assertEqual(LocmemClient.task_data, {})

    def test_len(self):
        LocmemClient.queues = {'all': ['a', 'b', 'c']}
        self.assertEqual(self.backend.len('all'), 3)
        self.assertEqual(self.backend.len('something'), 0)

    def test_drop_all(self):
        LocmemClient.queues = {'all': ['a', 'b', 'c']}
        LocmemClient.task_data = {
            'a': {'whatev': True},
            'b': 'grump',
            'd': 'another',
        }

        self.backend.drop_all('all')
        self.assertEqual(LocmemClient.queues, {'all': []})
        self.assertEqual(LocmemClient.task_data, {'d': 'another'})

    def test_push(self):
        self.assertEqual(LocmemClient.queues, {})
        self.assertEqual(LocmemClient.task_data, {})

        self.backend.push('all', 'hello', {'whee': 1})
        self.assertEqual(LocmemClient.queues, {
            'all': ['hello']
        })
        self.assertEqual(LocmemClient.task_data, {
            'hello': {'whee': 1},
        })

    def test_pop(self):
        self.backend.push('all', 'hello', {'whee': 1})

        data = self.backend.pop('all')
        self.assertEqual(data, {'whee': 1})
        self.assertEqual(LocmemClient.queues, {'all': []})
        self.assertEqual(LocmemClient.task_data, {})

    def test_get(self):
        self.backend.push('all', 'hello', {'whee': 1})
        self.backend.push('all', 'world', {'whee': 2})

        data = self.backend.get('all', 'world')
        self.assertEqual(data, {'whee': 2})
        self.assertEqual(LocmemClient.queues, {'all': ['hello']})
        self.assertEqual(LocmemClient.task_data, {
            'hello': {'whee': 1}
        })

        # Try a non-existent one.
        data = self.backend.get('all', 'nopenopenope')
        self.assertEqual(data, None)