← Back to branch summary

~unity-2d-team/unity-2d/Shell-MultiMonitor

« back to all changes in this revision

Viewing changes to grackle/tests/test_client.py

  • Committer: Aaron Bentley
  • Date: 2012-01-13 15:14:10 UTC
  • mto: (6.1.30 trunk)
  • mto: This revision was merged to the branch mainline in revision 45.
  • Revision ID: aaron@canonical.com-20120113151410-pexfwaq390oirhr2
Extract GrackleStore.

Show diffs side-by-side

added added

removed removed

Lines of Context:
grackle/tests/test_client.py
60
60
                    messages.extend(thread)
61
61
            else:
62
62
                messages.sort(key=lambda m: m[order])
63
 
        new_messages = []
64
 
        for message in messages:
65
 
            if (
66
 
                not parameters['include_hidden']
67
 
                and message.get('hidden', False)):
68
 
                continue
69
 
 
70
 
            if ('message_ids' in parameters and
71
 
                message['message_id'] not in parameters['message_ids']):
72
 
                continue
73
 
            message = dict(message)
74
 
            if 'headers' in parameters:
75
 
                headers = dict(
76
 
                    (k, v) for k, v in message['headers'].iteritems()
77
 
                    if k in parameters['headers'])
78
 
                message['headers'] = headers
79
 
            max_body = parameters.get('max_body_length')
80
 
            if max_body is not None:
81
 
                message['body'] = message['body'][:max_body]
82
 
            new_messages.append(message)
83
 
        messages = new_messages
 
63
        messages = [m for m in messages
 
64
                    if 'message_ids' not in parameters or
 
65
                    m['message_id'] in parameters['message_ids']]
84
66
        limit = parameters.get('limit', 100)
85
67
        memo = parameters.get('memo')
86
68
        message_id_indices = dict(
99
81
        else:
100
82
            next_memo = None
101
83
        messages = messages[start:end]
102
 
 
 
84
        new_messages = []
 
85
        for message in messages:
 
86
            message = dict(message)
 
87
            if 'headers' in parameters:
 
88
                headers = dict(
 
89
                    (k, v) for k, v in message['headers'].iteritems()
 
90
                    if k in parameters['headers'])
 
91
                message['headers'] = headers
 
92
            max_body = parameters.get('max_body_length')
 
93
            if max_body is not None:
 
94
                message['body'] = message['body'][:max_body]
 
95
            new_messages.append(message)
103
96
        response = {
104
 
            'messages': messages,
 
97
            'messages': new_messages,
105
98
            'next_memo': next_memo,
106
99
            'previous_memo': previous_memo
107
100
            }
320
313
        self.assertEqual({}, first_message['headers'])
321
314
 
322
315
    def test_get_messages_headers_exclude_headers(self):
323
 
        client = GrackleClient('localhost', 8441)
 
316
        client = GrackleClient('localhost', 8440)
324
317
        with fake_grackle_service(client,
325
318
            {'baz': [
326
319
                {'message_id': 'foo', 'headers': {'From': 'me'}}
332
325
        self.assertEqual({}, first_message['headers'])
333
326
 
334
327
    def test_get_messages_headers_include_headers(self):
335
 
        client = GrackleClient('localhost', 8442)
 
328
        client = GrackleClient('localhost', 8440)
336
329
        with fake_grackle_service(client,
337
330
            {'baz': [
338
331
                {'message_id': 'foo', 'headers': {'From': 'me', 'To': 'you'}}
344
337
        self.assertEqual({'From': 'me', 'To': 'you'}, first_message['headers'])
345
338
 
346
339
    def test_get_messages_max_body_length(self):
347
 
        client = GrackleClient('localhost', 8443)
 
340
        client = GrackleClient('localhost', 8440)
348
341
        with fake_grackle_service(client,
349
342
            {'baz': [
350
343
                {'message_id': 'foo', 'body': u'abcdefghi'}
353
346
        first_message = response['messages'][0]
354
347
        self.assertEqual('abc', first_message['body'])
355
348
 
356
 
    def test_include_hidden(self):
357
 
        client = GrackleClient('localhost', 8444)
358
 
        with fake_grackle_service(client,
359
 
            {'baz': [
360
 
                {'message_id': 'foo', 'hidden': True},
361
 
                {'message_id': 'bar', 'hidden': False}
362
 
            ]}):
363
 
            response = client.get_messages('baz', include_hidden=True)
364
 
            self.assertMessageIDs(['bar', 'foo'], response['messages'])
365
 
            response = client.get_messages('baz', include_hidden=False)
366
 
            self.assertMessageIDs(['bar'], response['messages'])
367